LatentUM:通过潜在空间统一模型释放交织跨模态推理的潜力 LatentUM: Unleashing the Potential of Interleaved Cross-Modal Reasoning via a Latent-Space Unified Model
统一语义空间表示所有模态,支持交织跨模态推理
前置知识
统一模型
统一模型是指能够在单一架构中处理和生成多种模态内容的深度学习系统,如文本、图像和视频。这类模型通常采用Transformer作为骨干网络,通过统一的token预测目标或结合自回归与扩散建模,实现跨模态的理解和生成能力。代表工作包括Chameleon、Emu3、Show-o等,它们旨在打破传统单模态模型的界限,实现真正的多模态智能。
本文的核心研究对象就是统一模型,理解其基本原理和面临的挑战是读懂LatentUM设计动机的基础。
CLIP特征
CLIP特征是通过对比学习训练得到的视觉表示向量,它们与语言表示在共享的语义空间中对齐。CLIP使用大量图像-文本对进行预训练,使得相似语义的图像和文本在特征空间中距离更近。这种特征天然具有跨模态对齐的特性,被广泛应用于视觉-语言任务中。
LatentUM使用CLIP特征作为视觉表示的基础,并通过量化将其离散化,这是实现语义对齐视觉token的关键技术。
世界建模
世界建模是指预测环境未来状态的能力,给定过去观测和动作,能够推理出环境随时间的演化规律。这是通用人工智能的核心能力之一,对于机器人导航、决策规划等任务至关重要。世界模型学习环境的动力学模型,可以用于模拟未来、进行规划和零样本泛化。
本文展示了LatentUM在视觉导航世界建模任务中的应用,验证了统一语义表示对时序推理的泛化能力。
KL散度
KL散度是衡量两个概率分布之间差异的指标,定义为DKL(P||Q)等于P(i)乘以log(P(i)/Q(i))的求和。它是非对称的,表示当使用分布Q来近似分布P时引入的信息损失。在机器学习中,KL散度常用于分布匹配、变分推断和强化学习中策略优化等场景。
本文的模型行为对齐量化使用KL散度作为损失函数,确保量化后的特征保持与原始特征相同的理解能力。
研究动机
现有统一模型面临的核心问题是视觉理解和生成使用分离的视觉表示,这些表示位于不同的特征空间。具体来说,理解任务需要语义特征,而生成任务通常追求像素级细节。这种分离导致模型无法直接对自己生成的视觉内容进行推理,必须先将生成的特征解码为像素,再重新编码为语义特征才能用于理解。这个像素空间中介步骤引入了两个严重问题:一是codec bias,即编码器-解码器对引入的有偏表示;二是跨模态误对齐,即像素空间的桥梁削弱了模态间的语义一致性。例如,在视觉空间规划任务中,现有方法如ThinkMorph和Uni-CoT需要反复进行像素解码和重新编码,计算效率低下且影响推理质量。
本文的目标是本文的具体目标是构建一个能够在共享语义潜在空间中表示所有模态的统一模型LatentUM,从而消除视觉理解和生成之间的像素空间中介。通过使用语义对齐的视觉表示,使模型能够直接对自己生成的视觉token进行推理,无需转换为像素。这种设计应该实现三个目标:提高计算效率,减少不必要的像素编解码;增强跨模态对齐,减轻codec bias;支持灵活的交织跨模态推理,如视觉链式思维、自反思的视觉生成以及世界建模。
与已有工作不同的是,本文的独特切入角度是重新审视统一模型的核心价值主张,不是将统一模型视为视觉生成器,而是将其定位为能够执行交织跨模态推理的系统。这一认识转变引导了全新的设计选择:与其追求像素级保真度,不如优先考虑语义正确性。具体来说,本文提出了三个创新点:第一,使用模型行为对齐量化将CLIP特征离散化,使量化特征保持与原始特征相同的视觉理解能力;第二,设计模态专家混合架构,用两个平行分支分别处理理解和生成任务,共享自注意力机制;第三,将像素解码器与核心模型解耦,使主模型从不优化像素重建,专注于语义空间。这与现有统一模型形成鲜明对比,后者大多专注于文本到图像生成任务,如TokenFlow-XL、Janus Pro、Bagel等,它们仍然依赖像素重建损失或VQVAE等基于像素的特征表示。
核心方法
LatentUM的整体思路基于一个核心直觉:对于跨模态推理任务,语义正确性比像素级保真度更重要。因此,LatentUM将所有模态表示在共享的语义潜在空间中,其中视觉信息以语义token的形式存在,与语言token在同一空间。技术路线分为三个层次:首先,通过模型行为对齐量化训练一个视觉tokenizer,将连续的CLIP特征离散化为视觉语义token;其次,构建基于自回归Transformer的统一模型,采用模态专家混合架构,包含理解和生成两个平行分支;最后,训练一个独立的扩散解码器,可选地将量化特征映射到像素空间用于可视化。整个框架的关键是量化特征保留理解能力而非重建像素,使模型能够直接推理自己的输出。
LatentUM的核心创新点在于三个层面的技术突破。第一是模型行为对齐量化,与传统VQVAE等基于像素重建的量化方法不同,MBAQ通过最小化VLM在原始特征和量化特征上的输出分布KL散度来训练量化器,确保量化特征保持视觉理解能力而非像素细节。第二是模态专家混合架构,每个Transformer层包含两个平行分支:理解分支和生成分支,它们共享自注意力机制但拥有独立的FFN和投影矩阵。这种设计避免了冲突的优化信号,同时通过共享注意力实现跨模态信息流动。第三是解耦的像素解码器,使用预训练的扩散Transformer将量化特征条件化映射到像素,但核心模型从不优化像素重建损失,保持潜在空间专注于语义。
方法步骤详情
LatentUM的完整流程包含四个主要步骤。第一步是视觉tokenizer训练,采用多码本量化将d维CLIP特征分割为C个块,每个块用独立的K大小码本量化,产生K的C次方的有效词汇表。量化器在LLaVA-v1.5-665K数据集上使用MBAQ损失训练,结合MCQ承诺损失加权lambda。第二步是LatentUMBase模型构建,基于InternVL3.5-4B架构,理解分支直接从预训练权重初始化并冻结,只训练生成分支。训练数据是32M来自BLIP3o的图像-文本对,使用多码本next token预测目标。第三步是像素解码器训练,独立微调MMDiT的条件分支将量化特征映射到像素。第四步是后训练激活交织推理能力,包括监督微调和使用GRPO强化学习通过自奖励机制优化生成质量。
技术新颖性
LatentUM的技术新颖性体现在多个维度。在表示学习方面,首次提出以VLM行为对齐作为量化目标,与传统特征重建目标形成本质区别。这一设计使量化特征能够被同一模型直接理解,无需像素中介。在架构设计方面,MoME的模态解耦但注意力共享的设计平衡了两个冲突需求:避免跨模态梯度干扰与保持跨模态信息流动。在训练策略方面,将像素重建完全解耦,核心模型从不优化像素损失,这是对主流统一模型训练范式的根本性突破。在应用范式方面,首次将统一模型定位为跨模态推理系统而非视觉生成器,并系统展示了在视觉空间规划、自反思生成和世界建模等任务上的能力。这些创新使得LatentUM在GenEval上达到0.85分数,在Visual Spatial Planning上达到0.99平均精度,验证了语义对齐表示的有效性。
实验结果
LatentUM的实验结果在四个层面验证了其设计有效性。在视觉理解方面,LatentUMBase在五个基准上达到竞争性性能:MME-P 1654优于TokenFlow XL的1551和Janus Pro的1444、POPE 88.9、SEED 76.3、MMBench 80.3、MMMU 54.6。更重要的是,在量化特征设置下性能下降很小,证明MBAQ确实保留了理解能力。在视觉生成方面,LatentUMBase在GenEval上达到0.85整体分数,在统一模型中最佳,尽管训练数据量最小。通过GRPO自反思后训练的LatentUMVis-Gen进一步将分数提升至0.87,使用外部像素奖励模型后达到0.92,建立统一模型SOTA。在跨模态推理方面,LatentUMVis-Plan在Visual Spatial Planning上表现卓越:粗粒度规划平均精度0.85,细粒度规划达到0.99,显著优于ThinkMorph的0.76。这验证了统一语义空间对推理任务的价值。在世界建模方面,LatentUMWM在NWM基准上达到ATE 1.34和RPE 0.34,超越统一基线Transfusion-RAE,接近专业NWM模型。消融研究进一步显示,使用语义特征比VQVAE特征在GenEval上有显著优势,MBAQ量化比MSE或余弦相似度重建在理解基准上一致更好。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| Visual Understanding (MMBench) | Accuracy | 80.3 (LatentUMBase) | 79.3 (Show-o2), 83.5 (BLIP3o 8B) | 优于大多数统一模型,接近更大模型 |
| Visual Generation (GenEval) | Overall Score | 0.92 (LatentUMVis-Gen + pixel-reward) | 0.84 (BLIP3o 8B), 0.82 (Bagel) | 统一模型中SOTA,提升约10% |
| Visual Spatial Planning | Average Accuracy | 0.99 (LatentUMVis-Plan fine-grained) | 0.76 (ThinkMorph), 0.76 (Mirage) | 大幅提升约30% |
| World Modeling (NWM) | ATE / RPE | 1.34 / 0.34 (LatentUMWM) | Transfusion-RAE / 0.52 (NWM*) | 超越统一基线,接近专业模型 |
局限与改进
作者承认了三个主要局限性。首先,模型仍限于固定分辨率生成和相对适中的预训练规模,限制了高分辨率任务的表现和大规模知识的获取。其次,当前世界建模设置在rollout期间仍依赖像素空间循环接口,而非完全潜在循环预测,这降低了纯语义推理的一致性和效率。第三,MBAQ目前仅对齐到单个VLM的行为,学到的语义表示的泛化性尚未充分探索,可能限制在不同理解任务上的性能。此外,我观察到LatentUM的像素解码器虽然独立训练,但在推理时需要将渲染的帧重新编码为下一个上下文,这引入了额外的计算开销和潜在的信息损失。另外,MoME架构的双分支设计虽然避免了梯度冲突,但参数量相对较大,部署成本较高。在长序列生成方面,文章未充分讨论上下文长度限制对复杂推理任务的影响。
独立分析的弱点
LatentUM的第一个弱点是固定分辨率限制。当前448乘448的分辨率对于高保真图像生成和细粒度视觉任务可能不足,特别是在处理文字、小物体和复杂纹理时。改进方向是扩展到可变分辨率生成,可能通过分层token表示或动态patch大小实现。第二个弱点是像素空间循环接口。当前世界建模需要在rollout时反复进行像素渲染和重新编码,破坏了纯语义推理的一致性。改进方向是开发完全潜在循环预测,让模型直接在语义token空间中递推未来状态,避免像素空间往返。第三个弱点是单VLM对齐限制。MBAQ目前仅对齐到InternVL3.5的行为,可能不能最优地泛化到所有理解任务。改进方向是对齐多个VLM的集成行为,或设计更通用的语义对齐目标。第四个弱点是长序列推理效率。虽然MoME减少了梯度冲突,但处理多步推理时的计算复杂度仍然较高。改进方向可以探索稀疏注意力或层次化推理机制。第五个弱点是缺乏实际部署验证。文章主要关注学术基准,未在真实应用场景中验证。改进方向应在机器人导航、交互式创作等实际任务中测试。第六个弱点是训练资源需求高,限制了学术社区的复现和改进。改进方向可以探索更高效的训练方法或蒸馏到更小模型。
未来方向
作者提出的未来研究方向包括四个方面。第一是扩展预训练数据和模型容量,通过更大规模的多模态数据提升模型性能和泛化能力。第二是扩展到可变分辨率和更长上下文生成,支持高分辨率图像和长视频任务。第三是改进长时序预测的一致性,这对世界建模和长期规划至关重要。第四是开发完全潜在的世界建模和规划管道,避免rollout期间的像素空间重新渲染。基于LatentUM的成果,可以延伸多个有前景的方向。在架构方面,可以探索更灵活的模态专家路由机制,如动态专家激活或软路由,平衡参数效率和性能。在应用方面,可以将交织推理范式扩展到更多任务,如多模态对话、代码生成与执行、科学推理等。在训练方面,可以探索更高效的强化学习方法,如直接偏好优化或离线RL,减少rollout计算成本。在理论方面,可以深入分析语义潜在空间的特性,如可解释性、解耦性和迁移性。在安全方面,需要研究如何控制生成内容的语义一致性,避免自反思循环中的错误放大。在效率方面,可以探索模型压缩和加速技术,使LatentUM适用于边缘设备部署。
复现评估
LatentUM的复现性评估显示中等难度。作者公开了代码,这是积极信号。然而,完整复现面临几个挑战。训练资源需求巨大:LatentUMBase预训练需要64块NVIDIA H100 GPU运行4天,像素解码器需要32块H100 GPU运行2天。这对学术研究机构来说是相当高的门槛。数据方面,使用了32M图像-文本对和LLaVA-v1.5-665K,这些数据集大多公开可用但需要整理。模型初始化依赖InternVL3.5-4B的预训练权重,这些权重应该可以获取。后训练数据大多有公开资源。实验方面,作者提供了详细的训练设置和超参数,但一些实现细节可能在附录中。评估方面,使用公开基准,便于比较。总体而言,有代码和权重的情况下复现中等难度,但完整训练需要大量计算资源,大多数研究者可能只能复现后训练或推理部分。改进复现性的建议是提供更多实现细节、检查点采样和轻量级训练配置。
论文图表
Table 3展示了GenEval2基准上的性能,该基准使用MLLM-as-judge提供更稳健评估。LatentUMVis-Gen在TIFAGM上达到31.3,显著优于Bagel + CoT的23.1和BLIP3o 8B的13.3;在TIFAAM上达到72.9,优于Bagel + CoT的70.9和BLIP3o 8B的59.4。这验证了基于模型自身理解能力的自反思奖励可以驱动显著的生成质量提升。
这个表格补充了GenEval的结果,使用更稳健的MLLM-as-judge评估进一步验证了自反思机制的有效性,是重要证据。