用对抗训练提升翻译模型,解决奖励模型失效问题。
RIVAL: Reinforcement Learning with Iterative and Adversarial Optimization for Machine Translation
- 设计对抗框架,让奖励模型与翻译模型轮流优化。
- 在字幕翻译任务上,相比基线提升显著,超越纯语言模型。
- 无需参考译文,结合人工评价与自动指标,稳定训练效果。
大型语言模型具备强大的多语言能力,将人类反馈强化学习(RLHF)应用于翻译任务展现出巨大潜力。然而我们发现,该范式在口语化字幕翻译任务中表现意外不佳。经研究发现,这是由于离线奖励模型随时间分布偏移,逐渐偏离在线语言模型,最终导致训练结果恶化。为此,本文提出RIVAL,一种对抗性训练框架,将过程建模为奖励模型与语言模型间的极小极大博弈。该框架迭代更新两者:奖励模型学习区分优质与劣质翻译(定性偏好奖励),语言模型则通过优化翻译来缩小差距。为稳定训练并提升泛化能力,还引入定量偏好奖励(如BLEU),实现无需参考译文的质量建模,且与人工评价对齐。大量实验表明,所提方法显著优于现有翻译基线。
原文摘要 · Abstract (English)
Large language models (LLMs) possess strong multilingual capabilities, and combining Reinforcement Learning from Human Feedback (RLHF) with translation tasks has shown great potential. However, we observe that this paradigm performs unexpectedly poorly when applied to colloquial subtitle translation tasks. In this work, we investigate this issue and find that the offline reward model (RM) gradually diverges from the online LLM due to distributional shift, ultimately leading to undesirable training outcomes. To address this, we propose RIVAL, an adversarial training framework that formulates the process as a min-max game between the RM and the LLM. RIVAL iteratively updates the both models, with the RM trained to distinguish strong from weak translations (qualitative preference reward), and the LLM trained to enhance its translation for closing this gap. To stabilize training and improve generalizability, we also incorporate quantitative preference reward (e.g., BLEU) into the RM, enabling reference-free quality modeling aligned with human evaluation. Through extensive experiments, we demonstrate that the proposed adversarial training framework significantly improves upon translation baselines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。