AI在线 AI在线

微软携手清华、北大推出奖励推理模型:根据 AI 任务复杂性动态分配计算资源

微软研究院联合清华大学、北京大学组建团队,推出奖励推理模型(Reward Reasoning Models,RRMs),通过显式推理过程动态分配计算资源,提升复杂任务评估效果。

科技媒体 marktechpost 今天(5 月 27 日)发布博文,报道称微软研究院联合清华大学、北京大学组建团队,推出奖励推理模型(Reward Reasoning Models,RRMs),通过显式推理过程动态分配计算资源,提升复杂任务评估效果。

AI在线援引博文介绍,强化学习(Reinforcement Learning,RL)已成为大语言模型(LLM)后训练的核心方法,通过人类反馈(RLHF)或可验证奖励(RLVR)提供监督信号。

然而,RLVR 在数学推理中虽有潜力,却因依赖可验证答案的训练查询而受限,难以应用于通用领域的大规模训练。

此外,现有奖励模型分为标量型和生成型两大类,均无法有效扩展测试时的计算资源。当前方法对所有输入统一分配计算资源,缺乏针对复杂查询进行细致分析的能力,导致评估效果不佳。

为解决上述问题,微软研究院、清华大学和北京大学的研究者联手推出奖励推理模型(RRMs)。RRMs 在给出最终奖励前执行显式推理过程,能够根据任务复杂性自适应分配额外计算资源。

这种方法通过“思维链”(Chain-of-Thought)推理,针对奖励不明显的复杂查询投入更多测试时计算资源。

RRMs 基于 Qwen2 模型,采用 Transformer-decoder 架构,将奖励建模转化为文本补全任务,生成推理过程后给出最终判断。

研究团队利用 RewardBench 库进行系统分析,评估指标包括指令遵循性、帮助性、准确性、无害性和细节水平。RRMs 还支持多响应评估,通过 ELO 评分系统和淘汰赛机制,结合多数投票提升计算资源利用率。

测试结果显示,RRMs 在 RewardBench 和 PandaLM Test 基准测试中表现突出。其中,RRM-32B 在推理类别中达到 98.6% 的准确率,与使用相同数据训练的 DirectJudge 模型相比,RRMs 展现出显著性能差距,证明其在复杂查询中有效利用测试时计算资源。

在奖励引导的最佳 N 推理(Best-of-N Inference)和后训练反馈中,RRMs 超越所有基线模型,且进一步提升多数投票机制效率。

研究还表明,随着模型规模从 7B、14B 到 32B 扩展,更长的推理时间始终带来准确性提升。RRMs 通过并行和顺序扩展方法高效利用计算资源,为传统标量奖励模型提供强大替代方案。

相关资讯

北大清华等联合发布 LLaVA-o1:首个自发性视觉 AI 模型,推理计算 Scaling 新思路

由北京大学、清华大学、鹏城实验室、阿里巴巴达摩院以及理海大学(Lehigh University)组成的研究团队,最新推出了 LLaVA-o1,这是首个具备自发性、类似于 GPT-o1 的系统性推理视觉语言模型。
11/19/2024 11:38:30 AM
故渊

BAAI、北大&港中文团队提出 SegVol:通用且可交互的医学体素分割模型

编辑 | ScienceAI上周,北京智源人工智能研究院(BAAI)、北京大学和香港中文大学的研究团队开源了 SegVol 医学通用分割模型。与过去一些很棒的 Medical SAM 工作不同,SegVol 是第一个能够同时支持 box,point 和 text prompt 进行任意尺寸原分辨率的 3D 体素分割模型。作为一个便捷的通用分割工具,研究人员将 SegVol 代码和模型开源到 GitHub:BAAI-DCAI/SegVol,欢迎大家使用。目前开源的模型权重文件包括(1)使用 96k CTs 预训练 2
11/28/2023 2:20:00 PM
ScienceAI

加速量子化学计算,字节&北大团队量子蒙特卡罗研究新进展登Nature子刊

编辑 | 萝卜皮基于神经网络的变分蒙特卡罗(NN-VMC)已成为一种有前途、有潜力的从头计算量子化学的尖端技术。然而,现有方法的高计算成本,阻碍了它们在现实化学问题中的应用。字节跳动与北京大学团队介绍了 NN-VMC 的最新研究进展,它实现了显著的加速率,从而极大地将 NN-VMC 的适用性扩展到更大的系统。该团队的关键设计是一个名为 Forward Laplacian 的计算框架,它通过有效的前向传播(forward)过程来计算与神经网络相关的 Laplacian(这是 NN-VMC 的瓶颈)。然后,研究人员证明
2/27/2024 3:55:00 PM
ScienceAI
  • 1