cs.DSAug 2, 2026

Active Regression for Single-Index Models with Unknown Link Functions

Authors: Chansophea Wathanak InYi LiWai Ming TaiXuan Wu

Organizations: School of Physical and Mathematical Sciences, Nanyang Technological University, Singapore · Independent Researcher · John Hopcroft Center for Computer Science, Shanghai Jiao Tong University, China

Abstract

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.

Explore similar work

Jul 7, 2026cs.LG

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

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.
Stephen Mussmann
May 14, 2026cs.LG

A Mutual Information Lower Bound for Multimodal Regression Active Learning

Active learning for continuous regression has lacked an acquisition function that targets epistemic uncertainty when the predictive distribution is multimodal: variance misses modal disagreement, and information-theoretic targets like BALD are designed for discrete outputs. We introduce a Two-Index framework that makes this separation explicit: one stochastic index selects among competing model hypotheses (epistemic source), while a second governs within-hypothesis randomness (aleatoric source). An entropy decomposition within the framework identifies the mutual information between the output and the epistemic index as a principled acquisition objective, and we prove this quantity vanishes as the model is trained on growing datasets, confirming that it captures exactly the uncertainty data can resolve. Because this mutual information is intractable for continuous outputs, we derive the Mutual Information Lower Bound (MI-LB) acquisition function, a closed-form approximation for Mixture Density Network ensembles. On benchmarks featuring multimodal systems, MI-LB matches or beats every baseline evaluated and is the only method to do so consistently -- geometric and Fisher-based baselines compete only when the input space already encodes the multimodality, and collapse otherwise.
Leonardo Ferreira Guilhoto, Akshat Kaushal, Paris Perdikaris
Jun 9, 2026cs.LG

Robust Regression of General ReLUs with Queries

We study the task of agnostically learning general (as opposed to homogeneous) ReLUs under the Gaussian distribution with respect to the squared loss. In the passive learning setting, recent work gave a computationally efficient algorithm that uses poly(d,1/ε)poly(d,1/ε) labeled examples and outputs a hypothesis with error O(opt)+εO(opt)+ε, where optopt is the squared loss of the best fit ReLU. Here we focus on the interactive setting, where the learner has some form of query access to the labels of unlabeled examples. Our main result is the first computationally efficient learner that uses dpolylog(1/ε)+O~(min{1/p,1/ε})d polylog(1/ε)+\tilde{O}(\min\{1/p, 1/ε\}) black-box label queries, where pp is the bias of the target function, and achieves error O(opt)+εO(opt)+ε. We complement our algorithmic result by showing that its query complexity bound is qualitatively near-optimal, even ignoring computational constraints. Finally, we establish that query access is essentially necessary to improve on the label complexity of passive learning. Specifically, for pool-based active learning, any active learner requires Ω~(d/ε)\tildeΩ(d/ε) labels, unless it draws a super-polynomial number of unlabeled examples.
Ilias Diakonikolas, Daniel M. Kane, Mingchen Ma