← 返回 2026-04-23

MMCORE:基于表征对齐潜在嵌入的多模态连接框架 MMCORE: MultiModal COnnection with Representation Aligned Latent Embeddings

Zijie Li, Yichun Shi, Jingxiang Sun, Ye Wang, Yixuan Huang, Zhiyao Guo, Xiaochen Lian, Peihao Zhu, Yu Tian, Zhonghua Zhai, Peng Wang 📅 2026-04-21 👍 4 2026-07-13 08:36
图像编辑 多模态生成 扩散模型 表征对齐 视觉语言模型

用可学习query token对齐MLLM与扩散模型的低成本统一多模态生成框架

前置知识

多模态大语言模型(MLLM)

在预训练大语言模型基础上加入视觉编码器(如ViT),通过图文对齐训练得到兼具理解与推理能力的视觉-语言模型。典型代表有GPT-4V、Qwen-VL、Seed-VLM等,可处理图文交错输入并进行视觉问答。

本文核心是用MLLM的语义表征驱动扩散生成,理解MLLM的因果注意力、token序列与文本潜空间是看懂查询token如何蒸馏视觉概念的前提。

扩散Transformer与MMDiT架构

扩散Transformer(DiT)用Transformer替换U-Net做去噪,MMDiT(多模态DiT)把文本与图像token在共享注意力流里融合,是Stable Diffusion 3、Flux等主流T2I模型的核心架构,擅长高保真图像合成。

本文用预训练MMDiT作为生成头并扩展到交错生成场景,其块因果注意力、独立Embedding Dropout策略都建立在MMDiT架构特性之上。

SigLIP/ViT语义视觉特征

ViT将图像切成patch做序列编码得到空间化视觉token,SigLIP是其对比学习变体,能产出富含高层语义(如物体、关系)的稠密向量,被广泛用作检索与对齐的语义锚点。

本文用冻结的ViT/SigLIP作为目标特征,通过余弦相似度损失拉齐MLLM输出与视觉语义空间,是连接理解与生成的关键'脚手架'。

Flow Matching与扩散损失

Flow Matching是一种从噪声到数据学习连续变换向量的训练范式,比DDPM更稳定、采样步数更少,本文扩散头即采用此目标做图像生成监督。

作者发现只用扩散损失监督query token会造成信号稀疏且收敛差,必须叠加ViT语义损失才能稳定训练,这是理解'两阶段'方案的基础。

研究动机

当前多模态生成领域存在两大主流范式的明显张力:自回归(AR)模型以LLM/VLM为代表,语义推理强但视觉质量落后;扩散/流匹配(FM)模型以MMDiT为代表,视觉保真度领先却难以承担复杂指令。Transfusion [48]、BAGEL [5] 等尝试在单一网络里把两者紧耦合,要么需要用LLM权重复制出生成分支,要么联合去噪特征与干净特征在同一前向中交替,导致训练效率显著退化、动辄消耗海量算力与私域数据,资源门槛居高不下,学术复现几乎不可能。即便走另一条更经济的路线——MetaQueries [22] 这类冻结MLLM + 轻量connector + 扩散解码器,仍然受限于固定query预算对长上下文捕捉不力、纯扩散损失带来的弱对齐以及MLLM冻结导致表征迁移不充分等问题,生成质量被SOTA明显拉开。从Fig.1(a)(b)的人类评测与Fig.2的DreamBench自动化评测看,在文本-图像对齐、编辑对齐、编辑一致性三个维度上,既有开源方案都显著落后于GPT-Image-1、Seedream 4.0级别的闭源系统。

本文的目标是本文目标非常明确:构建一套既能复用MLLM长上下文理解与推理能力、又能驱动扩散头输出高保真图像的统一多模态框架,同时把训练算力压到端到端方案的约30%。具体而言,作者希望让MLLM与扩散模型通过低干扰的查询接口对接:MLLM继续承担视觉问答与推理任务,扩散头负责像素级合成,二者解耦到不再需要显式connector模块,并保留对SFT、RLHF等后训练流程的友好性。在Fig.7、Fig.8的多组对照里,目标被细化为对反事实提示、空间关系、组合角色绑定、多图编辑等复杂场景达到或超过Seedream 4.0,并保持10+输入图像场景下的细粒度控制能力。

