跳转至

AdaThinking-E: One-Token Entropy Regulation for Adaptive Thinking

会议: ECCV2026
Paper: https://eccv.ecva.net/virtual/2026/poster/4501
PDF: https://media.eventhosts.cc/Conferences/ECCV2026/pdfs/6246.pdf
代码: https://github.com/PriNing/AdaThinking-E
领域: 多模态推理 / 文档理解
关键词: 自适应思考、模式切换、熵课程、奖励归因、文档问答

一句话总结

AdaThinking-E 将“是否思考”的首 token 与后续答案分开优化,用先高熵探索、后低熵收敛的课程学习文档问答中的模式选择,使 Qwen2.5 系模型在 OCR-Reasoning 上达到 26.5,同时在 CharXivRQ 上仅用平均 72 个输出 token 获得接近 R-4B 的表现。

研究背景与动机

文档问答并不总需要长推理链:从抬头读取发文部门与结合图表做计算,消耗的推理资源本应不同。统一开启思考会让简单抽取产生冗长输出,甚至引入原图没有支持的推断;统一关闭思考又会损害复杂计算。提示词开关把决定权留给用户,而预先标注题目难度、训练预算预测器,则需要外部规则或额外标注。

本文更具体的问题是:即使冷启动已经教会模型两种输出模式,直接用同一套强化学习奖励更新整条序列,也未必能学会合理选择。作者观察到 GRPO 和 DAPO 在这一设置下快速偏向“不思考”,并将其归因于模式决策与内容生成混在一起优化、以及长度相关的奖励归因问题。这是该实验设置下的观察,不宜推导成所有 GRPO 训练必然坍塌。

模式由输出开头的特殊 token 决定,因此真正需要探索的是这个局部决策,而不是让整条推理链始终保持高随机性。核心 idea:先给同一道题提供两种可用回答方式,再只在模式切换 token 上控制探索与收敛,并用各模式的答对情况校正选择,让其余 token 专注于答案质量。

方法详解

整体框架

输入是文档图像和问题,输出以 /no_think 或 /to_think 开头,之后是 <think>...</think><answer>...</answer>;不思考时思考标签内部为空。训练分为 AdaThinking-Doc 冷启动和强化学习两个阶段,推理时则由训练后的模型自行生成开关,无需在线运行额外的难度分类器。

强化学习对同一问题采样一组回答,统计模式、正确性和首 token 分布。首 token 接收熵课程与决策反馈形成的信号,其余 token 接收格式与准确率信号。下面画的是训练数据流,不是推理时必须运行的多个独立模型。

%%{init: {'flowchart': {'rankSpacing': 24, 'nodeSpacing': 28, 'padding': 6, 'wrappingWidth': 400}}}%%
flowchart TD
Input["公开训练集<br/>文档图像与问题"] --> Cold["双模式冷启动"]
Cold --> Sample["同题分组采样"]
Sample --> Entropy["首 token 熵课程"]
Entropy --> Feedback["决策反馈"]
Sample --> Quality["响应质量奖励"]
Feedback --> Update["按 token 角色更新模型"]
Quality --> Update

关键设计

1. 双模式冷启动:学习两种能力,而不是预先指定何时使用

AdaThinking-Doc 从公开数据集的训练集组织文档抽取与复杂推理样本。对每个数据点都构造思考和不思考两种回答,而不是把简单题只标成不思考、难题只标成思考。思考轨迹由 Seed1.5-VL 生成,再由 GPT-4o-mini 检查逻辑合理性;两种回答使用相同的思考和答案标签结构,只在模式 token 与思考内容上不同。

这样做的目的不是保证每题两种模式都一样好,而是避免强化学习开始之前某一分支已经不会生成。冷启动提供探索所需的行为覆盖,后续再用实际答对情况决定哪些输入值得思考。教师生成和检查仍会影响数据质量,因此“不预标难度”并不意味着训练数据完全没有外部影响。缓存未包含补充材料中的数据分布,不能据此填写数据集规模、各来源占比或过滤通过率。

