← 返回 2026-03-12

UniCom:通过压缩连续语义表示实现统一多模态建模 UniCom: Unified Multimodal Modeling via Compressed Continuous Semantic Representations

Yaqi Zhao, Wang Lin, Zijian Zhang, Miles Yang, Jingyuan Chen, Wentao Zhang, Zhao Zhong, Liefeng Bo 📅 2026-03-11 👍 4 2026-07-13 08:35
Transfusion 图像生成 图像编辑 扩散模型 统一多模态模型 连续表示学习

腾讯混元提出压缩通道维度的连续语义表示,打通统一多模态理解与生成。

前置知识

统一多模态模型 (Unified Multimodal Model)

一类试图把视觉理解(看图说话、VQA)和视觉生成(文生图、图像编辑)整合到同一个神经网络里的模型,例如 Chameleon、Emu3、Janus、Transfusion、Show-o。它们共享一套 tokenizer、一个 Transformer 主干,让两类任务共享表征、共享参数。

UniCom 的整个动机就是'怎么让统一模型同时不牺牲理解能力与生成质量',不熟悉这个方向就看不懂它为什么要折腾 token 化的方式。

离散 token 化 (Discrete Visual Tokenizer, 如 VQ-VAE / VA-VAE)

传统做法是用 VQ-VAE 之类把图像编码成一串离散 codebook 索引,可以直接塞进自回归 LLM。但量化本质上是不可逆的有损操作,会丢掉细粒度纹理、笔触和小字等高频细节,所以基于离散 token 的统一模型生成质量普遍不如专用扩散模型。

这是 UniCom 主要批评的对象之一,理解了'离散为什么会丢信息'才能体会它为什么坚持要用连续表示。

连续语义特征 (Continuous Semantic Embedding, CLIP/SigLIP)

CLIP、SigLIP 这类视觉-语言对比模型输出的高维连续向量(如 SigLIP2-SO400M 输出 1152 维),富含高层语义,但维度高、分布复杂、几何不光滑,直接送进扩散模型去生成分子非常难训,容易 FID 爆炸。

UniCom 的核心问题就是:怎么'既保留 SigLIP 的丰富语义,又让它能被扩散过程容易地采样'。这是它设计 compressor 的根本动机。

Transfusion 架构

Zhou et al. 2024 提出的一种混合架构:文本 token 用标准的 next-token 自回归 CE 损失,图像连续 latent 用 flow matching 扩散损失,通过统一的 Transformer 在同一序列上联合训练,靠 modality-aware attention mask 把两种监督分开。

UniCom 默认采用 Path I(Transfusion),并通过对照实验证明它比 Path II(MetaQuery 类查询式设计)收敛更快、编辑一致性更好——这是它一个重要结论。

Flow Matching / Rectified Flow

一类生成建模目标:给定数据 $x_1$ 和噪声 $\epsilon$,构造线性插值 $x_t = t x_1 + (1-t)\epsilon$,让网络预测速度场 $v_t = x_1 - \epsilon$,损失是 $\|v_\theta(x_t, t) - v_t\|^2$。相比 DDPM,它训练更稳定、采样步数更少。FLUX 和 SD3 都基于它。

UniCom 的生成器(扩散解码器 + 视觉生成分支)全部用 flow matching;它的 flow-matching 目标写作 $\mathcal{L}_{FM} = \mathbb{E}_{t,c,\tilde{z}_1,\epsilon} \|v_t - v_\theta(\tilde{z}_t, t; c)\|_2^2$,是公式 (5) 的核心。

MetaQuery / 查询式生成 (Query-based Generation)

以 BLIP3-o、MetaQuery-XL 为代表:让一个冻结的 MLLM 处理文本条件,输出一组可学习的 query token 的隐藏状态,再用一个小 connector 投影到扩散解码器的输入空间。生成路径和理解路径是解耦的,理解走一套视觉编码器、生成预测的是另一套视觉特征的重建。

