让小模型先学大模型的阅读基本功,再做任务表现更优。
Basic Reading Distillation
- 在无关下游任务的通用文本上训练小模型模仿大模型阅读行为
- 小模型在多个任务上超越或媲美20倍大的LLM
- 适合想高效部署大模型能力的工程应用与教育场景
大型语言模型(LLMs)在自然语言处理中表现出色,但计算资源消耗高,限制了实际部署。知识蒸馏和任务蒸馏虽能缩小模型规模,但均忽略对小模型进行与下游任务无关的通用文本基础阅读训练。本文提出基础阅读蒸馏(BRD),让小模型在每句话上模仿大模型的基础阅读行为,如命名实体识别、提问与回答。经过这种基础教育后,小模型在语言推理基准和BIG-bench任务上表现优异,可超越或媲美超过20倍大的LLM。分析显示,BRD有效改变小模型的概率分布,且与知识蒸馏和任务蒸馏具有正交性。
原文摘要 · Abstract (English)
Large language models (LLMs) have demonstrated remarkable abilities in various natural language processing areas, but they demand high computation resources which limits their deployment in real-world. Distillation is one technique to solve this problem through either knowledge distillation or task distillation. Both distillation approaches train small models to imitate specific features of LLMs, but they all neglect basic reading education for small models on generic texts that are \emph{unrelated} to downstream tasks. In this paper, we propose basic reading distillation (BRD) which educates a small model to imitate LLMs basic reading behaviors, such as named entity recognition, question raising and answering, on each sentence. After such basic education, we apply the small model on various tasks including language inference benchmarks and BIG-bench tasks. It shows that the small model can outperform or perform comparable to over 20x bigger LLMs. Analysis reveals that BRD effectively influences the probability distribution of the small model, and has orthogonality to either knowledge distillation or task distillation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。