Journal of Pathology Informatics (Jan 2022)

Number of necessary training examples for Neural Networks with different number of trainable parameters

  • Th.I. Götz,
  • S. Göb,
  • S. Sawant,
  • X.F. Erick,
  • T. Wittenberg,
  • C. Schmidkonz,
  • A.M. Tomé,
  • E.W. Lang,
  • A. Ramming

Journal volume & issue
Vol. 13
p. 100114

Abstract

Read online

In this work, the network complexity should be reduced with a concomitant reduction in the number of necessary training examples. The focus thus was on the dependence of proper evaluation metrics on the number of adjustable parameters of the considered deep neural network. The used data set encompassed Hematoxylin and Eosin (H&E) colored cell images provided by various clinics. We used a deep convolutional neural network to get the relation between a model’s complexity, its concomitant set of parameters, and the size of the training sample necessary to achieve a certain classification accuracy. The complexity of the deep neural networks was reduced by pruning a certain amount of filters in the network. As expected, the unpruned neural network showed best performance. The network with the highest number of trainable parameter achieved, within the estimated standard error of the optimized cross-entropy loss, best results up to 30% pruning. Strongly pruned networks are highly viable and the classification accuracy declines quickly with decreasing number of training patterns. However, up to a pruning ratio of 40%, we found a comparable performance of pruned and unpruned deep convolutional neural networks (DCNN) and densely connected convolutional networks (DCCN).

Keywords