UniCom 的 Path II 就是这个路线,但论文通过实验证明它比 Transfusion 收敛慢,且 query bottleneck 会丢空间细节(编辑一致性差),所以最终放弃了它。

LPIPS / 感知损失 (Perceptual Loss)

LPIPS 用预训练 VGG/AlexNet 提取特征,比较两图在多个层上的特征距离,比 L1/L2 更贴合人眼感知。公式 (2) 中 UniCom 用 $\mathcal{L}_{recon} = \mathcal{L}_{flow}(x, \hat{x}) + \lambda \cdot \mathcal{L}_{perc}(x, \hat{x})$ 联合训练压缩器和扩散解码器。

理解这个损失函数就能理解 compressor 怎么被训练成'既保留可重建像素信息、又留下有意义语义'的信息瓶颈。

研究动机

当前的统一多模态模型在'怎么把图像塞进 LLM'这一关键步骤上长期面临两难。一种主流路线——例如 Chameleon、Liquid、Emu3、Janus——把图像先用 VQ-VAE/VA-VAE 量化成离散 token 再走自回归,但量化是不可逆的有损操作,会丢掉笔触、小字、人脸身份这些细粒度信息,作者给出的实验证据是 X-Omni 在 ImageNet 重建上 PSNR 只有 15.66、SSIM 0.38,相比 FLUX VAE 的 33.65/0.93 差距巨大。另一种路线——VUGEN、Janus-Pro、TokenFlow——直接拿 SigLIP/CLIP 这类 1152 维的连续语义特征做目标,结果撞上'高维连续流形几何复杂、扩散难建模'的新问题:原始 SigLIP 特征直接训扩散,FID 高、收敛慢、训练不稳定(见 Fig 6 与 Fig 7)。第三种折中方案(如 BAGEL、Janus-Pro、Show-o2)把 VAE 潜空间和 ViT 语义特征拼接,但'理解'和'生成'活在两套表征空间里,深层统一受限。总结一句话:现有方法要么因离散化丢失细粒度,要么因高维连续难以生成,要么因两套表征无法真正统一。

本文的目标是UniCom 的目标非常明确:找到一种'既能保留 SigLIP 类语义特征的丰富信息、又能被扩散过程高效采样'的统一表征,并基于它构建一个真正共享表征空间的统一多模态模型。具体来说,作者要把 SigLIP2-SO400M 的 1152 维、1024 token 的高维连续特征,压缩成一个低维紧凑潜空间 $\tilde{Z} \subset \mathbb{R}^{N \times d}$(其中 $d \ll D$,最终选 $N=1024, d=64$),再让一个基于 Transfusion 的统一 Transformer 在这个紧凑空间上做条件生成,并在 ImageNet 重建、GenEval/DPG/WISE 文生图、ImgEdit/GEdit/KRIS/WorldEdit 图像编辑四个维度上达到或超过当前 SOTA。

与已有工作不同的是,UniCom 的独特切入角度是把'压缩'这件事做对、做透。已有工作通常只在'空间维度'压缩(把 token 数从 1024 砍到 256),但 UniCom 通过系统消融(Table 4)发现:通道维度的压缩(d 从 1152 压到 64)远比空间压缩(n 从 1024 降到 256)更保真——后者 PSNR 直接掉到 19.71,前者还维持在 22.17,几乎无损。在此基础上,再把'用什么模块做压缩'这件事系统化(MLP vs MHA,见 Fig 8 的 t-SNE),并给出两条生成路径(Transfusion vs MetaQuery)的对比消融,证明 Transfusion + 通道压缩才是当前最优组合。换句话说:别人在拼模块,UniCom 在研究'压缩到底该压哪一维、用什么压、压完怎么生成'这一组被忽视的设计维度,并通过完整的对照实验把它坐实。

核心方法

UniCom 的整体思路可以一句话概括:'拿 SigLIP2 抽连续语义特征 $\to$ 用一个浅层 Transformer 沿通道维度压缩到 $d=64$ 的紧凑潜空间 $\tilde{Z}$ $\to$ 冻结 compressor,用一个扩散解码器在 $\tilde{Z}$ 上重建像素以建立 $\tilde{Z}$ 与图像的稳定映射 $\to$ 再冻结 decoder,让一个基于 Qwen-2.5-7B 的统一 Transformer(Path I, Transfusion)在 $\tilde{Z}$ 上做 flow matching 条件生成。'直觉上可以把它想成'三明治':上半截是语言/语义(Qwen-2.5-7B + SigLIP2),下半截是像素(FLUX.1-dev 衍生的扩散解码器),中间夹一层 $N=1024, d=64$ 的连续潜空间 $\tilde{Z}$ 作为'通用接口'。技术路线上分四阶段:alignment $\to$ pre-training $\to$ continued training $\to$ supervised fine-tuning,全程 native resolution。

UniCom 相对已有方法的本质区别在于两点:(1) 别人压缩时只缩空间维度(少 token),UniCom 系统证明通道维度的压缩显著更优,并据此设计一个 attention-based 语义 compressor 把 SigLIP2 特征从 $D=1152$ 沿通道压到 $d=64$,同时保留全部 1024 个空间 token;(2) 别人要么坚持 query-based(MetaQuery/BLIP3-o),要么强行走离散 token 路线,UniCom 通过严格受控的对比消融证明 Transfusion 这条'统一 transformer + modality-aware mask + 文本 CE + 图像 FM'的路径在收敛速度、生成质量、编辑一致性上全面胜出,因此选它作默认方案。这两点的结合让 UniCom 成为第一个既能像 SOTA diffusion 一样生成(GenEval 0.87)、又能像 SOTA 编辑模型一样改图(WorldEdit 4.12 第一)、同时不依赖 VAE 来保持编辑一致性的统一多模态模型。

方法步骤详情

第一步是构建压缩潜空间 $\tilde{Z}$:输入图像 $x$ 经 SigLIP2-SO400M 得到 $Z = f_{und}(x) \in \mathbb{R}^{1024 \times 1152}$,再用 compressor $C_\phi$(浅层 Transformer)把通道压成 $\tilde{z} = C_\phi(Z) \in \mathbb{R}^{1024 \times 64}$,送入扩散解码器 $D_\psi$ 重建 $\hat{x}$;这一步的损失是 $\mathcal{L}_{recon} = \mathcal{L}_{flow}(x, \hat{x}) + \lambda \cdot \mathcal{L}_{perc}(x, \hat{x})$(公式 2),compressor 和 decoder 联合训练、encoder 冻结。第二步是建立生成能力:compressor 与 decoder 全部冻结,让基于 Qwen-2.5-7B-Instruct 的统一 Transformer 学习条件分布 $P(\tilde{z} | c)$。具体地,文本走标准 next-token CE,图像走 flow matching:在采样步 $t \sim U[0,1]$ 和噪声 $\epsilon \sim \mathcal{N}(0,I)$ 下构造 $\tilde{z}_t = t\tilde{z}_1 + (1-t)\epsilon$,目标速度 $v_t = \tilde{z}_1 - \epsilon$,损失 $\mathcal{L}_{FM} = \mathbb{E}_{t,c,\tilde{z}_1,\epsilon} \|v_t - v_\theta(\tilde{z}_t, t; c)\|_2^2$(公式 5);注意力掩码上对文本用因果 mask、对同一图像 token 之间用双向 mask,输入序列首尾用 [BOI]/[EOI] 标记图像边界。第三步是推理/编辑:T2I 时把噪声 $\tilde{z}_T$ 跟在文本后面,让模型 flow matching 出 $\tilde{z}_1$,再喂 decoder 出像素;编辑时把参考图像先用 compressor 压成 $\tilde{z}$ 再拼到文本后,让模型在已有 $\tilde{z}$ 基础上做条件化重建,全程不引入 VAE 潜变量。第四步(消融验证)系统地扫了四个维度——压缩目标形状 $(n, d)$、投影器架构(MLP vs MHA)、初始化(纯 LLM vs 预训练 VLM)、生成路径(Transfusion vs MetaQuery)——并用 ImageNet 重建指标(rFID/PSNR/SSIM)、GenEval/DPG 训练曲线、t-SNE 聚类结构共同支撑最终选型 $(n=1024, d=64, \text{MHA}, \text{VLM-initialized}, \text{Transfusion})$。

