3 papers
cs.LG2025
Energy-Weighted Flow Matching for Offline Reinforcement Learning
Shiyuan Zhang, Weitong Zhang, Quanquan Gu
This paper investigates energy guidance in generative modeling, where the target distribution is defined as , with $…
cs.LG2024
Uncertainty-Aware Reward-Free Exploration with General Function Approximation
Junkai Zhang, Weitong Zhang, Dongruo Zhou +1
Mastering multiple tasks through exploration and learning in an environment poses a significant challenge in reinforcement learning (RL). Unsupervised RL has been introduced to add…
cs.LG2024
Causal Graph ODE: Continuous Treatment Effect Modeling in Multi-agent Dynamical Systems
Zijie Huang, Jeehyun Hwang, Junkai Zhang +6
Real-world multi-agent systems are often dynamic and continuous, where the agents co-evolve and undergo changes in their trajectories and interactions over time. For example, the C…