小模型嵌入坍缩致性能下降,提出散度损失有效改善
Dispersion Loss Counteracts Embedding Condensation and Improves Generalization in Small Language Models
- 设计散度损失,强制小模型嵌入分散分布
- 实验显示该方法在10个基准上显著提效
- 适合追求高效小模型的开发者参考
大规模语言模型(LLM)通过不断增加参数量实现卓越性能,但扩展带来高昂计算成本。为理解模型规模效应,我们研究了大模型与小模型在表征上的差异,目标是将大模型的表征特性复制到小模型中。我们观察到一种几何现象,称为嵌入坍缩,即部分模型中的词元嵌入会聚集于狭窄的锥形子空间。通过对多个Transformer架构系统分析发现,小型模型如GPT2和Qwen3-0.6B存在严重坍缩,而大型模型如GPT2-xl和Qwen3-32B则更具抵抗力。此外,知识蒸馏无法可靠缓解此现象。为此,我们提出一种散度损失,在训练中显式鼓励嵌入分散。实验表明,该方法能有效缓解坍缩,恢复大模型中的分散模式,并在10个基准测试中提升性能。我们认为本工作为不增加参数量即可改进小规模Transformer提供了可解释的路径。
原文摘要 · Abstract (English)
Large language models (LLMs) achieve remarkable performance through ever-increasing parameter counts, but scaling incurs steep computational costs. To better understand LLM scaling, we study representational differences between LLMs and their smaller counterparts, with the goal of replicating the representational qualities of larger models in smaller models. We observe a geometric phenomenon which we term $\textbf{embedding condensation}$, where token embeddings collapse into a narrow cone-like subspace in some language models. Through systematic analyses across multiple Transformer families, we show that small models such as $\texttt{GPT2}$ and $\texttt{Qwen3-0.6B}$ exhibit severe condensation, whereas larger models such as $\texttt{GPT2-xl}$ and $\texttt{Qwen3-32B}$ are more resistant to this phenomenon. Additional observations show that embedding condensation is not reliably mitigated by knowledge distillation from larger models. To fight against it, we formulate a dispersion loss that explicitly encourages embedding dispersion during training. Experiments demonstrate that it mitigates condensation, recovers dispersion patterns seen in larger models, and yields performance gains across 10 benchmarks. We believe this work offers a principled path toward improving smaller Transformers without additional parameters.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。