Multi-Task Active Learning with Efficient Resource Allocation
stat.ME, stat.ML
Submitted: 2026-10-05
Updated: 2026-10-05
Code: https://github.com/zalandoresearch/fashion-mnist
License: http://creativecommons.org/licenses/by-nc-nd/4.0/
The gist: Many scientific studies allow costly auxiliary information to be collected during data labeling but not at deployment.
Terminology
Abstract
Many scientific studies allow costly auxiliary information to be collected during data labeling but not at deployment. Examples include diagnostic tests, laboratory assays, and expert evaluations. We study prediction under this deployment asymmetry, where auxiliary variables are selectively acquired during labeling under a budget constraint but systematically unavailable at prediction time, creating a missing-by-design problem that couples data acquisition, surrogate construction, and prediction. In this work, we introduce Active Learning with Cost-Adaptive Task Resource Allocations (ALCATRAs), a unified framework for selectively acquiring auxiliary information under resource constraints and leveraging that information to improve downstream prediction. ALCATRAs consists of two main components: a task-selection policy which strategically selects a sequence of cost-effective tasks for unlabeled data to perform, and a surrogate learning procedure which transfers knowledge from completed tasks to enhance model predictions. In theory, we show the effectiveness of surrogate models and sample-efficient task policies in improving the model's prediction error bound. Simulation studies and an application to the UCI heart disease cohort demonstrate improved sample efficiency of the proposed ALCATRAs framework relative to baselines under the studied settings.
Sources
- On the role of surrogates in the efficient estimation of treatment effects with limited outcome data
- Towards Cost Sensitive Decision Making
- Prediction Aided by Surrogate Training
- Reinforcement Learning with Efficient Active Feature Acquisition
- Adam: A Method for Stochastic Optimization
Related papers
- Doubly robust inference via calibration
- Bayesian Empirical Bayes: Simultaneous Inference from Probabilistic Symmetries
- Optimal Watermark Localization in Mixed-Source Large Language Model Texts
- Flexible Nonparametric Inference for Causal Effects under the Front-Door Model
- A Greedy PDE Router for Blending Neural Operators and Classical Methods
- Policy Learning with Weak Signals