与已有工作不同的是,MMCORE 的独特切入角度在于'查询token + 语义蒸馏 + 双通路条件'三件套:在MetaQueries的查询接口基础上做了三项关键改造——对MLLM做全参数联合微调而非冻结、对查询token施加与冻结ViT/SigLIP特征的余弦对齐损失、并让查询token与原始文本embedding并联作为扩散条件。这种设计既消除了固定query预算的瓶颈,又给了MLLM密集的中间监督信号,使视觉token在学扩散前先学到高层语义,再以双通路避免文本细节被截断。区别于BAGEL/Transfusion把AR与FM塞进一个网络,MMCORE保留模块化结构;区别于MetaQueries只用query或冻结MLLM,MMCORE通过loss $\mathcal{L}_{vis}$ 提供稳定的低方差梯度与可扩展的监督源,最终在Table 1里用完整消融把SFT前后绝对分数差推到 +25.9%(GPT-4o)和 +16.4%(Doubao),证实对齐策略比堆叠参数更关键。

核心方法

MMCORE 整体思路可概括为'用可学习查询token把MLLM语义蒸馏到扩散头条件空间,且不冻结MLLM'。架构(图5)分两段:底部多模态MLLM接收文本、可变长度图像token与可学习查询token $Q \in \mathbb{R}^{N \times D}$,通过自注意力把上下文压缩为 $K \times K$ 个视觉潜token;顶部是预训练MMDiT扩散头,接收历史帧VAE潜变量、文本embedding与视觉潜token作双重条件。训练也清晰两阶段(式2):阶段一先用$L_{llm}$ 与与冻结ViT特征的余弦相似度$L_{vis}$ 联合微调MLLM;阶段二冻结对齐后MLLM,单独继续预训练扩散头,配合独立Embedding Dropout缓解DiT文本偏置,最后做SFT 2k步 + RLHF贴近人类偏好。生成用块因果注意力(图6),当前帧只attend历史VAE与本帧文本/视觉token。

核心创新点是把'表征对齐'作为生成的前置约束:让MLLM在生成像素之前先蒸馏出与SigLIP/ViT对齐的视觉潜token,再用token + 文本双通路条件扩散头。这与已有方法的本质区别有三层。其一,对MetaQueries的扩展不是单纯放大query数或加connector,而是用$L_{vis}$ 给MLLM一个稠密、监督稳定的中间目标,使其在阶段一就能独立收敛,省去昂贵的端到端流匹配。其二,对MLLM做全参数联合微调而非冻结,这与绝大多数PEFT方案(如LoRA)不同,作者明确测出全微调把GPT-4o分数推到0.8114,超过LoRA。其三,双通路条件承认了ViT语义与文本词袋是互补信息源,而非互相替代:视觉token负责全局语义与跨模态定位,文本embedding保留细粒度词汇约束——这一点在Table 1末尾的失败案例中得到反向印证:强行把ViT embedding注入VAE的DiT条件反而让编辑对齐从55.2掉到30.62,说明两种视觉特征不能简单叠加。

方法步骤详情

①准备MLLM $\mathcal{F}_\theta$ 与冻结ViT $\Phi_{vit}$,后者patch池成$K^2$网格对query预算$N=K^2$,得$v=\Phi_{vit}(I)$。②MLLM追加可学query $Q$,图文batch练$L_{llm}$与对齐$\mathcal{L}_{vis}=\frac1N\sum_i(1-\frac{\mathcal{F}_\theta(Q_i)^\top v_i}{\|\mathcal{F}_\theta(Q_i)\|\|v_i\|})$,总$\mathcal{L}_{mllm}=\lambda_t L_{llm}+\lambda_a L_{vis}$,全参微调至收敛。③冻结MLLM做扩散训练:块因果mask(图6)使$t$帧仅看前$t-1$帧VAE与本帧文本+视觉token;文本drop率更高逼依赖MLLM后退火。④T2I+interleaved混合做CPT,约2k步SFT+RLHF。⑤推理:MLLM压缩≤10+张参考图为64视觉token($N=64$消融最优),与文本embedding并联送MMDiT按序生成。

技术新颖性

