pruning后语言模型易重复生成,该文提出新方法缓解此问题
FOCUS & RePAIR: Mitigating Text Degeneration via Token-Level Guidance for Pruned Large Language Models
- 将解码视为动态过程,分析重复成因
- 提出两种新目标函数,减少重复并提升生成质量
- 适合优化压缩后的大型语言模型生成效果
剪枝是压缩大语言模型的实用方法,但会加剧文本退化,尤其在困惑度和任务准确率基本不变的情况下仍出现重复循环。本文通过将解码视为进入并滞留于少数重复上下文的动态过程,对退化现象进行分层分析,将其分解为进入风险与持续风险,并发现持续性由采样集中合理候选词的‘逃逸质量’决定。基于此,提出两种剪枝后微调的词元级引导目标:FOCUS通过重加权蒸馏聚焦高置信度教师区域以抑制泄漏;RePAIR采用以起始点为中心的正负延续对与边际损失,促进合理替代项,防止过早陷入重复循环。在开放式续写和指令生成任务上的实验表明,两种方法均能稳定降低重复率并提升生成质量。
原文摘要 · Abstract (English)
Pruning is a practical approach to compress large language models (LLMs), but it can amplify text degeneration, especially repetition loops, even when perplexity and task accuracy remain largely unchanged. In this work, we present a token-level analysis of this failure mode by viewing decoding as a dynamical process that enters and persists in a small set of recurrent contexts. Our analysis decomposes degeneration into loop entry risk and loop persistence, and shows that persistence is controlled by the escape mass assigned to plausible alternatives within the token sampling set. Motivated by these findings, we propose two token-level guidance objectives for post-pruning fine-tuning. FOCUS reweights distillation toward high-confidence teacher regions to suppress leakage, while RePAIR uses onset-centered positive/negative continuation pairs with a margin loss to promote plausible alternatives and prevent early commitment to repetition loops. Experiments on open-ended continuation and instruction-based generation show that both methods consistently reduce repetition and improve generation quality.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。