技术新颖性

从技术新颖性看,UniCom 并不是发明了一个全新的网络模块,而是给出了一个此前被忽视但至关重要的设计原则:**通道维度的连续压缩优于空间维度的离散压缩**。它的实验证据相当扎实:Table 4 显示同样把信息压到约 1/18 的预算下,沿通道压到 d=64 仍能拿到 rFID 0.55、PSNR 22.17、SSIM 0.66,而沿空间把 token 数从 1024 砍到 256 时指标直接掉到 rFID 0.72、PSNR 20.29、SSIM 0.56;Fig 8 的 t-SNE 进一步说明 MHA compressor 压出的 64 维特征仍保持 SigLIP 的语义聚类结构,MLP 则把语义边界糊掉。第二点新颖性是把'理解阶段的语义先验'显式灌进生成阶段:Fig 7 表明把 Qwen-2.5-7B 在 SigLIP2 上预训练成 VLM 再接生成任务,比从纯 LLM 初始化收敛快约 3-4 倍(GenEval 上'3.4× faster' / '3.8× faster'),这是对'统一模型是否真有增益'这一长期争论的正面回答。第三点新颖性是 Method 章节给出了一条可复现的设计经验链——'特征形状 $\to$ 投影器 $\to$ 初始化 $\to$ 生成路径'——每一步都用消融数据说话,而不是凭直觉挑模块,这对后续做 unified model 的同行是直接可借鉴的工程手册。

Overview of the proposed framework. For a controlled comparison, both pathways are built upon the same compressed representations and jointly optimized with cross-entropy loss (Lce) and flow matching loss (Lfm).
Figure 2: Overview of the proposed framework. For a controlled comparison, both pathways are built upon the same compressed representations and jointly optimized with cross-entropy loss (Lce) and flow matching loss (Lfm).

实验结果

核心发现可分四块。**重建(Table 1)**:UniCom 在 ImageNet 验证集上,$d=64$ 版本 rFID=0.42、PSNR=22.28、SSIM=0.61,$d=1152$ 未压缩版 rFID=0.38、PSNR=22.60、SSIM=0.61——把通道压掉 18× 几乎无损;Fig 5 的可视化进一步说明它在文字小字符、人脸身份等高频细节上明显优于 X-Omni(PSNR 15.90)、UniTok、TokenFlow、MingTok、UniLIP 等基于语义特征的方法,甚至与专用 FLUX.1-dev VAE(PSNR 33.65)也并非代际差距。**文生图(Table 2)**:GenEval Overall 0.87、DPG-Bench 85.92、WISE Overall 0.58,全部进入 SOTA 第一梯队;WISE 上文化 0.55、生物 0.58、物理 0.66、化学 0.47,比 Janus-Pro(0.35)、BAGEL(0.52)、MetaQuery-XL(0.55)都更强,作者把这归因于 SigLIP 语义空间天然更适合编码文字。**图像编辑(Table 3)**:ImgEdit-Bench Overall 4.22(仅次于 TUNA 4.31、Qwen-Image 4.27)、GEdit-Bench G-Overall 7.32(仅次 Qwen-Image 7.56)、KRIS-Bench Overall 70.11(开源模型第一)、WorldEdit 4.12(全场第一);最关键的卖点是**不用 VAE 潜变量**就能保持编辑前后身份一致(KRIS Overall 70.11 vs BAGEL 56.21、OmniGen2 49.71),证明压缩语义特征确实保留住了细粒度结构。**消融(Table 4 + Table 5 + Fig 6/7/8)**:Table 4 量化了压缩形状的影响——保持 n=1024、压 d 到 64 比压 n 到 256 重建质量好得多;Fig 6 显示压缩到 d=64 让 T2I 训练收敛加速约 3.8×、最终 DPG 分数更高;Fig 7 左侧证明 VLM 初始化比纯 LLM 初始化 GenEval 高 0.2+、DPG 高 1.7+;Fig 7 右侧证明 Path I(Transfusion)比 Path II(MetaQuery)'3.5× faster';Table 5 在 6 个理解基准上证明 MHA projector 比 MLP projector 全面胜出(GQA 64.01 vs 62.80、MMMU 44.11 vs 43.00、OCRBench 36.00 vs 31.70),且与未压缩 SigLIP baseline 差距很小。

