← Search

Javier Antoran

10 accepted papers

2024

A Generative Model of Symmetry Transformations

NeurIPS 2024poster

Correctly capturing the symmetry transformations of data can lead to efficient models with strong generalization capabilities, though methods incorporating symmetries often require prior knowledge. While recent advancements have been made in learning those symmetries directly from the dataset, most…

2024

Improving Linear System Solvers for Hyperparameter Optimisation in Iterative Gaussian Processes

NeurIPS 2024poster

Scaling hyperparameter optimisation to very large datasets remains an open problem in the Gaussian process community. This paper focuses on iterative methods, which use linear system solvers, like conjugate gradients, alternating projections or stochastic gradient descent, to construct an estimate o…

2024

Stochastic Gradient Descent for Gaussian Processes Done Right

ICLR 2024poster

As is well known, both sampling from the posterior and computing the mean of the posterior in Gaussian process regression reduces to solving a large linear system of equations. We study the use of stochastic gradient descent for solving this linear system, and show that when done right---by which we…

2023

SE(3) Equivariant Augmented Coupling Flows

NeurIPS 2023spotlight

Coupling normalizing flows allow for fast sampling and density evaluation, making them the tool of choice for probabilistic modeling of physical systems. However, the standard coupling architecture precludes endowing flows that operate on the Cartesian coordinates of atoms with the SE(3) and permut…

2023

Sampling from Gaussian Process Posteriors using Stochastic Gradient Descent

NeurIPS 2023oral

Gaussian processes are a powerful framework for quantifying uncertainty and for sequential decision-making but are limited by the requirement of solving linear systems. In general, this has a cubic cost in dataset size and is sensitive to conditioning. We explore stochastic gradient algorithms as a…

2023

Sampling-based inference for large linear models, with application to linearised Laplace

ICLR 2023poster

Large-scale linear models are ubiquitous throughout machine learning, with contemporary application as surrogate models for neural network uncertainty quantification; that is, the linearised Laplace method. Alas, the computational cost associated with Bayesian linear models constrains this method's…

2022

Adapting the Linearised Laplace Model Evidence for Modern Deep Learning

ICML 2022spotlight

The linearised Laplace method for estimating model uncertainty has received renewed attention in the Bayesian deep learning community. The method provides reliable error bars and admits a closed-form expression for the model evidence, allowing for scalable selection of model hyperparameters. In this…

Cited by 39SourcePDFScholar
2021

Bayesian Deep Learning via Subnetwork Inference

ICML 2021spotlight

The Bayesian paradigm has the potential to solve core issues of deep neural networks such as poor calibration and data inefficiency. Alas, scaling Bayesian inference to large weight spaces often requires restrictive approximations. In this work, we show that it suffices to perform inference over a s…

2021

Getting a CLUE: A Method for Explaining Uncertainty Estimates

ICLR 2021oral

Both uncertainty estimation and interpretability are important factors for trustworthy machine learning systems. However, there is little work at the intersection of these two areas. We address this gap by proposing a novel method for interpreting uncertainty estimates from differentiable probabilis…

Cited by 151SourcePDFScholar
2020

Depth Uncertainty in Neural Networks

NeurIPS 2020poster

Existing methods for estimating uncertainty in deep learning tend to require multiple forward passes, making them unsuitable for applications where computational resources are limited. To solve this, we perform probabilistic reasoning over the depth of neural networks. Different depths correspond to…