技术新颖性体现在三处:其一,提出'语义蒸馏先于扩散监督'的训练范式,把视觉对齐loss以余弦相似度形式直接施加在MLLM的查询token上,使得MLLM与扩散头首次可以独立两阶段训练,避免昂贵联合计算,作者自报算力节省~70%。其二,块因果注意力设计是新颖的工程选择:明确把历史帧的VAE潜变量作为高密度细节承载,把当前帧视觉token作为高层语义承载,证实历史视觉token反而是干扰项,与直觉相反但有实证支持。其三,独立Embedding Dropout策略对DiT类模型特别有效:DiT对文本有强归纳偏置,简单拼接ViT特征会导致文本条件支配,联合复用文本与视觉token的同时通过差异化drop缓解偏置,是对Flow Matching社区方法 [7, 15] 的有益扩展。另外Table 1中Full Fine-Tuning + SFT 2k步超过基线25.9%的事实本身是新颖结论:少而精的SFT比继续大规模预训练更能拉齐人类审美。

Architecture of MMCORE for low-cost Unified Multi-modal Model
Figure 5: Architecture of MMCORE for low-cost Unified Multi-modal Model
Attention mask for diffusion-head training
Figure 6: Attention mask for diffusion-head training

实验结果

DreamBench人工评测(Fig.1)显示MMCORE全面领先:(a)T2I对齐84.42% vs Seedream 4.0约80、GPT-Image-1约78、Flux约69;(b)图像编辑对齐81.2% vs Seedream 4.0 79.88、GPT-Image-1 79.55;(c)编辑一致性75.82% vs Seedream 4.0 70.62、GPT-Image-1 68.89。Fig.2 AutoEval与人工一致。Fig.7反事实prompt对比中,MMCORE能正确表达'男人眼睛齐平女人嘴巴''珍珠+篮球空间排布'等空间关系。Fig.8多图编辑(≥4输入合成神话兽等)MMCORE部件归属性稳定。Table 1消融关键:6层connector让GPT-4o分从0.6791跃至0.7843(+10.5%);全微调推至0.8114(+19.5%);扩5×BS达0.8199;+SFT 2k步跃迁到0.8585(+25.9%)。反直觉发现:把视觉潜token注入VAE DiT条件反而让编辑对齐从55.2断崖跌至30.62,证实'条件不要混搭'。

Ablation on MLLM Architecture & Tuning Strategy
Table 1: Ablation on MLLM Architecture & Tuning Strategy
DreamBench AutoEval results of various models
Figure 2: DreamBench AutoEval results of various models
Precise/Reference image editing with MMCORE
Figure 4: Precise/Reference image editing with MMCORE
Text-to-image generation comparison against Seedream 4.0
Figure 7: Text-to-image generation comparison against Seedream 4.0
Single- and multi-image editing results comparison against Seedream 4.0
Figure 8: Single- and multi-image editing results comparison against Seedream 4.0
查看结构化数据
任务指标本文基线提升
DreamBench T2I对齐(人类评测Prompt-Image) 准确率(%),越高越好 84.42 Seedream 4.0约80、GPT-Image-1约78、Gemini约70、Qwen2.5 Image/Flux Kontext约69 +4.4 vs Seedream 4.0,+6.4 vs GPT-Image-1(来自Fig.1a)
DreamBench 图像编辑对齐(人类评测) 准确率(%),越高越好 81.2 Seedream 4.0 79.88、GPT-Image-1 79.55、Gemini约78、Flux Kontext约70 +1.32 vs Seedream 4.0
DreamBench 编辑一致性(人类评测) 准确率(%),越高越好 75.82 Seedream 4.0 70.62、GPT-Image-1 68.89、Gemini 68.01、Flux Kontext 47.87 +5.20 vs Seedream 4.0
Dreambench T2I(AutoEval,GPT-4o打分) 0-1归一化分数 0.8585(+SFT) 基线2层connector 0.6791 +25.9%(绝对)
Dreambench T2I(AutoEval,Doubao-VL2打分) 0-1归一化分数 0.8915(+SFT) 基线2层connector 0.7276 +16.4%(绝对)
Dreambench T2I不同connector深度(GPT-4o) 0-1归一化分数 6层 0.7843 2层 0.6791 +10.5%
Dreambench T2I 微调策略对比 0-1归一化分数(GPT-4o) 全参微调50k 0.8114;放大BS 0.8199;+SFT 0.8585 冻结VLM 2层 0.6791 +19.5%~+25.9%

