提出新框架MMAR,实现图像理解与生成无损融合。
MMAR: Towards Lossless Multi-Modal Auto-Regressive Probabilistic Modeling
- 用连续图像令牌避免离散化导致的信息损失。
- 在18个图像理解基准上超越多数现有模型,且生成质量高。
- 适合需要高保真图像理解与生成的多模态研究者。
近期多模态大模型的发展推动了联合概率模型的进步,使其兼具图像理解和生成能力。然而,我们发现现有方法在图像理解任务中存在信息丢失问题,原因在于图像离散化或扩散去噪过程。为此,我们提出一种新型多模态自回归(MMAR)概率建模框架。不同于离散化方法,MMAR采用连续值图像令牌,高效避免信息损失;不同于扩散方法,通过在每个自回归图像块嵌入上添加轻量级扩散头,将扩散过程与自回归主干解耦。这样,在从图像生成切换到理解任务时,主干模型对图像的隐藏表示不局限于最后的去噪步骤。为成功训练该方法,我们还提出了理论证明的数值稳定技术及平衡生成与理解目标的训练策略。在18个图像理解基准上的大量评估表明,MMAR显著优于大多数现有联合多模态模型,甚至超越使用预训练CLIP视觉编码器的方法。同时,MMAR具备高质量图像生成能力,且在更大数据和模型规模下表现出可扩展性。
原文摘要 · Abstract (English)
Recent advancements in multi-modal large language models have propelled the development of joint probabilistic models capable of both image understanding and generation. However, we have identified that recent methods suffer from loss of image information during understanding task, due to either image discretization or diffusion denoising steps. To address this issue, we propose a novel Multi-Modal Auto-Regressive (MMAR) probabilistic modeling framework. Unlike discretization line of method, MMAR takes in continuous-valued image tokens to avoid information loss in an efficient way. Differing from diffusion-based approaches, we disentangle the diffusion process from auto-regressive backbone model by employing a light-weight diffusion head on top each auto-regressed image patch embedding. In this way, when the model transits from image generation to understanding through text generation, the backbone model's hidden representation of the image is not limited to the last denoising step. To successfully train our method, we also propose a theoretically proven technique that addresses the numerical stability issue and a training strategy that balances the generation and understanding task goals. Extensive evaluations on 18 image understanding benchmarks show that MMAR significantly outperforms most of the existing joint multi-modal models, surpassing the method that employs pre-trained CLIP vision encoder. Meanwhile, MMAR is able to generate high quality images. We also show that our method is scalable with larger data and model size.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。