2. 首 token 熵课程:只对模式选择安排先探索后收敛

对每个问题按当前策略采样 \(G\) 条输出,首位置从模型预测分布中选择思考或不思考。作者用这两个模式概率的归一化熵描述选择的不确定性:两者接近时熵高,一者占优时熵低。该熵是模式决策分布的性质,不是最终答案是否正确的置信度,也不是整条生成序列的平均熵。

早期训练奖励高熵,让两种模式都有机会进入采样组;后期逐渐奖励低熵,让模型对同类输入形成稳定选择。转换系数 \(\alpha_k\) 随训练步数变化,总步数为 \(K\),\(\gamma\) 控制转折发生的相对位置,\(\beta\) 控制过渡陡峭程度。推理时不再执行这套课程,模型直接使用学到的首 token 分布。

熵奖励还包含鼓励两个合法模式 token 的总概率接近 1 的项,防止首位置生成其他标识。这里有两个不同问题:先保证输出的是合法开关,再让合法开关内部的分布从探索转向确定;单纯降低熵可能只会过早锁定一个错误分支。

缓存的原式 (1) 至 (4) 存在不同程度的提取损坏,尤其总目标、熵求和、调度函数和完整熵奖励无法逐符号核实。本笔记不猜写这些闭式表达,保留正文可确认的变量含义、作用位置和调度方向;下文仅列出仍可辨读的决策权重关系。

3. 决策反馈:用分组正确性给熵收敛指定方向

熵本身只告诉模型“选择要多确定”,没有说明“应该确定地选哪种模式”。作者因此把每条 rollout 按模式和最终答案是否正确分类,得到模式 \(m\) 的出现比例 \(r_m\) 与组内准确率 \(acc_m\)。准确率由 \(\epsilon_l\) 和 \(\epsilon_h\) 分成低、中、高三档,供准确率引导项 \(A\) 使用:两种模式都低准确率时鼓励思考;两种都在中档或都在高档时鼓励不思考。

另一项倾向引导 \(P\) 处理小样本导致的误导。例如思考模式答对 13/15,不思考模式只采到一次却答对 1/1,直接比较准确率会认为不思考更好,但这个判断可能只是采样噪声。当模式比例差 \(|r_{tk}-r_{ntk}|>\Delta_r\)、且两种模式均高准确率时,\(P\) 引导模型保留已有倾向。正文给出的权重关系为:

\[ w_m=\frac{acc_m}{acc_{\neg m}}+A(acc,m)+P(acc,m,\Delta_r). \]

其中 \(\neg m\) 表示另一种模式。该权重用于调节模式 token 的熵奖励,并不是奖励所有响应 token 多想或少想。\(A\)、\(P\) 的分段取值和边界处理被放到补充材料,而本缓存没有这些内容,尤其不能自行添加分母为零时的平滑常数。由此也能看出,这不是完全去除人为偏好的方法:它去掉了外部逐题难度标签,但仍内置“都能答对时更倾向省略思考”等规则。

4. 响应质量奖励:不要让答案内容替模式决策承担探索压力

首 token 之外的响应 token 使用内容相关的优势估计,目标是生成有效且正确的回答,不因当前选了哪种模式而改变答案标准。格式奖励 \(R_f\) 对合法模板给 1,否则给 0;准确率奖励 \(R_{acc}\) 对最终答案匹配真值给 1,否则给 0。格式约束尤其重要,因为首位置与后续位置承担不同角色,格式漂移会破坏这种分工。

这是一种奖励归因的拆分,而不是增加一个独立路由网络:同一模型的首 token 学习模式决策,后续 token 学习在给定模式下作答。它也不直接监督推理过程的每一步是否正确,最终答案奖励仍可能接受不忠实的思考轨迹。正文将具体损失算子与实现细节指向补充材料,因此这里不补写未经核验的 clipping、KL 或两种质量奖励的组合系数。

损失函数 / 训练策略

