arXiv:2609.05988cs.LGcs.AI2026-09

用分布对抗训练提升递归推理模型的稳定性和解题能力

DART: Distributional Adversarial Recurrent Training for Algorithm Learning

  • 用真实解附近的分布替代单一标签,提供更丰富的学习信号
  • 在迷宫、国际象棋和掩码数独任务中显著提升解的质量与稳定性
  • 适用于需要长期迭代推理的模型,对复杂任务更具鲁棒性

递归推理模型(RRMs)可通过隐藏空间中的迭代计算解决结构化问题,实现从简单到复杂的泛化。但随着任务难度增加,传统基于单点监督的训练方式面临挑战:有效解仅占解空间极小区域,无效解迅速泛滥。本文提出分布对抗递归训练(DART),将单点监督替换为围绕真实解的局部目标分布,并通过对抗目标使模型输出与该分布对齐。DART提供了更丰富的学习信号,促进向有效解的稳定迭代轨迹。在迷宫、国际象棋和掩码数独任务上,使用Deep Thinking Systems和Tiny Recursive Models等多个RRM,DART均提升了求解质量、稳定性和对分布偏移的鲁棒性。与标签平滑、高斯软标签及渐进训练对比表明,DART的效果不能仅由目标软化解释,且与稳定长时程递归的训练策略互补。结果表明DART是提升所评估递归推理模型鲁棒性的有前景方法。

原文摘要 · Abstract (English)

Recurrent reasoning models (RRMs) can solve structured problems, achieving easy-to-hard generalization through iterative computation in hidden space. These models are typically trained with instance-level supervision, which becomes increasingly problematic as task difficulty grows: valid solutions occupy a tiny region of the solution space, while invalid solutions proliferate rapidly. We propose Distributional Adversarial Recurrent Training (DART), a training framework that replaces single-point supervision with a local target distribution around the ground-truth solution and aligns model outputs with this distribution through an adversarial objective. DART provides a richer learning signal and encourages more stable iterative trajectories toward valid solutions. When evaluated on Maze, Chess, and masked Sudoku with multiple RRMs, including Deep Thinking Systems and Tiny Recursive Models, DART improves solution quality, stability, and robustness under the evaluated distribution shifts. Comparisons with label smoothing, Gaussian softened targets, and progressive training show that DART is not explained by target softening alone and is complementary to training schemes that stabilize long-horizon recurrence. These results identify DART as a promising approach for improving robustness across the evaluated recurrent reasoning models.

递归推理对抗训练强化学习模型鲁棒性

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