Deterministic Bounds and Random Estimates of Metric Tensors on Neuromanifolds
Ke Sun
摘要
The high-dimensional parameter space of deep neural networks --- the neuromanifold --- is endowed with a unique metric tensor defined by the Fisher information. Reliable and scalable computation of this metric tensor is valuable for theorists and practitioners. Focusing on neural classifiers, we return to a low-dimensional space of probability distributions, which we call the core space, and examine the spectrum and envelopes of its Fisher information matrix. We extend our discoveries there to deterministic bounds for the metric tensor on the neuromanifold. We introduce an unbiased random estimator based on Hutchinson's trace method and derive related bounds. It can be evaluated efficiently with a single backward pass per batch, with a standard deviation bounded by the true value up to scaling.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
它引用的顶会 Paper7
- wav2vec 2.0: A Framework for Self-Supervised Learning of Speech RepresentationsAlexei Baevski, Yuhao Zhou, Abdelrahman Mohamed, Michael AuliNeurIPS 2020 · 被引用 9,451 次
- ADAHESSIAN: An Adaptive Second Order Optimizer for Machine LearningZhewei Yao, Amir Gholami, Sheng Shen, Mustafa Mustafa 等AAAI 2021 · 被引用 358 次
- BackPACK: Packing more into BackpropFelix Dangel, Frederik Kunstner, Philipp HennigICLR 2020 · 被引用 114 次
- Catastrophic Fisher Explosion: Early Phase Fisher Matrix Impacts GeneralizationStanislaw Jastrzebski, Devansh Arpit, Oliver Åstrand, Giancarlo Kerg 等ICML 2021 · 被引用 78 次
- Tractable structured natural-gradient descent using local parameterizationsWu Lin, Frank Nielsen, Mohammad Emtiyaz Khan, Mark SchmidtICML 2021 · 被引用 36 次
相关 Paper
- Trade-Offs of Diagonal Fisher Information Matrix EstimatorsAlexander Soen, Ke SunNeurIPS 2024
- On the Variance of the Fisher Information for Deep LearningAlexander Soen, Ke SunNeurIPS 2021 · 被引用 31 次
- Learning Representation from Neural Fisher Kernel with Low-rank ApproximationRuixiang Zhang, Shuangfei Zhai, Etai Littwin, Joshua M. SusskindICLR 2022 · 被引用 5 次
- Fisher-Rao Sensitivity for Out-of-Distribution Detection in Deep Neural NetworksAnthony Nguyen, Antonin Bertrand, Sylvie Le Hégarat-Mascle, Emanuel Aldea 等ICLR 2026
- Spectral Bias Outside the Training Set for Deep Networks in the Kernel RegimeBenjamin Bowman, Guido F. MontúfarNeurIPS 2022 · 被引用 17 次
