Publications

You can also find my articles on my Google Scholar profile.

Machine Learning Conference and Journal Papers


Residual Stream Analysis with Multi-Layer SAEs

Published in ICLR, 2025

This paper introduces a novel approach to analyzing the residual stream in transformer models using multi-layer sparse autoencoders.

Recommended citation: Lawson T, Farnik L, Houghton C, Aitchison L. (2025). "Residual Stream Analysis with Multi-Layer SAEs." ICLR.

Using autodiff to estimate posterior moments, marginals and samples

Published in UAI, 2024

This paper presents methods for using automatic differentiation to estimate posterior moments, marginals and samples.

Recommended citation: Bowyer S, Heap T, Aitchison L. (2024). "Using autodiff to estimate posterior moments, marginals and samples." UAI.

TouchSDF: A DeepSDF Approach for 3D Shape Reconstruction using Vision-Based Tactile Sensing

Published in IEEE Robotics and Automation Letters, 2024

This paper presents a DeepSDF approach for 3D shape reconstruction using vision-based tactile sensing.

Recommended citation: Mauro C, Yijiong L, Church A, Alessio T, Aitchison L, Lepora NF. (2024). "TouchSDF: A DeepSDF Approach for 3D Shape Reconstruction using Vision-Based Tactile Sensing." IEEE Robotics and Automation Letters.

Position paper: Bayesian deep learning in the age of large-scale AI

Published in ICML, 2024

This position paper discusses Bayesian deep learning in the context of large-scale AI.

Recommended citation: Papamarkou T, Skoularidou M, Palla K, Aitchison L, et al. (2024). "Position paper: Bayesian deep learning in the age of large-scale AI." ICML.

Instruction Tuning With Loss Over Instructions

Published in NeurIPS, 2024

This paper presents a new approach to instruction tuning using loss over instructions.

Recommended citation: Shi Z, Yang AX, Wu B, Aitchison L, Yilmaz E, Lipani A. (2024). "Instruction Tuning With Loss Over Instructions." NeurIPS.

Bayesian low-rank adaptation for large language models

Published in ICML, 2024

This paper presents a Bayesian approach to low-rank adaptation for large language models.

Recommended citation: Yang AX, Robeyns M, Wang X, Aitchison L. (2024). "Bayesian low-rank adaptation for large language models." ICML.

Convolutional Deep Kernel Machines

Published in ICML, 2024

This paper extends deep kernel machines to handle convolutional architectures.

Recommended citation: Milsom E, Anson B, Aitchison L. (2024). "Convolutional Deep Kernel Machines." ICML.

Taylor TD-learning

Published in NeurIPS, 2023

This paper introduces Taylor TD-learning, a new approach to temporal difference learning.

Recommended citation: Garibbo M, Robeyns M, Aitchison L. (2023). "Taylor TD-learning." NeurIPS.

Massively parallel reweighted wake-sleep

Published in UAI, 2023

This paper presents a massively parallel implementation of the reweighted wake-sleep algorithm.

Recommended citation: Heap T, Leech G, Aitchison L. (2023). "Massively parallel reweighted wake-sleep." UAI.

A theory of representation learning gives a deep generalisation of kernel methods

Published in ICML, 2023

This paper develops a theory of representation learning that generalizes kernel methods, providing a deep generalization of traditional kernel approaches.

Recommended citation: Yang XY, Robeyns M, Milsom E, Anson B, Schoots N, Aitchison L. (2023). "A theory of representation learning gives a deep generalisation of kernel methods." ICML.

Data augmentation in Bayesian neural networks and the cold posterior effect

Published in UAI, 2022

This paper analyzes the relationship between data augmentation in Bayesian neural networks and the cold posterior effect.

Recommended citation: Nabarro S, Ganev S, Garriga-Alonso A, Fortuin V, van der Wilk M, Aitchison L. (2022). "Data augmentation in Bayesian neural networks and the cold posterior effect." UAI.

Bayesian Neural Network Priors Revisited

Published in ICLR, 2022

This paper reexamines and analyzes priors for Bayesian neural networks.

Recommended citation: Fortuin V, Garriga-Alonso A, Wenzel F, Rätsch G, Turner R, van der Wilk M, Aitchison L. (2022). "Bayesian Neural Network Priors Revisited." ICLR.

Gradient regularisation as approximate variational inference

Published in Entropy, 2021

This paper shows how gradient regularization can be understood as approximate variational inference.

Recommended citation: Unlu A, Aitchison L. (2021). "Gradient regularisation as approximate variational inference." Entropy.

Deep kernel processes

Published in ICML, 2021

This paper introduces deep kernel processes, a new class of probabilistic models.

Recommended citation: Aitchison L, Yang AX, Ober SW. (2021). "Deep kernel processes." ICML.

Tensor Monte Carlo: particle methods for the GPU era

Published in NeurIPS, 2019

This paper introduces Tensor Monte Carlo, adapting particle methods for modern GPU architectures.

Recommended citation: Aitchison L. (2019). "Tensor Monte Carlo: particle methods for the GPU era." NeurIPS.

Deep Convolutional Networks as shallow Gaussian Processes

Published in ICLR, 2019

This paper shows how deep convolutional networks can be understood as Gaussian processes.

Recommended citation: Garriga-Alonso A, Rasmussen CE, Aitchison L. (2019). "Deep Convolutional Networks as shallow Gaussian Processes." ICLR.

Other Publications


Postprocessing East African rainfall forecasts using a generative machine learning model

Published in Journal of Advances in Modeling Earth Systems, 2025

This paper presents a generative machine learning approach for postprocessing East African rainfall forecasts.

