arXiv:2606.09156cs.CV2026-06NeurIPS被引 6

一个模型搞定图文、深度图、图像编辑等多种生成任务

OmniGen-AR: AutoRegressive Any-to-Image Generation

论文配图:OmniGen-AR: AutoRegressive Any-to-Image Generation
图 1 · 摘自论文原文
  • 用统一框架处理多种输入条件,支持跨模态生成
  • 在GenEval和VBench上分别取得0.63和80.02的优异成绩
  • 创新设计分离注意力机制,防止条件信息泄露

自回归(AR)模型在视觉生成中展现出强大潜力,架构简单且优化目标明确。然而现有方法通常仅限于单模态输入(如文本),难以满足真实场景中多样化控制的需求。本文提出OmniGen-AR,一种统一的自回归框架,实现任意条件到图像的生成。通过共享视觉分词器与文本分词器,将多种视觉条件(如文本、分割图、深度图)统一表示,支持文本到图像、分割到图像、深度到图像、图像编辑、帧预测及文本到视频等任务。为防止条件令牌向内容令牌泄露信息,引入解耦因果注意力(DCA),在训练时分离条件与内容的因果掩码,不影响推理阶段的标准逐标记预测。该设计使OmniGen-AR在多个基准上达到新SOTA或竞争力水平,例如GenEval得分为0.63,VBench得分为80.02,证明其在灵活且高保真视觉生成方面的有效性。

原文摘要 · Abstract (English)

Autoregressive (AR) models have demonstrated strong potential in visual generation, offering superior performance with simple architectures and optimization objectives. However, existing methods are typically limited to single-modality conditions, e.g., text, restricting their applicability in real-world scenarios that demand image synthesis from diverse controls. In this work, we present OmniGen-AR, a unified autoregressive framework for Any-to-Image generation. By discretizing various visual conditions through a shared visual tokenizer and text prompts with a text tokenizer, OmniGen-AR supports a broad spectrum of conditional inputs within a single model, including text (text-to-image generation), spatial signals (segmentation-to-image and depth-to-image), and visual context (image editing, frame prediction, and text-to-video generation). To mitigate the risk of information leakage from condition tokens to content tokens, we introduce Disentangled Causal Attention (DCA), which separates the full-sequence causal mask into condition causal attention and content causal attention. It serves as a training-time regularizer without affecting the standard next-token prediction during inference. With this design, OmniGen-AR achieves new state-of-the-art or at least competitive results across a range of benchmark, e.g., 0.63 on GenEval and 80.02 on VBench, demonstrating its effectiveness in flexible and high-fidelity visual generation.

图像生成自回归模型多模态扩散模型

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