arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2505.15134cs.LGcs.AI
收藏

熵最小化在大语言模型推理中令人惊叹的有效性

The Unreasonable Effectiveness of Entropy Minimization in LLM Reasoning

University of Illinois Urbana-Champaign(伊利诺伊大学厄巴纳-香槟分校)

Shivam Agarwal, Zimin Zhang, Lifan Yuan, Jiawei Han, Hao Peng

AI总结:

本研究提出三种熵最小化方法,无需标注数据甚至无需参数更新即可有效激发预训练大语言模型的潜在推理能力,在多个基准上取得媲美甚至超越强基线与专有模型的性能。

摘要:

熵最小化(EM)训练模型将更多概率质量集中在其置信度最高的输出上。我们证明,仅使用这一简单目标,无需任何标注数据,就能大幅提升大语言模型(LLM)在高难度数学、物理和编程任务上的性能。我们探索了三种方法:(1)EM-FT:类似指令微调的方式最小化词元级熵,但训练数据为模型自身生成的无标注输出;(2)EM-RL:以负熵作为唯一最大化奖励的强化学习方法;(3)EM-INF:推理阶段的logit调整方法,无需任何训练数据或参数更新即可降低熵。在Qwen-7B模型上,无任何标注数据的EM-RL取得了与使用6万条标注样本训练的强基线RL方法(如GRPO和RLOO)相当甚至更优的性能。此外,EM-INF使Qwen-32B在高难度的SciCode基准上达到或超过了GPT-4o、Claude 3 Opus和Gemini 1.5 Pro等专有模型的性能,同时效率比自一致性和序列精炼方法高3倍。我们的发现表明,许多预训练LLM具备此前被低估的推理能力,仅通过熵最小化、无需任何标注数据甚至无需参数更新就能有效激发这些能力。

英文摘要:

Entropy minimization (EM) trains the model to concentrate even more probability mass on its most confident outputs. We show that this simple objective alone, without any labeled data, can substantially improve large language models' (LLMs) performance on challenging math, physics, and coding tasks. We explore three approaches: (1) EM-FT minimizes token-level entropy similarly to instruction finetuning, but on unlabeled outputs drawn from the model; (2) EM-RL: reinforcement learning with negative entropy as the only reward to maximize; (3) EM-INF: inference-time logit adjustment to reduce entropy without any training data or parameter updates. On Qwen-7B, EM-RL, without any labeled data, achieves comparable or better performance than strong RL baselines such as GRPO and RLOO that are trained on 60K labeled examples. Furthermore, EM-INF enables Qwen-32B to match or exceed the performance of proprietary models like GPT-4o, Claude 3 Opus, and Gemini 1.5 Pro on the challenging SciCode benchmark, while being 3x more efficient than self-consistency and sequential refinement. Our findings reveal that many pretrained LLMs possess previously underappreciated reasoning capabilities that can be effectively elicited through entropy minimization alone, without any labeled data or even any parameter updates.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