cd /news/machine-learning/learning-what-to-forget-distribution… · home › topics › machine-learning › article
[ARTICLE · art-143029] src=machinebrief.com ↗ pub= topic=machine-learning verified=true sentiment=· neutral

Learning What to Forget: Distributional Unlearning for LLM Representation Spaces

Researchers introduced Mamushi, a non-parametric framework for distributional unlearning that ranks forget examples using a probabilistic classifier whose Bayes-optimal logit equals the forget-to-retain log-density ratio, according to the arXiv paper 2609.38929v1. The authors prove that thresholding the population log-density ratio yields the optimal fixed-budget selection rule for their removal-preservation objective and establish a non-asymptotic transfer guarantee linking score-estimation and threshold-calibration errors to degradation from the population-optimal rule. In evaluations on toxic-language and topical-domain removal datasets across different representations, Mamushi achieved a more favorable removal-preservation trade-off than other baselines and reduced the number of forget examples required to reach a fixed forgetting target.

by read1 min views1 publishedOct 1, 2026

arXiv:2609.38929v1 Announce Type: new Abstract: Machine learning systems increasingly face the need to remove the influence of entire data domains, such as toxic language, harmful behavior, or topical content, rather than isolated records. Recent work formalizes this problem as \emph{distributional unlearning}: selecting a subset of a forget domain whose removal moves the training distribution away from an unwanted population while preserving proximity to the desired one. However, existing analyses often impose parametric assumptions to obtain tractable selection rules. These assumptions may be poorly suited to high-dimensional language-model representations. We introduce \textsc{Mamushi}, a framework for non-parametric distributional unlearning that ranks forget examples using a probabilistic classifier whose Bayes-optimal logit equals the forget-to-retain log-density ratio (up to an additive class-prior constant). We show that thresholding the population log-density ratio yields the optimal fixed-budget selection rule for our removal--preservation objective and establish a non-asymptotic transfer guarantee relating score-estimation and threshold-calibration errors to degradation from the population-optimal selection rule. Our empirical evaluation spans real-world datasets on toxic-language removal and topical-domain removal regimes using different representations, with \textsc{Mamushi} achieving a more favorable removal--preservation trade-off than other baselines. Our work shows that \textsc{Mamushi} can serve as an efficient selection approach for downstream machine unlearning procedures, reducing the number of forget examples required to reach a fixed forgetting target.

── more in #machine-learning 4 stories · sorted by recency
── more on @mamushi 3 stories trending now
sponsored brought to you by zahid.host 4,200+ EU-deployed projects
reading about agents? ship yours in a single git push.

Run your AI side-project on zahid.host

EU-based hosting, git-push deploys, automatic HTTPS, no cold starts. Free tier with a custom domain — perfect for shipping the agent you just read about.

$git push zahid main
→ Live at https://your-agent.zahid.host ✓
Get free account → Pricing
from €0/mo · no card required
LIVE [news/learning-what-to-for…] indexed:0 read:1min 2026-10-01 · —