← Search

Yanshu Zhang

3 accepted papers

2026

WIMLE: Uncertainty‑Aware World Models with IMLE for Sample‑Efficient Continuous Control

ICLR 2026poster

Model-based reinforcement learning promises strong sample efficiency but often underperforms in practice due to compounding model error, unimodal world models that average over multi-modal dynamics, and overconfident predictions that bias learning. We introduce WIMLE, a model-based method that exten…

Cited by 0SourcecodeScholar
2023

PAPR: Proximity Attention Point Rendering

NeurIPS 2023spotlight

Learning accurate and parsimonious point cloud representations of scene surfaces from scratch remains a challenge in 3D representation learning. Existing point-based methods often suffer from the vanishing gradient problem or require a large number of points to accurately model scene geometry and t…