Understanding NTK Variance in Implicit Neural Representations

📅 2025-12-17
📈 Citations: 0
✨ Influential: 0
📄 PDF
🤖 AI Summary
This work addresses the problem of excessively large neural tangent kernel (NTK) variance in implicit neural representations (INRs), which leads to slow convergence, severe spectral bias, and poor recovery of high-frequency details. We first establish a closed-form decomposition framework for NTK variance, revealing its dependence on pairwise input similarity and scaling terms. Based on this analysis, we provide a unified theoretical explanation for how architectural components—including positional encoding, spherical normalization, and Hadamard modulation—jointly suppress NTK variance and improve condition number. Through rigorous NTK-theoretic analysis, spectral bias diagnostics, and multi-task reconstruction experiments (e.g., image and signed distance function reconstruction), we demonstrate that our variance-regulation mechanism significantly enhances training stability and convergence speed, while achieving superior reconstruction fidelity—particularly in capturing high-frequency structural details.

Technology Category

Machine Learning: Kernel MethodsComputer Vision: Representation Learning for VisionNatural Language Processing: Sentence-level Semantics, Textual Inference, etc.

Application Category

Search and Retrieval-Augmented AI: Web query analysis, representation and understandingGraph Algorithms and Modeling for the Web: Representation, reconstruction, and subgraph or motif discovery in Web-related graphsSemantics and Knowledge: Methods, algorithms and applications for the development of semantic models, knowledge graphs and other forms of structured data models with machine-interpretable semantics
📝 Abstract
Implicit Neural Representations (INRs) often converge slowly and struggle to recover high-frequency details due to spectral bias. While prior work links this behavior to the Neural Tangent Kernel (NTK), how specific architectural choices affect NTK conditioning remains unclear. We show that many INR mechanisms can be understood through their impact on a small set of pairwise similarity factors and scaling terms that jointly determine NTK eigenvalue variance. For standard coordinate MLPs, limited input-feature interactions induce large eigenvalue dispersion and poor conditioning. We derive closed-form variance decompositions for common INR components and show that positional encoding reshapes input similarity, spherical normalization reduces variance via layerwise scaling, and Hadamard modulation introduces additional similarity factors strictly below one, yielding multiplicative variance reduction. This unified view explains how diverse INR architectures mitigate spectral bias by improving NTK conditioning. Experiments across multiple tasks confirm the predicted variance reductions and demonstrate faster, more stable convergence with improved reconstruction quality.
Problem

Research questions and friction points this paper is trying to address.

Analyzes how architectural choices affect NTK conditioning in INRs.
Explains mechanisms reducing NTK eigenvalue variance to mitigate spectral bias.
Demonstrates improved convergence and reconstruction via better NTK conditioning.
Innovation

Methods, ideas, or system contributions that make the work stand out.

Positional encoding reshapes input similarity to improve conditioning.
Spherical normalization reduces variance through layerwise scaling adjustments.
Hadamard modulation introduces similarity factors for multiplicative variance reduction.
C
Chengguang Ou
Fuzhou University
Y
Yixin Zhuang
Fuzhou University