Evaluation on image reconstruction ability on the ImageNet validation set. d denotes the feature dimension of compressed representations z. Models marked with † are re-evaluated using official checkpoints.
Table 1: Evaluation on image reconstruction ability on the ImageNet validation set. d denotes the feature dimension of compressed representations z. Models marked with † are re-evaluated using official checkpoints.
Image generation results on GenEval, DPG-Bench, and WISE. † refers to methods using LLM rewriters on GenEval. Abbreviations for WISE: Cult. (Cultural), Bio. (Biology), Phy. (Physics), Chem. (Chemistry).
Table 2: Image generation results on GenEval, DPG-Bench, and WISE. † refers to methods using LLM rewriters on GenEval. Abbreviations for WISE: Cult. (Cultural), Bio. (Biology), Phy. (Physics), Chem. (Chemistry).
Comparison of image editing capabilities. We evaluate on ImgEdit-Bench, GEdit-Bench, KRIS-Bench and WorldEdit.
Table 3: Comparison of image editing capabilities. We evaluate on ImgEdit-Bench, GEdit-Bench, KRIS-Bench and WorldEdit.
Reconstruction performance of different compressed feature shapes on ImageNet50k.
Table 4: Reconstruction performance of different compressed feature shapes on ImageNet50k.
Visual understanding performance on six benchmarks using compressed representations as visual inputs.
Table 5: Visual understanding performance on six benchmarks using compressed representations as visual inputs.
High-quality samples generated by UniCom. Built on compressed continuous representations, our unified multimodal model demonstrates exceptional capabilities in text-to-image generation, precise image editing, and fine-grained controllable generation.
Figure 1: High-quality samples generated by UniCom. Built on compressed continuous representations, our unified multimodal model demonstrates exceptional capabilities in text-to-image generation, precise image editing, and fine-grained controllable generation.
Comparison of the results for image editing, highlighting UniCom performance in tasks such as image manipulation, object swapping, and color adjustment.
Figure 3: Comparison of the results for image editing, highlighting UniCom performance in tasks such as image manipulation, object swapping, and color adjustment.
Overview of the proposed diffusion decoder and reconstruction analysis.
Figure 4: Overview of the proposed diffusion decoder and reconstruction analysis.
Visual comparison of image reconstruction results.
Figure 5: Visual comparison of image reconstruction results.
Feature dimension compression accelerates T2I training. Compressing to d = 64 achieves 3.8× faster convergence than original SigLIP features.
Figure 6: Feature dimension compression accelerates T2I training. Compressing to d = 64 achieves 3.8× faster convergence than original SigLIP features.
VLM initialization and Transfusion architecture improve generation. We report GenEval results during unify pre-training.
Figure 7: VLM initialization and Transfusion architecture improve generation. We report GenEval results during unify pre-training.
T-SNE visualization of feature distributions. We randomly select 6 distinct classes from ImageNet, sampling 150 images per class. MHA compression to 64-d preserves the structure, MLP projection yields scattered 64-d features, and FLUX-VAE has no semantic separability due to pixel-reconstruction focus.
Figure 8: T-SNE visualization of feature distributions. We randomly select 6 distinct classes from ImageNet, sampling 150 images per class. MHA compression to 64-d preserves the structure, MLP projection yields scattered 64-d features, and FLUX-VAE has no semantic separability due to pixel-reconstruction focus.
查看结构化数据
任务指标本文基线提升
图像重建(ImageNet val) rFID ↓ 0.42 (d=64) / 0.38 (d=1152) FLUX.1-dev VAE = 0.06;UniTok=0.38;X-Omni=8.30 相对 X-Omni 改进约 19.8×(0.42 vs 8.30),与 FLUX VAE 同量级且通道压缩 18× 几乎无损
图像重建(ImageNet val) PSNR ↑ 22.28 (d=64) / 22.60 (d=1152) FLUX.1-dev VAE=33.65;SD-VAE=28.62;UniTok=22.77;X-Omni=15.66 通道压缩 18× 仅损失 0.32 PSNR;远超 X-Omni (15.66) +6.62,逼近 UniTok 22.77
文生图 GenEval Overall ↑ 0.87 Mogao=0.89;BAGEL=0.88;Janus-Pro=0.80;SD3-Medium=0.74 进入开源第一梯队(仅次于 Mogao 0.89、BAGEL 0.88),反超 FLUX.1-Dev 0.82 与 OmniGen2 0.86
文生图 DPG-Bench Overall ↑ 85.92 X-Omni=87.65;Show-o2=86.14;BAGEL=85.07;Mogao=84.33 在统一多模态模型里位列第二,仅次于 X-Omni 1.73 分,明显高于 BLIP3-o/UniWorld 等
文生图 WISE Overall ↑ 0.58 MetaQuery-XL=0.55;BAGEL=0.52;FLUX.1-Dev=0.50;Janus-Pro=0.35 统一模型 SOTA,提升 +0.06 相对 BAGEL,文化/生物/物理/化学子项均领先
图像编辑 ImgEdit-Bench Overall ↑ 4.22 Qwen-Image=4.27;TUNA=4.31;OmniGen2=3.44;BAGEL=3.20 开源统一模型第一梯队,与最佳仅差 0.09,相对 BAGEL +1.02
图像编辑 GEdit-Bench G-Overall ↑ 7.32 Qwen-Image=7.56;FLUX.1 Kontext Pro=6.56;Step1X-Edit=6.70;BAGEL=6.52 仅次 Qwen-Image 0.24,超过 FLUX.1 Kontext Pro 0.76,超过 BAGEL 0.80
知识型编辑 KRIS-Bench Overall ↑ 70.11 (Fact 74.63 / Conc 69.48 / Proc 65.30) BAGEL=56.21;FLUX.1 Kontext Pro=54.17;OmniGen2=49.71;Step1X-Edit=43.29 开源统一模型第一,Fact 子项甚至超过商用 FLUX.1 Kontext Pro 17.41 分
图像编辑 WorldEdit Overall ↑ 4.12 FLUX.1 Kontext Pro=3.21;BAGEL=2.76;UniWorld-V1=3.26;AnyEdit=2.09 全场第一,相对 BAGEL +1.36、相对 FLUX.1 Kontext Pro +0.91,证明复杂世界知识编辑能力最强
压缩特征理解能力(视觉问答下游) GQA / MMMU / OCRBench GQA 64.01 / MMMU 44.11 / OCRBench 36.00 (MHA Proj.) Baseline 65.25 / 44.56 / 55.40;MLP Proj. 62.80 / 43.00 / 31.70 MHA 比 MLP 在 GQA +1.21、MMMU +1.11、OCRBench +4.30;与未压缩 baseline 差距 < 1.5 分,说明压缩几乎不损理解

