5 papers · 1 filter
Reflection-Driven Control for Trustworthy Code Agents
Bin Wang, Jiazheng Quan, Xingrui Yu +3
Contemporary large language model (LLM) agents are remarkably capable, but they still lack reliable safety controls and can produce unconstrained, unpredictable, and even actively…
Training Diffusion Policies via Prior-Mapping Co-Evolution
Chubin Zhang, Zhenglin Wan, Feng Chen +7
Reinforcement learning (RL) faces a persistent tension: policies that are stable to optimize (e.g., Gaussians) are often too simple to represent the multimodal action distributions…
FM-IRL: Flow-Matching for Reward Modeling and Policy Regularization in Reinforcement Learning
Zhenglin Wan, Jingxuan Wu, Xingrui Yu +4
Flow Matching (FM) has shown remarkable ability in modeling complex distributions and achieves strong performance in offline imitation learning for cloning expert behaviors. Howeve…
Letting Trajectories Spread: Quality-Preserving Control for Diverse Flow Matching
Jingxuan Wu, Zhenglin Wan, Xingrui Yu +4
Flow-based text-to-image models follow deterministic trajectories, making it costly to explore diverse modes under limited sampling budgets. Existing approaches to improving divers…
UniErase: Towards Balanced and Precise Unlearning in Language Models
Miao Yu, Liang Lin, Guibin Zhang +7
Large language models (LLMs) require iterative updates to address the outdated information problem, where LLM unlearning offers an approach for selective removal. However, mainstre…