MadryLab / trak
A fast, effective data attribution method for neural networks in PyTorch
☆207Updated 5 months ago
Alternatives and similar repositories for trak:
Users that are interested in trak are comparing it to the libraries listed below
- Influence Functions with (Eigenvalue-corrected) Kronecker-Factored Approximate Curvature☆147Updated 9 months ago
- `dattri` is a PyTorch library for developing, benchmarking, and deploying efficient data attribution algorithms.☆72Updated this week
- Influence Analysis and Estimation - Survey, Papers, and Taxonomy☆77Updated last year
- Data for "Datamodels: Predicting Predictions with Training Data"☆97Updated last year
- A simple PyTorch implementation of influence functions.☆85Updated 10 months ago
- AI Logging for Interpretability and Explainability🔬☆115Updated 11 months ago
- DataInf: Efficiently Estimating Data Influence in LoRA-tuned LLMs and Diffusion Models (ICLR 2024)☆64Updated 7 months ago
- Source code of "Task arithmetic in the tangent space: Improved editing of pre-trained models".☆101Updated last year
- ☆66Updated 3 years ago
- ☆60Updated 3 years ago
- ☆93Updated 3 weeks ago
- Conformal Language Modeling☆28Updated last year
- ☆92Updated 2 months ago
- Experiments and code to generate the GINC small-scale in-context learning dataset from "An Explanation for In-context Learning as Implici…☆106Updated last year
- A library for efficient patching and automatic circuit discovery.☆64Updated 2 weeks ago
- Sparse probing paper full code.☆56Updated last year
- Framework code with wandb, checkpointing, logging, configs, experimental protocols. Useful for fine-tuning models or training from scratc…☆150Updated 2 years ago
- Evaluate interpretability methods on localizing and disentangling concepts in LLMs.☆46Updated 7 months ago
- ☆88Updated 10 months ago
- ☆28Updated last year
- Official repository for CMU Machine Learning Department's 10732: Robustness and Adaptivity in Shifting Environments☆74Updated 2 years ago
- ☆93Updated last year
- ☆33Updated 4 months ago
- ☆47Updated last year
- ☆223Updated 7 months ago
- ☆54Updated 2 years ago
- ☆205Updated last year
- NeuroSurgeon is a package that enables researchers to uncover and manipulate subnetworks within models in Huggingface Transformers☆41Updated 2 months ago
- Röttger et al. (NAACL 2024): "XSTest: A Test Suite for Identifying Exaggerated Safety Behaviours in Large Language Models"☆95Updated 2 months ago
- Using sparse coding to find distributed representations used by neural networks.☆240Updated last year