Neighbour-Driven Gaussian Process Variational Autoencoders for Scalable Structured Latent Modelling

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Shi, Xinxing, Jiang, Xiaoyu, Álvarez, Mauricio A.
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866908490503553024
author Shi, Xinxing
Jiang, Xiaoyu
Álvarez, Mauricio A.
author_facet Shi, Xinxing
Jiang, Xiaoyu
Álvarez, Mauricio A.
contents Gaussian Process (GP) Variational Autoencoders (VAEs) extend standard VAEs by replacing the fully factorised Gaussian prior with a GP prior, thereby capturing richer correlations among latent variables. However, performing exact GP inference in large-scale GPVAEs is computationally prohibitive, often forcing existing approaches to rely on restrictive kernel assumptions or large sets of inducing points. In this work, we propose a neighbour-driven approximation strategy that exploits local adjacencies in the latent space to achieve scalable GPVAE inference. By confining computations to the nearest neighbours of each data point, our method preserves essential latent dependencies, allowing more flexible kernel choices and mitigating the need for numerous inducing points. Through extensive experiments on tasks including representation learning, data imputation, and conditional generation, we demonstrate that our approach outperforms other GPVAE variants in both predictive performance and computational efficiency.
format Preprint
id arxiv_https___arxiv_org_abs_2505_16481
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Neighbour-Driven Gaussian Process Variational Autoencoders for Scalable Structured Latent Modelling
Shi, Xinxing
Jiang, Xiaoyu
Álvarez, Mauricio A.
Machine Learning
Gaussian Process (GP) Variational Autoencoders (VAEs) extend standard VAEs by replacing the fully factorised Gaussian prior with a GP prior, thereby capturing richer correlations among latent variables. However, performing exact GP inference in large-scale GPVAEs is computationally prohibitive, often forcing existing approaches to rely on restrictive kernel assumptions or large sets of inducing points. In this work, we propose a neighbour-driven approximation strategy that exploits local adjacencies in the latent space to achieve scalable GPVAE inference. By confining computations to the nearest neighbours of each data point, our method preserves essential latent dependencies, allowing more flexible kernel choices and mitigating the need for numerous inducing points. Through extensive experiments on tasks including representation learning, data imputation, and conditional generation, we demonstrate that our approach outperforms other GPVAE variants in both predictive performance and computational efficiency.
title Neighbour-Driven Gaussian Process Variational Autoencoders for Scalable Structured Latent Modelling
topic Machine Learning
url https://arxiv.org/abs/2505.16481