Multi-agent reinforcement learning (MARL) commonly relies on a centralized critic to estimate the value function. However, learning such a critic from scratch is highly sample-inefficient and often lacks generalization across environments. At the same time, large vision-language-action models (VLAs) trained on internet-scale data exhibit strong multimodal reasoning and zero-shot generalization capabilities, yet directly deploying them for robotic execution remains computationally prohibitive, particularly in heterogeneous multi-robot systems with diverse embodiments and resource constraints. To address these challenges, we propose Multi-Agent Vision-Language-Critic Models (MA-VLCM), a framework that replaces the learned centralized critic in MARL with a pretrained vision-language model fine-tuned to evaluate multi-agent behavior. MA-VLCM acts as a centralized critic conditioned on natural language task descriptions, visual trajectory observations, and structured multi-agent state information. By eliminating critic learning during policy optimization, our approach significantly improves sample efficiency while producing compact execution policies suitable for deployment on resource-constrained robots. Results show good zero-shot return estimation on models with differing VLM backbones on in-distribution and out-of-distribution scenarios in multi-agent team settings
Figures & tables
Fig. 1: Overall architecture of the MA-VLCM training ( - - ) and inference (- -) directions. The MA-VLCM dataset ( D ) contains multi-modal, multi-agent dataset that is used to send trajectories of vision It , textual ℓ and agent observation ot and actions at data to train the VLCM. The main components that are trained are the GAT, Value head and LoRA adaptors on the vision encoder and language model. The output embeddings from the the encoders et and the GAT are concatentated before sending them to the language model. The output of the value head would be the estimated long-term return of the trajectory of data along with the language prompt sent during inference.
Fig. 2: (a) The Robotic Warehouse (RWARE) 3D environment created in Isaac Sim. (b) A rendering of rware-4ag grid environment with 4 agents and 4 requested boxes (in green) describing the rasterized semantic image, (c) and its equivalent BEV camera image collected from the same environment shown in (a).
Fig. 3: An unstructured offroad environment created in high-fidelity simulator with 3 Clearpath Jackal robots (left), and a corresponding rasterized top-down view (right), with the agents represented as color-coded markers moving to their target location on a traversability map generated from the high-fidelity simulator.
Parameter
Value
LoRA Rank ( r ), Scaling ( α )
r=16 , α=32
Graph Attention Network (GAT) Dim
256
Optimizer
AdamW
Learning Rate
3×10−4
Effective Batch Size
32
Video Clip Length ( H )
32 frames
TABLE I: Key MA-VLCM Hyperparameters
Metric
In-Distribution
Out-of-Distribution
RWARE
Offroad
RWARE
Offroad
GNN + Value Head + LLaVA-0.5B-OV (Baseline)
seconds/iteration ↓
0.20
Spearman ρ↑
0.80
0.82
0.78
0.86
MSE ↓
6.11
17.71
13.24
22.08
MAE ↓
5.10
2.04
5.22
3.03
TABLE II: Value Estimation: In-Distribution vs. Out-of-Distribution
Fig. 4: Value Estimation on IID vs OOD for RWARE environment with 0.5B VLCM
Fig. 5: Value Estimation on IID vs OOD for Offroad environment with 0.5B VLCM
Fig. 6: Box plot comparison for all 4 models on Spearman correlations and MSE values across both the environments
We study reinforcement learning (RL) methods for cooperative multi-agent Vision-Language-Action (VLA) models. This problem is challenging because VLAs are pretrained on large-scale single-agent data and therefore lack the fine-grained coordination skills required for inter-robot collaboration. Supervised fine-tuning (SFT) on multi-robot demonstrations partially bridges this gap, but its performance is bounded by the demonstration data and cannot improve from its own experience. We present a three-stage reinforced fine-tuning (RFT) pipeline for multi-agent VLAs. First, initialization-aware data collection sweeps over initial configurations and invokes human demonstrations only when the pretrained VLA repeatedly fails, yielding robustness to initialization shift with reduced human cost. Second, offline credit-filtered tuning assigns credit to individual agents and fine-tunes on per-agent trajectories with positive advantage rather than on entire joint rollouts. Third, we find existing online RL for VLAs are less effective for hard multi-agent tasks, which we attribute to noisy co-exploration and unstable updates. We instead use online latent-space fine tuning, which freeze the VLA and perform RL in its latent noise space. We evaluate our multi-agent VLA with both π0 and π0.5 backbones across 11 tasks in RoboTwin, RoboFactory and real-world manipulation with two Franka robots. Our multi-agent VLA improves the average success rate by +23.1%, +16.4%, and +44% on RoboTwin, RoboFactory, and real-world tasks, respectively. Code available at https://anonymous.4open.science/r/mavla_rft-2BC0/.
Ruixiao Xu, Wong Lik Hang Kenny, Zhiqian Liu +12
Beihang University · The Chinese University of Hong Kong · PKU-Psibot Lab +4
Vision-Language-Action (VLA) models show strong generalization for robotic control, but finetuning them with reinforcement learning (RL) is constrained by the high cost and safety risks of real-world interaction. Training VLA models in interactive world models avoids these issues but introduces several challenges, including pixel-level world modeling, multi-view consistency, and compounding errors under sparse rewards. Building on recent advances across large multimodal models and model-based RL, we propose VLA-MBPO, a practical framework to tackle these problems in VLA finetuning. Our approach has three key design choices: (i) adapting unified multimodal models (UMMs) for data-efficient world modeling; (ii) an interleaved view decoding mechanism to enforce multi-view consistency; and (iii) chunk-level branched rollout to mitigate error compounding. Theoretical analysis and experiments across simulation and real-world tasks demonstrate that VLA-MBPO significantly improves policy performance and sample efficiency, underscoring its robustness and scalability for real-world robotic deployment.
Zhilong Zhang, Haoxiang Ren, Yihao Sun +6
National Key Laboratory for Novel Software Technology, Nanjing University, Nanjing, China · School of Artificial Intelligence, Nanjing University, Nanjing, China · Mila - Quebec AI Institute +1
Reinforcement learning (RL) post-training of Vision-Language-Action (VLA) models has shown strong promise for robotic manipulation. Among RL methods, critic-based approaches rely on a value estimator that predominantly operates on single-frame observations or single-frame VLM backbone latents, which is a fundamental mismatch with the partially observable nature of robot control. A naive approach to incorporate observation history into the critic incurs exponential complexity with high-dimensional visual space, and still fails because pure scalar-return regression provides insufficient supervision for learning cross-temporal dynamics. We identify the root cause as a state approximation problem: without an explicit world modeling objective, the critic's representation cannot capture the temporal structure needed for accurate value estimation. To address this, we propose the World Critic Model (WCM), built on a lightweight LeJEPA architecture; WCM jointly predicts future latent state and estimates values, such that the critic's representation is explicitly trained to capture temporal dynamics rather than merely regress scalar returns. WCM integrates seamlessly into both on-policy and off-policy training pipelines and is compatible with state-of-the-art VLA backbones including Pi0, Pi0.5, and OpenVLA-OFT. Extensive experiments on 149 tasks across four benchmarks demonstrate that WCM consistently achieves state-of-the-art performance in both in-distribution and out-of-distribution settings, with particularly strong generalization gains. We further validate WCM on seven real-world manipulation tasks using OpenVLA-OFT and Pi0.5 with off-policy RL, confirming stable deployment across diverse settings.
Senyu Fei, Xiaopeng Yu, Siyin Wang +3
1Tongji University · 3Fudan University · 2Shanghai Innovation Institute