MedCAL-Bench: A Comprehensive Benchmark on Cold-Start Active Learning with Foundation Models for Medical Image Analysis
2508.03441v1
cs.CV
2025-08-09
Авторы:
Ning Zhu, Xiaochuan Ma, Shaoting Zhang, Guotai Wang
Резюме на русском
Задача Cold-Start Active Learning (CSAL) стремится выбирать информативные примеры для аннотации без предварительного знания, чтобы оптимизировать эффективность аннотации и повысить моделирование в условиях ограниченного бюджета аннотаций в области медицинского анализа изображений. Основная проблема заключается в неэффективности существующих CSAL-методов, которые опираются на Self-Supervised Learning (SSL) на целевой выборке, ограничиваясь недостаточно выработанными функциями выделения признаков. Нашим решением является MedCAL-Bench — первая систематическая система оценки на основе Foundation Models (FMs) для CSAL в медицинской области. Мы оценили 14 FMs и 7 CSAL-стратегий на 7 медицинских данных, затрагивающих классификацию и сегментацию различных модальностей. Мы показали, что большинство FMs эффективны в качестве функций выделения признаков, в том числе DINO-семья для сегментации. Также мы выявили, что различия в производительности FMs варьируются в зависимости от типа задачи, а также что разные стратегии выбора примеров подходят для различных данных. Основным выводом является, что MedCAL-Bench стал первым CSAL-бенчмарком, оценивающим как этапы выделения признаков, так и выбора примеров.
Abstract
Cold-Start Active Learning (CSAL) aims to select informative samples for
annotation without prior knowledge, which is important for improving annotation
efficiency and model performance under a limited annotation budget in medical
image analysis. Most existing CSAL methods rely on Self-Supervised Learning
(SSL) on the target dataset for feature extraction, which is inefficient and
limited by insufficient feature representation. Recently, pre-trained
Foundation Models (FMs) have shown powerful feature extraction ability with a
potential for better CSAL. However, this paradigm has been rarely investigated,
with a lack of benchmarks for comparison of FMs in CSAL tasks. To this end, we
propose MedCAL-Bench, the first systematic FM-based CSAL benchmark for medical
image analysis. We evaluate 14 FMs and 7 CSAL strategies across 7 datasets
under different annotation budgets, covering classification and segmentation
tasks from diverse medical modalities. It is also the first CSAL benchmark that
evaluates both the feature extraction and sample selection stages. Our
experimental results reveal that: 1) Most FMs are effective feature extractors
for CSAL, with DINO family performing the best in segmentation; 2) The
performance differences of these FMs are large in segmentation tasks, while
small for classification; 3) Different sample selection strategies should be
considered in CSAL on different datasets, with Active Learning by Processing
Surprisal (ALPS) performing the best in segmentation while RepDiv leading for
classification. The code is available at
https://github.com/HiLab-git/MedCAL-Bench.
Ссылки и действия
Дополнительные ресурсы: