English

Inter-Layer Hessian Analysis of Neural Networks with DAG Architectures

Machine Learning 2026-04-14 v1

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 H=HGN+HTH = H^{GN} + H^T 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 (Hv,wT ⁣ ⁣0H^{T}_{v,w}\!\equiv\!0 a.e., Hv,wf ⁣= ⁣Hv,wGN ⁣ ⁣0H^f_{v,w}\!=\!H^{GN}_{v,w}\!\succeq\!0); 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~R\mathcal{R}, geometric coupling~C\mathcal{C}, stable rank~D\mathcal{D}, GN-Gap) that are estimated stochastically in O(P)O(P) 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, 11{\sim}11M~parameters, Exp.\,6). When the architecture reduces to a single node, all definitions collapse to the standard Hessian θ2L(θ)Rp×p\nabla^2_\theta\mathcal{L}(\theta)\in\mathbb{R}^{p\times p}.

Keywords

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