|
SPECIAL ISSUE: ARTIFICIAL INTELLIGENCE AND MACHINE LEARNING TECHNOLOGIES
Barcodes as summary of loss function topology
S. A. Barannikovab, A. A. Korotinac, D. A. Oganesyana, D. I. Emtsevad, E. V. Burnaevac a Skolkovo Institute of Science and Technology, Moscow, Russia
b Université Paris Cité, Paris, France
c Artificial Intelligence Research Institute, Moscow, Russia
d Eidgenösische Technische Hochschule Zürich, Switzerland
Abstract:
We propose to study neural networks' loss surfaces by methods of topological data analysis. We suggest to apply barcodes of Morse complexes to explore topology of loss surfaces. An algorithm for calculations of the loss function’s barcodes of local minima is described. We have conducted experiments for calculating barcodes of local minima for benchmark functions and for loss surfaces of small neural networks. Our experiments confirm our two principal observations for neural networks' loss surfaces. First, the barcodes of local minima are located in a small lower part of the range of values of neural networks' loss function. Secondly, increase of the neural network’s depth and width lowers the barcodes of local minima. This has some natural implications for the neural network’s learning and for its generalization properties.
Citation:
S. A. Barannikov, A. A. Korotin, D. A. Oganesyan, D. I. Emtsev, E. V. Burnaev, “Barcodes as summary of loss function topology”, Dokl. RAN. Math. Inf. Proc. Upr., 514:2 (2023), 196–211; Dokl. Math., 108:suppl. 2 (2023), S333–S347
Linking options:
https://www.mathnet.ru/eng/danma465 https://www.mathnet.ru/eng/danma/v514/i2/p196
|
|