arXiv:2505.20288cs.CVcs.MM2025-05ICML被引 10

用低分辨率关键点引导多阶段图像生成,提升全局结构建模效率

Hierarchical Masked Autoregressive Models with Low-Resolution Token Pivots

  • 分阶段生成:先预测少量低分辨率图像块作结构引导
  • 生成质量优于传统方法,计算开销更低
  • 适合追求高效高质图像生成的研究者与开发者

自回归模型已成为视觉生成的强大范式。当前主流的逐标记预测通常基于单一尺度的密集图像标记序列,难以利用早期标记的全局上下文信息。本文提出分层自回归模型(Hi-MAR),通过少数低分辨率图像标记作为分层生成的枢纽,实现跨多尺度标记的层次化依赖建模。在第一阶段,模型学习预测少量低分辨率图像标记,作为反映全局结构的中间引导;这些标记在后续阶段提供全局结构感知,增强对典型密集图像标记的生成能力。此外,设计了新的扩散变压器头,以强化所有标记间的全局上下文信息,用于掩码标记预测。在类别条件生成和文本到图像生成任务上的大量实验表明,Hi-MAR性能优于典型自回归基线,同时计算成本更低。代码已开源。

原文摘要 · Abstract (English)

Autoregressive models have emerged as a powerful generative paradigm for visual generation. The current de-facto standard of next token prediction commonly operates over a single-scale sequence of dense image tokens, and is incapable of utilizing global context especially for early tokens prediction. In this paper, we introduce a new autoregressive design to model a hierarchy from a few low-resolution image tokens to the typical dense image tokens, and delve into a thorough hierarchical dependency across multi-scale image tokens. Technically, we present a Hierarchical Masked Autoregressive models (Hi-MAR) that pivot on low-resolution image tokens to trigger hierarchical autoregressive modeling in a multi-phase manner. Hi-MAR learns to predict a few image tokens in low resolution, functioning as intermediary pivots to reflect global structure, in the first phase. Such pivots act as the additional guidance to strengthen the next autoregressive modeling phase by shaping global structural awareness of typical dense image tokens. A new Diffusion Transformer head is further devised to amplify the global context among all tokens for mask token prediction. Extensive evaluations on both class-conditional and text-to-image generation tasks demonstrate that Hi-MAR outperforms typical AR baselines, while requiring fewer computational costs. Code is available at https://github.com/HiDream-ai/himar.

自回归生成多尺度建模图像生成高效模型

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