cs.MAApr 21, 2025

MARFT: Multi-Agent Reinforcement Fine-Tuning

Authors: Junwei Liao, Muning Wen, Jun Wang, Weinan Zhang

Organizations: Shanghai Jiao Tong University · Shanghai Innovation Institute · OPPO Research Institute

Abstract

Large Language Model (LLM)-based Multi-Agent Systems (LaMAS) offer a promising approach to collaborative tasks, but applying reinforcement learning to these systems remains challenging because classical multi-agent reinforcement learning assumptions do not fully capture their dependency structures, heterogeneous roles, and autoregressive outputs. This article provides a tutorial-style overview of the transition from reinforcement learning to reinforcement fine-tuning in single-agent and multi-agent settings, highlighting the differences between classical MARL and LLM-based multi-agent optimization. We introduce Multi-Agent Reinforcement Fine-Tuning (MARFT) and formalize LaMAS optimization as a Flexible Markov Game (Flex-MG). The MARFT framework combines predecessor-conditioned policy optimization, a centralized critic, parameter-efficient adapters, and token-level credit assignment. Our token-level analysis clarifies how within-action discounting affects returns and establishes consistency with action-level returns when the token discount is one. Experiments on fixed two- to four-agent mathematical and coding workflows compare MARFT with and without policy parameter sharing against IPPO. Both variants report higher mean evaluation scores in all 24 benchmark-team-size settings, while the variant without parameter sharing shows the clearest advantage in final training reward. Role-specification ablations examine how prescribed roles affect training. We conclude with application perspectives and open challenges, including dynamic workflows, sample efficiency, and system-level reliability. Code: https://github.com/jwliao-ai/MARFT.

Figures & tables

Explore similar work

CardsList
  1. LangMARL: Natural Language Multi-Agent Reinforcement Learning

    Apr 1, 2026Huaiyuan Yao, Longchao Da, Xiaoou Liu +3RL for LLM AgentsMulti-Agent Reinforcement Learning

  2. Dr. MAS: Stable Reinforcement Learning for Multi-Agent LLM Systems

    Feb 9, 2026Lang Feng, Longtao Zheng, Shuo He +2Multi-Agent LLM SystemsMulti-Agent System Optimization

  3. UnityMAS-O: A General RL Optimization Framework for LLM-Based Multi-Agent Systems

    May 26, 2026Yiqun Chen, Wei Yang, Erhan Zhang +14Multi-Agent LLM SystemsMulti-Agent System Optimization