MARFT: Multi-Agent Reinforcement Fine-Tuning
cs.MA, cs.AI, cs.LG, cs.RO
Submitted: 2025-04-21
Updated: 2026-10-12
Comments: 40 pages, 5 figures, 3 tables. Extended version of the paper accepted at DAI 2026. Code: https://github.com/jwliao-ai/MARFT
Code: https://github.com/jwliao-ai/MARFT
License: http://creativecommons.org/licenses/by/4.0/
The gist: 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
Terminology
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.
Sources
- Why Do Multi-Agent LLM Systems Fail?
- Octopus: On-device language model for function calling of software APIs
- AD-AutoGPT: An Autonomous GPT for Alzheimer's Disease Infodemiology
- DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning
- TinyAgent: Function Calling at the Edge
- Promptbreeder: Self-Referential Self-Improvement Via Prompt Evolution
- AReaL: A Large-Scale Asynchronous Reinforcement Learning System for Language Reasoning
- Accelerating scientific discovery with Co-Scientist
- OlympiadBench: A Challenging Benchmark for Promoting AGI with Olympiad-Level Bilingual Multimodal Scientific Problems
- Emergence of Locomotion Behaviours in Rich Environments
- Measuring Mathematical Problem Solving With the MATH Dataset
- REINFORCE++: Stabilizing Critic-Free Policy Optimization with Global Advantage Normalization
- OpenRLHF: An Easy-to-use, Scalable and High-performance RLHF Framework
- AgentCoder: Multi-Agent-based Code Generation with Iterative Testing and Optimisation
- LiveCodeBench: Holistic and Contamination Free Evaluation of Large Language Models for Code
- From LLMs to LLM-based Agents for Software Engineering: A Survey of Current, Challenges and Future
- Retrieval-Augmented Generation for Knowledge-Intensive NLP Tasks
- Solving Quantitative Reasoning Problems with Language Models
- Let's Verify Step by Step
- ToolSandbox: A Stateful, Conversational, Interactive Evaluation Benchmark for LLM Tool Use Capabilities
Related papers
- Highway Congestion Reduction through Reinforcement Learning Based Eulerian Headway Control
- ORACLE-SWE: Quantifying the Contribution of Critical Information Signals on SWE Agents
- You Only Align Once: Propagating Cooperative Behaviors in Multi-Agent Systems through Seed Agents
- Deny Without Disabling: Authorization-Paired Evaluation and Control for Multi-Agent Systems
- MA-SAPO: Multi-Agent Reasoning for Score-Aware Prompt Optimization
- PeroMAS: A Multi-agent System of Perovskite Material Discovery