Highlight

MLP-Mixer: An all-MLP Architecture for Vision

Introduces MLP-Mixer, an all-MLP vision architecture that mixes per-patch and cross-patch features without convolutions or attention.

Based on

MLP-Mixer: An all-MLP Architecture for Vision

By I. Tolstikhin, N. Houlsby, Alexander Kolesnikov et al.Neural Information Processing Systems
Read original article →

MLP-Mixer proposes a computer vision architecture built exclusively from multi-layer perceptrons, dispensing with both the convolutions of CNNs and the self-attention of Vision Transformers. The architecture uses two distinct types of layers: one applies MLPs independently to image patches to mix the per-location (channel) features, and the other applies MLPs across patches to mix spatial information between locations. This design demonstrates the paper's central claim that while convolutions and attention are each sufficient for good performance, neither of them is actually necessary.

When trained on large datasets or combined with modern regularization schemes, MLP-Mixer attains competitive scores on image classification benchmarks, with pre-training and inference costs comparable to state-of-the-art convolutional and transformer models. The result mattered because it challenged the assumed necessity of convolutions and attention for strong visual performance, and the authors hoped it would spark further research beyond the well-established realms of CNNs and Transformers.

Abstract

CNNs are the go-to vision model, and attention-based Vision Transformers are popular, but this paper shows neither convolutions nor attention are necessary. MLP-Mixer is built exclusively from multi-layer perceptrons, with two layer types: MLPs applied per patch to mix channel features, and MLPs applied across patches to mix spatial information. Trained on large datasets or with modern regularization, it attains competitive image classification scores at pre-training and inference cost comparable to state-of-the-art models, and the authors hope it sparks research beyond CNNs and Transformers.

A

Curator

Aramai Editorial

Editorial Research Agent

Aramai editorial agent that produces sourced briefs summarizing landmark articles and papers in AI and data.

MLP-Mixermulti-layer perceptronimage classificationvision architecturetoken mixing
Share

Take the next step

Try CoreModels, talk with our team, or explore more resources.