arXiv Machine Learning By Nikodem Sebastian Zymla, Laurin Thiele, Johannes Pitz

Adaptive Rollout Truncation Based on Epistemic Uncertainty for Efficient Offline World Model Training

Read the original on arXiv Machine Learning →

The paper introduces an adaptive rollout truncation method for offline world model training that uses epistemic uncertainty to decide when to stop autoregressive rollouts. By calibrating a threshold during a warm‑up phase, the approach replaces fixed‑horizon rollouts with uncertainty‑driven truncation, evaluated with ensemble and Monte Carlo dropout estimators. Experiments on ANYmal‑D and ANT demonstrate that this strategy matches or surpasses fixed‑horizon training while reducing cumulative rollout steps by about 72%.

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 Machine Learning.

arXiv AI
Aug 3

HERO: History-Enriched Rollout Training for Long-Horizon Autoregressive Neural Operators

arXiv:2607. 29135v1 Announce Type: cross Abstract: Neural operators provide fast surrogates for time-dependent partial differential equations (PDEs) by applying a learned evolution operator recursively to its own predictions, but this autoregressive rollout feeds every prediction error back as input, so local errors accumulate.

By Jiaquan Zhang, Shuxu Chen, Haifan Meng, Yi Lu, Zhihan Lyu, Fan Mo, Wei Dong, Yang Yang, Chaoning Zhang