Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

RISE: Self-Improving Robot Policy with Compositional World Model

About

Despite the sustained scaling on model capacity and data acquisition, Vision-Language-Action (VLA) models remain brittle in contact-rich and dynamic manipulation tasks, where minor execution deviations can compound into failures. While reinforcement learning (RL) offers a principled path to robustness, on-policy RL in the physical world is constrained by safety risk, hardware cost, and environment reset. To bridge this gap, we present RISE, a scalable framework of robotic reinforcement learning via imagination. At its core is a Compositional World Model that (i) predicts multi-view future via a controllable dynamics model, and (ii) evaluates imagined outcomes with a progress value model, producing informative advantages for the policy improvement. Such compositional design allows state and value to be tailored by best-suited yet distinct architectures and objectives. These components are integrated into a closed-loop self-improving pipeline that continuously generates imaginary rollouts, estimates advantages, and updates the policy in imaginary space without costly physical interaction. Across three challenging real-world tasks, RISE yields significant improvement over prior art, with more than +35% absolute performance increase in dynamic brick sorting, +45% for backpack packing, and +35% for box closing, respectively.

Jiazhi Yang, Kunyang Lin, Jinwei Li, Wencong Zhang, Tianwei Lin, Longyan Wu, Zhizhong Su, Hao Zhao, Ya-Qin Zhang, Li Chen, Ping Luo, Xiangyu Yue, Hongyang Li• 2026

Related benchmarks

TaskDatasetResultRank
Backpack PackingBackpack Packing Real-world 1.0 (test)
Success Rate85
6
Box ClosingBox Closing Real-world 1.0 (test)
Success Rate95
6
Dynamic Brick SortingDynamic Brick Sorting Real-world 1.0 (test)
Success Rate85
6
Dynamics ModelingReal-world tasks Experiment #1
PSNR23.9
4
Dynamics ModelingBridge dataset (Experiment #2)
PSNR23.68
4
Showing 5 of 5 rows

Other info

Follow for update