3 papers
cs.AI2026
World Action Planner: Generalizable Decision-Making with Action-Conditioned World Models
Xiangcheng Zhang, Yilun Du
Building generalizable agents for diverse applications remains a fundamental challenge. While imitation learning-based policies succeed in specific training environments, they ofte…
cs.LG2025
Inference-time Scaling of Diffusion Models through Classical Search
Xiangcheng Zhang, Haowei Lin, Haotian Ye +4
Classical search algorithms have long underpinned modern artificial intelligence. In this work, we tackle the challenge of inference-time control in diffusion models -- adapting ge…
cs.LG2024
Provable Risk-Sensitive Distributional Reinforcement Learning with General Function Approximation
Yu Chen, Xiangcheng Zhang, Siwei Wang +1
In the realm of reinforcement learning (RL), accounting for risk is crucial for making decisions under uncertainty, particularly in applications where safety and reliability are pa…