论文

LBLP:用不等式约束重构离线RL的贝尔曼最优性

Lifted Bellman Linear Programming for Offline Reinforcement Learning

精选理由

训练离线RL可以不用目标网络和EMA了,ALBUM单critic在OGBench追平FQL,显存最省

论文提出 Lifted Bellman Linear Program(LBLP),把贝尔曼最优性的线性规划刻画提升到 (Q,V) 联合空间,所有约束只涉及数据集内的状态-动作对,无需对行为策略动作做 off-policy 修正。其近似实现 ALBUM 将约束松弛为 hinge 惩罚并用 stop gradient 分离 K 步 rollout 目标,目标函数不含对自举目标的平方回归,因此训练不需要目标网络和 EMA 更新。在 OGBench 上,ALBUM 用单个 critic 配高斯策略,平均成绩追平 FQL,与近期的 action-chunking 方法相当。它同时是对比方法中参数量最少、峰值 GPU 显存占用最低的一个。

原文 · arXiv cs.LG

Lifted Bellman Linear Programming for Offline Reinforcement Learning

Offline reinforcement learning (RL) typically trains a critic by minimizing a regression loss against bootstrapped value targets stabilized by target networks with exponential moving average (EMA) updates. Multi-step targets incorporate behavior-policy actions and therefore require off-policy correction. We instead impose in-sample Bellman optimality on the critic through inequality constraints. We formulate the Lifted Bellman Linear Program (LBLP), which lifts the linear programming characterization of Bellman optimality to the joint $(Q,V)$ space so that every constraint involves only state-action pairs in the dataset. Its unique minimizer is the in-sample optimal pair, and constraints along $K$-step segments of dataset trajectories leave this minimizer unchanged for any rollout policy and horizon. Under deterministic dynamics, this minimizer lies between the best dataset return and the optimal value. Relaxing the constraints into hinge penalties recovers the same solution above a finite penalty coefficient in the tabular case. Approximate Lifted Bellman Unconstrained Minimization (ALBUM) implements this relaxation with neural networks and detaches the $K$-step rollout targets by stop gradient. Its objective contains no squared regression onto bootstrapped targets, so it can be trained without target networks or EMA updates. Under deterministic dynamics, the LBLP solution is a stationary point of the detached update under a coefficient condition independent of $γ$ and $K$, and the inequality constraints allow discounted returns along dataset trajectories to serve as lower bounds without off-policy correction or action chunking. On OGBench, ALBUM uses a single critic with a Gaussian policy, matches the average performance of FQL, and is comparable to recent action-chunking methods, while using the fewest parameters and the least peak GPU memory among all compared methods.