Inter-Layer Hessian Analysis of Neural Networks with DAG Architectures
Abstract
Modern automatic differentiation frameworks (JAX, PyTorch) return the Hessian of the loss function as a monolithic tensor, without exposing the internal structure of inter-layer interactions. This paper presents an analytical formalism that explicitly decomposes the full Hessian into blocks indexed by the DAG of an arbitrary architecture. The canonical decomposition separates the Gauss--Newton component (convex part) from the tensor component (residual curvature responsible for saddle points). For piecewise-linear activations (ReLU), the tensor component of the input Hessian vanishes ( a.e., ); the full parametric Hessian contains residual terms that do not reduce to the GGN. Building on this decomposition, we introduce diagnostic metrics (inter-layer resonance~, geometric coupling~, stable rank~, GN-Gap) that are estimated stochastically in time and reveal structural curvature interactions between layers. The theoretical analysis explains exponential decay of resonance in vanilla networks and its preservation under skip connections; empirical validation spans fully connected MLPs (Exp.\,1--5) and convolutional architectures (ResNet-18, M~parameters, Exp.\,6). When the architecture reduces to a single node, all definitions collapse to the standard Hessian .
Cite
@article{arxiv.2604.11639,
title = {Inter-Layer Hessian Analysis of Neural Networks with DAG Architectures},
author = {Maxim Bolshim and Alexander Kugaevskikh},
journal= {arXiv preprint arXiv:2604.11639},
year = {2026}
}
Comments
45 pages, 9 figures, 17 tables. Submitted to Neural Networks (Elsevier). Code: https://github.com/comiam/dag-hesse