STEAM: Self-Supervised Temporal Ensemble Advantage Modeling for Real-World Robot Learning

Abstract:Real-world robot learning increasingly relies on heterogeneous data, but demonstrations and rollouts often mix useful progress with stalls, corrections, and suboptimal behavior. Effective policy learning therefore requires frame-level advantages that distinguish reliable local progress from failures and regressions. We propose \textbf{S}elf-supervised \textbf{T}emporal \textbf{E}nsemble \textbf{A}dvantage \textbf{M}odeling (\textbf{STEAM}), a label-free method that learns such advantages from expert demonstrations. STEAM trains an ensemble of temporal-offset predictors on frame pairs within expert trajectories, using the normalized temporal offset between two frames as a self-supervised signal. Each predictor maps a frame pair to a distribution over temporal offsets, which is converted into a scalar advantage. STEAM then takes the minimum advantage across the ensemble to score mixed-quality rollout data conservatively. Across real-world bimanual towel folding, chip checkout, table clearing, and cola restocking tasks, STEAM identifies stalls, failures, and recoveries. When combined with CFGRL, STEAM achieves the highest policy success rate on all four tasks, averaging $83.5\%$ against $58.8\%$ for the best baseline. Project page: this https URL.

Robotics (cs.RO)

arXiv:2606.29834 [cs.RO]

(or arXiv:2606.29834v2 [cs.RO] for this version)

https://doi.org/10.48550/arXiv.2606.29834 Focus to learn more arXiv-issued DOI via DataCite

arXivLabs is a framework that allows collaborators to develop and share new arXiv features directly on our website.

Both individuals and organizations that work with arXivLabs have embraced and accepted our values of openness, community, excellence, and user data privacy. arXiv is committed to these values and only works with partners that adhere to them.

Have an idea for a project that will add value for arXiv's community? Learn more about arXivLabs.