Layout-Conditioned Autoregressive Text-to-Image Generation via Structured Masking
By: Zirui Zheng , Takashi Isobe , Tong Shen and more
Potential Business Impact:
Draws pictures from text and layout.
While autoregressive (AR) models have demonstrated remarkable success in image generation, extending them to layout-conditioned generation remains challenging due to the sparse nature of layout conditions and the risk of feature entanglement. We present Structured Masking for AR-based Layout-to-Image (SMARLI), a novel framework for layoutto-image generation that effectively integrates spatial layout constraints into AR-based image generation. To equip AR model with layout control, a specially designed structured masking strategy is applied to attention computation to govern the interaction among the global prompt, layout, and image tokens. This design prevents mis-association between different regions and their descriptions while enabling sufficient injection of layout constraints into the generation process. To further enhance generation quality and layout accuracy, we incorporate Group Relative Policy Optimization (GRPO) based post-training scheme with specially designed layout reward functions for next-set-based AR models. Experimental results demonstrate that SMARLI is able to seamlessly integrate layout tokens with text and image tokens without compromising generation quality. It achieves superior layoutaware control while maintaining the structural simplicity and generation efficiency of AR models.
Similar Papers
Text-Conditioned Background Generation for Editable Multi-Layer Documents
CV and Pattern Recognition
Makes documents look good and easy to read.
VideoMAR: Autoregressive Video Generatio with Continuous Tokens
CV and Pattern Recognition
Makes videos from pictures, faster and better.
Unleashing the Potential of Large Language Models for Text-to-Image Generation through Autoregressive Representation Alignment
CV and Pattern Recognition
Makes AI draw pictures that make sense.