K-文化的一切——从回归到 K-美妆,发送到您的邮箱订阅邮件

METAL MEDIA

The More Popular, The Harder to Forget: Adaptive Popularity for LLM Unlearning

arXiv:2608.142292026-08-13

越有名的事实越难从AI模型中删除,新方法AdaPop按知名度调节遗忘力度

让AI模型对特定信息进行遗忘训练时会遇到一个问题:越有名的事实在预训练中被记得越深,比冷门事实更难删除,但现有方法对所有待删除内容都施加同样强度的压力。AdaPop利用维基数据(Wikidata)站点链接数或大模型评分等外部知名度信号,为每个事实分配不同的遗忘力度,并配合一个每训练轮次自动调节保留惩罚的控制器,防止无关知识被误伤。在三种模型和两个基准测试中,面对改写过的提问时泄露的被删信息比其他方法少约5倍,面对刻意刁难式的提问时少约1.6倍。

METAL MEDIA 解读图

AdaPop工作原理:知名度感知遗忘与自动平衡

证据状态已报告实测结果

  1. 知名度评分用维基数据站点链接数或大模型评分等外部来源,为每条事实打知名度分数
  2. 转换为指数把知名度分数转换成一个指数(beta),对热门事实的每个词元施加更强遗忘压力,对冷门事实施加更弱压力
  3. 加权遗忘损失该指数与模型对词元的置信度结合,构成用于删除目标事实的训练信号权重
  4. 双重上升控制器每个训练轮次结束时检测保留知识受损程度,据此自动调高或调低保留惩罚强度
  5. 结果验证通过改写提问、刁难式提问以及内部隐藏状态对比,检验事实是否被真正删除而非仅在表面被掩盖
这是 METAL MEDIA 制作的解读图,并非论文作者提供的原图。

他们做了什么

  1. 问题:现有遗忘方法对所有待删除事实施加同样强度的梯度压力,但实际上越有名的事实在预训练中被记得越深、越难删除,而对冷门事实同样用力则会过度删除,损害其他需要保留的知识。
  2. 方法:AdaPop用维基数据站点链接数或大模型评分等外部信号衡量每条事实的知名度,把该分数转换成一个指数,让热门事实承受更强的删除压力、冷门事实承受更弱的压力;同时配合一个双重上升(dual-ascent)控制器,每训练轮次检测保留知识的损失情况并自动调整保留惩罚强度。
  3. 实验:在Llama-3.1-8B-Instruct、Qwen2.5-7B-Instruct、Gemma-7B-it三个模型和DUET、RWKU两个基准上,与GA、GD、NPO、WGA等现有方法进行了比较。
  4. 结果:面对改写过的提问,AdaPop泄露的被删信息比其他方法少约5倍;面对刻意刁难式的提问少约1.6倍。模型内部表征上,被删事实的隐藏状态相比遗忘前模型变化更大,而需要保留的知识表征几乎保持不变。
