工作日早上 7 点读 AI,周日早上 8 点读周报订阅邮件

METAL LAB

Towards Physics of Multimodal Pretraining: Knowledge Flow, Modality Synergy, Early Unification, and Recipes

arXiv:2608.050002026-08-04

一项控制实验研究:语言、图像理解与图像生成在同一模型中联合训练时如何互相促进或干扰

这篇论文系统研究了在单一"统一多模态"模型中同时预训练文本、图像理解与图像生成时,三种能力之间知识如何流动。研究同时使用大规模真实数据和基于CLEVR的可控合成数据,发现这三种能力之间的迁移是不对称的,且依赖于具体概念。基于这些发现,作者提炼出数据配比与架构设计的实用方案,并通过训练135亿参数的MoE模型(2万亿token)在更大规模上进行了验证。

METAL LAB 解读图

统一多模态预训练中的知识流动方向

证据状态已报告实测结果

  1. 语言数据增加DCLM语言数据可同时提升图像理解和图像生成
  2. 图像理解数据增加理解数据大幅提升生成,但使纯语言表现略微下降
  3. 图像生成数据增加生成数据对语言和理解只带来轻微波动,无明显趋势
  4. CLEVR概念迁移实验颜色/形状双向迁移失败;关系/大小/数量主要从理解迁移到生成
  5. 架构与时机设计(split_ffn、早期统一)共享注意力/归一化并分离前馈网络,加上尽早联合引入视觉,能产生协同效应
这是 METAL LAB 制作的解读图,并非论文作者提供的原图。

他们做了什么

  1. 提高语言数据比例能持续改善图像理解和图像生成表现,而提高图像理解数据比例能大幅提升图像生成,但会使纯语言表现略微下降。
  2. 增加图像生成数据对语言能力或图像理解能力既没有明显帮助也没有明显损害,只带来轻微波动。
  3. 在CLEVR合成测试平台上,颜色、形状等低层属性在理解与生成之间双向都无法迁移,而空间关系、大小、数量等结构性概念能从理解迁移到生成,但反方向大多失败。
  4. 共享注意力和归一化层、仅将前馈网络按模态分离(split_ffn)的设计,能在消除完全共享(dense)架构带来的模态竞争的同时保留跨模态协同效应。
  5. 从训练早期就引入视觉数据并与语言联合训练,效果优于延迟引入视觉或按顺序训练,延迟整合会引发一种"视觉懒惰"现象,即模型过度依赖语言先验而未充分优化视觉部分。
