Saved in:
Bibliographic Details
Main Authors: Jarl, Sanna, Bånkestad, Maria, Scragg, Jonathan J. S., Sjölund, Jens
Format: Preprint
Published: 2026
Subjects:
Online Access:https://arxiv.org/abs/2602.01898
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910008678023168
author Jarl, Sanna
Bånkestad, Maria
Scragg, Jonathan J. S.
Sjölund, Jens
author_facet Jarl, Sanna
Bånkestad, Maria
Scragg, Jonathan J. S.
Sjölund, Jens
contents Bayesian active learning relies on the precise quantification of predictive uncertainty to explore unknown function landscapes. While Gaussian process surrogates are the standard for such tasks, an underappreciated fact is that their posterior variance depends on the observed outputs only through the hyperparameters, rendering exploration largely insensitive to the actual measurements. We propose to inject observation-dependent feedback by warping the input space with a learned, monotone reparameterization. This mechanism allows the design policy to expand or compress regions of the input space in response to observed variability, thereby shaping the behavior of variance-based acquisition functions. We demonstrate that while such warps can be trained via marginal likelihood, a novel self-supervised objective yields substantially better performance. Our approach improves sample efficiency across a range of active learning benchmarks, particularly in regimes where non-stationarity challenges traditional methods.
format Preprint
id arxiv_https___arxiv_org_abs_2602_01898
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Observation-dependent Bayesian active learning via input-warped Gaussian processes
Jarl, Sanna
Bånkestad, Maria
Scragg, Jonathan J. S.
Sjölund, Jens
Machine Learning
68T05
Bayesian active learning relies on the precise quantification of predictive uncertainty to explore unknown function landscapes. While Gaussian process surrogates are the standard for such tasks, an underappreciated fact is that their posterior variance depends on the observed outputs only through the hyperparameters, rendering exploration largely insensitive to the actual measurements. We propose to inject observation-dependent feedback by warping the input space with a learned, monotone reparameterization. This mechanism allows the design policy to expand or compress regions of the input space in response to observed variability, thereby shaping the behavior of variance-based acquisition functions. We demonstrate that while such warps can be trained via marginal likelihood, a novel self-supervised objective yields substantially better performance. Our approach improves sample efficiency across a range of active learning benchmarks, particularly in regimes where non-stationarity challenges traditional methods.
title Observation-dependent Bayesian active learning via input-warped Gaussian processes
topic Machine Learning
68T05
url https://arxiv.org/abs/2602.01898