activity
20242026
collaborators
Showing cs.LGShow all

6 papers · 1 filter

cs.LG2026

DiffNash: Finding Nash Equilibrium in High-Dimensional Continuous Action Spaces with Diffusion Policies

Haichuan Wang, Anagha Satish, Jiale Zheng +4

Computing equilibria in high-dimensional continuous games requires learning and representing rich, multi-modal strategy distributions, something conventional neural policies strugg…

cs.LG2026

Reward Shaping for (Inference-Time) Alignment: A Stackelberg Game Perspective

Haichuan Wang, Tao Lin, Lingkai Kong +3

Existing alignment methods directly use the reward model learned from user preference data to optimize an LLM policy, subject to KL regularization with respect to the base policy.…

cs.LG2025

Generative AI Against Poaching: Latent Composite Flow Matching for Wildlife Conservation

Lingkai Kong, Haichuan Wang, Charles A. Emogor +3

Poaching poses significant threats to wildlife and biodiversity. A valuable step in reducing poaching is to forecast poacher behavior, which can inform patrol planning and other co…

cs.LG2025

Composite Flow Matching for Reinforcement Learning with Shifted-Dynamics Data

Lingkai Kong, Haichuan Wang, Tonghan Wang +2

Incorporating pre-collected offline data can substantially improve the sample efficiency of reinforcement learning (RL), but its benefits can break down when the transition dynamic…

cs.LG2025

Rule-Bottleneck Reinforcement Learning: Joint Explanation and Decision Optimization for Resource Allocation with Language Agents

Mauricio Tec, Guojun Xiong, Haichuan Wang +2

Deep Reinforcement Learning (RL) is remarkably effective in addressing sequential resource allocation problems in domains such as healthcare, public policy, and resource management…

cs.LG2024

Amelia: A Large Dataset and Benchmark for Airport Surface Movement Forecasting

Ingrid Navarro, Pablo Ortega-Kral, Jay Patrikar +6

Demand for air travel is rising, straining existing aviation infrastructure. In the US, more than 90% of airport control towers are understaffed, falling short of FAA and union sta…