让大模型反向思考,推理能力显著提升
Reverse Thinking Makes LLMs Stronger Reasoners
- 用教师模型生成正反向推理数据,训练学生模型双向推理
- 在12个数据集上平均提升13.53%,仅用10%数据超精调方法
- 适合需要强推理且数据稀缺的场景
反向思维在人类推理中至关重要。我们提出逆向增强思维(RevThink)框架,通过数据增强和多任务学习目标,使大语言模型具备反向推理能力。具体地,从教师模型收集结构化正向-反向推理数据,包含原始问题、正向推理、反向问题和反向推理。采用三项训练目标:(a) 从问题生成正向推理,(b) 从问题生成反向问题,(c) 从反向问题生成反向推理。在12个涵盖常识、数学和逻辑推理的数据集上,学生模型相比零样本性能平均提升13.53%,优于最强知识蒸馏基线6.84%。此外,仅使用10%原始正向推理数据,即超越使用10倍数据的标准微调方法,展现出优异的样本效率与跨分布泛化能力。
原文摘要 · Abstract (English)
Reverse thinking plays a crucial role in human reasoning. Humans can reason not only from a problem to a solution but also in reverse, i.e., start from the solution and reason towards the problem. This often enhances overall reasoning performance as it enables consistency checks between their forward and backward thinking. To enable Large Language Models (LLMs) to perform reverse thinking, we introduce Reverse-Enhanced Thinking (RevThink), a framework composed of data augmentation and learning objectives. In RevThink, we augment the dataset by collecting structured forward-backward reasoning from a teacher model, consisting of: (1) the original question, (2) forward reasoning, (3) backward question, and (4) backward reasoning. We then employ three objectives to train a smaller student model in a multi-task learning fashion: (a) generate forward reasoning from a question, (b) generate a backward question from a question, and (c) generate backward reasoning from the backward question. Experiments across 12 datasets covering commonsense, math, and logical reasoning show an average 13.53% improvement over the student model's zero-shot performance and a 6.84% improvement over the strongest knowledge distillation baselines. Moreover, our method demonstrates sample efficiency -- using only 10% of the correct forward reasoning from the training data, it outperforms a standard fine-tuning method trained on 10x more forward reasoning. RevThink also exhibits strong generalization to out-of-distribution held-out datasets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。