Figure 1: Impact of scaling language data on visual understanding and generation. Increasing the language ratio universally improves both vision capabilities.
Figure 1: Impact of scaling language data on visual understanding and generation. Increasing the language ratio universally improves both vision capabilities.
Figure 2: Impact of scaling visual understanding data. It significantly benefits visual generation but degrades pure language performance.
Figure 2: Impact of scaling visual understanding data. It significantly benefits visual generation but degrades pure language performance.
Table 1: Extensive grid search of data mixing ratios across three axes. We evaluate models on Language, Visual Understanding, and Visual Generation. The searches confirm that while language requires a dominant token share, visual capabilities peak at highly specific and asymmetrical ratios. The optimal configuration emerges in the "Next" sweep at a 70/25/5 split for Language, Understanding, and Generation respectively.
Mix %LanguageVisual UnderstandingVisual Generation
LUGPPL ↓Acc ↑Gen ↑Know ↑OCR ↑V-Ctr ↑Avg ↑DPG ↑GenEval ↑CLIP-Sim ↑DiffLoss ↓
Fix MM10454519.2741.8937.926.922.942.932.70.3260.1480.2560.2984
20404017.5144.4645.730.824.044.036.10.3950.1890.2730.2802
30353516.7345.0343.129.924.942.235.00.3610.1830.2730.2826
40303016.3145.7946.131.224.245.636.80.3310.1860.2690.2946
50252515.9846.5947.230.925.444.637.00.3850.2190.2740.2804
60202015.8146.7044.633.024.843.936.60.3870.2030.2750.2883
70151515.6846.9948.132.325.246.638.10.3990.2190.2730.2996
80101015.5746.8545.932.824.045.237.00.3880.2160.2710.2868
905515.4848.0843.731.821.446.335.80.3360.2040.2730.2894
Fix Lan5054516.0545.2643.829.423.743.135.00.3580.2060.2730.2785
50104016.0346.1845.931.024.245.636.70.4010.2210.2810.2801
50153516.0646.4246.632.123.445.436.90.3700.1930.2730.2834
50203016.0246.1746.831.223.545.236.70.3900.1990.2740.2909
50252515.9846.5947.230.925.444.637.00.3850.2190.2740.2804
50302016.0146.0545.531.125.244.736.60.3700.2080.2730.2893
50351516.0545.8547.032.523.846.037.30.3990.1990.2750.2821
50401016.0046.1447.732.926.545.738.20.4200.2160.2760.2874
5045516.0346.1446.932.325.846.137.80.3920.2000.2690.2931
Next7052515.6746.5547.032.923.943.536.80.3750.2040.2710.2823
70102015.7146.3446.830.622.744.336.10.3580.2060.2730.2855
70151515.6846.9948.132.325.246.638.10.3990.2190.2730.2996
70201015.6546.6548.031.926.346.338.10.4010.2210.2720.2934
7025515.6846.8648.332.725.847.138.50.4500.2370.2750.2868
Figure 3: Impact of scaling visual generation data. Adding visual generation causes minor fluctuations in language and most understanding tasks.
Figure 3: Impact of scaling visual generation data. Adding visual generation causes minor fluctuations in language and most understanding tasks.
Figure 4: Overview of the synthetic CLEVR testbed. We extend the standard CLEVR vocabulary across five conceptual axes: color, shape, spatial relation, size, and object count. To isolate knowledge flow, specific target concepts (highlighted in red) are systematically ablated from targeted modality training streams.
Figure 4: Overview of the synthetic CLEVR testbed. We extend the standard CLEVR vocabulary across five conceptual axes: color, shape, spatial relation, size, and object count. To isolate knowledge flow, specific target concepts (highlighted in red) are systematically ablated from targeted modality training streams.
Table 2: Scaling results and controlled baseline comparisons. We evaluate our model against three controlled baselines to validate our main design choices: data mixture recipes (Balanced Recipe), architecture design style (Dense Model), and vision alignment strategy (Late-Fusion).
ModelLanguageVisual UnderstandingVisual Generation
PPL ↓Acc ↑Gen ↑Know ↑OCR ↑V-Ctr ↑Avg ↑DPG ↑GenEval ↑CLIP-Sim ↑DiffLoss ↓
Balanced Recipe11.9752.8651.5038.9025.1550.1441.420.6760.4670.3100.261
Dense Model12.1452.0350.1236.6625.4349.7440.490.6670.4590.3080.266
Late-Fusion12.2551.7849.8937.0326.2249.5040.660.6720.4710.3080.269
Full11.6754.3153.6340.1127.2351.3343.080.6890.4820.3120.272
Figure 5: Zero-shot concept transfer results on CLEVR. Left (Color, Shape): Low-level (Color, Shape) attributes fail to transfer in either direction. Right (Relation, Size, Count): Structural concepts exhibit an asymmetric transfer. Understanding helps zero-shot generation, whereas generation largely fails to help understanding, with a minor exception for counting.
Figure 5: Zero-shot concept transfer results on CLEVR. Left (Color, Shape): Low-level (Color, Shape) attributes fail to transfer in either direction. Right (Relation, Size, Count): Structural concepts exhibit an asymmetric transfer. Understanding helps zero-shot generation, whereas generation largely fails to help understanding, with a minor exception for counting.
Figure 6: Concept recovery via fine-tuning. We measure how quickly models learn a missing low-level concept. Top row: Prior exposure via visual understanding provides no acceleration for color generation, but leaves a usable prior that accelerates shape generation. Bottom row: Prior exposure via visual generation acts as a booster, accelerating visual understanding learning across both color and shape.
Figure 6: Concept recovery via fine-tuning. We measure how quickly models learn a missing low-level concept. Top row: Prior exposure via visual understanding provides no acceleration for color generation, but leaves a usable prior that accelerates shape generation. Bottom row: Prior exposure via visual generation acts as a booster, accelerating visual understanding learning across both color and shape.

