VLAFlow: A Unified Training Framework for Vision-Language-Action Models via Co-training and Future Latent Alignment
- 01Action-Only Pre-training on Heterogeneous Data Can Make Your Robot Worse
- 02Language Descriptions and Future-State Prediction Are Complementary Regularizers That Stabilize Heterogeneous Training
- 03Future Latent Alignment Is Disproportionately Valuable for Cross-Platform Transfer
- 04The Loss Ratio Recipe Matters More Than Expected
- 05VLM-Side Adaptation Is Non-Negotiable for Downstream Performance
The One-Line Summary: This paper from Li Auto and Beijing University of Posts and Telecommunications runs the most controlled comparison yet of how you should pre-train a robot foundation model — and finds that simply scaling action-imitation data can actively hurt downstream performance. Adding language intent descriptions and/or future-state prediction as auxiliary training signals is what actually makes heterogeneous robot data transferable.
1. Key Themes
Action-Only Pre-training on Heterogeneous Data Can Make Your Robot Worse
The paper's most operationally important finding is that the dominant pre-training paradigm — just predict actions from observations — is fragile when trained on diverse, multi-robot datasets. The authors define "negative transfer" as a pre-trained model performing worse than a model with no pre-training at all, and they document it happening systematically.
From Section 4.2 (Table 2): MindPI (Full PT) scores 68.2% RT-1 Visual Matching and 55.5% RT-1 Visual Augmentation — worse than the no-pre-training baseline of 75.7% and 60.4% respectively. The paper states directly: "MindPI (Full PT) improves over the no-pretraining baseline on WidowX but degrades substantially on RT-1 VM/VA, reflecting the instability and potential negative transfer of action-only pre-training under heterogeneous robot data."
The data ablations in Section 4.5 (Table 11) make this concrete: using the RoboCOIN subset drops WidowX average success from a 63.0% no-PT baseline to just 40.4%. Different data subsets don't just help different amounts — some actively destroy transfer capability.
Language Descriptions and Future-State Prediction Are Complementary Regularizers That Stabilize Heterogeneous Training
The paper's core contribution is demonstrating that two auxiliary signals act as "intermediate constraints" that smooth out the chaos of training on data from dozens of different robots, action spaces, and sampling frequencies.
Language supervision (MindLPI) forces the model to verbalize what it's doing ("move forward 12 cm, close gripper"), preserving the VLM's semantic grounding. Future latent prediction (MindWPI) forces the model to predict what the world will look like after an action, using a frozen V-JEPA 2 encoder. The combined MindLWPI achieves 99.1% LIBERO average, 74.8% LIBERO-Plus Total, and 75.5% WidowX — the best across-the-board result in Section 4.2.
The paper frames this as a "meta-action space" in Section 4.7: "Language descriptions explicitly express high-level action intent, while future latent features capture action-induced state changes. When combined, these two forms of supervision better smooth the optimization space between action intent and real control signals."
Future Latent Alignment Is Disproportionately Valuable for Cross-Platform Transfer
While the combined MindLWPI wins overall, the pure future-latent model MindWPI actually beats it on the RT-1 platform specifically (86.7% VM vs. 84.4% VM). This isn't noise — it points to a meaningful insight about what future-state prediction is actually buying you.
From Section 4.3 (Table 5): MindWPI scores 86.7% RT-1 VM and 71.1% RT-1 VA, outperforming every public baseline including π₀ (58.8% VM), OpenVLA-OFT (63.0% VM), and SpatialVLA (75.1% VM). The paper explains: "MindWPI achieves the best RT-1 VM/VA results, showing that future latent alignment provides an effective training signal for modeling action outcomes and state transitions under cross-platform distribution shifts."
The Loss Ratio Recipe Matters More Than Expected — and Has Unintuitive Optimal Values
This is a practitioner-critical finding buried in the ablations. Pre-training with a 1:1 action-to-latent loss ratio, then fine-tuning with a 0.1:1 ratio, dramatically outperforms other configurations.
From Section 4.5 (Table 8): Using a 1:1 fine-tuning ratio from the same pre-trained checkpoint yields only 61.5% WidowX, while the recommended 0.1:1 fine-tuning ratio yields 74.5% — a 13 percentage point swing from a single hyperparameter. Even dropping the latent loss entirely at fine-tuning time (70.8%) beats using it equally-weighted. The paper concludes: "A weak latent loss during pre-training does not provide a sufficiently strong state-transition constraint, while the balanced 1:1 pre-training objective better couples action modeling with future-state prediction."
VLM-Side Adaptation Is Non-Negotiable for Downstream Performance
The LoRA experiments in Section 4.6 reveal a sharp asymmetry that has direct implications for deployment engineering. Adapting only the action expert — the "cheap" fine-tuning path — performs poorly. Adapting the VLM is where the leverage is.
From Table 12: Action-only LoRA at r=512 (103.8M parameters) achieves 78.6% LIBERO average. VLM-only LoRA at r=256 (94.4M parameters) achieves 97.1% — nearly matching full fine-tuning at 97.5% (MindPI Full PT) with a similar parameter count. The paper notes: "For VLM-based VLA models, downstream adaptation requires not only updating the action expert, but also providing sufficient low-rank adaptation capacity for vision-language representations."
2. Contrarian Perspectives
More Heterogeneous Pre-training Data Is Not Automatically Better — It Can Be Actively Harmful
The conventional wisdom in foundation model scaling is that more diverse data helps generalization. This paper challenges that directly in the context of robot pre-training.
The data source ablation in Section 4.5 (Table 11) shows the RoboCOIN subset dropping performance from 63.0% (no PT) to 40.4% — a 22.6 point regression — on action-only pre-training. The paper argues this isn't a data quality problem per se: "Negative transfer is not caused by robot-data pre-training per se, but by the difficulty of aligning heterogeneous embodiments, action definitions, task distributions, and visual domains under a purely low-dimensional action-supervision objective."
The implication is that companies scaling up robot data collection without investing in training paradigm design may be building toward a performance cliff, not a performance gain.
Freezing the VLM During Pre-training Is a Surprisingly Competitive Strategy — But Not for the Reason You'd Think
Most practitioners freeze the VLM backbone during robot pre-training to "protect" language capability. The paper validates this intuition but reveals the mechanism is more nuanced than simply "don't corrupt the VLM."
From Section 4.2 (Table 2): MindPI (Frozen VLM) achieves 74.9% LIBERO-Plus Total — higher than MindPI (Full PT) at 68.8%, and competitive with MindLPI (72.3%). The paper notes in Section 4.3: "MindPI (Frozen VLM) better preserves the generalization capability of the original VLM." However, Section 1 also states that freezing "does not fully exploit robot data to learn action-related state changes" — meaning it's a defensive strategy, not an offensive one. The real gains come from adding auxiliary supervision that lets you update the full model without destroying generalization.
Gradient Flow Into the VLM From Action Loss Is a Feature, Not a Bug
Many VLA training pipelines use stop-gradient to prevent low-level action supervision from contaminating the VLM's language representations. The paper's ablation shows this is backwards.
From Section 4.5 (Table 7): MindLPI with stop-gradient on the action-to-VLM gradient achieves only 45.8% LIBERO-Plus Total. Without stop-gradient, it achieves 72.3% — a 26.5 point improvement. The paper concludes: "Backpropagation from the action loss into the VLM is not merely noise; it may provide action-outcome alignment signals for vision-language representations." This directly challenges the common engineering practice of isolating the VLM from action-level supervision.
3. Companies Identified
Physical Intelligence (π₀, π₀.5) The direct architectural and paradigm reference point for this entire paper. VLAFlow uses a "π₀-style architecture" as its shared backbone, and both π₀ and π₀.5 appear as public baselines throughout. π₀ scores 58.8% RT-1 VM vs. MindWPI's 86.7% (Table 5). π₀.5 represents the "VLM co-training paradigm" that MindLPI is designed to study. The paper cites both extensively in Sections 1 and 2.1.
"The conventional action-modeling paradigm represented by π₀ usually predicts action chunks directly from visual observations and language instructions... Under such action-only modeling, however, the pre-training data distribution often plays a decisive role in downstream performance." (Section 1)
Li Auto (Li Auto Inc.) The primary industry sponsor of this research. Five of the seven authors are affiliated with Li Auto Inc. This is notable — a major Chinese automotive company is doing serious VLA foundation model research, suggesting autonomous/embodied AI investment at the infrastructure level well beyond cars.
Affiliation listed as "Li Auto Inc." for authors Fengfa Li, Hongjin Ji, Lei Ren, Kun Zhan, and Yan Xie (Author list, Abstract)
Google DeepMind / Google (RT-1, RT-2, Open X-Embodiment) RT-1 and RT-2 are cited as foundational prior work (Section 2.1), and RT-1 is one of two robot platforms in the SimplerEnv benchmark used throughout the paper. The Open X-Embodiment dataset, a Google-led initiative, is a core component of the OXEMix training corpus.
"Open X-Embodiment aggregates large-scale demonstration trajectories across many robot platforms and is currently one of the most widely used cross-embodiment pre-training corpora." (Section 2.1)
NVIDIA (GR00T-N1) Cited as a public baseline in Table 3 (LIBERO), scoring 93.9% average vs. MindLWPI's 99.1%. Relevant as a competing foundation model approach.
"GROOT-N1 [5] 94.4 97.6 93.0 90.6 93.9" (Table 3, Section 4.3)
Meta AI (V-JEPA, V-JEPA 2) V-JEPA 2 is the frozen latent feature extractor used in MindWPI and MindLWPI — it is a load-bearing component of the paper's best-performing paradigms. The paper's future latent alignment results are directly dependent on Meta's self-supervised video model.
"We use a frozen V-JEPA 2 model as the latent feature extractor... Given current and future frames, we extract latent representations z_cur and z_fut." (Section 3.5)
Alibaba / Qwen Team (Qwen3-VL-4B-Instruct) The VLM backbone for all four paradigms. All performance results in this paper are built on top of Qwen3-VL-4B-Instruct.
"In our implementation, we use Qwen3-VL-4B-Instruct as the VLM backbone." (Section 3.2)
4. People Identified
Lei Ren — Li Auto Inc. Project leader and corresponding author. The driving force behind the VLAFlow framework and the controlled comparison methodology. Based at Li Auto's AI research division.
Listed as "Project leader" and "†Corresponding author" in the author list.
Guoyang Xia — Li Auto Inc. / Beijing University of Posts and Telecommunications Equal first author. PhD student split between BUPT and Li Auto, suggesting this is coming from an industry-academia collaboration model becoming common in Chinese Physical AI.
"Guoyang Xia¹·², *Equal contribution." (Author list)
Fengfa Li — Li Auto Inc. Equal first author alongside Xia. Together they are the primary architects of the VLAFlow framework.
"Fengfa Li¹, *Equal contribution." (Author list)
Fangxiang Feng — Beijing University of Posts and Telecommunications Co-corresponding author from the academic side, from BUPT's School of Artificial Intelligence. Represents the academic grounding of this work.
"Fangxiang Feng²,‡ Corresponding author." (Author list)
Mido Assran et al. (Meta AI) — V-JEPA 2 Not authors of this paper, but their V-JEPA 2 model is a critical enabling technology for MindWPI and MindLWPI. The success of future latent alignment in this paper is a downstream validation of Meta's self-supervised video pre-training approach.
"V-JEPA 2 further demonstrates the potential of scaled self-supervised video pre-training and lightweight robot-data post-training for robot control." (Section 2.2, citing Assran et al. [2025])
5. Operating Insights
Do Not Deploy Action-Only Pre-training on a Heterogeneous Data Corpus Without Auxiliary Supervision — Validate Transfer Before Committing
If you are building a foundation model on mixed robot data (different arms, grippers, sampling rates, task types), action-only pre-training is not a safe default. The paper documents that it can produce negative transfer — making your downstream fine-tuned model worse than if you had never pre-trained at all. This is not a theoretical risk; it shows up concretely across multiple benchmarks.
The practical decision rule: if your pre-training corpus mixes more than one robot platform or action space definition, default to MindLWPI-style training with both language action descriptions and future latent prediction. The engineering overhead is manageable — language supervision uses a lightweight autoregressive head that is dropped at inference time, and V-JEPA 2 latent features can be compressed 4x with average pooling before being fed to the action expert.
"Action-only full-parameter pre-training can damage transfer ability on downstream tasks with large distribution gaps... This phenomenon indicates that the key to VLA pre-training lies not only in data scale, but also in whether the pre-training objective can learn transferable intermediate representations from highly heterogeneous data." (Section 1)
Set Your Fine-tuning Loss Ratio to 0.1:1 (Latent:Action) Even If You Pre-trained at 1:1
This is a directly actionable engineering recipe. The paper shows a 13-point performance swing from this single hyperparameter. The intuition: during pre-training, you need strong latent supervision to build representations that understand state transitions. During fine-tuning on a target platform, you want the latent loss as a light regularizer — not a competing objective — so the model can specialize to the new embodiment without fighting itself.
"Starting from the same act+lat (1:1) pre-trained checkpoint, a 1:1 FT ratio performs poorly, likely because the latent constraint becomes too strong during target-domain adaptation and competes with the action objective... our default MindWPI recipe uses a balanced action-latent objective during pre-training and a weaker future latent regularizer during downstream fine-tuning, namely act+lat (1:1) for pre-training and λ_lat^ft : λ_act^ft = 0.1:1 for fine-tuning." (Section 4.5)
When Fine-tuning for Production Deployment, Allocate LoRA Budget to the VLM — Not the Action Expert
For teams that need parameter-efficient fine-tuning (edge deployment, frequent task updates, limited GPU budget), the paper provides a clear allocation rule: VLM-side LoRA at r=256 (94M parameters) nearly matches full fine-tuning on LIBERO. Action-expert-only LoRA at twice the parameter count (r=512, 104M parameters) falls far short. If you can only adapt one side, adapt the VLM.
"VLM-side LoRA already approaches full fine-tuning with around 100M trainable parameters, while both-side LoRA achieves the highest average score under a larger parameter budget. This suggests that for VLM-based VLA models, downstream adaptation requires not only updating the action expert, but also providing sufficient low-rank adaptation capacity for vision-language representations." (Section 4.6)
6. Overlooked Insights
Platform-Specific Fine-tuning Recovers Substantially More Value From Pre-training Than Mixed Fine-tuning — and the Gap Grows With Pre-training Quality
The paper includes a calibration table (Section 4.4, Table 6) that most readers will skip because it's framed as a "methodology note." But it reveals something important: for pre-trained models, fine-tuning on mixed multi-platform data actively suppresses the gains from pre-training compared to fine-tuning on single-platform data.
For MindPI without pre-training, mixed vs. Bridge-only fine-tuning is nearly the same (63.0% vs. 59.6%, a -3.4 point gap). But for MindPI with pre-training, mixed fine-tuning scores 55.5% vs. Bridge-only at 65.9% — a +10.4 point gap in favor of platform-specific fine-tuning. The paper attributes this to "cross-platform gradient conflicts" during adaptation, consistent with the multi-task gradient conflict literature. The implication for operators: the more capable your pre-trained model, the more important it is to use deployment-specific (not blended) fine-tuning data. Mixing data at fine-tuning time can hide pre-training value entirely.
"For pre-trained models, the model has already acquired a certain cross-embodiment action prior from OXEMix; applying Bridge and RT-1 simultaneously during downstream adaptation may introduce cross-platform gradient conflicts or compromise optima, thereby weakening adaptation to a specific target platform." (Section 4.4)
The Future-Frame Offset Hyperparameter Has a Performance Cliff at 32 Frames — With No Published Guidance in Prior Work
The paper tests two future-frame offsets for the latent prediction target: 8 frames ahead and 32 frames ahead. The 32-frame offset drops WidowX performance from 73.4% to 59.9% in the no-pre-training ablation — a 13.5 point collapse (Section 4.5, Table 9).
This matters because no prior VLA work using future prediction has systematically characterized this sensitivity. If you're predicting too far into the future, the latent target captures state changes caused by many subsequent actions, not the current action chunk, and the supervision signal becomes uninformative for learning immediate control. The paper notes: "A future-frame interval that is too long may introduce state changes weakly associated with the current action." Teams implementing future latent prediction need to tune this parameter carefully relative to their robot's control frequency and action chunk length — and the default of T=16 action steps with an 8-frame lookahead appears to be a meaningful design choice, not an arbitrary one.
"Offset 32 is clearly weaker than the default Offset 8, suggesting that a future-frame interval that is too long may introduce state changes weakly associated with the current action." (Section 4.5, Table 9)