Recommended citation: Antonio B, McRae ATT, MacLeod D, Cooper FC, Marsham J, Aitchison L, Palmer TN, Watson PAG. (2025). "Postprocessing East African rainfall forecasts using a generative machine learning model." Journal of Advances in Modeling Earth Systems.

Relating Human Error-Based Learning to Modern Deep RL Algorithms

Published in Neural Computation, 2024

This paper relates human error-based learning to modern deep reinforcement learning algorithms.

Recommended citation: Garibbo M, Ludwig CJH, Lepora NF, Aitchison L. (2024). "Relating Human Error-Based Learning to Modern Deep RL Algorithms." Neural Computation.

Deep Learning for Downscaling Tropical Cyclone Rainfall to Hazard-relevant Spatial Scales

Published in Journal of Geophysical Research: Atmospheres, 2023

This paper presents deep learning methods for downscaling tropical cyclone rainfall predictions.

Recommended citation: Vosper E, Watson P, Harris L, McRae A, Santos-Rodriguez R, Aitchison L, Mitchell D. (2023). "Deep Learning for Downscaling Tropical Cyclone Rainfall to Hazard-relevant Spatial Scales." Journal of Geophysical Research: Atmospheres.

Mass mask-wearing notably reduces COVID-19 transmission

Published in PNAS, 2022

This paper demonstrates the effectiveness of mass mask-wearing in reducing COVID-19 transmission.

Recommended citation: Leech G, Rogers-Smith C, Sandbrink JB, Snodin B, Zinkov R, Rader B, Brownstein JS, Gal Y, Bhatt S, Sharma M, Mindermann S, Brauner JM, Aitchison L. (2022). "Mass mask-wearing notably reduces COVID-19 transmission." PNAS.

Understanding the effectiveness of government interventions against the resurgence of COVID-19 in Europe

Published in Nature Communications, 2021

This paper analyzes the effectiveness of different government interventions in controlling COVID-19 resurgence in Europe.

Recommended citation: Sharma M, Mindermann S, Rogers-Smith C, Leech G, Snodin B, Ahuja J, Sandbrink JB, Monrad JT, Altman G, Dhaliwal G, Finnveden L, Norman AJ, Oehm SB, Sandkühler JF, Aitchison L, Gavenciak T, Mellan T, Kulveit J, Chindelevitch L, Flaxman S, Gal Y, Mishra S, Bhatt S, Brauner JM. (2021). "Understanding the effectiveness of government interventions against the resurgence of COVID-19 in Europe." Nature Communications.
Download Paper

Synaptic plasticity as Bayesian inference

Published in Nature Neuroscience, 2021

This paper shows how synaptic plasticity can be understood as a form of Bayesian inference.

Recommended citation: Aitchison L, Jegminat J, Menendez J, Pfister JP, Pouget A, Latham PE. (2021). "Synaptic plasticity as Bayesian inference." Nature Neuroscience.

Model-based Bayesian inference of neural activity and connectivity from all-optical interrogation of a neural circuit

Published in Advances in Neural Information Processing Systems, 2017

This paper presents methods for inferring neural activity and connectivity from calcium imaging data.

Recommended citation: Aitchison L, Russell L, Packer AM, Yan J, Castonguay P, Häusser M, Turaga SC. (2017). "Model-based Bayesian inference of neural activity and connectivity from all-optical interrogation of a neural circuit." Advances in Neural Information Processing Systems.

With or without you: predictive coding and Bayesian inference in the brain

Published in Current Opinion in Neurobiology, 2017

This paper discusses the relationship between predictive coding and Bayesian inference in neural computation.

Recommended citation: Aitchison L, Lengyel M. (2017). "With or without you: predictive coding and Bayesian inference in the brain." Current Opinion in Neurobiology.

Confidence matching in group decision-making

Published in Nature Human Behaviour, 2017

This paper examines how confidence matching affects group decision-making.

Recommended citation: Bang D, Aitchison L, Moran R, Castanon SH, Rafiee B, Mahmoodi A, Lau JYF, Latham PE, Bahrami B, Summerfield C. (2017). "Confidence matching in group decision-making." Nature Human Behaviour.

Active dendritic integration as a mechanism for robust and precise grid cell firing

Published in Nature Neuroscience, 2017

This paper shows how active dendritic integration contributes to robust and precise grid cell firing.

Recommended citation: Schmidt-Hieber C, Toleikyte G, Aitchison L, Roth A, Clark BA, Branco T, Häusser M. (2017). "Active dendritic integration as a mechanism for robust and precise grid cell firing." Nature Neuroscience.

Zipf’s law arises naturally when there are underlying, unobserved variables

Published in PLoS Computational Biology, 2016

This paper provides a theoretical explanation for the emergence of Zipf’s law in natural phenomena.

Recommended citation: Aitchison L, Corradi N, Latham PE. (2016). "Zipf's law arises naturally when there are underlying, unobserved variables." PLoS Computational Biology.

Doubly Bayesian analysis of confidence in perceptual decision-making

Published in PLoS Computational Biology, 2015

This paper presents a Bayesian analysis of confidence in perceptual decision-making.

Recommended citation: Aitchison L, Bang D, Bahrami B, Latham PE. (2015). "Doubly Bayesian analysis of confidence in perceptual decision-making." PLoS Computational Biology.

Fast sampling-based inference in balanced neuronal networks

Published in NeurIPS, 2014

This paper presents methods for fast sampling-based inference in balanced neural networks.

Recommended citation: Hennequin G, Aitchison L, Lengyel M. (2014). "Fast sampling-based inference in balanced neuronal networks." NeurIPS.