English

Autoregressive Pre-Training on Pixels and Texts

Computation and Language 2024-10-04 v3 Computer Vision and Pattern Recognition

Abstract

The integration of visual and textual information represents a promising direction in the advancement of language models. In this paper, we explore the dual modality of language--both visual and textual--within an autoregressive framework, pre-trained on both document images and texts. Our method employs a multimodal training strategy, utilizing visual data through next patch prediction with a regression head and/or textual data through next token prediction with a classification head. We focus on understanding the interaction between these two modalities and their combined impact on model performance. Our extensive evaluation across a wide range of benchmarks shows that incorporating both visual and textual data significantly improves the performance of pixel-based language models. Remarkably, we find that a unidirectional pixel-based model trained solely on visual data can achieve comparable results to state-of-the-art bidirectional models on several language understanding tasks. This work uncovers the untapped potential of integrating visual and textual modalities for more effective language modeling. We release our code, data, and model checkpoints at \url{https://github.com/ernie-research/pixelgpt}.

Keywords

Cite

@article{arxiv.2404.10710,
  title  = {Autoregressive Pre-Training on Pixels and Texts},
  author = {Yekun Chai and Qingyi Liu and Jingwu Xiao and Shuohuan Wang and Yu Sun and Hua Wu},
  journal= {arXiv preprint arXiv:2404.10710},
  year   = {2024}
}

Comments

EMNLP 2024

R2 v1 2026-06-28T15:56:04.463Z