游乐游手机版
首页/科技数码/文章详情

清华团队研发TDRM算法:用时间差分优化奖励函数

时间:2025-10-09 18:13
奖励模型(RM)在基于大语言模型(LLM)的强化学习(RL)和推理时验证中都占据核心地位,并已在数学求解、代码生成和指令遵循等任务中展现出卓越性能。 然而,现有奖励模型存在一个关键缺陷——缺乏时间

前沿研究突破:时间差分奖励模型TDRM重塑LLM强化学习范式

奖励模型作为大语言模型强化学习与推理验证的核心组件,在数学推导、编程辅助和指令响应等关键领域展现出巨大潜力。但传统奖励模型存在一个致命短板——时间维度上的评估连续性缺失。这一问题直接导致策略优化方向偏差和训练过程波动加剧。

研究发现,当前模型在长序列推理过程中,单步奖励评估往往与其他步骤隔离割裂,产生误导性训练信号。特别是在复杂思维链场景下,模型需完成多步推演后才能获得终端奖励,致使各推理步骤价值难以准确判断。

针对这一核心挑战,清华大学联合加州理工研发团队创新性提出TDRM框架——运用时间差分(TD)训练方法构建具备时间一致性的奖励模型,显著提升模型训练稳定性。

值得关注的是,该项目已全面开源包括核心代码、训练数据和模型参数,为学界提供重要研究资源。

\

TDRM框架的技术创新

TDRM引入三重技术突破:首先是建立基于n步时间差分的过程奖励模型,其次在强化学习阶段实现实时在线优化,最终通过可验证奖励的线性整合形成完整解决方案。

与传统方法不同,TDRM创造性将时间差分技术应用于奖励模型本身的构建过程,而非仅用于数据预处理。这种革新思路带来了更平滑的奖励空间和更密集的训练信号。

TDRM框架架构

关键技术特征

平滑性优化: 通过局部普希茨连续分析验证,TDRM相比传统方法显著降低27%的评估敏感度,确保推理微调不会引发奖励值剧烈波动。

创新的奖励机制: 引入余弦奖励函数与TD-λ算法的结合范式,既考虑单步推理正确性,又兼顾整体推理长度因素,实现精细化的奖励分配。

在线强化学习: 采用动态状态值计算方法,使模型能够实时适应训练轨迹变化,大幅提升价值预测的准确性。

实验结果验证

在MATH-500等五个高难度数学数据集上的测试表明,TDRM展现出卓越性能:Best-of-N采样方法中最高提升6.6%准确率,树搜索场景更是实现23.7%的性能飞跃。

强化学习场景下,TDRM仅需5%的训练数据量就能达到传统方法同等效果,在GLM4-9B等主流模型架构上都取得了明显的策略优化。

性能对比结果

这项研究不仅解决了奖励模型的时间一致性问题,更为复杂推理任务的自动化评估开辟了新方向。开源资源的释放将加速该技术在更广泛领域的应用落地。

来源:https://36kr.com/p/3501437759003785
上一篇2025年Q3教育科技投融资复盘:AI平台驱动1.5亿美元交易潮 下一篇诺贝尔奖得主奥马尔·亚吉的科研传奇之路
本站内容用于信息整理与展示,如有侵权或内容问题请及时联系处理。

相关推荐

补充同频道和同主题内容,方便继续浏览更多相关内容。

同类最新

继续查看同栏目最近更新的文章。

更多
泰坦军团战魂KG277VPLUS双模显示器27英寸4K165Hz/520Hz仅1888元
科技数码 · 2026-07-03

泰坦军团战魂KG277VPLUS双模显示器27英寸4K165Hz/520Hz仅1888元

泰坦军团“战魂KG277VPLUS”27英寸显示器发售,支持4K165Hz与FHD520Hz双模切换,定价1888元。采用FastIPS面板,97%DCI-P3色域,配备升降支架及双HDMI2 1和双DP1 4接口。

苹果调价影响消费需求 2026年全球笔电出货量或降13.6%
科技数码 · 2026-07-03

苹果调价影响消费需求 2026年全球笔电出货量或降13.6%

迈入2026年,DRAM与NAND闪存的供应持续紧张及价格不断攀升,正逐步传导至终端消费市场。可以预见,下半年市场环境将更加严峻。上半年多家PC厂商已陆续上调产品定价,最终连苹果也不得不跟进,宣布提升iPad、Mac及家居设备的价格,以应对存储成本的快速上涨。 TrendForce分析指出,苹果全面

苹果iPhone 18 Pro自研C2芯片或不支持5G毫米波
科技数码 · 2026-07-03

苹果iPhone 18 Pro自研C2芯片或不支持5G毫米波

苹果自研C2芯片仅支持Sub-6GHz,不支持5G毫米波。因此,美版iPhone18Pro继续采用高通基带方案以支持毫米波,而其他地区版本则搭载苹果自研C2芯片。这一差异将导致在毫米波覆盖的市场中,用户峰值速率可能显著低于美版用户。

纳睿雷达推出睿宸超精细化短时临近AI气象大模型
科技数码 · 2026-07-03

纳睿雷达推出睿宸超精细化短时临近AI气象大模型

纳睿雷达近日释放了一项重磅成果。2026年7月1日,公司正式对外发布了两款自主研发的全新产品:一款是“WDSPT0152型”S波段全极化多功能有源相控阵雷达,另一款则是名为“睿宸”的超精细化短时临近AI气象大模型。从产品战略来看,此次发布直指气象监测与灾害预警领域的技术制高点。 先来看这款S波段雷达

南航国际创新港一期交付 四大专业园区打造空天产业强磁场
科技数码 · 2026-07-03

南航国际创新港一期交付 四大专业园区打造空天产业强磁场

近日,南京航空航天大学与六合区深度合作的标杆项目——南航国际创新港一期正式交付投用。两个地块陆续启用,成功串联起高校科研能量、地方产业载体与市场创新主体,为南京打造全国领先的航空航天产业创新中心、助力江苏布局商业航天全产业链,提供了坚实的物理支撑。 该创新港一期位于六合区雄州街道,分为3号和4号两个