Figure 1: Overview of AdaPop. The popularity score reweights per-fact token contributions to the forget loss; the dual-ascent controller adjusts the retain penalty each epoch to maintain retain quality (Algorithm 1).
Figure 1: Overview of AdaPop. The popularity score reweights per-fact token contributions to the forget loss; the dual-ascent controller adjusts the retain penalty each epoch to maintain retain quality (Algorithm 1).
Table 1: ROUGE-L Recall and Cosine Similarity for Llama-3.1-8B-Instruct at lr=10−4. ROUGE-L: mean ± std over three seeds; Cosine Similarity: single seed. F. ↓ = forget; R. ↑ = retain. w/o MU = the original checkpoint, evaluated with no unlearning applied. Underline = best among NPO, WGA, AdaPop. †Unusable: generation failure (GA: joint output and retain collapse; GD: severe retain degradation with collapse-style generation artefacts; see Appendix G).
Bench.AlgorithmRougeCos Sim
F. ↓R. ↑F. ↓R. ↑
w/o MU0.9390.9680.8740.883
AdaPop0.043 ± .0070.959 ± .0090.3690.971
GA†0.000 ± .0000.000 ± .0000.0930.093
GD†0.021 ± .0030.853 ± .0110.1200.897
NPO0.670 ± .0950.996 ± .0050.6840.998
DUETWGA0.036 ± .0020.995 ± .0030.4420.996
w/o MU0.7550.8270.7760.823
AdaPop0.078 ± .0050.972 ± .0080.1250.977
GA†0.002 ± .0000.001 ± .0000.0630.067
GD†0.026 ± .0050.759 ± .0210.0960.829
NPO0.540 ± .0340.957 ± .0050.4030.967
RWKUWGA0.095 ± .0090.977 ± .0030.2470.984
Figure 2: Wikidata popularity-score distributions for the forget sets of DUET and RWKU. DUET spans nearly two orders of magnitude (69–3,763; median 1,090), providing a direct test of popularity-sensitive methods; RWKU is narrower and skewed lower (median 130).
Figure 2: Wikidata popularity-score distributions for the forget sets of DUET and RWKU. DUET spans nearly two orders of magnitude (69–3,763; median 1,090), providing a direct test of popularity-sensitive methods; RWKU is narrower and skewed lower (median 130).
Table 2: ROUGE-L Recall and Cosine Similarity for Qwen2.5-7B-Instruct at lr=10−4. Notation as in Table 1.
Bench.AlgorithmRougeCos Sim
F. ↓R. ↑F. ↓R. ↑
w/o MU0.9210.8860.8640.861
AdaPop0.056 ± .0070.950 ± .0080.4560.964
GA†0.000 ± .0000.000 ± .0000.0580.092
GD†0.022 ± .0040.757 ± .0150.0940.841
NPO0.684 ± .0230.973 ± .0020.5760.979
DUETWGA0.058 ± .0020.987 ± .0090.4770.999
w/o MU0.5700.6660.3220.347
AdaPop0.016 ± .0070.855 ± .0030.1780.911
GA†0.001 ± .0000.000 ± .0000.0710.075
GD†0.014 ± .0030.462 ± .0180.1500.640
NPO0.271 ± .0100.783 ± .0120.4900.870
RWKUWGA0.038 ± .0160.890 ± .0050.2120.931
Figure 3: AdaPop on DUET (Llama) under three popularity signals: Wikidata score, LLM-as-Judge (3-seed mean), and measured corpus frequency (infini-gram over Pile-train). Left: forget ROUGE-L (↓). Right: holdout ROUGE-L (↑). Each point averages a rare-tier and a popular-tier run at that learning rate. All signals use the anchors of Appendix B, (a,b)=(58.7,0.796). Rates above 5×10−4 are omitted: at 10−3 every signal has at least one collapsed tier.
Figure 3: AdaPop on DUET (Llama) under three popularity signals: Wikidata score, LLM-as-Judge (3-seed mean), and measured corpus frequency (infini-gram over Pile-train). Left: forget ROUGE-L (↓). Right: holdout ROUGE-L (↑). Each point averages a rare-tier and a popular-tier run at that learning rate. All signals use the anchors of Appendix B, (a,b)=(58.7,0.796). Rates above 5×10−4 are omitted: at 10−3 every signal has at least one collapsed tier.
Table 3: ROUGE-L Recall and Cosine Similarity for Gemma-7B-it at lr=10−4. Notation as in Table 1.
Bench.AlgorithmRougeCos Sim
F. ↓R. ↑F. ↓R. ↑
w/o MU0.8920.9260.5860.541
AdaPop0.023 ± .0040.976 ± .0110.2060.976
GA†0.000 ± .0000.000 ± .0000.0960.102
GD†0.044 ± .0070.598 ± .0240.0690.626
NPO0.626 ± .0710.968 ± .0070.3940.956
DUETWGA0.050 ± .0020.996 ± .0040.2370.997
w/o MU0.4710.5510.3280.349
AdaPop0.034 ± .0140.948 ± .0090.0680.961
GA†0.000 ± .0000.000 ± .0000.0000.000
GD†0.013 ± .0030.334 ± .0130.1150.532
NPO0.341 ± .0130.773 ± .0100.2400.840
RWKUWGA0.040 ± .0030.950 ± .0070.1350.965
Figure 4: Internal representation metrics at lr=10−4, across three models. Each row corresponds to one metric; each column corresponds to one data split. For the forget split, deeper erasure corresponds to lower ΔLP, higher ΔRank, lower Hid.Cos, and higher KL; directional desiderata are reversed for the retain split. Marker shape indicates benchmark (DUET vs. RWKU); colour indicates model family.
Figure 4: Internal representation metrics at lr=10−4, across three models. Each row corresponds to one metric; each column corresponds to one data split. For the forget split, deeper erasure corresponds to lower ΔLP, higher ΔRank, lower Hid.Cos, and higher KL; directional desiderata are reversed for the retain split. Marker shape indicates benchmark (DUET vs. RWKU); colour indicates model family.
Table 4: Robustness at lr=10−4. Top: paraphrase ROUGE-L on the DUET forget split (lower is better). Bottom: RWKU Level-3 adversarial-attack ROUGE-L; lower indicates stronger resistance to knowledge recovery. Underline = best among NPO, WGA, AdaPop. GA and GD collapse and are excluded from forget-quality ranking.
AlgoLlamaQwenGemma
DUET paraphrase forget ROUGE-L ↓
AdaPop0.0450.0740.027
GA†0.0000.0000.000
GD†0.0150.0450.117
NPO0.6960.6050.568
WGA0.0420.1040.050
RWKU adversarial-attack forget ROUGE-L ↓
AdaPop0.2620.1440.195
GA†0.0000.0010.000
GD†0.2540.1710.103
NPO0.6570.4000.485
WGA0.3960.2110.239
Figure 5: Component ablation on DUET. Top row: rare-fact paraphrase forget ROUGE-L (↓) and retain ROUGE-L (↑) over lr∈[2×10−5,8×10−4]. Bottom row: popular-fact paraphrase forget ROUGE-L (↓) and retain ROUGE-L (↑) over lr∈[2×10−4,8×10−3].
Figure 5: Component ablation on DUET. Top row: rare-fact paraphrase forget ROUGE-L (↓) and retain ROUGE-L (↑) over lr∈[2×10−5,8×10−4]. Bottom row: popular-fact paraphrase forget ROUGE-L (↓) and retain ROUGE-L (↑) over lr∈[2×10−4,8×10−3].
Table 5: DUET paraphrase forget ROUGE-L (↓) split by popularity tier at lr=10−4. Underline = best among NPO, WGA, AdaPop. GA is omitted (all values 0.000 under collapse); †GD’s low rare-tier values coincide with retain collapse (Table 1).
AlgoLlamaQwenGemma
Pop.RarePop.RarePop.Rare
AdaPop0.0400.0490.0550.0920.0280.026
GD†0.0270.0030.0860.0030.2320.002
NPO0.8540.5380.7780.4320.7960.339
WGA0.0670.0170.1940.0140.0960.003
Figure 6: ROUGE-L (solid) and Cosine Similarity (dashed) on the merged DUET forget (left) and retain (right) splits across learning rates, for Llama (top), Qwen (middle), and Gemma (bottom).
Figure 6: ROUGE-L (solid) and Cosine Similarity (dashed) on the merged DUET forget (left) and retain (right) splits across learning rates, for Llama (top), Qwen (middle), and Gemma (bottom).
Table 6: Internal metrics at lr=10−4, averaged across Llama, Qwen, and Gemma. Directional arrows are shown in each section header. Underline = best among NPO, WGA, AdaPop. GA and GD values are reported for reference but indicate model collapse and are not ranked. Per-model breakdown in Appendix K.
AlgoΔLPΔRankHid.CosKL
DUET — forget split (ΔLP↓, ΔRank↑, Hid.Cos↓, KL↑)
AdaPop−160¯+53,760¯0.70242.5
GA†−3,106+106,8260.352479.6
GD†−5,917+104,1340.1921006.4
NPO−42+8,8240.8615.4
WGA−52−3,7140.72916.2
RWKU — forget split (ΔLP↓, ΔRank↑, Hid.Cos↓, KL↑)
AdaPop−23¯+4,169¯0.4849.1
GA†−6,083+96,2540.285953.1
GD†−6,842+96,8810.1751139.7
NPO−18+2,7390.8832.9
WGA−3.5−11,0100.5416.7
Retain split — DUET (ΔLP↑, ΔRank↓, Hid.Cos↑, KL↓)
AdaPop+21−13,7700.8756.5
GA†−2,750+103,1040.348511.4
GD†−251−8,7510.76048.2
NPO+21−13,6600.9107.5
WGA+23¯−13,930¯0.8807.5
Retain split — RWKU (ΔLP↑, ΔRank↓, Hid.Cos↑, KL↓)
AdaPop+33−12,1900.8836.2
GA†−6,106+98,8020.289955.8
GD†−540−3,1220.78397.1
NPO+31−11,7270.8697.4
WGA+34¯−12,202¯0.8756.8
Figure 7: ROUGE-L (solid) and Cosine Similarity (dashed) on the RWKU forget (left) and retain (right) splits across learning rates, for Llama (top), Qwen (middle), and Gemma (bottom).
Figure 7: ROUGE-L (solid) and Cosine Similarity (dashed) on the RWKU forget (left) and retain (right) splits across learning rates, for Llama (top), Qwen (middle), and Gemma (bottom).
Table 7: MMLU accuracy and HellaSwag (HS) normalised accuracy at lr=10−4. w/o MU = the original checkpoint, with no unlearning applied. Values averaged across DUET and RWKU checkpoints.
AlgoLlamaQwenGemma
MMLUHSMMLUHSMMLUHS
w/o MU0.650.730.710.680.470.64
AdaPop0.650.760.710.680.520.62
GA0.230.330.230.430.250.26
GD0.640.590.710.650.470.37
NPO0.650.720.700.680.510.63
WGA0.660.760.710.700.520.68
Figure 8: DUET forget and retain splits separated by popularity tier (Llama). Top row: rare facts. Bottom row: popular facts. The order-of-magnitude difference in the learning rate required to achieve comparable forgetting across the two tiers is direct evidence of the popularity gap.
Figure 8: DUET forget and retain splits separated by popularity tier (Llama). Top row: rare facts. Bottom row: popular facts. The order-of-magnitude difference in the learning rate required to achieve comparable forgetting across the two tiers is direct evidence of the popularity gap.
Table 8: AdaPop coefficient sensitivity. Each row perturbs a and/or b by ±20% from the analytically derived baseline. F. ↓ = forget ROUGE-L; R. ↑ = retain ROUGE-L. Δ values are signed differences from the baseline.
ConfigabF. ↓R. ↑ΔF.ΔR.
baseline58.700.79600.0450.96100
+a,+b70.440.95520.0380.975−0.007+0.014
+a,−b70.440.63680.0440.994−0.001+0.033
−a,+b46.960.95520.0240.920−0.021−0.041
−a,−b46.960.63680.0340.991−0.011+0.030
Figure 9: Forget (left) and retain (right) ROUGE-L Recall per training epoch on DUET, Llama-3.1-8B-Instruct, at lr=10−4.
Figure 9: Forget (left) and retain (right) ROUGE-L Recall per training epoch on DUET, Llama-3.1-8B-Instruct, at lr=10−4.
Table 9: Discordant cases between Wikidata score and LLM-judge scores on DUET. ROUGE-L: Llama-3.1-8B-Instruct recall before unlearning. Top group: Wikidata-popular facts recalled at ROUGE-L 1.0 despite low LLM-judged salience, indicating deep memorisation. Bottom group: Wikidata-rare facts with higher LLM scores but variable recall, consistent with shallower encoding.
QuestionAnswerWikidataLLMROUGE-L
Wikidata popular, LLM rare
What is the country of Ensenada?Mexico2113301.0
What is the country of Tourcoing?France2112401.0
What is the country of Qus?Egypt3763501.0
Wikidata rare, LLM popular
What is the instance of Hua Hin?seaside resort14890000.5
What is the instance of Weitra?municipality of Austria14150000.7
What is the located in the administrative territorial entity of Ollantaytambo?Urubamba Province16540001.0
Table 10: Agreement of each popularity proxy with measured corpus frequency on DUET, counted over Pile-train (383B tokens) with infini-gram. Pearson is computed in log space. The two proxies agree with each other at Spearman 0.596 and assign the same rare/popular label to 84% of facts.
ProxyPearsonSpearman
Wikidata score0.9700.839
LLM judge0.6770.659
Table 11: AdaPop under corrupted popularity scores (DUET, Llama-3.1-8B-Instruct, lr=10−4). Forget ROUGE-L is reported per popularity tier. Retain degrades by 0.04 even when every label is inverted.
Score noiseF. rare ↓F. pop. ↓R. ↑
none0.050.040.96
log-normal, σ=1.00.020.050.96
labels 50% swapped0.020.050.97
labels 100% swapped0.010.050.92
Table 12: ROUGE-L Recall for additional baselines, Llama-3.1-8B-Instruct at lr=10−4. w/o MU = the original checkpoint, with no unlearning applied; AdaPop reproduced from Table 1 for reference. F. ↓ = forget; R. ↑ = retain. Underline = best among non-collapsed methods. †Model collapse.
Bench.AlgorithmROUGE-L
F. ↓R. ↑
w/o MU0.9390.968
AdaPop0.0430.959
UNDIAL0.8840.998
RMU0.8820.998
PDU0.4760.954
NPO-SAM0.6840.970
SimNPO0.3400.999
SatImp0.1080.995
Adaptive RMU0.9330.963
AltPO0.1811.000
FLAT†0.0010.001
TPO0.3970.997
DUETCE-U†0.0000.000
w/o MU0.7550.827
AdaPop0.0780.972
UNDIAL0.5680.939
RMU0.8010.986
PDU0.2010.909
NPO-SAM0.5290.885
SimNPO0.5730.987
SatImp0.4170.985
Adaptive RMU0.1620.819
AltPO0.2000.988
FLAT†0.0040.003
TPO0.3410.986
RWKUCE-U†0.0000.000
Table 13: ROUGE-L Recall for additional baselines, Qwen2.5-7B-Instruct at lr=10−4. Notation as in Table 12.
Bench.AlgorithmROUGE-L
F. ↓R. ↑
w/o MU0.9210.886
AdaPop0.0560.950
UNDIAL0.7970.930
RMU0.7340.983
PDU0.6320.869
NPO-SAM0.2110.769
SimNPO0.2950.991
SatImp0.0700.985
Adaptive RMU0.5620.831
AltPO0.2470.999
FLAT†0.0000.001
TPO0.4600.999
DUETCE-U†0.0010.001
w/o MU0.5700.666
AdaPop0.0160.855
UNDIAL0.3130.668
RMU0.4070.922
PDU0.4550.710
NPO-SAM0.2980.596
SimNPO0.3380.934
SatImp0.1870.891
Adaptive RMU0.1180.466
AltPO0.1700.989
FLAT†0.0040.003
TPO0.3820.934
RWKUCE-U†0.0030.001
Table 14: ROUGE-L Recall for additional baselines, Gemma-7B-it at lr=10−4. Notation as in Table 12.
Bench.AlgorithmROUGE-L
F. ↓R. ↑
w/o MU0.8920.926
AdaPop0.0230.976
UNDIAL0.8190.980
RMU0.8510.992
PDU0.6770.934
NPO-SAM0.5390.777
SimNPO0.5810.997
SatImp0.0700.998
Adaptive RMU0.2930.828
AltPO0.2621.000
FLAT†0.0010.001
TPO0.8990.999
DUETCE-U†0.0000.000
w/o MU0.4710.551
AdaPop0.0340.948
UNDIAL0.4530.830
RMU0.5470.968
PDU0.4040.825
NPO-SAM0.2710.644
SimNPO0.3730.969
SatImp0.1020.964
Adaptive RMU0.0910.535
AltPO0.1720.989
FLAT†0.0050.004
TPO0.5940.971
RWKUCE-U†0.0000.000

