SYNTOLOGY HomeExplorerAtlasCodeMethodologyAboutDevelopersFeedPricing
Paper · 2406.04657 · 2024

From Spikes to Heavy Tails: Unveiling the Spectral Evolution of Neural Networks

arXiv · PDF · Open in the Atlas

Code that ran

We lifted 6 functions out of this paper's own repositories and ran 6 of them in a sandbox. "Ran" means the function executed on a synthesized input and returned a value. It is not a reproduction of the paper's results.

RepositoryRoleRan
kvignesh1420/single-index-ht canonical 6 of 6
FunctionStatusWhere it lives
get_student_model Ran kvignesh1420/single-index-ht/src/models.py
pointer only (licence: NONE) · get_code("af651907e36ef135")
get_teacher_model Ran kvignesh1420/single-index-ht/src/models.py
pointer only (licence: NONE) · get_code("38801e12bc08eee3")
prepare_data_hash Ran kvignesh1420/single-index-ht/src/utils.py
pointer only (licence: NONE) · get_code("ccf42d5c20819210")
prepare_dataloaders Ran kvignesh1420/single-index-ht/src/data.py
pointer only (licence: NONE) · get_code("cfd4a149f17e8111")
prepare_student_model_hash Ran kvignesh1420/single-index-ht/src/utils.py
pointer only (licence: NONE) · get_code("ce2e93b940fa25bb")
prepare_teacher_model_hash Ran kvignesh1420/single-index-ht/src/utils.py
pointer only (licence: NONE) · get_code("6fa0051cd08bf372")

Repositories linked to this paper

Some links come from the archived Papers with Code dataset (CC BY-SA 4.0): attribution and licence.

Abstract

Training strategies for modern deep neural networks (NNs) tend to induce a heavy-tailed (HT) empirical spectral density (ESD) in the layer weights. While previous efforts have shown that the HT phenomenon correlates with good generalization in large NNs, a theoretical explanation of its occurrence is still lacking. Especially, understanding the conditions which lead to this phenomenon can shed light on the interplay between generalization and weight spectra. Our work aims to bridge this gap by presenting a simple, rich setting to model the emergence of HT ESD. In particular, we present a theory-informed setup for 'crafting' heavy tails in the ESD of two-layer NNs and present a systematic analysis of the HT ESD emergence without any gradient noise. This is the first work to analyze a noise-free setting, and we also incorporate optimizer (GD/Adam) dependent (large) learning rates into the HT ESD analysis. Our results highlight the role of learning rates on the Bulk+Spike and HT shape of the ESDs in the early phase of training, which can facilitate generalization in the two-layer NN. These observations shed light on the behavior of large-scale NNs, albeit in a much simpler setting.

For agents

The same record, over MCP at https://syntology.ai/mcp:

get_harvested_code_for_paper("2406.04657")
get_code_for_paper("2406.04657")
have("2406.04657")

Connect an agent — have() is free.