用时序差分学习压缩大模型,只关注高频词提升效率
Language Model Distillation: A Temporal Difference Imitation Learning Perspective
- 基于教师模型分布稀疏性,仅在高频词上进行时序差分学习
- 在小模型上实现与大模型相当的性能,推理速度更快
- 适合需要轻量级部署的NLP应用,如移动端或实时系统
大型语言模型在众多自然语言任务中取得显著进展,但其庞大的规模常带来高昂的计算成本。模型蒸馏已成为将大模型压缩为更小、更高效模型的常用方法。现有许多语言模型蒸馏方法可从模仿学习或逆强化学习的角度理解为行为克隆。这一视角启发了后续研究,采用(逆)强化学习技术,包括行为克隆变体和时序差分学习方法。本文不提出新的具体时序差分算法,而是通过利用教师模型的分布稀疏性,构建一个通用的时序差分蒸馏框架。研究表明,语言模型通常将大部分概率质量集中在一小部分标记上。基于此,我们设计了一个在缩减动作空间(词汇子集)上运行的时序差分学习框架,并展示了实用算法的推导过程及性能提升效果。
原文摘要 · Abstract (English)
Large language models have led to significant progress across many NLP tasks, although their massive sizes often incur substantial computational costs. Distillation has become a common practice to compress these large and highly capable models into smaller, more efficient ones. Many existing language model distillation methods can be viewed as behavior cloning from the perspective of imitation learning or inverse reinforcement learning. This viewpoint has inspired subsequent studies that leverage (inverse) reinforcement learning techniques, including variations of behavior cloning and temporal difference learning methods. Rather than proposing yet another specific temporal difference method, we introduce a general framework for temporal difference-based distillation by exploiting the distributional sparsity of the teacher model. Specifically, it is often observed that language models assign most probability mass to a small subset of tokens. Motivated by this observation, we design a temporal difference learning framework that operates on a reduced action space (a subset of vocabulary), and demonstrate how practical algorithms can be derived and the resulting performance improvements.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。