Conservative Offline Robot Policy Learning via Posterior-Transition Reweighting

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Zhang, Wanpeng, Luo, Hao, Zheng, Sipeng, Feng, Yicheng, Xu, Haiweng, Xi, Ziheng, Xu, Chaoyi, Yuan, Haoqi, Lu, Zongqing
Natura: Preprint
Pubblicazione: 2026
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866912971042586624
author Zhang, Wanpeng
Luo, Hao
Zheng, Sipeng
Feng, Yicheng
Xu, Haiweng
Xi, Ziheng
Xu, Chaoyi
Yuan, Haoqi
Lu, Zongqing
author_facet Zhang, Wanpeng
Luo, Hao
Zheng, Sipeng
Feng, Yicheng
Xu, Haiweng
Xi, Ziheng
Xu, Chaoyi
Yuan, Haoqi
Lu, Zongqing
contents Offline post-training adapts a pretrained robot policy to a target dataset by supervised regression on recorded actions. In practice, robot datasets are heterogeneous: they mix embodiments, camera setups, and demonstrations of varying quality, so many trajectories reflect recovery behavior, inconsistent operator skill, or weakly informative supervision. Uniform post-training gives equal credit to all samples and can therefore average over conflicting or low-attribution data. We propose Posterior-Transition Reweighting (PTR), a reward-free and conservative post-training method that decides how much each training sample should influence the supervised update. For each sample, PTR encodes the observed post-action consequence as a latent target, inserts it into a candidate pool of mismatched targets, and uses a separate transition scorer to estimate a softmax identification posterior over target indices. The posterior-to-uniform ratio defines the PTR score, which is converted into a clipped-and-mixed weight and applied to the original action objective through self-normalized weighted regression. This construction requires no tractable policy likelihood and is compatible with both diffusion and flow-matching action heads. Rather than uniformly trusting all recorded supervision, PTR reallocates credit according to how attributable each sample's post-action consequence is under the current representation, improving conservative offline adaptation to heterogeneous robot data.
format Preprint
id arxiv_https___arxiv_org_abs_2603_16542
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Conservative Offline Robot Policy Learning via Posterior-Transition Reweighting
Zhang, Wanpeng
Luo, Hao
Zheng, Sipeng
Feng, Yicheng
Xu, Haiweng
Xi, Ziheng
Xu, Chaoyi
Yuan, Haoqi
Lu, Zongqing
Robotics
Offline post-training adapts a pretrained robot policy to a target dataset by supervised regression on recorded actions. In practice, robot datasets are heterogeneous: they mix embodiments, camera setups, and demonstrations of varying quality, so many trajectories reflect recovery behavior, inconsistent operator skill, or weakly informative supervision. Uniform post-training gives equal credit to all samples and can therefore average over conflicting or low-attribution data. We propose Posterior-Transition Reweighting (PTR), a reward-free and conservative post-training method that decides how much each training sample should influence the supervised update. For each sample, PTR encodes the observed post-action consequence as a latent target, inserts it into a candidate pool of mismatched targets, and uses a separate transition scorer to estimate a softmax identification posterior over target indices. The posterior-to-uniform ratio defines the PTR score, which is converted into a clipped-and-mixed weight and applied to the original action objective through self-normalized weighted regression. This construction requires no tractable policy likelihood and is compatible with both diffusion and flow-matching action heads. Rather than uniformly trusting all recorded supervision, PTR reallocates credit according to how attributable each sample's post-action consequence is under the current representation, improving conservative offline adaptation to heterogeneous robot data.
title Conservative Offline Robot Policy Learning via Posterior-Transition Reweighting
topic Robotics
url https://arxiv.org/abs/2603.16542