arXiv AI By Jorge de Freitas, Christos Ziakas, Alessandra Russo

Generating Critic-Guided Synthetic Experience in Latent World Models

Read the original on arXiv AI →

The paper introduces Critic-Guided Synthetic Experience (CGSE), a method that uses a frozen latent world model to generate new transitions offline. CGSE samples candidate actions from an offline policy, refines them to maximize a first‑order approximation of the Q‑function within a trust region, and then uses the critic to select the best action for each imagined transition. The resulting synthetic transitions are added to the replay buffer, improving an offline policy’s success rate by 7.1% on three OGBench manipulation tasks.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv AI.

arXiv AI
Aug 19

Q-Learning With World Models

The paper introduces QWM, a framework that integrates world models with standard Q‑learning to perform test‑time search over imagined trajectories. By training the policy and value function solely on real transitions, QWM avoids compounding model bias while still benefiting from predictive search. Experiments on the Robomimic and LIBERO manipulation benchmarks show that QWM outperforms strong prior state‑of‑the‑art methods in both sample efficiency and performance.

By Perry Dong, Yueru Jia, Chelsea Finn, Dorsa Sadigh
arXiv AI
4d ago

MAC-v2: Long-horizon model-based value expansion from pixels

MAC‑v2 is a new framework for long‑horizon model‑based value expansion from pixel observations in offline reinforcement learning. It trains a reconstruction‑free latent world model that learns representations through prediction and control objectives, avoiding the need to encode control‑irrelevant visual details. By using action chunking, MAC‑v2 reduces recursive model predictions, mitigating error accumulation and achieving strong performance on visual benchmarks such as OGBench, MMBench2, Robomimic, and LIBERO‑10.

By Kwanyoung Park, Sergey Levine
arXiv Computer Vision
Oct 23

DRILL: Training World Models to Improve Policies, Not Predict Pixels

The paper introduces DRILL, a bilevel training framework that optimizes world models for policy improvement rather than pixel accuracy. By reweighting prediction errors with policy visitation and confidence (DRILL‑VWI) or using a full meta‑gradient approach (DRILL‑IMG), the authors demonstrate significant gains in downstream policy success rates across multiple manipulation simulators. The work shows that world models can be more effectively trained using the policy gradients they generate, rather than relying solely on pixel‑level prediction metrics.

By Zhaolu Kang, Tailong Luo, Lei Wei, Guangyuan Dong, Sihan Liu, Yingjie He, Siheng Wang, Shuaibo Li, Rongchao Zhang, Zhen Tian, Richeng Xuan, Zhichao Hu