Saved in:
Bibliographic Details
Main Authors: Wu, Sean, Hong, Jordan, Bai, Keyu, Bachmann, Gregor
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2410.09655
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910648166776832
author Wu, Sean
Hong, Jordan
Bai, Keyu
Bachmann, Gregor
author_facet Wu, Sean
Hong, Jordan
Bai, Keyu
Bachmann, Gregor
contents Due to their weak inductive bias, Multi-Layer Perceptrons (MLPs) have subpar performance at low-compute levels compared to standard architectures such as convolution-based networks (CNN). Recent work, however, has shown that the performance gap drastically reduces as the amount of compute is increased without changing the amount of inductive bias. In this work, we study the converse: in the low-compute regime, how does the incremental increase of inductive bias affect performance? To quantify inductive bias, we propose a "soft MLP" approach, which we coin Interpolated MLP (I-MLP). We control the amount of inductive bias in the standard MLP by introducing a novel algorithm based on interpolation between fixed weights from a prior model with high inductive bias. We showcase our method using various prior models, including CNNs and the MLP-Mixer architecture. This interpolation scheme allows fractional control of inductive bias, which may be attractive when full inductive bias is not desired (e.g. in the mid-compute regime). We find experimentally that for Vision Tasks in the low-compute regime, there is a continuous and two-sided logarithmic relationship between inductive bias and performance when using CNN and MLP-Mixer prior models.
format Preprint
id arxiv_https___arxiv_org_abs_2410_09655
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Interpolated-MLPs: Controllable Inductive Bias
Wu, Sean
Hong, Jordan
Bai, Keyu
Bachmann, Gregor
Machine Learning
Due to their weak inductive bias, Multi-Layer Perceptrons (MLPs) have subpar performance at low-compute levels compared to standard architectures such as convolution-based networks (CNN). Recent work, however, has shown that the performance gap drastically reduces as the amount of compute is increased without changing the amount of inductive bias. In this work, we study the converse: in the low-compute regime, how does the incremental increase of inductive bias affect performance? To quantify inductive bias, we propose a "soft MLP" approach, which we coin Interpolated MLP (I-MLP). We control the amount of inductive bias in the standard MLP by introducing a novel algorithm based on interpolation between fixed weights from a prior model with high inductive bias. We showcase our method using various prior models, including CNNs and the MLP-Mixer architecture. This interpolation scheme allows fractional control of inductive bias, which may be attractive when full inductive bias is not desired (e.g. in the mid-compute regime). We find experimentally that for Vision Tasks in the low-compute regime, there is a continuous and two-sided logarithmic relationship between inductive bias and performance when using CNN and MLP-Mixer prior models.
title Interpolated-MLPs: Controllable Inductive Bias
topic Machine Learning
url https://arxiv.org/abs/2410.09655