arXiv:2506.02519cs.CL2025-06ACL被引 3

让小模型通过优选推理路径协作,无需依赖大模型即可提升推理能力。

Learning Together to Perform Better: Teaching Small-Scale LLMs to Collaborate via Preferential Rationale Tuning

  • 用多路径推理+偏好优化,让小模型自主选择最佳解题思路。
  • 在5个数据集上超越多个基线,1B到8B参数模型均有效。
  • 适合无权使用大模型的商业场景,可扩展至不同模型家族。

大型语言模型(如GPT-4)展现出通过生成逐步推理过程解决复杂问题的能力。以往研究利用这一能力提升较小、成本更低的语言模型(如70亿参数)。然而,由于大型模型预训练数据缺乏透明性,涉及版权与法律问题,使其难以在商业环境中使用。目前较少关注不依赖大模型知识蒸馏来提升小模型内在推理能力的方法。为此,我们提出COLLATE——一种可训练框架,通过从多样化推理路径中筛选出能最大化真实答案概率的推理过程,来优化小模型输出。该框架使同一模型多次生成不同行为的推理路径,并通过偏好优化训练模型选择最优方案。COLLATE在数学问题求解、自然语言推理和常识推理三个领域的5个数据集上优于多个可训练及提示基线方法。我们验证了其在不同模型家族、10亿至80亿参数规模下的有效性,并通过消融实验证明了由下游任务引导的多推理提供者带来的优势。代码已公开于https://github.com/Sohanpatnaik106/collate。

原文摘要 · Abstract (English)

LLMssuch as GPT-4 have shown a remarkable ability to solve complex questions by generating step-by-step rationales. Prior works have utilized this capability to improve smaller and cheaper LMs (say, with 7B parameters). However, various practical constraints, such as copyright and legal issues, owing to lack of transparency in the pre-training data of large (often closed) models, prevent their use in commercial settings. Little focus has been given to improving the innate reasoning ability of smaller models without distilling information from larger LLMs. To address this, we propose COLLATE, a trainable framework that tunes a (small) LLM to generate those outputs from a pool of diverse rationales that selectively improves the downstream task. COLLATE enforces multiple instances of the same LLM to exhibit distinct behavior and employs them to generate rationales to obtain diverse outputs. The LLM is then tuned via preference optimization to choose the candidate rationale which maximizes the likelihood of ground-truth answer. COLLATE outperforms several trainable and prompting baselines on 5 datasets across 3 domains: maths problem solving, natural language inference, and commonsense reasoning. We show the eff icacy of COLLATE on LLMs from different model families across varying parameter scales (1B to 8B) and demonstrate the benefit of multiple rationale providers guided by the end task through ablations. Code is released here (https://github.com/Sohanpatnaik106/collate).

小模型推理偏好优化多路径协作

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