解决大模型与小模型协作中的双向知识迁移难题。
Federated Reasoning Distillation Framework with Model Learnability-Aware Data Allocation
- 根据模型可学习性差距动态分配高价值数据样本。
- 通过对比学习对齐推理路径概率,提升小模型本地适应能力。
- 适配现有框架,适合资源受限场景下的模型协作。
数据分配在联邦大语言模型(LLM)与小语言模型(SLM)的推理协同中至关重要。然而,现有方法未能解决协同中的一个未充分探索的挑战:双向模型可学习性差距——客户端的SLM无法识别符合其可学习性约束的高回报样本,难以从LLM有效获取知识;而LLM也难以选择能带来新知识的样本。此外,这些协同框架还面临领域无关的推理迁移问题,现有方法无法灵活适应本地领域数据,导致SLM难以从通用LLM中有效习得逐步推理能力。为此,我们提出LaDa:一种基于模型可学习性感知的数据分配的联邦推理蒸馏框架。该框架引入可学习性感知的数据过滤器,根据每个SLM与LLM对之间的可学习性差距,自适应分配高回报样本,促进双向知识迁移。我们进一步设计了一种领域自适应推理蒸馏方法,通过在筛选出的高回报样本上对联合推理路径概率进行对比蒸馏学习,使SLM能够捕捉本地数据分布下的底层推理模式。LaDa可作为插件模块集成至现有协同框架中,依据模型可学习性差距动态调整知识迁移策略。
原文摘要 · Abstract (English)
Data allocation plays a critical role in federated large language model (LLM) and small language models (SLMs) reasoning collaboration. Nevertheless, existing data allocation methods fail to address an under-explored challenge in collaboration: bidirectional model learnability gap, where client-side SLMs cannot identify high-reward samples matching their learnability constraints for effective knowledge transfer from LLMs, while LLMs struggle to select samples contributing novel knowledge beyond their existing data. Furthermore, these collaboration frameworks face another key challenge: domain-agnostic reasoning transfer, where existing reasoning transfer methods fail to flexibly adapt to the local domain data, preventing SLMs from effectively acquiring step-by-step reasoning abilities within from general LLM. To address these challenges, we propose LaDa, a federated reasoning distillation framework with model learnability-aware data allocation. It introduces a model learnability-aware data filter that adaptively allocates high-reward samples based on the learnability gap between each SLM and LLM pair, effectively facilitating bidirectional knowledge transfer. We further design a domain adaptive reasoning distillation method that aligns joint probabilities of reasoning paths on filtered high-reward samples through contrastive distillation learning between SLM and LLM, enabling SLM to capture underlying reasoning patterns under local data distribution. LaDa operates as a plug-in module for existing collaboration frameworks, adapting knowledge transfer based on model learnability gaps.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。