PromptGate Client Adaptive Vision Language Gating for Open Set Federated Active Learning
Organizations: University of Bonn, University Hospital Bonn, Clinic for Diagnostic and Interventional Radiology, 53127 Bonn, Germany
Abstract
Deploying medical AI across resource-constrained institutions demands data-efficient learning pipelines that respect patient privacy. Federated Learning (FL) enables collaborative medical AI without centralising data, yet real-world clinical pools are inherently open-set, containing out-of-distribution (OOD) noise such as imaging artifacts and wrong modalities. Standard Active Learning (AL) query strategies mistake this noise for informative samples, wasting scarce annotation budgets. We propose PromptGate, a dynamic VLM-gated framework for Open-Set Federated AL that purifies unlabeled pools before querying. PromptGate introduces a federated Class-Specific Context Optimization: lightweight, learnable prompt vectors that adapt a frozen BiomedCLIP backbone to local clinical domains and aggregate globally via FedAvg -- without sharing patient data. As new annotations arrive, prompts progressively sharpen the ID/OOD boundary, turning the VLM into a dynamic gatekeeper that is strategy-agnostic: a plug-and-play pre-selection module enhancing any downstream AL strategy. Experiments on distributed dermatology and breast imaging benchmarks show that while static VLM prompting degrades to 50% ID purity, PromptGate maintains 95% purity with 98% OOD recall.
Figures & tables
| Strategies | Random | Entropy | FEAL | PAL | LfOSA | OpenPath* | Avg |
|---|---|---|---|---|---|---|---|
| FedISIC (R = 5) | |||||||
| Coldstart | 61.7 (60.2) | 59.1 (62.0) | 45.9 (57.6) | 61.1 (50.5) | 52.1 (55.7) | 84.1 (60.5) | 60.7 (57.8) |
| Baseline | 60.9 (59.5) | 72.9 (61.5) | 78.7 (58.0) | 73.1 (49.7) | 76.1 (54.1) | 83.8 (61.0) | 74.3 (57.3) |
| Ours (Mixed) | 97.4 (64.4) | 98.9 ( 66.1 ) | 98.7 (57.2) | 98.7 (50.4) | 98.6 (54.9) | 86.5 (63.5) | 96.5 (59.4) |
| Ours (Global) | 98.1 (59.8) | 99.0 (63.7) | 98.9 (58.5) | 98.7 ( 53.1 ) | 99.0 (58.3) | 86.6 ( 64.4 ) | 96.7 (59.6) |
| Ours (Local) | 98.2 ( 65.3 ) | 99.5 (64.2) | 98.8 ( 60.1 ) | 98.7 (50.2) | 99.2 ( 60.0 ) | 86.5 (62.2) | 96.8 ( 60.3 ) |
| Variant | Met. (%) | C0 | C1 | C2 | C3 | Avg |
|---|---|---|---|---|---|---|
| Baseline | ID BMA | 30.6 | 27.6 | 31.2 | 26.6 | 27.5 |
| Acc. | 83.2 | 87.9 | 85.8 | 82.4 | 84.4 | |
| OOD Recall | 0.6 | 0.9 | 0.0 | 0.0 | 0.5 | |
| Ours (Mixed) | ID BMA | 35.7 | 64.4 | 43.2 | 35.3 | 42.0 |
| Acc. | 99.5 | 99.6 | 99.7 | 99.7 | 99.6 | |
| OOD Recall | 97.6 | 99.7 | 99.3 | 97.6 | 98.2 |
Active Learning
Active learning selects the examples, labels, features, or experiments most useful for improving a model under limited budgets.
See all 288 papers