NARAIM: Native Aspect Ratio Autoregressive Image Models

Daniel Gallo Fernández, Robert van der Klis, Răzvan-Andrei Matișan, Janusz Partyka, Samuele Papa, Efstratios Gavves, Phillip Lippe

Keywords: self-supervised learning, computer vision, autoregressive image models, scalable pre-training, native aspect ratio

TL;DR: We extend autoregressive image model pre-training to native aspect ratios and highlight its benefit.

Abstract: While vision transformers are able to solve a wide variety of computer vision tasks, no pre-training method has yet demonstrated the same scaling laws as observed in language models. Autoregressive models show promising results, but are commonly trained on images that are cropped or transformed into square images, which distorts or destroys information present in the input. To overcome this limitation, we propose NARAIM, a vision model pre-trained with an autoregressive objective that uses images in their native aspect ratio. By maintaining the native aspect ratio, we preserve the original spatial context, thereby enhancing the model's ability to interpret visual information. In our experiments, we show that maintaining the aspect ratio improves performance on a downstream classification task.

Submission Number: 19