用两阶段训练让自回归图像模型更精准地理解多模态指令。
MENTOR: Efficient Multimodal-Conditioned Tuning for Autoregressive Vision Generation Models
- 采用两阶段训练,实现输入与图像的逐标记对齐。
- 在DreamBench++上超越基线,保持概念一致性和提示遵循性。
- 无需额外模块,训练高效,适合多任务场景使用。
近期文本到图像模型虽生成质量高,但在精确视觉控制、多模态输入平衡及复杂多模态图像生成的训练成本方面仍存挑战。为此,我们提出MENTOR,一种高效的自回归(AR)框架,用于多模态条件微调。MENTOR结合自回归图像生成器与两阶段训练范式,实现无需辅助适配器或交叉注意力模块的细粒度、逐标记对齐。两阶段训练包括:(1) 多模态对齐阶段,建立像素级与语义级对齐;(2) 多模态指令微调阶段,平衡多模态输入融合并增强生成可控性。尽管模型规模小、基础组件非最优且训练资源有限,MENTOR在DreamBench++基准上表现优异,优于多个竞争基线,在概念保留和提示遵循方面领先。此外,本方法相较扩散模型在图像重建保真度、任务泛化能力与训练效率上均有提升。数据集、代码与模型已公开于:https://github.com/HaozheZhao/MENTOR。
原文摘要 · Abstract (English)
Recent text-to-image models produce high-quality results but still struggle with precise visual control, balancing multimodal inputs, and requiring extensive training for complex multimodal image generation. To address these limitations, we propose MENTOR, a novel autoregressive (AR) framework for efficient Multimodal-conditioned Tuning for Autoregressive multimodal image generation. MENTOR combines an AR image generator with a two-stage training paradigm, enabling fine-grained, token-level alignment between multimodal inputs and image outputs without relying on auxiliary adapters or cross-attention modules. The two-stage training consists of: (1) a multimodal alignment stage that establishes robust pixel- and semantic-level alignment, followed by (2) a multimodal instruction tuning stage that balances the integration of multimodal inputs and enhances generation controllability. Despite modest model size, suboptimal base components, and limited training resources, MENTOR achieves strong performance on the DreamBench++ benchmark, outperforming competitive baselines in concept preservation and prompt following. Additionally, our method delivers superior image reconstruction fidelity, broad task adaptability, and improved training efficiency compared to diffusion-based methods. Dataset, code, and models are available at: https://github.com/HaozheZhao/MENTOR
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。