Feature-level Site Leakage Reduction for Cross-Hospital Chest X-ray Transfer via Self-Supervised Learning
Organizations: University of Mentouri Brothers Constantine 1, Constantine, Algeria · University of Paris-VIII-Vincennes-Saint-Denis, France · LRIA Laboratory, University of Science and Technology Houari Boumediene (USTHB), Bab-Ezzouar, Algeria · LSIACIO Laboratory, University of Mentouri Brothers Constantine 1, Constantine, Algeria
Abstract
Cross-hospital failure in chest X-ray models is often attributed to domain shift, yet most work assumes invariance without measuring it. This paper studies how to measure site leakage directly and how that measurement changes conclusions about transfer methods. We study multi-site self-supervised learning (SSL) and feature-level adversarial site confusion for cross-hospital transfer. We pretrain a ResNet-18 on NIH and CheXpert without pathology labels. We then freeze the encoder and train a linear pneumonia classifier on NIH only, evaluating transfer to RSNA. We quantify site leakage using a post hoc linear probe that predicts acquisition site from frozen backbone features and projection features . Across 3 random seeds, multi-site SSL improves RSNA AUC from 0.6736 0.0148 (ImageNet initialization) to 0.7804 0.0197. Adding adversarial site confusion on reduces measured leakage but does not reliably improve AUC and increases variance. On , site probe accuracy drops from 0.9890 0.0021 (SSL-only) to 0.8504 0.0051 (CanonicalF), where chance is 0.50. On , probe accuracy drops from 0.8912 0.0092 to 0.7810 0.0250. These results show that measuring leakage changes how transfer methods should be interpreted: multi-site SSL drives transfer, while adversarial confusion exposes the limits of invariance assumptions.
Figures & tables
| Characteristic | NIH ChestX-ray14 | CheXpert | RSNA Pneumonia Challenge |
|---|---|---|---|
| Origin | NIH Clinical Center, USA | Stanford Hospital, USA | RSNA Pneumonia Detection Challenge (compiled from multiple institutions) |
| Modality | Chest X-ray (2D) | Chest X-ray (2D) | Chest X-ray (2D) |
| Labeling | Image-level labels (includes pneumonia) | Labels extracted from radiology reports (CheXpert labeler); uncertainty present | Pneumonia annotations for challenge (used here for evaluation) |
| Views used in this work | Frontal only | Frontal only | Frontal only (as provided in split) |
| Role in this paper | Supervised source site for linear head; also part of SSL pool | Unlabeled SSL pretraining site; extra target evaluation on CheXpert-valid | Held-out target evaluation for transfer (unseen during supervised training/selection) |
| Size used in this work | NIH-train: 26,454 images; NIH-val: 2,940 images | Train: 191,027 images; Valid: 202 images (pneumonia task) | 30,227 images |
| Method | NIH val AUC | RSNA AUC | CheXpert valid AUC |
|---|---|---|---|
| ImageNet Frozen | 0.5787 0.0789 | 0.6736 0.0148 | 0.5939 0.0352 |
| SSL-only Frozen | 0.6220 0.0408 | 0.7804 0.0197 | 0.7309 0.0554 |
| CanonicalF Frozen | 0.5923 0.0152 | 0.7241 0.0956 | 0.7998 0.0399 |
| Stage | Probe on | Probe on |
|---|---|---|
| Before SSL (ImageNet) | 0.9962 0.0004 | 0.9815 0.0030 |
| After SSL-only | 0.9890 0.0021 | 0.8912 0.0092 |
| After CanonicalF | 0.8504 0.0051 | 0.7810 0.0250 |
Cross-Domain Transfer Learning
Cross-domain transfer learning reuses knowledge across datasets, populations, devices, or tasks when their distributions and feature spaces differ.
See all 203 papers