← Search

Boxuan Zhang

5 accepted papers

2026

Mixture-of-World Models: Scaling Multi-Task Reinforcement Learning with Modular Latent Dynamics

ICLR 2026poster

A fundamental challenge in multi-task reinforcement learning (MTRL) is achieving sample efficiency in visual domains where tasks exhibit significant heterogeneity in both observations and dynamics. Model-based RL (MBRL) offers a promising path to sample efficiency through world models, but standard…

Cited by 0SourceScholar
2025

CoT-UQ: Improving Response-wise Uncertainty Quantification in LLMs with Chain-of-Thought

ACL 2025finding

Large language models (LLMs) excel in many tasks but struggle to accurately quantify uncertainty in their generated responses. This limitation makes it challenging to detect misinformation and ensure reliable decision-making. Existing uncertainty quantification (UQ) methods for LLMs are primarily pr…

2025

DyMoDreamer: World Modeling with Dynamic Modulation

NeurIPS 2025poster

A critical bottleneck in deep reinforcement learning (DRL) is sample inefficiency, as training high-performance agents often demands extensive environmental interactions. Model-based reinforcement learning (MBRL) mitigates this by building world models that simulate environmental dynamics and genera…

Cited by 0SourceScholar
2025

Stochastic Trajectory Optimization for Robotic Skill Acquisition From a Suboptimal Demonstration

RA-L 2025

Learning from Demonstration (LfD) has emerged as a crucial method for robots to acquire new skills. However, when given suboptimal task trajectory demonstrations with shape characteristics reflecting human preferences but subpar dynamic attributes such as slow motion, robots not only need to mimic t

Cited by 0SourcecodeScholar
2024

What If the Input is Expanded in OOD Detection?

NeurIPS 2024poster

Out-of-distribution (OOD) detection aims to identify OOD inputs from unknown classes, which is important for the reliable deployment of machine learning models in the open world. Various scoring functions are proposed to distinguish it from in-distribution (ID) data. However, existing methods genera…