跳到正文
arXiv:cs.AI· Yaomin Wang, Jianting Pan, Ran Tian, Xiaoyang Li, Yu Zhang, Hengle Qin, Tianshu YU·· 20 小时前

AdaGamma:面向强化学习时间自适应的状态依赖折扣

AdaGamma: State-Dependent Discounting for Temporal Adaptation in Reinforcement Learning

AI 导读

AdaGamma 是一个在深度 actor-critic 强化学习中实现状态依赖折扣的框架,通过学习有界折扣函数 γ_φ(s),并用回报一致性目标训练,避免 TD-collapse。该方法可集成进 SAC 和 PPO,在连续控制基准上取得提升,并在 JD Logistics 平台为期四周的在线 A/B 测试中获得统计显著收益。

正文

View PDF HTML (experimental)

Abstract:The discount factor in reinforcement learning controls both the effective planning horizon and the strength of bootstrapping, yet most deep RL methods use a single fixed value across all states. State-dependent discounting is a natural way to adapt temporal credit assignment to local state structure, but naively learning a discount function inside deep actor--critic methods creates a self-referential target: the learned discount scales the same bootstrap term against which it is trained. Consequently, TD-error-based objectives can admit a degenerate shortcut, reducing target dependence on downstream value estimates instead of learning meaningful temporal structure. We propose \textbf{AdaGamma}, a practical framework for state-dependent discounting in deep actor--critic reinforcement learning. AdaGamma learns a bounded discount function \(\gamma_\phi(s)\) and trains it using a return-consistency objective that matches a one-step adaptive bootstrap to a multi-step return computed with a lagged, stop-gradient reference discount. This removes the direct target manipulation path and encourages the discount to encode state-dependent value propagation. We analyze the Bellman operator induced by a fixed state-dependent discount in finite tabular MDPs, proving contraction, exact soft policy iteration, and Lipschitz sensitivity under standard boundedness assumptions. Empirically, AdaGamma integrates into SAC and PPO, avoids TD-collapse, improves continuous-control benchmarks, and achieves statistically significant gains in a four-week online A/B test on the JD Logistics platform.
Comments: 29 pages, 6 figures
Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI)
Cite as: arXiv:2605.06149 [cs.LG]
  (or arXiv:2605.06149v2 [cs.LG] for this version)
  https://doi.org/10.48550/arXiv.2605.06149

arXiv-issued DOI via DataCite

Submission history

From: Yaomin Wang [view email]
[v1] Thu, 7 May 2026 12:42:46 UTC (11,758 KB)
[v2] Sun, 11 Oct 2026 03:53:21 UTC (11,840 KB)

来源:arXiv:cs.AI · arxiv.org