冷启动采用 SWIFT,学习率 1e-6、batch size 4、训练 1 epoch。强化学习采用 VeRL,学习率 1e-6、batch size 32,每个问题采样 16 条回答;正文同时报告 1 epoch 和 2000 steps,硬件为 8 张 NVIDIA A100。由于缓存缺少数据规模和补充材料,不能进一步推断两个训练长度描述的换算关系。

训练段将基座写作 QwenVL-2.5-7B 和 QwenVL-3-8B,主结果表的原始 Qwen2.5 基线行却写成 QwenVL2.5-8B,命名存在不一致。以下将本文模型明确记作 Qwen2.5 系 7B 与 Qwen3 系 8B,不把表中命名差异默默改成新的事实。所有消融均使用 Qwen2.5-VL-7B。

评测使用 VLMEvalKit,需要大模型判断的任务由 GPT-4o-mini 评估。固定 NT/TK 的结果通过在输入阶段主动插入模式 token 得到,ATK 则让模型自己选择。比较输出长度时,作者给所有问题加入只用单词或短语回答的要求,以减少答案正文长度的差异,但这不等同于控制了所有模型的训练预算与解码成本。

实验关键数据

主实验

下表选取原表 1 的五项基准,保留论文报告的分数尺度,数值越高越好;不同基准评分协议不同,不跨列求平均,也不把所有分数笼统称作同一种准确率。NT 为不思考,TK 为思考,ATK 为自适应思考。

模型 / 模式 DocVQA ChartQA CharXivRQ CharXivDQ OCR-Reasoning
R-4B / ATK 94.1 87.2 56.8 82.9 22.2
AdaThinking-E Qwen2.5 7B / NT 96.3 88.4 55.2 82.6 21.7
AdaThinking-E Qwen2.5 7B / TK 95.1 88.5 56.2 83.3 26.1
AdaThinking-E Qwen2.5 7B / ATK 96.3 89.1 56.7 83.5 26.5
Qwen3 8B / TK 95.3 88.6 53.0 85.9 48.4
AdaThinking-E Qwen3 8B / ATK 96.4 90.1 57.3 86.6 49.7

Qwen2.5 系自适应模型在 OCR-Reasoning 上比自身固定 TK 高 0.4 分,比 R-4B 高 4.3 分;但 CharXivRQ 的 56.7 仍略低于 R-4B 的 56.8,因此应描述为接近,而不是全面超过。跨模型比较还混有模型规模和训练来源差异,自身三种模式的比较更直接反映模式选择效果。

效率证据来自正文对图 3 的说明:CharXivRQ 上本文思考比例为 27%、平均输出 72 token,R-4B 为 82% 和 366 token;CharXivDQ 上思考比例分别为 11% 和 45%。OCR-Reasoning 上本文自适应模式的思考比例约 62%,相对自身 TK 少生成平均 88 token;这里的 88 是差值,不能误写成总输出长度。

消融实验

原表 2 的决策反馈消融同时报告分数与思考比例,不能只看“更少思考”就判断更好。以下三行均为 Qwen2.5-VL-7B 设置。

决策反馈配置 ChartQA 分数 ChartQA 思考率 (%) OCR-Reasoning 分数 OCR-Reasoning 思考率 (%)
去掉 A、P,只用准确率反馈 88.1 39.9 25.7 83.4
A、P;阈值 0.3/0.7,比例差阈值 0.625 88.4 3.4 25.9 59.7
A、P;阈值 0.3/0.7,比例差阈值 0.375 89.1 24.3 26.5 61.9

最后一行相对去掉 A、P,提高 ChartQA 1.0 分、OCR-Reasoning 0.8 分,同时分别减少思考率 15.6 和 21.5 个百分点。比例差阈值提高到 0.625 虽使 ChartQA 更少思考,却降到 88.4,说明倾向判定会实质改变效率与效果的取舍。

原表 3 检验熵课程,以下保留冷启动、两种固定熵方向以及主配置,避免把“动态课程有效”与“只要低熵就有效”混为一谈。

