Amortized Safe Active Learning for Real-Time Data Acquisition: Pretrained Neural Policies from Simulated Nonparametric Functions

Cen-You Li, Marc Toussaint, Barbara Rakitsch, Christoph Zimmer

TL;DR: A method to simulate and meta-learn active learning and safe active learning data selection policies for real-time deployment.

Abstract:

Safe active learning (AL) is a sequential scheme for learning unknown systems while respecting safety constraints during data acquisition.
Existing methods often rely on Gaussian processes (GPs) to model the task and safety constraints, requiring repeated GP updates and constrained acquisition optimization—incurring in significant computations which are challenging for real-time decision-making. We propose amortized AL for regression and amortize safe AL, replacing expensive online computations with a pretrained neural policy. Inspired by recent advances in amortized Bayesian experimental design, we turn GPs into a pretraining simulator. We train our policy prior to the AL deployment on simulated nonparametric functions, using Fourier feature-based GP sampling and a differentiable acquisition objective that is safety-aware in the safe AL setting. At deployment, our policy selects informative and (if desired) safe queries via a single forward pass, eliminating GP inference and acquisition optimization. This leads to magnitudes of speed improvements while preserving learning quality. Our framework is modular and, without the safety component, yields real-time unconstrained AL for time-sensitive tasks.

Submission Number: 221