4 papers · 1 filter
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…
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…