研究结果

  • 面对改写过的提问,AdaPop泄露的被删信息比同类方法少约5倍;面对刻意刁难式的提问少约1.6倍。
  • 在三个模型(Llama、Qwen、Gemma)和两个基准(DUET、RWKU)、固定学习率条件下,在未崩溃的稳定方法中,AdaPop在全部六种模型-基准组合上被删内容的余弦相似度最低,在其中五种组合上ROUGE-L也最低。
  • 通用能力指标MMLU准确率与遗忘前模型相比差距保持在0.05以内。
  • 维基数据分数、大模型评分、以及基于Pile-train语料用infini-gram实测的频率三种知名度信号,在84%的事实上对稀有/热门的分类判断一致;即使把所有知名度标签完全反转,保留性能也只下降了0.04。

可应用场景

  • 因隐私、法律或安全原因需要从已部署语言模型中删除特定事实的服务场景
  • 利用维基数据等现成的知名度信号,自动校准不同知识删除力度的研究或工程实践
  • 设计检验被删信息是否会在改写或刁难式提问下重新浮现的遗忘效果评估流程

局限与待验证事项

  • 实验仅在三个7-8B规模模型(Llama-3.1-8B-Instruct、Qwen2.5-7B-Instruct、Gemma-7B-it)和两个基准(DUET、RWKU)上进行,更大规模或不同结构的模型未经验证。
  • 仅使用了LoRA微调,完全微调因参考已有研究结果被排除,未在本研究中单独验证。
  • 主要依赖的知名度信号是维基数据分数;在没有维基数据覆盖的事实上使用大模型评分作为替代信号,其在同一档次内排序事实的能力相对较弱。
  • 在Llama模型配合DUET基准的组合中,WGA方法在表层ROUGE-L指标上略微更低,是AdaPop总体优势中的一个例外。

