用动态奖励平衡创意写作的主观质量与客观约束。
RLMR: Reinforcement Learning with Mixed Rewards for Creative Writing
- 根据生成样本质量动态调整约束奖励权重,实现自适应优化。
- 在WriteEval上手动评估胜率达72.75%,指令遵循率提升至86.65%。
- 适合需要兼顾创意表达与格式规范的文本生成场景。
大型语言模型广泛应用于创意写作。此类任务需在主观写作质量(如文采、情感表达)与客观约束遵循(如格式要求、字数限制)间取得平衡。现有方法难以兼顾:单一奖励策略无法同时提升两类能力,固定权重混合奖励则缺乏场景适应性。为此,我们提出强化学习混合奖励(RLMR),采用写作质量评分模型(主观)与约束验证模型(客观)动态混合奖励。约束奖励权重依据采样组内写作质量自动调节,使违反约束的样本在GRPO中获得负优势,从而被训练时惩罚,这是本方法的核心创新。我们在8B至72B参数量的多种模型上进行自动化与人工评估,并构建真实写作基准WriteEval进行综合测试。结果表明,该方法在指令遵循(IFEval从83.36%提升至86.65%)与写作质量(WriteEval人工专家两两对比胜率72.75%)上均实现稳定提升。据我们所知,RLMR是首个在在线强化学习中结合主观偏好与客观验证的工作,为多维度创意写作优化提供了有效方案。
原文摘要 · Abstract (English)
Large language models are extensively utilized in creative writing applications. Creative writing requires a balance between subjective writing quality (e.g., literariness and emotional expression) and objective constraint following (e.g., format requirements and word limits). Existing methods find it difficult to balance these two aspects: single reward strategies fail to improve both abilities simultaneously, while fixed-weight mixed-reward methods lack the ability to adapt to different writing scenarios. To address this problem, we propose Reinforcement Learning with Mixed Rewards (RLMR), utilizing a dynamically mixed reward system from a writing reward model evaluating subjective writing quality and a constraint verification model assessing objective constraint following. The constraint following reward weight is adjusted dynamically according to the writing quality within sampled groups, ensuring that samples violating constraints get negative advantage in GRPO and thus penalized during training, which is the key innovation of this proposed method. We conduct automated and manual evaluations across diverse model families from 8B to 72B parameters. Additionally, we construct a real-world writing benchmark named WriteEval for comprehensive evaluation. Results illustrate that our method achieves consistent improvements in both instruction following (IFEval from 83.36% to 86.65%) and writing quality (72.75% win rate in manual expert pairwise evaluations on WriteEval). To the best of our knowledge, RLMR is the first work to combine subjective preferences with objective verification in online RL training, providing an effective solution for multi-dimensional creative writing optimization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。