Autoregressive Image Generation with Linear Complexity: A Spatial-Aware Decay Perspective

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Mao, Yuxin, Qin, Zhen, Zhou, Jinxing, Deng, Hui, Shen, Xuyang, Fan, Bin, Zhang, Jing, Zhong, Yiran, Dai, Yuchao
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866915369502900224
author Mao, Yuxin
Qin, Zhen
Zhou, Jinxing
Deng, Hui
Shen, Xuyang
Fan, Bin
Zhang, Jing
Zhong, Yiran
Dai, Yuchao
author_facet Mao, Yuxin
Qin, Zhen
Zhou, Jinxing
Deng, Hui
Shen, Xuyang
Fan, Bin
Zhang, Jing
Zhong, Yiran
Dai, Yuchao
contents Autoregressive (AR) models have garnered significant attention in image generation for their ability to effectively capture both local and global structures within visual data. However, prevalent AR models predominantly rely on the transformer architectures, which are beset by quadratic computational complexity concerning input sequence length and substantial memory overhead due to the necessity of maintaining key-value caches. Although linear attention mechanisms have successfully reduced this burden in language models, our initial experiments reveal that they significantly degrade image generation quality because of their inability to capture critical long-range dependencies in visual data. We propose Linear Attention with Spatial-Aware Decay (LASAD), a novel attention mechanism that explicitly preserves genuine 2D spatial relationships within the flattened image sequences by computing position-dependent decay factors based on true 2D spatial location rather than 1D sequence positions. Based on this mechanism, we present LASADGen, an autoregressive image generator that enables selective attention to relevant spatial contexts with linear complexity. Experiments on ImageNet show LASADGen achieves state-of-the-art image generation performance and computational efficiency, bridging the gap between linear attention's efficiency and spatial understanding needed for high-quality generation.
format Preprint
id arxiv_https___arxiv_org_abs_2507_01652
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Autoregressive Image Generation with Linear Complexity: A Spatial-Aware Decay Perspective
Mao, Yuxin
Qin, Zhen
Zhou, Jinxing
Deng, Hui
Shen, Xuyang
Fan, Bin
Zhang, Jing
Zhong, Yiran
Dai, Yuchao
Computer Vision and Pattern Recognition
Artificial Intelligence
Multimedia
Autoregressive (AR) models have garnered significant attention in image generation for their ability to effectively capture both local and global structures within visual data. However, prevalent AR models predominantly rely on the transformer architectures, which are beset by quadratic computational complexity concerning input sequence length and substantial memory overhead due to the necessity of maintaining key-value caches. Although linear attention mechanisms have successfully reduced this burden in language models, our initial experiments reveal that they significantly degrade image generation quality because of their inability to capture critical long-range dependencies in visual data. We propose Linear Attention with Spatial-Aware Decay (LASAD), a novel attention mechanism that explicitly preserves genuine 2D spatial relationships within the flattened image sequences by computing position-dependent decay factors based on true 2D spatial location rather than 1D sequence positions. Based on this mechanism, we present LASADGen, an autoregressive image generator that enables selective attention to relevant spatial contexts with linear complexity. Experiments on ImageNet show LASADGen achieves state-of-the-art image generation performance and computational efficiency, bridging the gap between linear attention's efficiency and spatial understanding needed for high-quality generation.
title Autoregressive Image Generation with Linear Complexity: A Spatial-Aware Decay Perspective
topic Computer Vision and Pattern Recognition
Artificial Intelligence
Multimedia
url https://arxiv.org/abs/2507.01652