Search papers, labs, and topics across Lattice.
This paper introduces counterfactual transport flows, a novel framework for offline reinforcement learning that refines candidate trajectories using historical data as feedback. By constructing local preference pairs from offline trajectories with higher task-specific returns, the method allows for conservative refinement that balances the preservation of original behavior with the potential for improvement. Experiments on D4RL benchmarks demonstrate significant enhancements in trajectory performance while maintaining interpretability in the refinement process.
Offline RL can now refine trajectories conservatively without risking extrapolation, thanks to a novel framework that leverages local preference pairs for targeted improvements.
Offline reinforcement learning (RL) offers a path to policy improvement from logged data alone, using historical returns or other measurable outcomes as world feedback. A key difficulty is improving observed behavior without extrapolating beyond what the offline data supports. We propose \emph{counterfactual transport flows}, a source-conditioned trajectory refinement framework for offline decision-making guided by world feedback. Given a low-feedback candidate trajectory, we construct local preference pairs from offline data by retrieving nearby trajectories in latent trajectory space with higher task-specific feedback, and use them as weak supervision for conservative refinement. The framework learns instance-specific refinement directions: at inference time, a refinement strength parameter controls how far the candidate trajectory is transported, enabling a trade-off between preserving the original behavior and applying stronger improvement. Experiments on D4RL benchmarks, including AntMaze and MuJoCo tasks, show that our method improves behavior from historical returns as world feedback, while providing interpretable trajectory-level refinement paths.
