Towards Flexible Inductive Bias via Progressive Reparameterization Scheduling

TL;DR

This paper introduces Progressive Reparameterization Scheduling (PRS), a method to dynamically adjust inductive biases between convolution and self-attention in models, improving performance across different data scales.

Contribution

The paper proposes a novel reparameterization approach that interpolates inductive biases between CNNs and ViTs, enabling flexible adaptation to data scale variations.

Findings

PRS outperforms previous methods on small-scale datasets like CIFAR-100.

Fourier analysis reveals how inductive bias effectiveness varies with data scale.

Reparameterization accelerates the transition from convolution to self-attention during training.

Abstract

There are two de facto standard architectures in recent computer vision: Convolutional Neural Networks (CNNs) and Vision Transformers (ViTs). Strong inductive biases of convolutions help the model learn sample effectively, but such strong biases also limit the upper bound of CNNs when sufficient data are available. On the contrary, ViT is inferior to CNNs for small data but superior for sufficient data. Recent approaches attempt to combine the strengths of these two architectures. However, we show these approaches overlook that the optimal inductive bias also changes according to the target data scale changes by comparing various models' accuracy on subsets of sampled ImageNet at different ratios. In addition, through Fourier analysis of feature maps, the model's response patterns according to signal frequency changes, we observe which inductive bias is advantageous for each data scale. The more convolution-like inductive bias is included in the model, the smaller the data scale is required where the ViT-like model outperforms the ResNet performance. To obtain a model with flexible inductive bias on the data scale, we show reparameterization can interpolate inductive bias between convolution and self-attention. By adjusting the number of epochs the model stays in the convolution, we show that reparameterization from convolution to self-attention interpolates the Fourier analysis pattern between CNNs and ViTs. Adapting these findings, we propose Progressive Reparameterization Scheduling (PRS), in which reparameterization adjusts the required amount of convolution-like or self-attention-like inductive bias per layer. For small-scale datasets, our PRS performs reparameterization from convolution to self-attention linearly faster at the late stage layer. PRS outperformed previous studies on the small-scale dataset, e.g., CIFAR-100.