局限与改进

作者在论文中明确承认与隐含的局限可以分几类。**作者承认的部分**:第一,论文虽然证明 channel compression 在其设定下最优,但目前 compressor 和 decoder 必须联合预训练、然后固定下来再训生成器,这意味着 $\tilde{Z}$ 的几何一旦定型就被锁死,无法在后续训练中继续优化(Section 3.2 末尾)。第二,Path II(MetaQuery)只在小规模实验中被对比,且作者通过给 Path II 额外加了 SigLIP 统一特征空间 + I2I 重建(受 RECA 启发)的'补丁'才让它跑通——这暗示 query-based 路线本身的特征对齐问题是结构性挑战,不容易干净解决(Section 4.4)。第三,Table 5 显示即使 MHA projector,压缩后 OCRBench 仍从 55.40 掉到 36.00,说明对'超细粒度文字'这类任务,连续压缩仍有不可忽视的信息损失,需要靠 sequence concatenation(64.22 / 60.26 / 73.11 / 45.00 / 60.88 / 48.10)来部分补救。**我自己观察到的局限**:(1) 整体管线依赖 SigLIP2-SO400M 和 FLUX.1-dev 两个强基座,作者给的训练数据混合、超参、batch size 都在 Appendix,但正文没有完整给出训练 token 数与总机时,复现门槛高。(2) Table 1 显示 UniCom 的 PSNR(22.28)仍明显低于专用 VAE(FLUX VAE 33.65),意味着它在'像素级重建'上其实不是最优——它赢在'语义保真 + 可生成性'的折中,而不是像素精度。(3) Path I(Transfusion)对显存压力大,因为每个图像都要处理 1024 个 64 维 latent 并和文本 token 拼接,序列长度是 Path II(少量 query)的几十倍;论文没有给出推理延迟数据。(4) 在 WorldEdit 这种场景复杂、需要世界知识的任务上虽然拿了第一(4.12),但 GenEval 上仅 0.87,说明复杂 prompt 组合(color-attribute)仍有提升空间。(5) Compressor 推理时被冻结,意味着不同下游任务不能自适应调整特征几何。

