Improving the Linearized Laplace Approximation via Quadratic Approximations

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Jiménez, Pedro, Ortega, Luis A., Morales-Álvarez, Pablo, Hernández-Lobato, Daniel
Formato: Preprint
Publicado: 2026
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866910010116669440
author Jiménez, Pedro
Ortega, Luis A.
Morales-Álvarez, Pablo
Hernández-Lobato, Daniel
author_facet Jiménez, Pedro
Ortega, Luis A.
Morales-Álvarez, Pablo
Hernández-Lobato, Daniel
contents Deep neural networks (DNNs) often produce overconfident out-of-distribution predictions, motivating Bayesian uncertainty quantification. The Linearized Laplace Approximation (LLA) achieves this by linearizing the DNN and applying Laplace inference to the resulting model. Importantly, the linear model is also used for prediction. We argue this linearization in the posterior may degrade fidelity to the true Laplace approximation. To alleviate this problem, without increasing significantly the computational cost, we propose the Quadratic Laplace Approximation (QLA). QLA approximates each second order factor in the approximate Laplace log-posterior using a rank-one factor obtained via efficient power iterations. QLA is expected to yield a posterior precision closer to that of the full Laplace without forming the full Hessian, which is typically intractable. For prediction, QLA also uses the linearized model. Empirically, QLA yields modest yet consistent uncertainty estimation improvements over LLA on five regression datasets.
format Preprint
id arxiv_https___arxiv_org_abs_2602_03394
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Improving the Linearized Laplace Approximation via Quadratic Approximations
Jiménez, Pedro
Ortega, Luis A.
Morales-Álvarez, Pablo
Hernández-Lobato, Daniel
Machine Learning
Deep neural networks (DNNs) often produce overconfident out-of-distribution predictions, motivating Bayesian uncertainty quantification. The Linearized Laplace Approximation (LLA) achieves this by linearizing the DNN and applying Laplace inference to the resulting model. Importantly, the linear model is also used for prediction. We argue this linearization in the posterior may degrade fidelity to the true Laplace approximation. To alleviate this problem, without increasing significantly the computational cost, we propose the Quadratic Laplace Approximation (QLA). QLA approximates each second order factor in the approximate Laplace log-posterior using a rank-one factor obtained via efficient power iterations. QLA is expected to yield a posterior precision closer to that of the full Laplace without forming the full Hessian, which is typically intractable. For prediction, QLA also uses the linearized model. Empirically, QLA yields modest yet consistent uncertainty estimation improvements over LLA on five regression datasets.
title Improving the Linearized Laplace Approximation via Quadratic Approximations
topic Machine Learning
url https://arxiv.org/abs/2602.03394