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
- ☆32May 24, 2023Updated 2 years ago
- Influence Functions with (Eigenvalue-corrected) Kronecker-Factored Approximate Curvature☆184Jun 24, 2025Updated 8 months ago
- ModelDiff: A Framework for Comparing Learning Algorithms☆58Aug 15, 2023Updated 2 years ago
- ☆51Jan 24, 2024Updated 2 years ago
- `dattri` is a PyTorch library for developing, benchmarking, and deploying efficient data attribution algorithms.☆110Feb 10, 2026Updated 2 weeks ago
- DataInf: Efficiently Estimating Data Influence in LoRA-tuned LLMs and Diffusion Models (ICLR 2024)☆79Oct 3, 2024Updated last year
- Code for "Tracing Knowledge in Language Models Back to the Training Data"☆39Dec 27, 2022Updated 3 years ago
- Code for the paper "The Journey, Not the Destination: How Data Guides Diffusion Models"☆25Dec 12, 2023Updated 2 years ago
- ☆32Feb 11, 2025Updated last year
- 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
- 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)☆239Feb 14, 2022Updated 4 years ago
- Distilling Model Failures as Directions in Latent Space☆47Feb 8, 2023Updated 3 years ago
- [ICML 2024] LESS: Selecting Influential Data for Targeted Instruction Tuning☆512Oct 20, 2024Updated last year
- AI Logging for Interpretability and Explainability🔬☆140Jun 7, 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
- Intriguing Properties of Data Attribution on Diffusion Models (ICLR 2024)☆37Jan 23, 2024Updated 2 years ago
- ☆118Feb 11, 2025Updated last year
- Dataset Interfaces: Diagnosing Model Failures Using Controllable Counterfactual Generation☆45Feb 27, 2023Updated 3 years ago
- ☆213Oct 10, 2022Updated 3 years ago
- Attribute (or cite) statements generated by LLMs back to in-context information.☆322Oct 8, 2024Updated last year
- ☆43Oct 13, 2023Updated 2 years ago
- ☆15Jul 24, 2022Updated 3 years ago
- Sparse and discrete interpretability tool for neural networks☆64Feb 12, 2024Updated 2 years ago
- Code for T-MARS data filtering☆35Aug 23, 2023Updated 2 years ago
- A toolkit for quantitative evaluation of data attribution methods.☆56Jul 14, 2025Updated 7 months ago
- Debiasing Through Data Attribution☆12May 23, 2024Updated last year
- Official code for the paper: "Metadata Archaeology"☆19May 10, 2023Updated 2 years ago
- Code for the paper "Evading Black-box Classifiers Without Breaking Eggs" [SaTML 2024]☆21Apr 15, 2024Updated last year
- OpenDataVal: a Unified Benchmark for Data Valuation in Python (NeurIPS 2023)☆100Feb 4, 2025Updated last year
- DSIR large-scale data selection framework for language model training☆270Apr 7, 2024Updated last year
- Code for coreset selection methods☆255Feb 27, 2023Updated 3 years ago
- A simple PyTorch implementation of influence functions.☆92Jun 17, 2024Updated last year
- Pytorch implementation of DoReMi, a method for optimizing the data mixture weights in language modeling datasets☆350Dec 26, 2023Updated 2 years ago
- ☆25Nov 14, 2022Updated 3 years ago
- A codebase that makes differentially private training of transformers easy.☆183Dec 9, 2022Updated 3 years ago
- [ICML 2024] Selecting High-Quality Data for Training Language Models☆201Dec 8, 2025Updated 2 months ago