让小模型互相讨论,通过选择性优化提升推理能力。
It Helps to Take a Second Opinion: Teaching Smaller LLMs to Deliberate Mutually via Selective Rationale Optimisation
- 两个变体小模型相互生成并优化推理过程,增强多样性。
- 在5个数据集上最高提升5%,优于单模型自反思。
- 适用于4B至14B参数的多种模型,支持商用部署。
大型语言模型(如GPT-4)可通过逐步生成和自我修正推理链来处理复杂任务。较小的语言模型(SLMs,通常<13B参数)常通过知识蒸馏使用大模型生成的数据进行改进。然而,由于API成本、版权、法律与伦理政策等实际限制,商业应用中难以使用大型(常为黑箱)模型训练小模型。现有方法在提升小模型自主探索推理路径并自我评估方面进展有限。为此,我们提出COALITION——一种可训练框架,使同一小模型的两个变体相互交互,协同生成并优化针对目标任务的推理链。两变体在生成与优化阶段表现出不同行为,以产生多样化的候选推理链。模型通过选择性推理优化(SRO)训练,偏好生成能最大化获得真实答案概率的推理链。推理时,控制器选择合适变体完成推理生成与优化。在涵盖数学问题、常识推理和自然语言蕴含的五个数据集上,COALITION相比多个基线最高提升5%。消融实验表明,双变体间交叉沟通优于单模型自反思。我们还验证了COALITION在4B至14B参数量级及Mistral、Llama、Qwen、Phi等多种模型家族中的适用性。代码已开源:https://github.com/Sohanpatnaik106/coalition。
原文摘要 · Abstract (English)
Very large language models (LLMs) such as GPT-4 have shown the ability to handle complex tasks by generating and self-refining step-by-step rationales. Smaller language models (SLMs), typically with < 13B parameters, have been improved by using the data generated from very-large LMs through knowledge distillation. However, various practical constraints such as API costs, copyright, legal and ethical policies restrict using large (often opaque) models to train smaller models for commercial use. Limited success has been achieved at improving the ability of an SLM to explore the space of possible rationales and evaluate them by itself through self-deliberation. To address this, we propose COALITION, a trainable framework that facilitates interaction between two variants of the same SLM and trains them to generate and refine rationales optimized for the end-task. The variants exhibit different behaviors to produce a set of diverse candidate rationales during the generation and refinement steps. The model is then trained via Selective Rationale Optimization (SRO) to prefer generating rationale candidates that maximize the likelihood of producing the ground-truth answer. During inference, COALITION employs a controller to select the suitable variant for generating and refining the rationales. On five different datasets covering mathematical problems, commonsense reasoning, and natural language inference, COALITION outperforms several baselines by up to 5%. Our ablation studies reveal that cross-communication between the two variants performs better than using the single model to self-refine the rationales. We also demonstrate the applicability of COALITION for LMs of varying scales (4B to 14B parameters) and model families (Mistral, Llama, Qwen, Phi). We release the code for this work at https://github.com/Sohanpatnaik106/coalition.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。