Resmlp: Feedforward networks for image classification with data-efficient training

H Touvron, P Bojanowski, M Caron… - IEEE transactions on …, 2022 - ieeexplore.ieee.org
IEEE transactions on pattern analysis and machine intelligence, 2022ieeexplore.ieee.org
We present ResMLP, an architecture built entirely upon multi-layer perceptrons for image
classification. It is a simple residual network that alternates (i) a linear layer in which image
patches interact, independently and identically across channels, and (ii) a two-layer feed-
forward network in which channels interact independently per patch. When trained with a
modern training strategy using heavy data-augmentation and optionally distillation, it attains
surprisingly good accuracy/complexity trade-offs on ImageNet. We also train ResMLP …
We present ResMLP, an architecture built entirely upon multi-layer perceptrons for image classification. It is a simple residual network that alternates (i) a linear layer in which image patches interact, independently and identically across channels, and (ii) a two-layer feed-forward network in which channels interact independently per patch. When trained with a modern training strategy using heavy data-augmentation and optionally distillation, it attains surprisingly good accuracy/complexity trade-offs on ImageNet. We also train ResMLP models in a self-supervised setup, to further remove priors from employing a labelled dataset. Finally, by adapting our model to machine translation we achieve surprisingly good results. We share pre-trained models and our code based on the Timm library.
ieeexplore.ieee.org
Showing the best result for this search. See all results