Papers
Topics
Authors
Recent
Search
2000 character limit reached

NTK-ECRN: Eig-Controlled Residual Networks

Updated 10 February 2026
  • NTK-ECRN is a deep residual network that explicitly controls the NTK spectrum using Fourier features, variable layer scaling, and stochastic depth.
  • The architecture stabilizes optimization by bounding eigenvalue growth, ensuring improved generalization and reliable performance across regression and classification tasks.
  • Empirical results show NTK-ECRN outperforms standard models with lower error and higher accuracy, bridging infinite-width theory and practical neural architectures.

The NTK-Eigenvalue-Controlled Residual Network (NTK-ECRN) is a deep residual architecture designed for explicit, layerwise control over the spectral properties of its Neural Tangent Kernel (NTK). By integrating Fourier feature embeddings, residual blocks with variable scaling, and stochastic depth, NTK-ECRN enables analytic and empirical study of NTK dynamics, particularly the evolution and conditioning of its eigenvalues during training. This approach both extends infinite-width neural tangent theory and yields practical, robust architectures for deep learning across regression and classification settings (Mysore et al., 9 Dec 2025).

1. Architecture and Core Design Components

NTK-ECRN is an LL-layer residual network with finite (potentially large) hidden width nn, instantiated as

h(l)=h(l1)+αlσ(Wlh(l1)+bl)h^{(l)} = h^{(l-1)} + \alpha_l \, \sigma\big(W^l h^{(l-1)} + b^l\big)

for l=1,,Ll=1,\ldots,L, where αl>0\alpha_l > 0 introduces explicit per-block scaling. The network features three principal components:

  • Fourier Feature Embeddings: The input xRdx \in \mathbb{R}^d is mapped via fixed or trainable frequencies BRdf×dB \in \mathbb{R}^{d_f \times d} to

ϕ(x)=[sin(2πBx),cos(2πBx)]R2df\phi(x) = [\sin(2\pi Bx), \cos(2\pi Bx)] \in \mathbb{R}^{2d_f}

to amplify high-frequency modes in the input and mitigate the NTK's standard spectral bias.

  • Residual Scaling: Layerwise αl\alpha_l control the magnitude of each block's update, directly modulating the NTK's spectral increments and eigenvalue growth.
  • Stochastic Depth: Each residual block is dropped with probability plp_l, leading to

nn0

with nn1 serving as a regularizer and source of NTK stability.

Parameters are initialized (“standard NTK initialization”) as nn2 and nn3, which enforces kernel convergence in the infinite-width regime. The output is computed by a final linear layer.

2. NTK Definition and Kernel Spectral Evolution

The NTK at training iteration nn4 is

nn5

For a dataset, the Gram matrix nn6 encodes the NTK between all pairs of training points. The growth of the NTK norm and its eigenvalues is constrained by the architecture: nn7 iterated over all blocks and steps,

nn8

Eigenvalues evolve according to: nn9 with h(l)=h(l1)+αlσ(Wlh(l1)+bl)h^{(l)} = h^{(l-1)} + \alpha_l \, \sigma\big(W^l h^{(l-1)} + b^l\big)0 the Jacobian of block h(l)=h(l1)+αlσ(Wlh(l1)+bl)h^{(l)} = h^{(l-1)} + \alpha_l \, \sigma\big(W^l h^{(l-1)} + b^l\big)1. Per Weyl's inequality, for all h(l)=h(l1)+αlσ(Wlh(l1)+bl)h^{(l)} = h^{(l-1)} + \alpha_l \, \sigma\big(W^l h^{(l-1)} + b^l\big)2,

h(l)=h(l1)+αlσ(Wlh(l1)+bl)h^{(l)} = h^{(l-1)} + \alpha_l \, \sigma\big(W^l h^{(l-1)} + b^l\big)3

ensuring that increments in the NTK have bounded impact on all eigenmodes.

3. Spectral Shaping, Generalization, and Optimization Stability

Modulation of the NTK spectrum has several key consequences:

  • Generalization: Decomposing outputs along NTK eigenvectors, convergence under gradient flow is

h(l)=h(l1)+αlσ(Wlh(l1)+bl)h^{(l)} = h^{(l-1)} + \alpha_l \, \sigma\big(W^l h^{(l-1)} + b^l\big)4

with a bound on generalization error: h(l)=h(l1)+αlσ(Wlh(l1)+bl)h^{(l)} = h^{(l-1)} + \alpha_l \, \sigma\big(W^l h^{(l-1)} + b^l\big)5 where h(l)=h(l1)+αlσ(Wlh(l1)+bl)h^{(l)} = h^{(l-1)} + \alpha_l \, \sigma\big(W^l h^{(l-1)} + b^l\big)6 accounts for finite-width effects. Larger eigenvalues along informative directions reduce the penalty and yield better interpolation.

  • Stability/Conditioning: Ensuring moderate condition number h(l)=h(l1)+αlσ(Wlh(l1)+bl)h^{(l)} = h^{(l-1)} + \alpha_l \, \sigma\big(W^l h^{(l-1)} + b^l\big)7 is essential for stable optimization. Control of h(l)=h(l1)+αlσ(Wlh(l1)+bl)h^{(l)} = h^{(l-1)} + \alpha_l \, \sigma\big(W^l h^{(l-1)} + b^l\big)8 and h(l)=h(l1)+αlσ(Wlh(l1)+bl)h^{(l)} = h^{(l-1)} + \alpha_l \, \sigma\big(W^l h^{(l-1)} + b^l\big)9 prevents runaway behavior ("edge-of-stability": rapid l=1,,Ll=1,\ldots,L0 spikes) and thus secures robust gradient descent.
  • Fourier and residual scaling roles: Fourier features flatten the initial eigenvalue decay (enhancing representation of high frequencies), while increasing l=1,,Ll=1,\ldots,L1 selectively boosts high-frequency modes at the cost of possible spectral instability if not carefully capped.

