cs.LGJul 7, 2026

The Approximation Ratio for the Risk of Myopic Bayesian Active Learning for Linear Regression

Authors: Stephen Mussmann

Organizations: School of Computer Science Georgia Institute of Technology

Abstract

Active learning studies the fundamental question: what data should we choose to observe? The greedy algorithm in optimal experiment design is a common heuristic and also equivalent to myopic Bayesian active learning for linear regression, the common framework where long-term planning is replaced with the one-step optimal choice. In this work, we prove a first-of-its-kind approximation ratio for the greedy algorithm's risk that is tight up to an absolute constant. The approximation ratio is linear in the maximum initial leverage score (MILS), a newly identified quantity fundamental to the greedy algorithm's performance. Finally, we illustrate the results with simple numerical simulations.

Explore similar work

Jun 12, 2026cs.LG

A Complexity Measure for Active Learning in Multi-group Mean Estimation

We study a \emph{max-risk} objective for active learning in a multi-group mean estimation dd-armed bandits: a learner adaptively allocates a budget of TT samples across dd groups to minimize the worst-case uncertainty index maxk[d]σk2/nk\max_{k\in[d]}σ_k^2/n_k, where σkσ_k is the standard deviation of the distribution of arm dd, and nkn_k is the number of times arm dd is sampled. We develop a local minimax framework and prove the first general lower bound for this objective, valid for any finite-variance hypothesis class. The bound separates difficulty into three orthogonal factors: a \emph{budget} term, a \emph{heteroscedasticity} index measuring how unevenly the uncertainty is spread across arms, and a model-dependent complexity measure, the \emph{Variance Local Curvature} (VLC\mathrm{VLC}), which captures how much information a local change of variance creates inside the hypothesis class. For smooth classes, the VLC\mathrm{VLC} is a reparametrization of a variance--Fisher information, with closed-form values for common families. Benchmarking against the strongest available upper bound shows near-optimality up to logarithmic factors in broad regimes, and pinpoints a systematic gap in highly heterogeneous instances. Our proof introduces two key ingredients: a loss-induced 1\ell_1 geometry on the decision space, and a representation-based instance generator that reduces hard-instance construction to an explicit random matrix calculation.
Abdellah Aznag, Rachel Cummings, Adam N. Elmachtoub
Aug 3, 2026cs.LG

BRiG-AFA: Bellman Risk-to-Go Learning for Non-Myopic Active Feature Acquisition

Active feature acquisition (AFA) asks which unobserved feature to measure next for each test instance under a budget. Greedy rules are easy to train but can overlook context features whose value is realized only through later acquisitions, while reinforcement-learning and generative approaches introduce difficult optimization or conditional-density estimation. We introduce \method, a deployable, supervised alternative that learns a separate candidate-conditioned risk-to-go function for every remaining budget. Starting from the one-step terminal classification risk, the functions are fitted backward with Bellman targets; inference greedily minimizes the learned terminal risk using only observed values, the mask, candidate identity, and remaining budget. A controlled non-myopic benchmark shows the expected mechanism: at budgets two and three, \method improves accuracy over its one-step ablation by 4.84±2.174.84\pm2.17 and 4.39±1.104.39\pm1.10 percentage points (mean ±\pm standard error over five seeds). On Fashion-MNIST with 20 candidate pixels, it improves accuracy at every nontrivial reported budget on average, including 10.20±0.7410.20\pm0.74 points at four acquisitions; its mean paired gain across budgets {2,4,8,12,16}\{2,4,8,12,16\} is 3.50±0.373.50\pm0.37 points. A three-seed MiniBooNE study is mixed at small budgets but positive at 8 and 16 acquisitions, identifying a current boundary rather than supporting a universal claim. These results establish a reproducible mechanism-level case for direct Bellman risk regression and delimit the experiments still needed for state-of-the-art comparison.
Jiaorong Feng, Qian Li, Ying Li
Aug 2, 2026cs.DS

Active Regression for Single-Index Models with Unknown Link Functions

This paper studies active regression for single-index models under general p\ell_p-loss with an unknown 11-Lipschitz link function ff, formulated as minf,xf(Ax)bpp\min_{f,x} \|f(Ax)-b\|_p^p with full access to AA but coordinate-query access to bb. Prior work established upper bounds for known link functions for all p1p\geq 1 and for unknown link functions only in the p=2p=2 case, together with lower bounds for p2p\leq 2. This work addresses the more challenging setting of unknown link functions and general p1p \geq 1. A non-adaptive sampling algorithm is presented that achieves a (1+ε)(1+ε)-approximation using O(dp/21/εp2polylog(n/ε))O(d^{p/2\vee 1}/ε^{p\vee 2}\operatorname{poly}\log(n/ε)) queries. Nearly tight lower bounds are also established for p>2p>2. These results close much of the remaining gap in active p\ell_p-regression for single-index models.
Chansophea Wathanak In, Yi Li, Wai Ming Tai +1