Local properties of neural networks through the lens of layer-wise Hessians

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Bolshim, Maxim, Kugaevskikh, Alexander
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912695202086912
author Bolshim, Maxim
Kugaevskikh, Alexander
author_facet Bolshim, Maxim
Kugaevskikh, Alexander
contents We introduce a methodology for analyzing neural networks through the lens of layer-wise Hessian matrices. The local Hessian of each functional block (layer) is defined as the matrix of second derivatives of a scalar function with respect to the parameters of that layer. This concept provides a formal tool for characterizing the local geometry of the parameter space. We show that the spectral properties of local Hessians, such as the distribution of eigenvalues, reveal quantitative patterns associated with overfitting, underparameterization, and expressivity in neural network architectures. We conduct an extensive empirical study involving 111 experiments across 37 datasets. The results demonstrate consistent structural regularities in the evolution of local Hessians during training and highlight correlations between their spectra and generalization performance. These findings establish a foundation for using local geometric analysis to guide the diagnosis and design of deep neural networks. The proposed framework connects optimization geometry with functional behavior and offers practical insight for improving network architectures and training stability.
format Preprint
id arxiv_https___arxiv_org_abs_2510_17486
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Local properties of neural networks through the lens of layer-wise Hessians
Bolshim, Maxim
Kugaevskikh, Alexander
Machine Learning
68T07, 68T05, 65K10, 90C30
I.2.6; G.1.6; I.5.1
We introduce a methodology for analyzing neural networks through the lens of layer-wise Hessian matrices. The local Hessian of each functional block (layer) is defined as the matrix of second derivatives of a scalar function with respect to the parameters of that layer. This concept provides a formal tool for characterizing the local geometry of the parameter space. We show that the spectral properties of local Hessians, such as the distribution of eigenvalues, reveal quantitative patterns associated with overfitting, underparameterization, and expressivity in neural network architectures. We conduct an extensive empirical study involving 111 experiments across 37 datasets. The results demonstrate consistent structural regularities in the evolution of local Hessians during training and highlight correlations between their spectra and generalization performance. These findings establish a foundation for using local geometric analysis to guide the diagnosis and design of deep neural networks. The proposed framework connects optimization geometry with functional behavior and offers practical insight for improving network architectures and training stability.
title Local properties of neural networks through the lens of layer-wise Hessians
topic Machine Learning
68T07, 68T05, 65K10, 90C30
I.2.6; G.1.6; I.5.1
url https://arxiv.org/abs/2510.17486