研究结果

  • 将语言数据比例从0%提高到80%,使图像理解在四个评测维度(General、Knowledge、OCR & Chart、Vision-Centric)上均单调提升,同时条件与无条件的扩散损失都下降。
  • 提高图像理解数据比例显著改善了图像生成指标和扩散损失,但纯语言基准分数和困惑度略有变差。
  • 增加图像生成数据比例只使语言准确率/困惑度以及图像理解四个维度出现轻微波动,没有明显趋势。
  • 在CLEVR概念消融实验中,颜色和形状在理解与生成两个方向上的零样本迁移完全失败,表现降至无接触对照组水平;而空间关系、大小、数量能从理解迁移到生成,但反方向大多失败,仅计数概念有轻微例外。
  • 架构实验表明,共享注意力和归一化、仅分离前馈网络(split_ffn)消除了完全共享(dense)架构中的竞争同时保留了协同效应,且这一结果在RAE、Raw Pixels、CLIP+VAE、AR(UniTok)四种视觉分词方案上都成立;延长纯语言预训练阶段会使视觉理解和生成性能持续下降,联合训练在几乎所有指标上都优于顺序训练。
Figure 7: Overview of data complexity progressions. Examples of visual (top) and language (bottom) data used to evaluate the impact of task complexity on modality interactions, ranging from simple synthetic patterns to complex real-world distributions.
Figure 7: Overview of data complexity progressions. Examples of visual (top) and language (bottom) data used to evaluate the impact of task complexity on modality interactions, ranging from simple synthetic patterns to complex real-world distributions.
Figure 8: Impact of task complexity on modality interaction. Left: Escalating visual task complexity gradually turns synergy into competition. Simple visual tasks (e.g., backgrounds, noise) improve language modeling, whereas complex visual distributions (SSTK, video) degrade text perplexity. Right: Introducing language universally aids visual generation, but the simplest linguistic distribution provides the maximum synergistic boost.
Figure 8: Impact of task complexity on modality interaction. Left: Escalating visual task complexity gradually turns synergy into competition. Simple visual tasks (e.g., backgrounds, noise) improve language modeling, whereas complex visual distributions (SSTK, video) degrade text perplexity. Right: Introducing language universally aids visual generation, but the simplest linguistic distribution provides the maximum synergistic boost.

可应用场景

  • 在设计统一多模态预训练时,参考语言/理解/生成大致70/25/5的数据配比方案
  • 考虑采用共享注意力与归一化、仅按模态分离前馈网络的split_ffn式架构
  • 在设计训练课程时,决定何时以及如何(联合还是顺序)引入视觉数据
Figure 9: Impact of parameter sharing on cross-modal performance. Fully shared (dense) parameters force modality competition, degrading both language and vision. Decoupling solely the FFNs (split_ffn) perfectly mitigates this competition while leveraging shared attention to foster strong synergy. Decoupling attention (split_ffn_attn) or normalization (split_ffn_norm) significantly diminishes these improvements, and fully isolating all parameters (split_all) yields identical results to baselines.
Figure 9: Impact of parameter sharing on cross-modal performance. Fully shared (dense) parameters force modality competition, degrading both language and vision. Decoupling solely the FFNs (split_ffn) perfectly mitigates this competition while leveraging shared attention to foster strong synergy. Decoupling attention (split_ffn_attn) or normalization (split_ffn_norm) significantly diminishes these improvements, and fully isolating all parameters (split_all) yields identical results to baselines.
Figure 10: Impact of vision encoder designs on modality synergy. Left: The impact of pairing pure background images with language across different encoder configurations on language perplexity (Δ PPL). Right: The relative change in diffusion loss (%) for conditional and unconditional generation when paired with simple language. Modality synergy consistently occurs across all four visual tokenization designs.
Figure 10: Impact of vision encoder designs on modality synergy. Left: The impact of pairing pure background images with language across different encoder configurations on language perplexity (Δ PPL). Right: The relative change in diffusion loss (%) for conditional and unconditional generation when paired with simple language. Modality synergy consistently occurs across all four visual tokenization designs.

