通过插值控制神经网络归纳偏置,提升低算力下MLP性能
Interpolated-MLPs: Controllable Inductive Bias
- 用预训练模型权重插值生成带可控偏置的MLP
- 低算力下性能随归纳偏置呈双侧对数关系提升
- 适合需要灵活调节模型复杂度的研究者
由于归纳偏置较弱,多层感知机(MLP)在低算力条件下表现不如基于卷积的网络(CNN)。然而,已有研究表明,当计算量增加时,性能差距会显著缩小。本文反其道而行之:在低算力场景下,逐步增加归纳偏置会对性能产生何种影响?为此,我们提出一种“软MLP”方法——插值MLP(I-MLP),通过在高归纳偏置的先验模型(如CNN、MLP-Mixer)与标准MLP之间进行权重插值,实现对归纳偏置的连续调控。实验表明,在视觉任务的低算力条件下,使用CNN和MLP-Mixer作为先验模型时,性能与归纳偏置之间存在连续且双侧的对数关系。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。