---
title: NTK Eigenfunctions in Deep Learning
url: https://www.emergentmind.com/topics/ntk-eigenfunctions
type: topic
---

# NTK Eigenfunctions in Deep Learning

Neural Tangent Kernel (NTK) eigenfunctions are the spectral modes of the integral operator induced by the NTK, providing a functional basis in which the inductive biases and learning dynamics of infinite-width neural networks can be analyzed. These eigenfunctions underpin the spectral decomposition of training dynamics and generalization, and play a central role in understanding phenomena such as shortcut learning in deep neural networks. Theoretical analysis, especially in the case where input distributions are clustered or imbalanced, reveals how shortcut features emerge as top NTK eigenfunctions, with large eigenvalues and disproportionate influence over function fitting—even after the application of regularization techniques aimed at margin control [2602.03066].

## 1. Formal Definition of NTK Eigenfunctions

Let $\rho$ be a probability measure on the input space $X$. In the infinite-width (lazy training) limit, a neural network’s NTK becomes a fixed, positive semi-definite kernel function $K : X \times X \to \mathbb{R}$. This kernel induces a compact Hilbert-Schmidt integral operator,
$$
T_K[g](x) := \int_X K(x, x')\,g(x')\,d\rho(x').
$$
An NTK eigenfunction (or feature) is defined as any function $\phi_i \in L^2(X, \rho)$ that satisfies
$$
\int_X K(x, x') \,\phi_i(x')\,d\rho(x') = \lambda_i\, \phi_i(x),
$$
where $\lambda_i \geq 0$ is the corresponding NTK eigenvalue and $\|\phi_i\|^2 = \int\phi_i^2\,d\rho = 1$. Under mean squared error (MSE) loss, the solution after infinite time admits a decomposition in the NTK eigenbasis:
$$
f_\infty(x) = \sum_{i} \frac{1}{\lambda_i}\langle\phi_i, y\rangle\,\phi_i(x).
$$
Each NTK feature $\phi_i$ acts as a coordinate axis in function space, weighted by both label signal and the inverse NTK eigenvalue [2602.03066].

## 2. Closed-Form Eigenfunctions for Linear Networks

For a linear network, the NTK simplifies to $K(x, y) = \langle x, y \rangle$. Consider the input distribution $p(x) = \sum_{k=1}^K \pi_k N(\mu_k, \sigma_k^2 I)$, a Gaussian mixture with cluster weights $\pi_k$. Seeking eigenfunctions linear in $x$, i.e., $\phi(x) = x \cdot v$ for unit vector $v$, the action of the NTK operator is:
$$
T_K[\phi](x) = x \cdot \Big[\sum_k \pi_k (\sigma_k^2 v + (\mu_k \mu_k^T) v)\Big] = \lambda (x \cdot v).
$$
By setting $M = \sum_k \pi_k \mu_k \mu_k^T$ and $S = \sum_k \pi_k \sigma_k^2$, one establishes that $v$ must be an eigenvector of $M$ with eigenvalue $a$, and the associated NTK eigenvalue is $\lambda = S + a$.

The normalized eigenfunctions are
$$
\phi_i(x) = \frac{x^T v_i}{\sqrt{\lambda_i}}, \quad \lambda_i = S + a_i, \quad M v_i = a_i v_i.
$$
If one mixture component dominates (large $\pi_k$), the corresponding $v_k$ has the largest $a_k$ and thus $\lambda_k$, marking it as a “shortcut” direction [2602.03066].

## 3. Influence of Cluster Imbalance and Variance

When the cluster weights $\pi_k$ are imbalanced, shortcut eigenfunctions arise. Spectral bias implies that, during gradient flow, modes with large $\lambda_i$ are learned fastest, so shortcut-aligned features associated with large $\pi_k$ emerge early. At convergence, the solution for linear MSE-regression has the form:
$$
f^*(x) = \sum_k w_k (x \cdot v_k),
$$
with
$$
w_k = \frac{\sum_{j \in pos} \pi_j \mu_j \cdot v_k - \sum_{j \not\in pos} \pi_j \mu_j \cdot v_k}{S + v_k^T M v_k}.
$$
If the cluster means $\mu_j$ are orthogonal, this simplifies to
$$
w_k = \pm \frac{\pi_k \|\mu_k\|}{\sum_i \pi_i \sigma_i^2 + \pi_k \|\mu_k\|^2}.
$$
The weight $w_k$ grows monotonically in $\pi_k$. The denominator $S = \sum \pi_i \sigma_i^2$ depends on within-cluster variance: nonzero variances amplify the dominance of shortcut features, whereas if $\sigma_i^2 \to 0$ all weights share a denominator, suppressing shortcut amplification [2602.03066].

## 4. Robustness of NTK Spectral Bias to Margin Control

Previous work suggested that maximal $\ell_2$ margin bias (the result of cross-entropy optimization on separable data) is responsible for shortcut learning. Introducing an SD regularization term, $(\lambda/2)\|f(x)\|^2$, controls the margin, shifting solutions toward ridge regression. However, both cross-entropy with SD and MSE with SD approach the same optimal ratios of feature weights $w_i/w_j$ as $\lambda \to \infty$, matching the MSE solution. Thus, shortcut bias, favored by high $\pi_k$ clusters, persists even under aggressive margin control; it is an intrinsic property of the NTK spectrum, not merely a consequence of margin-seeking dynamics [2602.03066].

## 5. Empirical Observations in Nonlinear Architectures

The dominance of shortcut NTK eigenfunctions extends beyond linear models. Two-layer ReLU networks, trained on synthetic Patched-MNIST and Colored-MNIST datasets, exhibit top empirical NTK eigenfunctions that align with spurious features (patch or color), verified via saliency maps of $f_i(x)$. Lower-order eigenfunctions correspond to semantically meaningful digit shapes.

A metric termed “availability,” $A(g) = g^T K g / (\|g\|^2 \|K\|_F)$, quantifies NTK alignment; shortcut labels typically have much higher availability than core labels throughout training, reflecting their spectral accessibility.

For pretrained ResNet-18 on Waterbirds, CelebA, and Dogs-vs-Cats, empirical NTK analysis confirms that shortcut labels maintain higher NTK availability regardless of the loss function (cross-entropy, MSE, SD). Notably, altering the shortcut-strength (e.g., patch size in Patched-MNIST) scales shortcut availability. Pretraining (e.g., on ImageNet) can sometimes invert these spectral relationships, making core-label features easier to access and revealing the role of initialization [2602.03066].

## 6. Summary and Implications

The spectral decomposition of the NTK, and the resulting dominance of shortcut-aligned eigenfunctions, offers a unifying mechanism for shortcut learning across architectures and datasets. Primary determinants are the presence of high-mixing-weight (large $\pi_k$) clusters and nonzero within-cluster variances. Neural networks preferentially fit the highest-eigenvalue NTK features early and retain them after training, even in the presence of strong regularization or margin controls. These findings, established analytically for linear Gaussian mixtures and empirically validated in complex networks, underscore the robustness of NTK-based spectral bias as an explanatory paradigm for shortcut learning in deep learning [2602.03066].

Source: https://www.emergentmind.com/topics/ntk-eigenfunctions