arXiv · 2504.14416
Exploring Pseudo-Token Approaches in Transformer Neural Processes
Abstract
Neural Processes (NPs) have gained attention in meta-learning for their ability to quantify uncertainty, together with their rapid prediction and adaptability. However, traditional NPs are prone to underfitting. Transformer Neural Processes (TNPs) significantly outperform existing NPs, yet their applicability in real-world scenarios is hindered by their quadratic computational complexity relative to both context and target data points. To address this, pseudo-token-based TNPs (PT-TNPs) have emerged as a novel NPs subset that condense context data into latent vectors or pseudo-tokens, reducing computational demands. We introduce the Induced Set Attentive Neural Processes (ISANPs), employing Induced Set Attention and an innovative query phase to improve querying efficiency. Our evaluations show that ISANPs perform competitively with TNPs and often surpass state-of-the-art models in 1D regression, image completion, contextual bandits, and Bayesian optimization. Crucially, ISANPs offer a tunable balance between performance and computational complexity, which scale well to larger datasets where TNPs face limitations.
Explore related subjects
Keep this discovery
Jose Lara-Rangel, Nanze Chen, Fengzhe Zhang. 2025-04-19. Exploring Pseudo-Token Approaches in Transformer Neural Processes. https://arxiv.org/abs/2504.14416
Cite the original work for its findings. Save a collection to share your selection of sources.