通过衰减奖励建模,避免大模型迭代优化中的重复提示失效问题。
Progressive Content Refinement with Decaying Reward Joint LinUCB

- 用期望最大化算法同时估计提示效果和奖励衰减参数。
- 在情感反转和GSM8K任务上超越强基线,提升显著。
- 适合需要持续优化提示的长周期生成任务使用。
迭代优化已显著提升大语言模型性能,但现有方法如基于反馈的Self-Refine或传统多臂老虎机策略常依赖静态选项,忽略奖励饱和效应,导致对相同提示或动作的过度利用,造成奖励随时间递减。为此,我们提出一种新颖的上下文相关多臂老虎机算法,显式建模奖励衰减。利用期望最大化(EM)算法,该方法同步估计每个动作(提示)的专属价值与衰减参数。通过将提示嵌入为动作,实现动作价值与衰减参数的联合学习,区别于传统的分离式线性置信上界(LinUCB)框架。在情感反转和GSM8K基准测试上的实验表明,该方法显著优于多个强基线。消融研究进一步证实,在老虎机框架中引入奖励衰减建模对于缓解过度利用、优化迭代优化过程至关重要。
原文摘要 · Abstract (English)
Iterative refinement has significantly enhanced Large Language Model (LLM) performance; however, existing methods ranging from feedback-based Self-Refine to traditional bandit approaches often rely on static options or overlook the saturation effect. This neglect leads to over-exploitation, where the continuous use of identical prompts or arms results in diminishing rewards over time. To address this challenge, we propose a novel contextual bandit algorithm that explicitly incorporates reward decay modeling. Utilizing an Expectation-Maximization (EM) algorithm, our method simultaneously estimates both arm-specific and decay parameters. Furthermore, by embedding prompts as arms, we facilitate the joint learning of arm values, distinguishing our approach from the traditional disjoint Linear Upper Confidence Bound (LinUCB) framework. Experimental results on Sentiment Reversal and GSM8K benchmarks demonstrate that our method achieves significant performance gains over strong baselines. Finally, our ablation study confirms that the integration of reward decay modeling within the bandit framework is crucial for mitigating over-exploitation and optimizing the iterative refinement process.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。