Latent Bottlenecked Attentive Neural Processes
Not on the map. Why?
Abstract (arXiv)
Neural Processes (NPs) are popular methods in meta-learning that can estimate predictive uncertainty on target datapoints by conditioning on a context dataset. Previous state-of-the-art method Transformer Neural Processes (TNPs) achieve strong performance but require quadratic computation with respect to the number of context datapoints, significantly limiting its scalability. Conversely, existing sub-quadratic NP variants perform significantly worse than that of TNPs. Tackling this issue, we propose Latent Bottlenecked Attentive Neural Processes (LBANPs), a new computationally efficient sub-quadratic NP variant, that has a querying computational complexity independent of the number of context datapoints. The model encodes the context dataset into a constant number of latent vectors on which self-attention is performed. When making predictions, the model retrieves higher-order information from the context dataset via multiple cross-attention mechanisms on the latent vectors. We empirically show that LBANPs achieve results competitive with the state-of-the-art on meta-regression, image completion, and contextual multi-armed bandits. We demonstrate that LBANPs can trade-off the computational cost and performance according to the number of latent vectors. Finally, we show LBANPs can scale beyond existing attention-based NP variants to larger dataset settings.
Key facts. Cited by 26 papers in the map (data through 14 Oct 2026). Builds on 20 of its 24 references that are in the map. First posted on arXiv 15 Nov 2022.
Builds on
20 of 24 references.
References · 20
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2014Deep Learning Face Attributes in the Wildcited by 3,693
- 2017Deep Setscited by 1,421
- 2021Perceiver: General Perception with Iterative Attentioncited by 1,104
- 2021Perceiver IO: A General Architecture for Structured Inputs & Outputscited by 629
- 2018Conditional Neural Processescited by 426
- 2020Visual Transformers: Token-based Image Representation and Processing for Computer Visioncited by 334
- 2018Neural Processescited by 311
- 2018Deep Bayesian Bandits Showdown: An Empirical Comparison of Bayesian Deep Networks for Thompson Samplingcited by 169
- 2019Convolutional Conditional Neural Processescited by 142
- 2022Transformer Neural Processes: Uncertainty-Aware Meta Learning Via Sequence Modelingcited by 127
- 2021Coordination Among Neural Modules Through a Shared Global Workspacecited by 85
8 more
- 2019Sequential Neural Processescited by 55
- 2020Bootstrapping Neural Processescited by 40
- 2021The Gaussian Neural Processcited by 34
- 2020Equivariant Learning of Stochastic Fields: Gaussian Processes and Steerable Conditional Neural Processescited by 29
- 2022The Neural Process Family: Survey, Applications and Perspectivescited by 23
- 2019Meta-Learning surrogate models for sequential decision makingcited by 14
- 2021On Contrastive Representations of Stochastic Processescited by 12
- 2019Recurrent Neural Processescited by 10
Built on by
Citing papers per year: 2023: 5; 2024: 10; 2025: 7; 2026: 4.
26 papers.
Sort:
- 2025AB-UPT: Scaling Neural CFD Surrogates for High-Fidelity Automotive Aerodynamics Simulations via Anchored-Branched Universal Physics Transformerscited by 53
- 2024TuneTables: Context Optimization for Scalable Prior-Data Fitted Networkscited by 47
- 2024Amortized Probabilistic Conditioning for Optimization, Simulation and Inferencecited by 23
- 2024Translation Equivariant Transformer Neural Processescited by 13
- 2025Estimating Interventional Distributions with Uncertain Causal Graphs through Meta-Learningcited by 13
- 2023Beyond Unimodal: Generalising Neural Processes for Multimodal Uncertainty Estimationcited by 11
- 2023Memory Efficient Neural Processes via Constant Memory Attention Blockcited by 9
- 2024Approximately Equivariant Neural Processescited by 8
- 2025ALINE: Joint Amortization for Bayesian Inference and Active Data Acquisitioncited by 8
- 2025Dimension Agnostic Neural Processescited by 5
- 2024Spectral Convolutional Conditional Neural Processescited by 4
- 2024In-Context In-Context Learning with Transformer Neural Processescited by 4
14 more
- 2024Gridded Transformer Neural Processes for Large Unstructured Spatio-Temporal Datacited by 4
- 2025Efficient Autoregressive Inference for Transformer Probabilistic Modelscited by 4
- 2023Adaptive Conditional Quantile Neural Processescited by 3
- 2024System-Aware Neural ODE Processes for Few-Shot Bayesian Optimizationcited by 3
- 2023Tree Cross Attentioncited by 2
- 2024Parallelized Spatiotemporal Bindingcited by 2
- 2025Exploring Pseudo-Token Approaches in Transformer Neural Processescited by 2
- 2026Incremental Transformer Neural Processescited by 2
- 2023Constant Memory Attention Blockcited by 1
- 2025Scalable Spatiotemporal Inference with Biased Scan Attention Transformer Neural Processescited by 1
- 2024Exploring Probabilistic Models for Semi-supervised Learningcited by 0
- 2026Amortising Inference and Meta-Learning Priors in Neural Networkscited by 0
- 2026PRIM: Meta-Learned Bayesian Root Cause Analysiscited by 0
- 2026Revisiting Neural Processes via Fourier Transform and Volterra Seriescited by 0
- 2026Revisiting Neural Processes via Fourier Transform and Volterra Series31 May 2026
- 2026PRIM: Meta-Learned Bayesian Root Cause Analysis9 May 2026
- 2026Incremental Transformer Neural Processes21 Feb 2026
- 2026Amortising Inference and Meta-Learning Priors in Neural Networks9 Feb 2026
- 2025Efficient Autoregressive Inference for Transformer Probabilistic Models10 Oct 2025
- 2025Estimating Interventional Distributions with Uncertain Causal Graphs through Meta-Learning7 Jul 2025
- 2025Scalable Spatiotemporal Inference with Biased Scan Attention Transformer Neural Processes10 Jun 2025
- 2025ALINE: Joint Amortization for Bayesian Inference and Active Data Acquisition8 Jun 2025
- 2025Exploring Pseudo-Token Approaches in Transformer Neural Processes19 Apr 2025
- 2025Dimension Agnostic Neural Processes28 Feb 2025
- 2025AB-UPT: Scaling Neural CFD Surrogates for High-Fidelity Automotive Aerodynamics Simulations via Anchored-Branched Universal Physics Transformers13 Feb 2025
- 2024Amortized Probabilistic Conditioning for Optimization, Simulation and Inference20 Oct 2024
14 more
- 2024Gridded Transformer Neural Processes for Large Unstructured Spatio-Temporal Data9 Oct 2024
- 2024In-Context In-Context Learning with Transformer Neural Processes19 Jun 2024
- 2024Approximately Equivariant Neural Processes19 Jun 2024
- 2024Translation Equivariant Transformer Neural Processes18 Jun 2024
- 2024System-Aware Neural ODE Processes for Few-Shot Bayesian Optimization4 Jun 2024
- 2024Spectral Convolutional Conditional Neural Processes19 Apr 2024
- 2024Exploring Probabilistic Models for Semi-supervised Learning5 Apr 2024
- 2024Parallelized Spatiotemporal Binding26 Feb 2024
- 2024TuneTables: Context Optimization for Scalable Prior-Data Fitted Networks17 Feb 2024
- 2023Tree Cross Attention29 Sep 2023
- 2023Constant Memory Attention Block21 Jun 2023
- 2023Adaptive Conditional Quantile Neural Processes30 May 2023
- 2023Memory Efficient Neural Processes via Constant Memory Attention Block23 May 2023
- 2023Beyond Unimodal: Generalising Neural Processes for Multimodal Uncertainty Estimation4 Apr 2023
- 2025AB-UPT: Scaling Neural CFD Surrogates for High-Fidelity Automotive Aerodynamics Simulations via Anchored-Branched Universal Physics Transformerscited by 53
- 2024TuneTables: Context Optimization for Scalable Prior-Data Fitted Networkscited by 47
- 2024Amortized Probabilistic Conditioning for Optimization, Simulation and Inferencecited by 23
- 2024Translation Equivariant Transformer Neural Processescited by 13
- 2025Estimating Interventional Distributions with Uncertain Causal Graphs through Meta-Learningcited by 13
- 2023Beyond Unimodal: Generalising Neural Processes for Multimodal Uncertainty Estimationcited by 11
- 2023Memory Efficient Neural Processes via Constant Memory Attention Blockcited by 9
- 2024Approximately Equivariant Neural Processescited by 8
- 2025ALINE: Joint Amortization for Bayesian Inference and Active Data Acquisitioncited by 8
- 2025Dimension Agnostic Neural Processescited by 5
- 2024Spectral Convolutional Conditional Neural Processescited by 4
- 2024In-Context In-Context Learning with Transformer Neural Processescited by 4
12 most cited of 26 All 26 (JSON)
- 2026Revisiting Neural Processes via Fourier Transform and Volterra Series31 May 2026
- 2026PRIM: Meta-Learned Bayesian Root Cause Analysis9 May 2026
- 2026Incremental Transformer Neural Processes21 Feb 2026
- 2026Amortising Inference and Meta-Learning Priors in Neural Networks9 Feb 2026
- 2025Efficient Autoregressive Inference for Transformer Probabilistic Models10 Oct 2025
- 2025Estimating Interventional Distributions with Uncertain Causal Graphs through Meta-Learning7 Jul 2025
- 2025Scalable Spatiotemporal Inference with Biased Scan Attention Transformer Neural Processes10 Jun 2025
- 2025ALINE: Joint Amortization for Bayesian Inference and Active Data Acquisition8 Jun 2025
- 2025Exploring Pseudo-Token Approaches in Transformer Neural Processes19 Apr 2025
- 2025Dimension Agnostic Neural Processes28 Feb 2025
- 2025AB-UPT: Scaling Neural CFD Surrogates for High-Fidelity Automotive Aerodynamics Simulations via Anchored-Branched Universal Physics Transformers13 Feb 2025
- 2024Amortized Probabilistic Conditioning for Optimization, Simulation and Inference20 Oct 2024
12 newest of 26 All 26 (JSON)