6 papers
Scheduling Thoughts: Learning the Order of Thought in Diffusion Language Models
Jiawei Xu, Minghui Liu, Aakriti Agrawal +2
Masked diffusion language models decode by iteratively unmasking tokens, where the unmasking order defines an "order of thought" that strongly influences generation quality yet is…
Guava: Distilling Frontier VLMs into a Compact Agent through a Robotic Manipulation Harness
Haowen Liu, Xirui Li, Shaoxiong Yao +5
Language models trained on large-scale vision-language data have demonstrated strong potential for embodied agents. Harnessing models through embodied tools use offers a promising…
GenFlowRL: Shaping Rewards with Generative Object-Centric Flow in Visual Reinforcement Learning
Kelin Yu, Sheng Zhang, Harshit Soora +4
Recent advances have shown that video generation models can enhance robot learning by deriving effective robot actions through inverse dynamics. However, these methods heavily depe…
PEnGUiN: Partially Equivariant Graph NeUral Networks for Sample Efficient MARL
Joshua McClellan, Greyson Brothers, Furong Huang +1
Equivariant Graph Neural Networks (EGNNs) have emerged as a promising approach in Multi-Agent Reinforcement Learning (MARL), leveraging symmetry guarantees to greatly improve sampl…
Statistical Guarantees for Lifelong Reinforcement Learning using PAC-Bayes Theory
Zhi Zhang, Chris Chow, Yasi Zhang +7
Lifelong reinforcement learning (RL) has been developed as a paradigm for extending single-task RL to more realistic, dynamic settings. In lifelong RL, the "life" of an RL agent is…
Boosting Sample Efficiency and Generalization in Multi-agent Reinforcement Learning via Equivariance
Joshua McClellan, Naveed Haghani, John Winder +2
Multi-Agent Reinforcement Learning (MARL) struggles with sample inefficiency and poor generalization [1]. These challenges are partially due to a lack of structure or inductive bia…