Tsinghua NeurIPS Paper: First-Order Action Supervision Boosts Offline RL, Especially With Little Data

量子位 · wechat · 2026-10-10

A NeurIPS 2026 paper from Tsinghua AIR, 'Higher-Order Action Supervision Makes A Strong Policy Class,' argues that jittery robot control and overreactive driving stem from abrupt changes between consecutive actions, yet standard policy learning only supervises single-point actions. Without changing the policy network, the authors add a first-order constraint: the policy's directional derivative along state trajectories should fit the action difference a′−a in the data.

The idea uniformly covers deterministic (TD3+BC), Gaussian stochastic (via Ornstein-Uhlenbeck temporal correlation), and flow-matching policies, and injects first-order information into value learning through a continuous-time HJB framework — adding gradient-alignment terms on top of the Bellman residual — retrofitting TD3+BC, ReBRAC, IQL and IFQL.

Experiments span 50 OGBench tasks and 10k-transition D4RL sets: ReBRAC jumps from 26 to 61 on antmaze-giant-navigate; on walker2d-expert, TD3+BC goes from 2.7 to 82.3 and pure behavior cloning from 26.7 to 91.1. Rollouts show smoother action changes. The takeaway: local variation in continuous trajectories is especially valuable when samples are scarce, as a plug-and-play loss term.

Original post →

More from Embodied

Embodied channel →