局限与待验证事项

  • 主要控制实验基于一个特定的1.5B至2.3B规模骨干模型以及特定数据集(SSTK、DCLM),尚未验证是否能推广到其他架构或数据组合。
  • 基于CLEVR得出的结论来自简化的合成环境,是否适用于更复杂的真实图像和语言分布仍需进一步检验。
  • 大规模验证仅限于一组135亿参数MoE模型、2万亿token的训练,在其他规模或模型系列上的可复现性仍需更多实验。
  • 生成质量评判依赖另一个模型Qwen3-VL-8B-Instruct作为自动裁判,该裁判模型本身可能存在的偏差未被单独验证。
Figure 11: Timing of unification training. The x-axis represents the number of pure language tokens consumed before visual data is introduced to the training mix. While extending the initial pure language phase yields marginal improvements in unimodal text metrics like language accuracy and perplexity, it triggers a steep and consistent decline in performance across all visual understanding and visual generation benchmarks.
Figure 11: Timing of unification training. The x-axis represents the number of pure language tokens consumed before visual data is introduced to the training mix. While extending the initial pure language phase yields marginal improvements in unimodal text metrics like language accuracy and perplexity, it triggers a steep and consistent decline in performance across all visual understanding and visual generation benchmarks.
Figure 12: Impact of sequential versus joint pretraining across various modality orderings. The charts display the performance of six distinct sequential training paths. Solid bars denote strict sequential training, while patterned bars indicate training with a 12.5% replay buffer of previously seen modalities. The horizontal dashed line represents the simultaneous joint training baseline. The results clearly show that joint training dominates all sequential approaches across almost every metric. Although data replay slightly mitigates catastrophic forgetting, it fails to match the cross-modal synergies.
Figure 12: Impact of sequential versus joint pretraining across various modality orderings. The charts display the performance of six distinct sequential training paths. Solid bars denote strict sequential training, while patterned bars indicate training with a 12.5% replay buffer of previously seen modalities. The horizontal dashed line represents the simultaneous joint training baseline. The results clearly show that joint training dominates all sequential approaches across almost every metric. Although data replay slightly mitigates catastrophic forgetting, it fails to match the cross-modal synergies.

为什么重要

随着行业转向在单一网络中同时处理文本和图像的统一模型,团队此前主要靠经验法则来决定数据配比、架构和训练顺序。这项研究用受控实验取代了这种猜测,为数据配比、架构设计和训练时机提供了具体可用的方案。

本文术语

  • 统一多模态预训练 · 从一开始就在同一个模型中联合训练文本生成、图像理解和图像生成
  • 早期统一(Early Unification) · 从训练最初阶段就把视觉数据与语言数据一起引入,让两种模态共同演化
  • 视觉懒惰(Vision Laziness) · 延迟引入视觉数据导致模型未充分优化视觉部分、而过度依赖语言先验的现象
  • split_ffn · 共享注意力和归一化层,但为每种模态单独保留前馈网络的架构设计
  • CLEVR · 一种可以精确控制颜色、形状、数量、空间位置的合成三维场景数据集

论文原文摘要(英文)

Vision offers a critical axis for advancing foundation models, driving a shift towards natively unified multimodal pretraining. Despite this momentum, the design space and the fundamental mechanisms of how modalities interact during unified training remain underexplored. We provide empirical clarity through a systematic exploration of multimodal pretraining. Our controlled experiments on both synthetic and large-scale real-world datasets yield four key insights into the physics of multimodal pre

作者 · Junlin Han

在 arXiv 阅读

最新论文

全部论文 →

METAL LAB 最新报道

图片来源: Junlin Han et al., arXiv:2608.05000, arxiv-nonexclusive