Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Scaling MLPs: A Tale of Inductive Bias

About

In this work we revisit the most fundamental building block in deep learning, the multi-layer perceptron (MLP), and study the limits of its performance on vision tasks. Empirical insights into MLPs are important for multiple reasons. (1) Given the recent narrative "less inductive bias is better", popularized due to transformers eclipsing convolutional models, it is natural to explore the limits of this hypothesis. To that end, MLPs offer an ideal test bed, as they lack any vision-specific inductive bias. (2) MLPs have almost exclusively been the main protagonist in the deep learning theory literature due to their mathematical simplicity, serving as a proxy to explain empirical phenomena observed for more complex architectures. Surprisingly, experimental datapoints for MLPs are very difficult to find in the literature, especially when coupled with large pre-training protocols. This discrepancy between practice and theory is worrying: Do MLPs reflect the empirical advances exhibited by practical models? Or do theorists need to rethink the role of MLPs as a proxy? We provide insights into both these aspects. We show that the performance of MLPs drastically improves with scale (95% on CIFAR10, 82% on CIFAR100, 58% on ImageNet ReaL), highlighting that lack of inductive bias can indeed be compensated. We observe that MLPs mimic the behaviour of their modern counterparts faithfully, with some components in the learning setting however exhibiting stronger or unexpected behaviours. Due to their inherent computational efficiency, large pre-training experiments become more accessible for academic researchers. All of our experiments were run on a single GPU.

Gregor Bachmann, Sotiris Anagnostidis, Thomas Hofmann• 2023

Related benchmarks

TaskDatasetResultRank
Image ClassificationImageNet
Top-1 Accuracy51.5
324
Image ClassificationCIFAR10
Accuracy95.5
240
Image ClassificationImageNet-ReaL
Precision@157.9
195
Image ClassificationSTL-10
Top-1 Accuracy92.2
128
Image ClassificationTinyImageNet
Accuracy73.1
108
Image ClassificationCIFAR100
Top-1 Accuracy82.6
42
Image Classification64x64 images
Latency (ms)21.2
4
Showing 7 of 7 rows

Other info

Code

Follow for update