用自动生成的细粒度奖励模型提升大模型翻译质量。
MT-RewardTree: A Comprehensive Framework for Advancing LLM-Based Machine Translation via Reward Modeling
- 用近似蒙特卡洛树搜索自动生成翻译过程中的细粒度偏好对。
- 在相同输入前缀下,模型在词级和序列级评测中均达顶尖水平。
- 无需额外对齐训练即可实现测试时优化,适合工业级翻译系统。
过程奖励模型(PRMs)在大语言模型的复杂推理任务中表现优异,但在机器翻译(MT)领域仍缺乏系统方法与评估基准。为此,本文提出 extbf{MT-RewardTree},一个用于构建、评估与部署机器翻译过程奖励模型的完整框架。不同于传统的人工标注偏好对,我们提出一种基于近似蒙特卡洛树搜索(MCTS)的自动方法,生成细粒度的词级别偏好对,显著降低人工标注成本。我们建立了首个面向机器翻译的奖励模型评估基准,并系统比较了不同架构,发现词级监督能有效捕捉细粒度偏好。实验表明,在相同输入前缀下,我们的 MT-PRM-Qwen-2.5-3B 模型在词级与序列级评估中均达到当前最优性能。此外,我们展示了实际应用场景:无需额外对齐训练,该模型可实现测试时对齐,并显著提升假设集成效果。本工作为奖励模型在机器翻译研究中的作用提供了重要洞见。代码与数据已公开于 https://sabijun.github.io/MT_RewardTreePage。
原文摘要 · Abstract (English)
Process reward models (PRMs) have shown success in complex reasoning tasks for large language models (LLMs). However, their application to machine translation (MT) remains underexplored due to the lack of systematic methodologies and evaluation benchmarks. To address this gap, we introduce \textbf{MT-RewardTree}, a comprehensive framework for constructing, evaluating, and deploying process reward models in MT. Unlike traditional vanilla preference pair construction, we propose a novel method for automatically generating token-level preference pairs using approximate Monte Carlo Tree Search (MCTS), which mitigates the prohibitive cost of human annotation for fine-grained steps. Then, we establish the first MT-specific reward model benchmark and provide a systematic comparison of different reward modeling architectures, revealing that token-level supervision effectively captures fine-grained preferences. Experimental results demonstrate that our MT-PRM-Qwen-2.5-3B achieves state-of-the-art performance in both token-level and sequence-level evaluation given the same input prefix. Furthermore, we showcase practical applications where PRMs enable test-time alignment for LLMs without additional alignment training and significantly improve performance in hypothesis ensembling. Our work provides valuable insights into the role of reward models in MT research. Our code and data are released in \href{https://sabijun.github.io/MT_RewardTreePage/}{https://sabijun.github.io/MT\_RewardTreePage}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。