English

Convolutional Initialization for Data-Efficient Vision Transformers

Computer Vision and Pattern Recognition 2024-01-24 v1

Abstract

Training vision transformer networks on small datasets poses challenges. In contrast, convolutional neural networks (CNNs) can achieve state-of-the-art performance by leveraging their architectural inductive bias. In this paper, we investigate whether this inductive bias can be reinterpreted as an initialization bias within a vision transformer network. Our approach is motivated by the finding that random impulse filters can achieve almost comparable performance to learned filters in CNNs. We introduce a novel initialization strategy for transformer networks that can achieve comparable performance to CNNs on small datasets while preserving its architectural flexibility.

Keywords

Cite

@article{arxiv.2401.12511,
  title  = {Convolutional Initialization for Data-Efficient Vision Transformers},
  author = {Jianqiao Zheng and Xueqian Li and Simon Lucey},
  journal= {arXiv preprint arXiv:2401.12511},
  year   = {2024}
}

Comments

14 pages, 9 figures, 8 tables