Optimizing Neural Networks with Kronecker-factored Approximate Curvature
Not on the map. Why?
Abstract (arXiv)
We propose an efficient method for approximating natural gradient descent in neural networks which we call Kronecker-Factored Approximate Curvature (K-FAC). K-FAC is based on an efficiently invertible approximation of a neural network's Fisher information matrix which is neither diagonal nor low-rank, and in some cases is completely non-sparse. It is derived by approximating various large blocks of the Fisher (corresponding to entire layers) as being the Kronecker product of two much smaller matrices. While only several times more expensive to compute than the plain stochastic gradient, the updates produced by K-FAC make much more progress optimizing the objective, which results in an algorithm that can be much faster than stochastic gradient descent with momentum in practice. And unlike some previously proposed approximate natural-gradient/Newton methods which use high-quality non-diagonal curvature matrices (such as Hessian-free optimization), K-FAC works very well in highly stochastic optimization regimes. This is because the cost of storing and inverting K-FAC's approximation to the curvature matrix does not depend on the amount of data used to estimate it, which is a feature typically associated only with diagonal or low-rank approximations to the curvature matrix.
Key facts. Cited by 776 papers in the map (data through 7 Oct 2026). First posted on arXiv 19 Mar 2015.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2018: 1; 2019: 0; 2020: 0; 2021: 82; 2022: 80; 2023: 95; 2024: 119; 2025: 156; 2026: 243.
776 papers.
Sort:
By area:
- 2023A Comprehensive Survey of Continual Learning: Theory, Method and Applicationcited by 860
- 2021A Survey of Uncertainty in Deep Neural Networkscited by 675
- 2021Merging Models with Fisher-Weighted Averagingcited by 613
- 2021BRECQ: Pushing the Limit of Post-Training Quantization by Block Reconstructioncited by 464
- 2023Symbolic Discovery of Optimization Algorithmscited by 423
- 2021Laplace Redux -- Effortless Bayesian Deep Learningcited by 359
- 2022Optimal Brain Compression: A Framework for Accurate Post-Training Quantization and Pruningcited by 317
- 2023Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-trainingcited by 260
- 2021Fishr: Invariant Gradient Variances for Out-of-Distribution Generalizationcited by 186
- 2024SOAP: Improving and Stabilizing Shampoo using Adamcited by 179
- 2022If Influence Functions are the Answer, Then What is the Question?cited by 142
- 2024Adam-mini: Use Fewer Learning Rates To Gain Morecited by 134
38 more
- 2022Towards Efficient and Scalable Sharpness-Aware Minimizationcited by 129
- 2024Why Transformers Need Adam: A Hessian Perspectivecited by 123
- 2023Structured Pruning for Deep Convolutional Neural Networks: A surveycited by 122
- 2023Noise Is Not the Main Factor Behind the Gap Between SGD and Adam on Transformers, but Sign Descent Might Becited by 114
- 2021Sharpness-Aware Minimization Improves Language Model Generalizationcited by 108
- 2021Scalable Marginal Likelihood Estimation for Model Selection in Deep Learningcited by 105
- 2018Multi-Agent Generative Adversarial Imitation Learningcited by 99
- 2024What is Your Data Worth to GPT? LLM-Scale Data Valuation with Influence Functionscited by 88
- 2022Deep Unlearning via Randomized Conditionally Independent Hessianscited by 83
- 2025NorMuon: Making Muon more efficient and scalablecited by 74
- 2024Continual Learning and Catastrophic Forgettingcited by 73
- 2022Make Sharpness-Aware Minimization Stronger: A Sparsified Perturbation Approachcited by 70
- 2023A Distributed Data-Parallel PyTorch Implementation of the Distributed Shampoo Optimizer for Training Neural Networks At-Scalecited by 69
- 2024Fisher Flow Matching for Generative Modeling over Discrete Datacited by 68
- 2021Spectral Normalisation for Deep Reinforcement Learning: an Optimisation Perspectivecited by 64
- 2021Stochastic Training is Not Necessary for Generalizationcited by 63
- 2021Estimating High Order Gradients of the Data Distribution by Denoisingcited by 63
- 2021Training Multi-Layer Over-Parametrized Neural Network in Subquadratic Timecited by 62
- 2022Fast Finite Width Neural Tangent Kernelcited by 62
- 2022Adaptive Second Order Coresets for Data-efficient Machine Learningcited by 60
- 2024MARS: Unleashing the Power of Variance Reduction for Training Large Modelscited by 59
- 2022VeLO: Training Versatile Learned Optimizers by Scaling Upcited by 57
- 2022Bayesian Model Selection, the Marginal Likelihood, and Generalizationcited by 56
- 2021Recent Advances of Continual Learning in Computer Vision: An Overviewcited by 55
- 2025Gradient Alignment in Physics-informed Neural Networks: A Second-Order Optimization Perspectivecited by 55
- 2025ASGO: Adaptive Structured Gradient Optimizationcited by 55
- 2021M-FAC: Efficient Matrix-Free Approximations of Second-Order Informationcited by 54
- 2024A New Perspective on Shampoo's Preconditionercited by 54
- 2021Analytic Insights into Structure and Rank of Neural Network Hessian Mapscited by 52
- 2021Rapid training of deep neural networks without skip connections or normalization layers using Deep Kernel Shapingcited by 51
- 2023Kronecker-Factored Approximate Curvature for Modern Neural Network Architecturescited by 49
- 2021Interactive Label Cleaning with Example-based Explanationscited by 48
- 2021Noether's Learning Dynamics: Role of Symmetry Breaking in Neural Networkscited by 47
- 2021Natural continual learning: success is a journey, not (just) a destinationcited by 46
- 2022A Self-Attention Ansatz for Ab-initio Quantum Chemistrycited by 45
- 2023Deep Transformers without Shortcuts: Modifying Self-attention for Faithful Signal Propagationcited by 44
- 2023Towards Sustainable Learning: Coresets for Data-efficient Deep Learningcited by 44
- 2023Incorporating Neuro-Inspired Adaptability for Continual Learning in Artificial Intelligencecited by 43
50 most cited of 776 All 776 (JSON)
- 2026SoftServe: A Scalable Quasi-Newton Method for Deep Learning1 Oct 2026
- 2026ITC-MoE: Importance-guided Token-aware Compression for MoE Diffusion Language Models1 Oct 2026
- 2026The Row Normalization Puzzle in Muon30 Sep 2026
- 2026Distilling Diffusion Score Discrepancy for Efficient Training Data Attribution30 Sep 2026
- 2026dattri-LLM: A Unified and Efficient Library for Training Data Attribution at LLM Scale30 Sep 2026
- 2026ShamAN-Q: Shampoo Augmented NanoQuant for Sub-1-bit LLM Weights29 Sep 2026
- 2026Security-Enhanced Seed-Based Weight Quantization for Large Language Models29 Sep 2026
- 2026Scaling Influence Functions in LLMs through Eigenbasis-Corrected One-Bit Gradient Projection29 Sep 2026
- 2026Normalize-Then-Precondition: A Hierarchical Approach to Marginal Scale and Interaction Geometry for LLM Training29 Sep 2026
- 2026Normative Loss Landscape Navigation: A Trajectory-Based Approach to Mitigating Forgetting in Incremental Learning28 Sep 2026
- 2026Commutator Memory: Sparse, Path-Local Reading and Steering in Language Models28 Sep 2026
- 2026Beyond One Epoch: Uncertainty-Weighted Sensitivity Regularization for Recommendation Models28 Sep 2026
38 more
- 2026The cost of useful natural gradient updates27 Sep 2026
- 2026A Mesoscopic View of Transformer Weights Through Row and Column Scale Fields25 Sep 2026
- 2026Convergence guarantees for Muon: New parameter regimes and generalizations24 Sep 2026
- 2026On the second-order optimization for spiking neural networks24 Sep 2026
- 2026RGSQ: Riemannian Geometry-Sensitive Quantization for Large Vision-Language Models21 Sep 2026
- 2026FlashBoB: I/O-Efficient Exact Backward-over-Backward for Softmax Attention21 Sep 2026
- 2026Stochastic Reconfiguration as Statistical Filtering for Overparameterized Neural Quantum States20 Sep 2026
- 2026Optimizers for Diffusion Models: A Controlled Benchmark19 Sep 2026
- 2026COREM: Cosine-Relation Momentum Reshaping with Stateful Writeback18 Sep 2026
- 2026Not All Layers Need Tuning: Diagnosing and Directing Adaptation in Vision-Language-Action Models16 Sep 2026
- 2026Geometry of learning dynamics: Gradient descent versus natural gradient on the ridge of optimization15 Sep 2026
- 2026Optimal Pruning for Neural Architectures using Fisher Information Distances14 Sep 2026
- 2026Benchmarking Optimizers to Solve Inverse Problems with Differentiable Physics Simulators12 Sep 2026
- 2026Muon-C: Operator-Aligned Muon for Convolutional Kernels9 Sep 2026
- 2026Beyond the Matrix Sign: Quadratic Spectral Descent7 Sep 2026
- 2026Mind the Approximation: Fisher-Weighted SVD Compression for ViTs7 Sep 2026
- 2026All for 1-Bit: Towards Genuine 1-Bit Post-Training Quantization for LLMs5 Sep 2026
- 2026TaRA: Training-Aware Low-Rank Adaptation Initialization2 Sep 2026
- 2026Scalable Kronecker-Fisher Approximation: Efficient Hessian Analysis for Billion-Parameter Language Models Compression2 Sep 2026
- 2026Frozen Cores Need Task Signal: Fisher-Whitened Cross-Covariance for Low-Resource LLM Adaptation1 Sep 2026
- 2026Normalized Low-Rank Adaptation31 Aug 2026
- 2026REAL-Q: E2E LLM Quantization via Dynamic Gradient Descent30 Aug 2026
- 2026Blog: Survey of Optimizers28 Aug 2026
- 2026AIM: Anchor Identity Features, Then Match for Multimodal Large Language Model Unlearning28 Aug 2026
- 2026Unifying Detection and Adaptation in Task-Free Continual Learning27 Aug 2026
- 2026Spectral Allocation: Why Muon Outperforms Adam, and How to Improve Muon26 Aug 2026
- 2026Guided Riemannian Optimization (GuRO): Bridging Model Predictive Control and Decision Transformers24 Aug 2026
- 2026Toward a First-Principles Update Geometry for the Language-Model Head23 Aug 2026
- 2026Improved denoising diffusion probabilistic models with efficient non-diagonal covariance modeling22 Aug 2026
- 2026Loss-Parameterized Fisher Width Along Learning Trajectories21 Aug 2026
- 2026Activation-Keyed Momentum: An Anisotropic Momentum Update via the Delta Rule19 Aug 2026
- 2026Detecting Contaminated Code-Generation Prompt Batches via Influence Functions14 Aug 2026
- 2026SoftWater: Class-Aware Rate Allocation for Softmax Quantization12 Aug 2026
- 2026MOON: Multi-Objective OrthoNormalized Updates for Multitask Learning12 Aug 2026
- 2026Fisher8: Stabilizing Neural Heteroscedastic Regression via Output-Layer Fisher Geometry11 Aug 2026
- 2026Second-Order Muon Done Right: A Principled Marriage of Spectral Geometry and Curvature10 Aug 2026
- 2026Tracing sources of epistemic uncertainty in deep learning predictions: homo- and hetero-scedastic linearized estimators7 Aug 2026
- 2026MALT: Lightweight Curvature-Aware Muon via Diagonal Preconditioning5 Aug 2026
50 newest of 776 All 776 (JSON)
- 2022Deep Unlearning via Randomized Conditionally Independent Hessianscited by 83
- 2024Unified Gradient-Based Machine Unlearning with Remain Geometry Enhancementcited by 43
- 2024Data Attribution for Text-to-Image Models by Unlearning Synthesized Imagescited by 22
- 2025Geometric-disentangelment Unlearningcited by 4
- 2026Gauss-Newton Unlearning for the LLM Eracited by 4
- 2025How does Bayesian Sampling help Membership Inference Attacks?cited by 2
- 2025Text-Guided Image Invariant Feature Learning for Robust Image Watermarkingcited by 1
- 2025IMU: Influence-guided Machine Unlearningcited by 1
- 2026Robust LLM Unlearning Against Relearning Attacks: The Minor Components in Representations Mattercited by 1
- 2026Distilling Linearized Behavior into Non-Linear Fine-Tuning for Effective Task Arithmeticcited by 1
- 2025How to Protect Models against Adversarial Unlearning?cited by 0
- 2026Why Some Models Resist Unlearning: A Linear Stability Perspectivecited by 0
12 most cited of 14 All 14 (JSON)
- 2026The Forgetting-Retention Dilemma: Certified Unlearning Theory in Continual Learning29 Jun 2026
- 2026Detectability in Diversity: Improved Canary Crafting for Privacy Auditing in One Run26 May 2026
- 2026Distilling Linearized Behavior into Non-Linear Fine-Tuning for Effective Task Arithmetic18 May 2026
- 2026Robust LLM Unlearning Against Relearning Attacks: The Minor Components in Representations Matter12 May 2026
- 2026Gauss-Newton Unlearning for the LLM Era11 Feb 2026
- 2026Why Some Models Resist Unlearning: A Linear Stability Perspective3 Feb 2026
- 2025Geometric-disentangelment Unlearning21 Nov 2025
- 2025IMU: Influence-guided Machine Unlearning3 Aug 2025
- 2025How to Protect Models against Adversarial Unlearning?15 Jul 2025
- 2025Text-Guided Image Invariant Feature Learning for Robust Image Watermarking18 Mar 2025
- 2025How does Bayesian Sampling help Membership Inference Attacks?10 Mar 2025
- 2024Unified Gradient-Based Machine Unlearning with Remain Geometry Enhancement29 Sep 2024
12 newest of 14 All 14 (JSON)
- 2024Fisher Flow Matching for Generative Modeling over Discrete Datacited by 68
- 2024Influence Functions for Scalable Data Attribution in Diffusion Modelscited by 31
- 2024Towards Faster Training of Diffusion Models: An Inspiration of A Consistency Phenomenoncited by 14
- 2026SPARE: Self-distillation for PARameter-Efficient Removalcited by 1
- 2026Source-Prior-Driven Selective Adaptation for Efficient Diffusion Model Finetuningcited by 0
- 2026Improved denoising diffusion probabilistic models with efficient non-diagonal covariance modelingcited by 0
- 2026Optimizers for Diffusion Models: A Controlled Benchmarkcited by 0
- 2026Distilling Diffusion Score Discrepancy for Efficient Training Data Attributioncited by 0
- 2026ITC-MoE: Importance-guided Token-aware Compression for MoE Diffusion Language Modelscited by 0
- 2026ITC-MoE: Importance-guided Token-aware Compression for MoE Diffusion Language Models1 Oct 2026
- 2026Distilling Diffusion Score Discrepancy for Efficient Training Data Attribution30 Sep 2026
- 2026Optimizers for Diffusion Models: A Controlled Benchmark19 Sep 2026
- 2026Improved denoising diffusion probabilistic models with efficient non-diagonal covariance modeling22 Aug 2026
- 2026Source-Prior-Driven Selective Adaptation for Efficient Diffusion Model Finetuning23 Jul 2026
- 2026SPARE: Self-distillation for PARameter-Efficient Removal4 Feb 2026
- 2024Influence Functions for Scalable Data Attribution in Diffusion Models17 Oct 2024
- 2024Fisher Flow Matching for Generative Modeling over Discrete Data23 May 2024
- 2024Towards Faster Training of Diffusion Models: An Inspiration of A Consistency Phenomenon14 Mar 2024
- 2025Rethinking LLM Training through Information Geometry and Quantum Metricscited by 3
- 2026Geometric Self-Distillation for Reasoning Generalizationcited by 2
- 2025From Memorization to Reasoning in the Spectrum of Loss Curvaturecited by 1
- 2025Optimal Sparsity of Mixture-of-Experts Language Models for Reasoning Taskscited by 0
- 2026LASER: Loss-Aware Singular-value Decomposition and Rank Allocation for Efficient Low-Precision Vision-Language Modelscited by 0
- 2026Geometric Self-Distillation for Reasoning Generalization7 Jul 2026
- 2026LASER: Loss-Aware Singular-value Decomposition and Rank Allocation for Efficient Low-Precision Vision-Language Models30 May 2026
- 2025From Memorization to Reasoning in the Spectrum of Loss Curvature28 Oct 2025
- 2025Optimal Sparsity of Mixture-of-Experts Language Models for Reasoning Tasks26 Aug 2025
- 2025Rethinking LLM Training through Information Geometry and Quantum Metrics18 Jun 2025
- 2024Post-hoc Probabilistic Vision-Language Modelscited by 13
- 2026AIM: Anchor Identity Features, Then Match for Multimodal Large Language Model Unlearningcited by 1
- 2026Curvature-Guided Mixing for MLLM Adaptationcited by 0
- 2026RGSQ: Riemannian Geometry-Sensitive Quantization for Large Vision-Language Modelscited by 0
- 2026RGSQ: Riemannian Geometry-Sensitive Quantization for Large Vision-Language Models21 Sep 2026
- 2026AIM: Anchor Identity Features, Then Match for Multimodal Large Language Model Unlearning28 Aug 2026
- 2026Curvature-Guided Mixing for MLLM Adaptation23 Jun 2026
- 2024Post-hoc Probabilistic Vision-Language Models8 Dec 2024
- 2025HEAPr: Hessian-based Efficient Atomic Expert Pruning in Output Spacecited by 3
- 2026SPHERE: Mitigating the Loss of Spectral Plasticity in Mixture-of-Experts for Deep Reinforcement Learningcited by 2
- 2026Geometric Metrics for MoE Specialization: From Fisher Information to Early Failure Detectioncited by 0
- 2026MONA: Muon Optimizer with Nesterov Acceleration for Scalable Language Model Trainingcited by 0
- 2026MONA: Muon Optimizer with Nesterov Acceleration for Scalable Language Model Training26 May 2026
- 2026SPHERE: Mitigating the Loss of Spectral Plasticity in Mixture-of-Experts for Deep Reinforcement Learning6 May 2026
- 2026Geometric Metrics for MoE Specialization: From Fisher Information to Early Failure Detection16 Apr 2026
- 2025HEAPr: Hessian-based Efficient Atomic Expert Pruning in Output Space26 Sep 2025
- 2023A Comprehensive Survey of Continual Learning: Theory, Method and Applicationcited by 860
- 2021A Survey of Uncertainty in Deep Neural Networkscited by 675
- 2021Merging Models with Fisher-Weighted Averagingcited by 613
- 2021BRECQ: Pushing the Limit of Post-Training Quantization by Block Reconstructioncited by 464
- 2023Symbolic Discovery of Optimization Algorithmscited by 423
- 2021Laplace Redux -- Effortless Bayesian Deep Learningcited by 359
- 2022Optimal Brain Compression: A Framework for Accurate Post-Training Quantization and Pruningcited by 317
- 2023Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-trainingcited by 260
- 2021Fishr: Invariant Gradient Variances for Out-of-Distribution Generalizationcited by 186
- 2024SOAP: Improving and Stabilizing Shampoo using Adamcited by 179
- 2022If Influence Functions are the Answer, Then What is the Question?cited by 142
- 2024Adam-mini: Use Fewer Learning Rates To Gain Morecited by 134
12 most cited of 731 All 731 (JSON)
- 2026SoftServe: A Scalable Quasi-Newton Method for Deep Learning1 Oct 2026
- 2026The Row Normalization Puzzle in Muon30 Sep 2026
- 2026dattri-LLM: A Unified and Efficient Library for Training Data Attribution at LLM Scale30 Sep 2026
- 2026ShamAN-Q: Shampoo Augmented NanoQuant for Sub-1-bit LLM Weights29 Sep 2026
- 2026Security-Enhanced Seed-Based Weight Quantization for Large Language Models29 Sep 2026
- 2026Scaling Influence Functions in LLMs through Eigenbasis-Corrected One-Bit Gradient Projection29 Sep 2026
- 2026Normalize-Then-Precondition: A Hierarchical Approach to Marginal Scale and Interaction Geometry for LLM Training29 Sep 2026
- 2026Normative Loss Landscape Navigation: A Trajectory-Based Approach to Mitigating Forgetting in Incremental Learning28 Sep 2026
- 2026Commutator Memory: Sparse, Path-Local Reading and Steering in Language Models28 Sep 2026
- 2026Beyond One Epoch: Uncertainty-Weighted Sensitivity Regularization for Recommendation Models28 Sep 2026
- 2026The cost of useful natural gradient updates27 Sep 2026
- 2026A Mesoscopic View of Transformer Weights Through Row and Column Scale Fields25 Sep 2026
12 newest of 731 All 731 (JSON)