vix.ing · top · new · best · stats

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

2021/05/07 by Hugo Touvron, Piotr Bojanowski, Touvron, Hugo +21 · 29 citations
Computer Science · Engineering · Mathematics · Medicine · Neuroscience · #Algorithm #Artificial intelligence #Artificial neural network #Brain Tumor Detection and Classification #COVID-19 diagnosis using AI #Code (set theory) #Computer Vision and Pattern Recognition (cs.CV) #Computer network #Computer science #Contextual image classification #Digital Imaging for Blood Diseases #Engineering #FOS: Computer and information sciences #Feed forward #Image (mathematics) #Independent and identically distributed random variables #Layer (electronics) #Machine learning #Mathematics #Network architecture #Pattern recognition (psychology) #Perceptron #Residual #Set (abstract data type) #Simple (philosophy) #Statistics #Translation (biology) #cs.CV

paper · pdf · open access · doi:10.48550/arxiv.2105.03404

published in arXiv (Cornell University) (Cornell University)

openalex publication_date 2021/05/07 · arxiv created 2021/06/10 · arxiv updated 2021/06/11 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/28

Abstract

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.

Cited by

Related