独立分析的弱点

独立分析的弱点和对应改进方向有四点。**(1) 压缩后的 $\tilde{Z}$ 与像素重建之间存在不可控的失配**:当 SigLIP2 编码的特征本身就含歧义(比如遮挡、低光、运动模糊),compressor 会把这个歧义一并压进去,decoder 重建出的图像可能就丢掉关键身份信息;KRIS-Bench Fact 74.63 看起来高,但 Proc 仅 65.30 说明对'多步推理式编辑'仍有缺口。改进方向是引入 iterative refinement 让 compressor 的输出在生成阶段可被二次校准,或者加一个 confidence token 提示模型'哪些 patch 的语义不可靠'。**(2) 缺少对超长 prompt / 多轮对话的系统评测**:Table 2/3 都是单轮编辑、单条 prompt 评测,但真实用户场景常常是'先把这件衣服换成红色,然后把背景改成雪山'这样的多轮复合指令;由于 UniCom 在编辑时只把压缩 $\tilde{z}$ 拼到文本后,理论上支持多轮,但论文没有给出多轮一致性指标。改进方向是补一个 multi-turn editing benchmark(如 MUREdit 类),并显式追踪身份一致性随轮次的衰减曲线。**(3) Path II(MetaQuery)虽然输了,作者却没有深入分析 query 数 $M$ 与生成质量的关系曲线**;Fig 7 只给了一个固定的 query 设置就下结论'query-based 收敛慢',不够公平。改进方向是补一组 query 数从 16 到 256 的扫描,看是否存在某个甜蜜点让 Path II 追上 Path I。**(4) 对'高质量文字渲染'的来源没有做受控实验**:Table 2 显示 WISE 上化学/物理 0.47/0.66 很强,作者归因于 SigLIP 语义空间更适合文本,但没有 ablation 把 SigLIP2 替换成 CLIP-ViT-L 或 DINOv2 看是否同样成立。改进方向是补一组'不同视觉编码器 $\times$ 同样压缩配置'的对照,把'语义先验'这一变量隔离出来。**(5) 推理时 $\tilde{z}_T$ 是从 $\mathcal{N}(0, I)$ 采样的纯噪声,没有利用任何结构先验**:相比 FLUX 直接在 VAE 潜空间采样(已有相当结构),UniCom 要从零学 1024×64 的 64K 维联合分布,理论上需要更多采样步数,论文没给 CFG / sampler 步数的消融。改进方向是引入一个轻量级 prior(autoregressive 的 token-level 粗生成 + diffusion 的细节细化)来降低 $\tilde{z}$ 空间的采样难度。

