用强化学习让大模型更准地生成数字结果
Beyond Token-level Supervision: Unlocking the Potential of Decoding-based Regression via Reinforcement Learning
- 把生成过程当作决策过程,用序列级奖励引导整体数值准确
- 在表格和代码指标预测上,精度超越现有最佳方法
- 适合需要高精度数值预测的通用场景
基于解码的回归将回归任务重新定义为序列生成问题,已成为利用大语言模型进行数值预测的有前景范式。然而,其进展受限于离散标记级目标(如交叉熵)与连续数值之间的不匹配。依赖标记级约束的方法往往无法捕捉目标值的整体量级,限制了精度与泛化能力。本文提出通过强化学习解锁基于解码回归的潜力。我们将生成过程建模为马尔可夫决策过程,利用序列级奖励强制全局数值一致性。在表格回归和代码度量回归上的大量实验表明,所提方法(特别是ReMax和GRPO)持续优于最先进的标记级基线和传统回归头,展现出引入序列级信号的优势。分析进一步显示,强化学习显著提升了采样效率与预测精度,确立了基于解码的回归作为通用数值预测的鲁棒且准确范式。
原文摘要 · Abstract (English)
Decoding-based regression, which reformulates regression as a sequence generation task, has emerged as a promising paradigm of applying large language models for numerical prediction. However, its progress is hindered by the misalignment between discrete token-level objectives (e.g., cross-entropy) and continuous numerical values. Existing approaches relying on token-level constraints often fail to capture the global magnitude of the target value, limiting their precision and generalization. In this paper, we propose to unlock the potential of decoding-based regression via Reinforcement Learning (RL). We formulate the generation process as a Markov Decision Process, utilizing sequence-level rewards to enforce global numerical coherence. Extensive experiments on tabular regression and code metric regression demonstrate that our method (specifically with ReMax and GRPO) consistently outperforms both state-of-the-art token-level baselines and traditional regression heads, showing the superiority of introducing sequence-level signals. Our analysis further reveals that RL significantly enhances sampling efficiency and predictive precision, establishing decoding-based regression as a robust and accurate paradigm for general-purpose numerical prediction.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。