RUS  ENG
Full version
JOURNALS // Doklady Rossijskoj Akademii Nauk. Mathematika, Informatika, Processy Upravlenia // Archive

Dokl. RAN. Math. Inf. Proc. Upr., 2023 Volume 514, Number 2, Pages 196–211 (Mi danma465)

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.

UDC: 004.8

Presented: A. L. Semenov
Received: 02.09.2023
Revised: 08.09.2023
Accepted: 18.10.2023

DOI: 10.31857/S268695432360177X


 English version:
Doklady Mathematics, 2023, 108:suppl. 2, S333–S347

Bibliographic databases:


© Steklov Math. Inst. of RAS, 2024