A fast, effective data attribution method for neural networks in PyTorch
☆232Nov 18, 2024Updated last year
Alternatives and similar repositories for trak
Users that are interested in trak are comparing it to the libraries listed below
Sorting:
- Data for "Datamodels: Predicting Predictions with Training Data"☆97May 25, 2023Updated 2 years ago
- Influence Analysis and Estimation - Survey, Papers, and Taxonomy☆87Feb 27, 2024Updated 2 years ago
- ☆51Jan 24, 2024Updated 2 years ago
- Influence Functions with (Eigenvalue-corrected) Kronecker-Factored Approximate Curvature☆185Jun 24, 2025Updated 8 months ago
- DataInf: Efficiently Estimating Data Influence in LoRA-tuned LLMs and Diffusion Models (ICLR 2024)☆81Oct 3, 2024Updated last year
- Code for "Tracing Knowledge in Language Models Back to the Training Data"☆39Dec 27, 2022Updated 3 years ago
- `dattri` is a PyTorch library for developing, benchmarking, and deploying efficient data attribution algorithms.☆117Mar 13, 2026Updated last week
- Code for the paper "The Journey, Not the Destination: How Data Guides Diffusion Models"☆25Dec 12, 2023Updated 2 years ago
- ☆33Feb 11, 2025Updated last year
- ModelDiff: A Framework for Comparing Learning Algorithms☆58Aug 15, 2023Updated 2 years ago
- Code for NeurIPS'23 paper "A Bayesian Approach To Analysing Training Data Attribution In Deep Learning"☆17Jan 12, 2024Updated 2 years ago
- Implementation of Estimating Training Data Influence by Tracing Gradient Descent (NeurIPS 2020)☆240Feb 14, 2022Updated 4 years ago
- Intriguing Properties of Data Attribution on Diffusion Models (ICLR 2024)☆37Jan 23, 2024Updated 2 years ago
- This is a PyTorch reimplementation of Influence Functions from the ICML2017 best paper: Understanding Black-box Predictions via Influence…☆344Oct 29, 2023Updated 2 years ago
- [ICML 2024] LESS: Selecting Influential Data for Targeted Instruction Tuning☆516Oct 20, 2024Updated last year
- AI Logging for Interpretability and Explainability🔬☆140Jun 7, 2024Updated last year
- An implementation of online data mixing for the Pile dataset, based on the GPT-NeoX library.☆13Jan 9, 2024Updated 2 years ago
- ☆44Oct 13, 2023Updated 2 years ago
- Debiasing Through Data Attribution☆12May 23, 2024Updated last year
- Official repository for MATES: Model-Aware Data Selection for Efficient Pretraining with Data Influence Models [NeurIPS 2024]☆79Nov 14, 2024Updated last year
- A simple and efficient baseline for data attribution☆11Nov 10, 2023Updated 2 years ago
- A simple PyTorch implementation of influence functions.☆92Jun 17, 2024Updated last year
- ☆65Jan 13, 2022Updated 4 years ago
- A toolkit for quantitative evaluation of data attribution methods.☆57Jul 14, 2025Updated 8 months ago
- OpenDataVal: a Unified Benchmark for Data Valuation in Python (NeurIPS 2023)☆100Feb 4, 2025Updated last year
- Distilling Model Failures as Directions in Latent Space☆48Feb 8, 2023Updated 3 years ago
- Code for paper: “What Data Benefits My Classifier?” Enhancing Model Performance and Interpretability through Influence-Based Data Selecti…☆23May 17, 2024Updated last year
- ☆15Jul 24, 2022Updated 3 years ago
- Code for coreset selection methods☆253Feb 27, 2023Updated 3 years ago
- ☆213Oct 10, 2022Updated 3 years ago
- Implementation of TSDS: Data Selection for Task-Specific Model Finetuning. An optimal-transport framework for selecting domain-specific a…☆18Dec 25, 2024Updated last year
- Code for the ICLR 2020 Paper, "A Theory of Usable Information under Computational Constraints"☆30Jul 8, 2020Updated 5 years ago
- Dataset Interfaces: Diagnosing Model Failures Using Controllable Counterfactual Generation☆45Feb 27, 2023Updated 3 years ago
- ☆40Aug 20, 2025Updated 7 months ago
- [ICML 2024] Selecting High-Quality Data for Training Language Models☆201Dec 8, 2025Updated 3 months ago
- Decomposing and Editing Predictions by Modeling Model Computation☆139Jun 12, 2024Updated last year
- Code for the paper "Evading Black-box Classifiers Without Breaking Eggs" [SaTML 2024]☆21Apr 15, 2024Updated last year
- ☆119Feb 11, 2025Updated last year
- ☆62Jun 8, 2021Updated 4 years ago