The NTK-ECRN advances over classical ResNets and FC architectures by offering explicit and quantitative eigenvalue control:

  • In overparameterized ResNets, the skip-connection structure was shown to constrain the operator norm of layer propagation, giving width requirements polynomial rather than exponential in depth and maintaining a strictly positive smallest eigenvalue at initialization and during training (Li et al., 2020).
  • Spectral analysis of the residual NTK (ResNTK) in the infinite-width limit demonstrates that the kernel is diagonalized by spherical harmonics, with eigenvalues decaying as l=1,,Ll=1,\ldots,L2 in input dimension l=1,,Ll=1,\ldots,L3. The "spikiness" of the spectrum is controlled by the skip-to-residual weight l=1,,Ll=1,\ldots,L4; constant l=1,,Ll=1,\ldots,L5 induces spike-like sharpening as depth grows, whereas scaling l=1,,Ll=1,\ldots,L6 ensures a depth-invariant, stable spectrum (Belfer et al., 2021).
  • At finite width, fluctuations around the infinite-width kernel (and its spectrum) are l=1,,Ll=1,\ldots,L7, and the condition number remains tightly controlled if the sum of l=1,,Ll=1,\ldots,L8 and depth l=1,,Ll=1,\ldots,L9 are chosen to satisfy αl>0\alpha_l > 00. The standard "FixUp" scaling αl>0\alpha_l > 01 achieves this flat spectrum, while intentionally larger αl>0\alpha_l > 02 can be used to adjust spectral decay or condition number (Littwin et al., 2020).

5. Empirical Performance, Metrics, and Spectrum Evolution

The performance of NTK-ECRN is validated empirically against MLPs, ResNet-18, and infinite-width predictors. Representative results include:

Model MSE (↓) αl>0\alpha_l > 03 (↑) Accuracy (%) (↑) CE Loss (↓)
MLP (512) αl>0\alpha_l > 04 αl>0\alpha_l > 05 αl>0\alpha_l > 06 αl>0\alpha_l > 07
ResNet-18 αl>0\alpha_l > 08 αl>0\alpha_l > 09 xRdx \in \mathbb{R}^d0 xRdx \in \mathbb{R}^d1
Standard NTK xRdx \in \mathbb{R}^d2 xRdx \in \mathbb{R}^d3 xRdx \in \mathbb{R}^d4 xRdx \in \mathbb{R}^d5
NTK-ECRN xRdx \in \mathbb{R}^d6 xRdx \in \mathbb{R}^d7 xRdx \in \mathbb{R}^d8 xRdx \in \mathbb{R}^d9

In CIFAR-10 (5,000 images), NTK-ECRN achieves BRdf×dB \in \mathbb{R}^{d_f \times d}0 accuracy and BRdf×dB \in \mathbb{R}^{d_f \times d}1 cross-entropy loss, outperforming all baselines. Empirical kernel evolution exhibits smooth, predictable growth in the largest eigenvalue and linear BRdf×dB \in \mathbb{R}^{d_f \times d}2 scaling, in contrast to the instability and sharp spectral spikes observed in standard architectures (Mysore et al., 9 Dec 2025).

6. Significance, Extensions, and Limitations

NTK-ECRN establishes a functional bridge between infinite-width kernel theory and practical, scalable architectures. Fourier feature embedding shapes the initial spectrum; residual scaling constrains kernel drift and eigenvalue growth; and stochastic depth both regularizes and renders analytic study tractable. These architectural controls enable:

  • Adaptation of spectral properties during training, promising for high-frequency learning tasks
  • Consistent generalization and stability across depths and widths
  • Potential extensions, including adaptive per-layer scaling based on live NTK estimates or integration with standard normalization techniques

A central limitation remains the non-negligible effect of finite-width-induced fluctuations as network width decreases. Existing theoretical bounds for BRdf×dB \in \mathbb{R}^{d_f \times d}3 become less sharp in narrow settings, motivating further work on non-asymptotic kernel evolution (Mysore et al., 9 Dec 2025, Li et al., 2020, Belfer et al., 2021, Littwin et al., 2020).

Topic to Video (Beta)

No one has generated a video about this topic yet.

Whiteboard

No one has generated a whiteboard explanation for this topic yet.

Follow Topic

Get notified by email when new papers are published related to NTK-Eigenvalue-Controlled Residual Networks.