未来方向

作者在第 5 节 Conclusion 明确点出的方向是**把 UniCom 框架拓展到视频生成和更广义的多模态推理**——'Future work could focus on further optimizing the compression module and expanding the application of this framework to other domains, such as video generation and multimodal reasoning.'这其实揭示了三条具体的研究线:(1) 把 compressor 从 image-level 拓展到 video-level,需要解决时序压缩与跨帧注意力,工程上挑战是 1024 token 已经让图像序列很长,视频再乘以帧数会爆显存,可能要引入时序维度的二次压缩;(2) 把 $\tilde{Z}$ 作为'通用中间表征'用到其他模态(音频、3D、深度图),这要求 compressor 架构对模态无关;(3) 把'理解→生成'的单向路径拓展为'理解⇔生成'的闭环,让生成的图像反馈回理解模块,形成 reasoning-by-generating 的能力。基于这篇工作的成果还可以自然延伸的方向有:把 Path I 改成 sparse mixture-of-experts 来降低文本/图像互不干扰;用蒸馏把 Qwen-2.5-7B 换到 3B 量级做移动端部署;以及把'通道压缩 + Transfusion'这套范式应用到多语言文生图(中文 prompt 直接生成)。最后,WorldEdit 4.12 第一但 GenEval 仅 0.87 的现象提示下一步应该专门训练'复杂知识型文生图',把 SigLIP 语义空间与世界知识显式对齐。

复现评估

复现评估需要从三个维度看。**(1) 开源情况**:论文正文没明确写'release checkpoint',但作者团队是 Tencent Hunyuan,加上文末给出 Project Page https://miazhao7708.github.io/UniComPage/,很可能后续会放出权重;如果不放出,单从论文复现难度很大。**(2) 数据**:训练用了 SigLIP2-SO400M-Patch16-NaFlex 作为语义编码器、FLUX.1-dev 作为扩散解码器 backbone、Qwen-2.5-7B-Instruct 作为 LLM,训练数据混合和四阶段(alignment / pre-training / continued training / SFT)的细节被推到 Appendix Section A;用户文本丰富的数据被明确指出对文字渲染有显著增益,但具体数据来源未在正文披露。**(3) 算力**:模型至少包含 Qwen-2.5-7B(7B 参数)+ SigLIP2-SO400M(约 400M)+ FLUX.1-dev 衍生 decoder + 浅层 Transformer compressor,且要跑四阶段训练 + ablation,再加上 GenEval/DPG/WISE/ImgEdit/GEdit/KRIS/WorldEdit 6 个 benchmark 的评测——保守估计需要 64-128 张 A100/H100 训 2-4 周,小团队基本不可能复现;如果只是想验证'通道压缩优于空间压缩'的核心结论,则只需训 compressor + decoder 这一小段(论文给出了具体配置),单卡 8×A100 大约可以复现 Table 4 的几个对比。**(4) 评测门槛**:6 个 benchmark 大多需要申请 API 或特殊权重(KRIS、WorldEdit 是较新的),且 GEdit/GPT-4o judge 涉及 GPT-4o 调用,复现总成本数千美元起。综合来看,**论文给出的核心 idea(通道压缩 + Transfusion)相对易复现**,但要复现 SOTA 数字本身门槛很高。