熵配置 ChartQA 分数 ChartQA 思考率 (%) OCR-Reasoning 分数 OCR-Reasoning 思考率 (%)
仅冷启动 87.6 9.2 21.6 10.4
始终鼓励高熵,gamma=1.0 88.1 75.6 23.7 48.4
始终鼓励低熵,gamma=0.0 87.8 0.0 18.7 0.0
动态课程,beta=10,gamma=0.4 89.1 24.3 26.5 61.9

关键发现

  • 始终低熵时两项任务思考率均为 0.0%,OCR-Reasoning 比动态课程低 7.8 分。这直接支持“需要先探索”的论点,但不是所有设置中都具有同样降幅的证明。
  • 冷启动并不自动带来合适的模式分配:OCR-Reasoning 思考率从 10.4% 增加到 61.9% 时分数提高。本文并不是在所有任务上一味减少思考。
  • 表 3 另有 beta=10、gamma=0.5 的配置在 OCR-Reasoning 达到 26.9,但思考率为 75.6%;主配置 26.5/61.9 不是纯准确率最优点,而是更节省思考的选择。

亮点与洞察

  • 把“决策 token”和“执行 token”区分开,比对整段文本统一加熵更贴近控制问题。可迁移的原则是把探索压力放在真正决定计算路径的位置,而非所有生成位置。
  • 同题双模式冷启动使强化学习能比较两种行为,而不是继承教师的逐题难度分类。但只有保留足够采样覆盖,组内比较才可能可靠。
  • 思考比例和平均输出长度一起报告,揭示了分数接近时的不同代价。论文的价值不只是缩短回答,而是让简单抽取与复杂推理获得不同的计算配置。

局限与展望

  • 作者没有单列完整局限章节,以下主要是基于实验与缓存可见信息的审读判断。实验集中于文档理解和两种 Qwen 系基座,不能直接外推到视频、具身操作或通用智能体。
  • “不依赖外部难度标签”不等于没有人工设计:准确率分档、比例差阈值以及 A、P 的方向规则均引入先验。可进一步检验这些阈值在新领域是否需要重新调节。
  • 熵低只代表模式选择稳定,不能证明模型知道最优计算策略;组内样本少时,准确率比例也可能不可靠。还需有重复随机种子、置信区间和分布外校准实验来支撑更强的稳定性结论。
  • 首 token 的奖励分离不意味着共享网络参数完全互不干扰,最终答案奖励也不保证推理忠实。可进一步测量两类更新的梯度冲突,以及错误模式是否仍通过内容学习被强化。
  • token 节省不等于实测延迟、能耗或训练成本节省。缓存未报告这些指标,且缺少补充材料中的损失细节、反馈分段取值和数据配比,不能仅凭本笔记完整复现训练。

相关工作与启发

  • 对比 GRPO / DAPO:本文保留分组采样训练思路,但按 token 角色拆分信号,并对开关增加动态熵课程。模式坍塌对照说明的是作者双模式设置中的失效现象,不是对这些算法普遍能力的否定。
  • 对比 R-4B / Keye-VL:它们也是自适应思考模型;本文的区别在于围绕模式 token 的概率分布安排探索与收敛,而不是仅把模式选择当作整条回答的一部分。R-4B 的 CharXivRQ 对照尤其说明接近的分数可以对应很不同的生成开销。
  • 对比固定开关与长度压缩:外部提示开关需要用户提前决定,压缩推理链则主要调整思考后的长度。本文先解决是否进入思考分支;未来可以在分支内部结合长度控制,但这是延伸方向,并非文中已验证的模块。

评分

  • 新颖性: 4/5。模式 token 的定向熵课程与奖励归因拆分有明确针对性,但仍建立在既有双模式训练与分组 RL 上。
  • 实验充分度: 4/5。主结果、固定模式比较、决策反馈与熵课程消融形成较完整证据链,缺少延迟与多次运行不确定性报告。
  • 写作质量: 3/5。主要思路易理解,但“自主发现”的表述强于其启发式反馈实际支持范围,且模型命名与部分文字增益需要对照表格审读。
  • 价值: 4/5。对文档推理服务的计算分配有实用意义,是否能跨领域复用仍需进一步验证。