arXiv:2509.25837cs.LGcs.AI2025-09被引 5

用新方法让小模型更精准模仿大模型,提升推理效率

Distillation of Large Language Models via Concrete Score Matching

  • 基于离散分数匹配设计新目标,保留大模型的原始逻辑差异
  • 在多个大模型上测试,小模型性能超越现有方法,且多样性更好
  • 适合需要高效部署大模型的应用场景,如移动端或实时系统

大语言模型性能优异但部署成本高,知识蒸馏成为高效推理的关键。现有蒸馏方法多通过Softmax匹配师生概率,导致重要logit信息模糊;直接蒸馏logit虽缓解此问题,却忽略logit平移不变性,限制解空间。本文提出混凝土分数蒸馏(CSD),一种离散分数匹配目标,克服Softmax平滑与解空间受限双重缺陷。我们解决了自回归大模型中离散分数匹配的训练不稳定性与二次复杂度问题,使CSD能灵活加权所有词表对之间的相对logit差异。框架内包含模式聚焦与模式覆盖两类实例。在GPT-2-1.5B、OpenLLaMA-7B和GEMMA-7B-IT上进行任务无关指令跟随与任务特定蒸馏实验,结果表明CSD持续优于近期蒸馏方法,在保真度与多样性间取得良好权衡,且与在线策略技术结合可带来互补增益,证明其在大模型蒸馏中的可扩展性与有效性。

原文摘要 · Abstract (English)

Large language models (LLMs) deliver remarkable performance but are costly to deploy, motivating knowledge distillation (KD) for efficient inference. Existing KD objectives typically match student and teacher probabilities via softmax, which blurs valuable logit information. While direct logit distillation (DLD) mitigates softmax smoothing, it fails to account for logit shift invariance, thereby restricting the solution space. We propose Concrete Score Distillation (CSD), a discrete score-matching objective that overcomes both softmax-induced smoothing and restrictions on the optimal solution set. We resolve the training instability and quadratic complexity of discrete score-matching in autoregressive LLMs, and the resulting CSD objective aligns relative logit differences across all vocabulary pairs between student and teacher with flexible weighting. We provide both mode-seeking and mode-covering instances within our framework and evaluate CSD on task-agnostic instruction-following and task-specific distillation using GPT-2-1.5B, OpenLLaMA-7B, and GEMMA-7B-IT. Experiments show that CSD consistently surpasses recent KD objectives, achieves favorable fidelity-diversity trade-offs, and yields complementary gains when combined with on-policy techniques, demonstrating its scalability and effectiveness for LLM distillation. Code: https://github.com/aailab-kaist/CSD.

知识蒸馏大模型压缩生成模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。