Evaluate whether an unlearning method removes a fact from a model's actual weights
Measure how much forgotten personal information resurfaces after further training
Compare an unlearning method's effect on forget, retain, and general capability
Implement and benchmark a new custom unlearning method against provided baselines
| mcgill-nlp/lacuna | 0xallam/my-recipe | 0xhassaan/nn-from-scratch | |
|---|---|---|---|
| Stars | 0 | — | 0 |
| Language | Python | Python | Python |
| Last pushed | — | 2022-11-22 | — |
| Maintenance | — | Dormant | — |
| Setup difficulty | hard | moderate | moderate |
| Complexity | 5/5 | 2/5 | 4/5 |
| Audience | researcher | general | developer |
Figures from each repo's GitHub metadata at analysis time.
Requires an Ampere-or-newer GPU, CUDA 12.6, and downloading tens of gigabytes of model artifacts.
LACUNA is a research testbed for studying machine unlearning, which is the problem of trying to make an AI language model forget specific pieces of information it learned during training. What makes LACUNA different is that it starts by deliberately planting made-up personal information, like fake names and email addresses, into known and exact locations inside the internal numbers of a language model, so researchers know precisely where each fact lives before they try to remove it. This lets a researcher measure something that is normally impossible to check: whether an unlearning method actually erases a fact from the specific internal parameters that stored it, or whether it merely stops the model from saying that fact out loud while the information is technically still there and can resurface later. The project calls this second, weaker outcome a resurfacing risk, since a stronger prompting method might bring the hidden fact back out. This particular repository is the evaluation release, meaning it assumes you already have one of the provided models, built on the OLMo architecture at either 1 billion or 7 billion parameters, and it walks through the steps of applying your own or an included unlearning method to it, then scoring the result. Scoring happens along three angles: whether the target information is gone while other knowledge stays intact, whether the actual correct internal weights were changed, and how much of the forgotten information comes back if the model is trained a little further on related data. Running it requires a capable GPU, a specific version of Python, and downloading large model files that can take up many tens of gigabytes of disk space, so this is squarely a tool for machine learning researchers working on model safety and privacy, not something a casual user would install. A companion branch of this same repository contains the full pipeline used to build these models from scratch, while this repository focuses only on evaluating unlearning methods against them.
A research testbed for testing whether AI unlearning methods truly erase planted facts from a model's weights, not just its outputs.
Mainly Python. The stack also includes Python, PyTorch, CUDA.
Setup difficulty is rated hard, with roughly 1h+ to a first successful run.
Mainly researcher.
This repo across BitVibe Labs
double-check against the repo, no cap.