为什么重要

随着AI系统因隐私、安全或合规原因需要删除特定信息的场景增多,关键在于这些信息是否真正从模型内部被清除,而不是仅仅在输出层面被掩盖。这项研究表明,利用事实的知名度信息可以实现更彻底的删除,同时更少损害其他需要保留的知识。

本文术语

  • 遗忘训练(unlearning) · 不重新训练整个模型,只把已训练模型中的特定知识去除的操作
  • 维基数据站点链接分数 · 根据一个维基数据条目被多少语言版本的维基百科链接,来衡量该事实的知名度
  • 双重上升控制器(dual-ascent controller) · 根据每个训练轮次观察到的知识损害程度,自动调节删除力度与保留惩罚之间平衡的机制
  • ROUGE-L · 衡量生成答案与正确答案词序重叠程度的指标,对待删除内容越低越好,对需保留内容越高越好
  • LLM-as-Judge · 让大语言模型对某个事实的知名程度打分,作为一种知名度信号

论文原文摘要(英文)

Popular facts are memorised more deeply during pretraining and resist removal longer than rare ones, yet existing LLM unlearning methods apply uniform gradient pressure regardless of training-data frequency. We propose the AdaPop (Adaptive Popularity) method, which combines local token confidence with a per-fact popularity-dependent exponent derived from an external proxy (e.g., Wikidata sitelinks, LLM-as-Judge), and automates the forget-retain balance via a dual-ascent controller that adjusts the retain penalty each epoch. Across three model families and two benchmarks, AdaPop leaks ~5x less forgotten content than competing methods under paraphrased queries and ~1.6x less under adversarial reformulations. We support our analysis with internal metrics: under our method, forget-set hidden states move further from the pre-unlearning model's states than under other methods, while retain-set representations remain close.

作者 · Anna Borisiuk

在 arXiv 阅读

最新论文

全部论文 →

METAL MEDIA 最新报道

图片来源: Anna Borisiuk et al., arXiv:2608.14229, cc-by-nc-nd-4.0