局限与改进

作者明确三点局限:(1)与Nano-Banana-pro [4]、GPT-image 1.5 [20]等闭源旗舰仍有差距,怀疑来自MLLM本身——加入SoTA VLM的prompt rewrite后效果显著变好,下游生成瓶颈在MLLM语义深度。(2)'理解-生成 trade-off'——把生成能力灌入MLLM轻微损害VQA/OCR等纯理解能力,作者认为是课程调度问题而非架构缺陷,期待把生成目标引入预训练早期解决。(3)视觉token冗余,目前只是文本条件补丁而非替代,必须与ViT encoder + diffusion decoder配合才能驱动高质量生成,作者称之为'下一个Omni-Tokenizer'的必要性。我还观察到:所有量化指标基于内部DreamBench,缺少GenEval/DPG-Bench/T2I-CompBench等公开基准横向对位;且Table 1里SFT 2k步带来最大跳变,但SFT数据集未公开。

独立分析的弱点

(1)视觉token冗余:当前query token像补丁而非完整替代文本,必须与ViT + diffusion decoder共存,模型体积未真正变小;改进方向是取消分立ViT encoder,让MLLM输出可重建+语义统一的token。(2)评估基准不公开:所有量化结果都来自内部DreamBench,缺乏GenEval、DPG-Bench、T2I-CompBench等公认基准;改进方向是把同一套权重送公开Arena盲评。(3)VAE+ViT混搭会崩:作者报告编辑对齐从55.2骤降到30.62却无修复方案;改进方向是引入显式模态解耦adapter,让两种特征先投影到正交子空间再融合。(4)训练曲线公开度低:Table 1只有终态分数,'~30%算力'的论断难以核验;改进方向是开源recipe并补充scaling law。(5)集成方法有限:缺乏DPO/过程奖励模型等现代RLHF技术,且作者自评gap主要在MLLM端;改进方向是在MLLM侧引入更多RLHF信号。

未来方向

(1)Omni-Tokenizer:作者明确定义下一个核心工作为'既能像素级重建又能语义推理'的统一视觉token,把MLLM→扩散头→VAE三段流水线收敛到单一自回归模型,逼近'一次前向统一完成理解与生成'的终极目标。(2)预训练早注入生成目标:作者认为理解-生成trade-off是课程问题而非架构限制,未来应在MLLM预训练中段就加入视觉token对齐任务,使生成与语义同步演化。(3)缩小与闭源旗舰差距:把更大MLLM(推测百亿到千亿参)与CPT+SFT+RLHF联合训练,并引入更强prompt rewriting模型作为思维链增强器。(4)工程方向:替代余弦对齐损失为MLP head + 双向KL,把ViT distillation从单层patch升级到多层深度特征融合;把块因果注意力扩展到视频帧预测,做时序版MMCORE。(5)数据方向:SFT只用2k步带来巨大提升,说明高质量SFT数据稀缺,未来可对反事实prompt、空间推理、计数prompt构造合成课程让对齐损失自然下降。

复现评估

复现评估不容乐观:本文ByteDance Seed内部作品,明确写到'由于开源与内部SoTA的基础设施差距显著,所有实验使用内部高性能管线',意味着模型权重、训练数据、SFT集合、prompt rewriting模型全部未公开。从算力体量推断MLLM规模数十亿参数以上,全参微调加5×BS需千卡集群(H100/GB200),单实验室难复现。块因果注意力细节、独立Dropout退火表、$\lambda_t/\lambda_a$超参只在正文与附录提及,无配置文件。下游评估依赖自建DreamBench,外部难跑通。Table 1消融方法论清晰但无法在公开权重上验证。复现难度'极高'——需重建MLLM推理引擎、T2I扩散管线、AutoEval模型三块基础设施,重训数周才有相似规模结果。学术读者更现实的选择是在MetaQueries、SEED-Bench等更小公开子实验上验证趋势,但本文未提供这些模型。论文符号系统一致、式1式2给出可对齐的目标函数,理论上可在小规模上验证'6层connector优于2层''SFT优于继续预训练'等趋势,但绝对分数不可对位。