A PyTorch implementation for the ICLR'23 paper Metadata Archaeology: Unearthing Data Subsets by Leveraging Training Dynamics.
Check the OpenReview page for detailed discussions on the paper.
The code can be executed using the train.sh script provided.
The code also requires memorization/consistency scores to compute typical/atypical examples.
Configure the data_dir variable in the script to define the path to these scores.
We use consistency scores by default. Consistency scores can be downloaded from the official website: https://pluskid.github.io/structural-regularity/
Similarly, memorization scores can also be downloaded from the official website: https://pluskid.github.io/influence-memorization/
Surfaced examples can be accessed through Google Drive. Also checkout our webpage for more details.
As loss trajectories themselves as of great interest, the computed loss trajectories on CIFAR-10, CIFAR-100 and ImageNet are directly available for download through
Google Drive.
Please refer to the main script metadata_archaeology.py for details regarding how to load these files.
@article{siddiqui2022metadataarchaeology,
title={Metadata Archaeology: Unearthing Data Subsets by Leveraging Training Dynamics},
author={Siddiqui, Shoaib Ahmed and Rajkumar, Nitarshan and Maharaj, Tegan and Krueger, David and Hooker, Sara},
journal={arXiv preprint},
year={2022},
url={https://arxiv.org/abs/2209.10015}
}
In case of any issues, feel free to drop me an email or open an issue on the repository.
Email: [email protected]
MIT