VALIANT: Prompt Instability for Active Learning in Black-Box Medical Imaging

Authors

  • Dwarikanath Mahapatra Department of Computer Science, Khalifa University, Abu Dhabi, UAE
  • Behzad Bozorgtabar University of Southern Denmark, Odense, Denmark EPFL, Lausanne, Switzerland
  • Sudipta Roy Jio University, Navi Mumbai, India
  • Imran Razzak MBZUAI, Abu Dhabi, UAE
  • Mauricio Reyes University of Bern, Switzerland Dept. of Radiation Oncology, Inselspital, Bern University Hospital, Switzerland

DOI:

https://doi.org/10.1609/aaai.v40i10.37734

Abstract

The deployment of large, black-box foundation models for medical image classification is often hindered by the high cost of acquiring large, task-specific labeled datasets for fine-tuning. While active learning (AL) presents a promising solution, many state-of-the-art AL methods are computationally expensive or require full access to internal model parameters. We present VALIANT (Visual Adaptation and Learning Integration for Active learNing Tasks), a new active learning framework designed to efficiently adapt black-box foundation models by overcoming these limitations. VALIANT introduces a lightweight Visual Prompt Decoder (VIPD), trained via unsupervised Zero-Order Optimization (ZOO), to generate task-specific visual prompts without internal model access. Our core contribution is a perturbation-based ranking strategy that leverages this VIPD to formulate a computationally efficient, gradient-aware informativeness metric. This metric, which we term prompt instability, identifies the most impactful samples for the labeling budget. VALIANT further enhances this process by incorporating anatomical information from unsupervised segmentation maps to generate more discriminative visual prompts. Extensive evaluations on multiple medical datasets demonstrate VALIANT’s superior performance and significant reduction in labeling costs compared to a range of existing active learning techniques, positioning it as a scalable and practical solution for medical image analysis.

Published

2026-03-14

How to Cite

Mahapatra, D., Bozorgtabar, B., Roy, S., Razzak, I., & Reyes, M. (2026). VALIANT: Prompt Instability for Active Learning in Black-Box Medical Imaging. Proceedings of the AAAI Conference on Artificial Intelligence, 40(10), 7901-7909. https://doi.org/10.1609/aaai.v40i10.37734

Issue

Section

AAAI Technical Track on Computer Vision VII