Reinforcement Learning under a Multi-agent Predictive State Representation Model: Method and Theory
Zhi Zhang, Zhuoran Yang, Han Liu, Pratap Tokekar, Furong Huang
Abstract
This paper proposes a new algorithm for learning the optimal policies under a novel multi-agent predictive state representation reinforcement learning model. Compared to the state-of-the-art methods, the most striking feature of our approach is the introduction of a dynamic interaction graph to the model, which allows us to represent each agent's predictive state by considering the behaviors of its ``neighborhood'' agents. Methodologically, we develop an online algorithm that simultaneously learns the predictive state representation and agent policies. Theoretically, we provide an upper bound of the -norm of the learned predictive state representation. Empirically, to demonstrate the efficacy of the proposed method, we provide thorough numerical results on both a MAMuJoCo robotic learning experiment and a multi-agent particle learning environment.
Ask about this paper
Ask your agent about it.
Lune has read the top-tier papers around this one, so every answer names the papers it rests on.
Cited by top-tier papers5
- Optimistic MLE: A Generic Model-Based Algorithm for Partially Observable Sequential Decision MakingQinghua Liu, Praneeth Netrapalli, Csaba Szepesvári, Chi JinSTOC 2023 · 7 citations
- On the Role of Information Structure in Reinforcement Learning for Partially-Observable Sequential Teams and GamesAwni Altabaa, Zhuoran YangNeurIPS 2024 · 5 citations
- PAC Reinforcement Learning for Predictive State RepresentationsWenhao Zhan, Masatoshi Uehara, Wen Sun, Jason D. LeeICLR 2023 · 1 citation
- Provable Benefits of Multi-task RL under Non-Markovian Decision Making ProcessesRuiquan Huang, Yuan Cheng, Jing Yang, Vincent Tan et al.ICLR 2024
- From Embedding to Control: Representations for Stochastic Multi-Object SystemsXiaoyuan Cheng, Yiming Yang, Wei Jiang, Chenyang Yuan et al.ICLR 2026
Related papers
- Recursive Reasoning Graph for Multi-Agent Reinforcement LearningXiaobai Ma, David Isele, Jayesh K. Gupta, Kikuo Fujimura et al.AAAI 2022 · 8 citations
- Local Policies for Graph-Structured Markov Decision ProcessesFathima Faizal, Asuman Ozdaglar, Martin WainwrightICML 2026
- Multi-Agent Actor-Critic with Hierarchical Graph Attention NetworkHeechang Ryu, Hayong Shin, Jinkyoo ParkAAAI 2020 · 143 citations
- Model-based Reinforcement Learning for Parameterized Action SpacesRenhao Zhang, Haotian Fu, Yilin Miao, George KonidarisICML 2024 · 8 citations
- Graph Diffusion for Robust Multi-Agent CoordinationXianghua Zeng, Hang Su, Zhengyi Wang, Zhiyuan LinICML 2025
