arXiv:2411.19722cs.LGcs.AI2024-11ICLR被引 44

JetFormer统一生成图像与文本,无需预训练组件。

JetFormer: An Autoregressive Generative Model of Raw Images and Text

  • 仅用自回归解码器+归一化流模型,直接建模原始数据。
  • 生成图像质量媲美依赖预训练编码器的基线模型。
  • 首个实现高保真图像生成与强似然上界联合的模型。

消除建模约束并统一跨域架构是大模型进展的关键驱动力。然而,多数模型仍依赖多个独立训练的组件,如模态专用编码器和解码器。本文进一步简化图像与文本的联合生成建模,提出一种仅含自回归解码器的Transformer——JetFormer,可直接最大化原始数据的似然,无需任何预训练组件,同时理解并生成文本与图像。具体而言,我们利用归一化流模型获得软令牌图像表示,并与自回归多模态Transformer联合训练。该归一化流模型在推理时兼具图像编码(感知)与解码(生成)功能。JetFormer在文本到图像生成质量上达到与近期基于VQ-VAE和VAE的基线相当的水平。这些基线依赖预训练图像自编码器,需复杂混合损失(包括感知损失)。同时,JetFormer展现出稳健的图像理解能力。据我们所知,它是首个能生成高保真图像并提供强大似然上界的模型。

原文摘要 · Abstract (English)

Removing modeling constraints and unifying architectures across domains has been a key driver of the recent progress in training large multimodal models. However, most of these models still rely on many separately trained components such as modality-specific encoders and decoders. In this work, we further streamline joint generative modeling of images and text. We propose an autoregressive decoder-only transformer - JetFormer - which is trained to directly maximize the likelihood of raw data, without relying on any separately pretrained components, and can understand and generate both text and images. Specifically, we leverage a normalizing flow model to obtain a soft-token image representation that is jointly trained with an autoregressive multimodal transformer. The normalizing flow model serves as both an image encoder for perception tasks and an image decoder for image generation tasks during inference. JetFormer achieves text-to-image generation quality competitive with recent VQ-VAE- and VAE-based baselines. These baselines rely on pretrained image autoencoders, which are trained with a complex mixture of losses, including perceptual ones. At the same time, JetFormer demonstrates robust image understanding capabilities. To the best of our knowledge, JetFormer is the first model that is capable of generating high-fidelity images and producing strong log-likelihood bounds.

图像生成自回归模型多模态归一化流

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。