Stability of Accuracy for the Training of DNNs Via the Uniform Doubling Condition

TL;DR

This paper investigates the stability of accuracy during deep neural network training, introducing a uniform doubling condition on training data that guarantees accuracy stability across time and extends results to broader activation functions.

Contribution

It introduces a uniform doubling condition on training data that ensures accuracy stability and extends stability results to piecewise linear activation functions beyond absolute value.

Findings

Uniform doubling condition guarantees accuracy stability over training

Stability results extended to Leaky ReLU and other piecewise linear activations

Provides sufficient data conditions for high accuracy persistence

Abstract

We study the stability of accuracy during the training of deep neural networks (DNNs). In this context, the training of a DNN is performed via the minimization of a cross-entropy loss function, and the performance metric is accuracy (the proportion of objects that are classified correctly). While training results in a decrease of loss, the accuracy does not necessarily increase during the process and may sometimes even decrease. The goal of achieving stability of accuracy is to ensure that if accuracy is high at some initial time, it remains high throughout training. A recent result by Berlyand, Jabin, and Safsten introduces a doubling condition on the training data, which ensures the stability of accuracy during training for DNNs using the absolute value activation function. For training data in $\mathbb{R}^n$, this doubling condition is formulated using slabs in $\mathbb{R}^n$ and depends on the choice of the slabs. The goal of this paper is twofold. First, to make the doubling condition uniform, that is, independent of the choice of slabs. This leads to sufficient conditions for stability in terms of training data only. In other words, for a training set $T$ that satisfies the uniform doubling condition, there exists a family of DNNs such that a DNN from this family with high accuracy on the training set at some training time $t_0$ will have high accuracy for all time $t>t_0$. Moreover, establishing uniformity is necessary for the numerical implementation of the doubling condition. The second goal is to extend the original stability results from the absolute value activation function to a broader class of piecewise linear activation functions with finitely many critical points, such as the popular Leaky ReLU.