用统一自回归框架同时实现图像理解与生成,效果超越现有方法。
Harmonizing Visual Representations for Unified Multimodal Understanding and Generation
- 采用掩码自回归编码器,共享同一视觉表示
- 在生成和理解任务上均达顶尖性能,生成超越基准
- 适合需要统一多模态模型的研究者与开发者
将视觉理解与生成统一于单一多模态框架仍具挑战,因两者需不同粒度的表征。现有基于向量量化(VQ)或变分自编码器(VAE)的方法侧重图像内在特征而弱化语义,影响理解性能。受掩码图像建模(MIM)及掩码自回归(MAR)图像生成启发,我们发现MAR编码器在预训练后具备优异线性探测准确率和对视觉概念的精确响应,表明其可胜任理解任务。基于此,提出 extit{Harmon}:一个共享MAR编码器的统一自回归框架。通过三阶段训练逐步优化理解与生成能力,在GenEval、MJHQ30K和WISE基准上达成当前最优生成效果,且在图像理解任务上达到专用语义编码器方法(如Janus)的水平。代码与模型将开源于https://github.com/wusize/Harmon。
原文摘要 · Abstract (English)
Unifying visual understanding and generation within a single multimodal framework remains a significant challenge, as the two inherently heterogeneous tasks require representations at different levels of granularity. Current approaches that utilize vector quantization (VQ) or variational autoencoders (VAE) for unified visual representation prioritize intrinsic imagery features over semantics, compromising understanding performance. In this work, we take inspiration from masked image modelling (MIM) that learns rich semantics via a mask-and-reconstruct pre-training and its successful extension to masked autoregressive (MAR) image generation. A preliminary study on the MAR encoder's representation reveals exceptional linear probing accuracy and precise feature response to visual concepts, which indicates MAR's potential for visual understanding tasks beyond its original generation role. Based on these insights, we present \emph{Harmon}, a unified autoregressive framework that harmonizes understanding and generation tasks with a shared MAR encoder. Through a three-stage training procedure that progressively optimizes understanding and generation capabilities, Harmon achieves state-of-the-art image generation results on the GenEval, MJHQ30K and WISE benchmarks while matching the performance of methods with dedicated semantic encoders (e.g., Janus) on image understanding benchmarks. Our code and models will be available at https://github.com/wusize/Harmon.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。