OUIDecay: Adaptive Layer-wise Weight Decay for CNNs Using Online Activation Patterns

TL;DR

OUIDecay introduces an adaptive, layer-wise weight decay method for CNNs that uses online activation patterns to improve regularization without validation data, outperforming fixed decay in various benchmarks.

Contribution

The paper presents OUIDecay, a novel activation-based, online, layer-wise weight decay scheduler that adapts during training without validation data, enhancing CNN regularization.

Findings

OUIDecay achieves the best mean validation loss in 7 out of 8 settings.

Activation-driven decay outperforms fixed and gradient-based methods.

The approach is lightweight and suitable for online training.

Abstract

Weight decay remains one of the most widely used regularization mechanisms for training convolutional neural networks, yet it is still commonly applied as a fixed coefficient shared by all layers throughout training. This uniform treatment ignores that different layers may follow different structural dynamics and therefore may require different regularization strengths. In this work, we propose OUIDecay, an adaptive layer-wise and time-dependent weight decay scheduler for CNNs driven by the Overfitting-Underfitting Indicator (OUI), an activation-based metric previously shown to provide early information about regularization quality. OUIDecay uses a lightweight batch-based formulation of OUI to monitor the structural behavior of each layer online and periodically rescales its weight decay relative to the other layers in the network. Unlike gradient-based adaptive decay methods, our approach relies on functional information extracted from activation patterns and does not require validation data. Experiments on EfficientNet-B0 with Stanford Cars, ResNet50 with Food101, DenseNet121 with CIFAR100, and MobileNetV2 with CIFAR10 show that OUIDecay achieves the best mean best-validation-loss in 7 out of 8 evaluated settings. These results indicate that activation-driven weight decay adaptation is a practical and effective alternative to fixed decay and gradient-based adaptive decay, while keeping the method lightweight and suitable for online use.