Emergence of hierarchical modes from deep learning
Abstract
Large-scale deep neural networks consume expensive training costs, but the training results in less-interpretable weight matrices constructing the networks. Here, we propose a mode decomposition learning that can interpret the weight matrices as a hierarchy of latent modes. These modes are akin to patterns in physics studies of memory networks, but the least number of modes increases only logarithmically with the network width and even becomes a constant when the width grows further. The mode decomposition learning not only saves a significant large amount of training costs but also explains the network performance with the leading modes, displaying a striking piecewise power-law behavior. The modes specify a progressively compact latent space across the network hierarchy, making a more disentangled subspace compared to standard training. Our mode decomposition learning is also studied in an analytic online learning setting, which reveals multiple stages of learning dynamics with a continuous specialization of hidden nodes. Therefore the proposed mode decomposition learning points to a cheap and interpretable route towards the magical deep learning.
- Publication:
-
Physical Review Research
- Pub Date:
- April 2023
- DOI:
- 10.1103/PhysRevResearch.5.L022011
- arXiv:
- arXiv:2208.09859
- Bibcode:
- 2023PhRvR...5b2011L
- Keywords:
-
- Computer Science - Machine Learning;
- Condensed Matter - Disordered Systems and Neural Networks;
- Condensed Matter - Statistical Mechanics;
- Computer Science - Neural and Evolutionary Computing;
- Quantitative Biology - Neurons and Cognition
- E-Print:
- 5 pages +11 pages (SM), 4+10 figures, revised version to the journal