Regularisation in neural networks: a survey and empirical analysis of approaches

TL;DR

This paper reviews various regularisation techniques for neural networks, categorizes them, and empirically tests their effectiveness across multiple datasets and architectures, revealing dataset-dependent results.

Contribution

It provides a comprehensive taxonomy of regularisation methods and empirically evaluates their practical impact, highlighting when and where they improve neural network performance.

Findings

Regularisation effectiveness varies by dataset type.

Batch normalization benefits image datasets.

Regularisation improves numeric datasets performance.

Abstract

Despite huge successes on a wide range of tasks, neural networks are known to sometimes struggle to generalise to unseen data. Many approaches have been proposed over the years to promote the generalisation ability of neural networks, collectively known as regularisation techniques. These are used as common practice under the assumption that any regularisation added to the pipeline would result in a performance improvement. In this study, we investigate whether this assumption holds in practice. First, we provide a broad review of regularisation techniques, including modern theories such as double descent. We propose a taxonomy of methods under four broad categories, namely: (1) data-based strategies, (2) architecture strategies, (3) training strategies, and (4) loss function strategies. Notably, we highlight the contradictions and correspondences between the approaches in these broad classes. Further, we perform an empirical comparison of the various regularisation techniques on classification tasks for ten numerical and image datasets applied to the multi-layer perceptron and convolutional neural network architectures. Results show that the efficacy of regularisation is dataset-dependent. For example, the use of a regularisation term only improved performance on numeric datasets, whereas batch normalisation improved performance on image datasets only. Generalisation is crucial to machine learning; thus, understanding the effects of applying regularisation techniques, and considering the connections between them is essential to the appropriate use of these methods in practice.