← Search

Zhengmao Zhu

3 accepted papers

2026

DIVE: Scaling Diversity in Agentic Task Synthesis for Generalizable Tool Use

ICML 2026poster

Recent work increasingly synthesizes agentic tasks for post-training tool-using LLMs, yet robust generalization under shifts in tasks and toolsets remains an open challenge. We trace this brittleness to insufficient diversity in synthesized training tasks. Scaling diversity is difficult because trai…

Cited by 0SourceScholar
2023

Adversarial Counterfactual Environment Model Learning

NeurIPS 2023spotlight

An accurate environment dynamics model is crucial for various downstream tasks in sequential decision-making, such as counterfactual prediction, off-policy evaluation, and offline reinforcement learning. Currently, these models were learned through empirical risk minimization (ERM) by step-wise fit…

2023

Learning World Models with Identifiable Factorization

NeurIPS 2023poster

Extracting a stable and compact representation of the environment is crucial for efficient reinforcement learning in high-dimensional, noisy, and non-stationary environments. Different categories of information coexist in such environments -- how to effectively extract and disentangle the informati…

Cited by 16SourcePDFScholar