Efficient Chest X-ray Representation Learning via Semantic-Partitioned Contrastive Learning
Organizations: Shenzhen University of Advanced Technology, Shenzhen, China
Abstract
Self-supervised learning (SSL) has emerged as a powerful paradigm for Chest X-ray (CXR) analysis under limited annotations. Yet, existing SSL strategies remain suboptimal for medical imaging. Masked image modeling allocates substantial computation to reconstructing high-frequency background details with limited diagnostic value. Contrastive learning, on the other hand, often depends on aggressive augmentations that risk altering clinically meaningful structures. We introduce Semantic-Partitioned Contrastive Learning (S-PCL), an efficient pre-training framework tailored for CXR representation learning. Instead of reconstructing pixels or relying on heavy augmentations, S-PCL randomly partitions patch tokens from a single CXR into two non-overlapping semantic subsets. Each subset provides a complementary but incomplete view. The encoder must maximize agreement between these partitions, implicitly inferring global anatomical layout and local pathological cues from partial evidence. This semantic partitioning forms an internal bottleneck that enforces long-range dependency modeling and structural coherence. S-PCL eliminates the need for hand-crafted augmentations, auxiliary decoders, and momentum encoders. The resulting architecture is streamlined, computationally efficient, and easy to scale. Extensive experiments on large-scale CXR benchmarks, including ChestX-ray14, CheXpert, RSNA Pneumonia and SIIM-ACR Pneumothorax, show that S-PCL achieves competitive performance while attaining the lowest GFLOPs and superior accuracy among existing SSL approaches.
Figures & tables
| Method | GFLOPs | GPU Hrs | ChestX-ray14 | CheXpert | RSNA Pneu. | ||||||
|---|---|---|---|---|---|---|---|---|---|---|---|
| 1% | 10% | 100% | 1% | 10% | 100% | 1% | 10% | 100% | |||
| Training with clinical reports | |||||||||||
| M3AE Chen et al. (2022) | 34.4 | 710 | - | - | - | 86.2 | 87.3 | 87.9 | 89.0 | 90.8 | 92.3 |
| REFERS Zhou et al. (2022) | 54.8 | - | 76.7 | 80.9 | 84.7 | 87.2 | 88.1 | 88.2 | 89.4 | 91.6 | 92.7 |
| MRM Zhou et al. (2023) | 13.1 | 800 | 79.4 | 84.0 | 85.9 | 88.5 | 88.5 | 88.7 | 91.3 | 92.7 | 93.3 |
| Training without clinical reports | |||||||||||
| Method | Backbone | Atelectasis | Cardiomegaly | Consolidation | Edema | Effusion | mAUC (%) |
|---|---|---|---|---|---|---|---|
| Label-Assemble Kang et al. (2023) | DN121 | 82.1 | 85.9 | 94.4 | 89.2 | 93.6 | 89.0 |
| Medical MAE Xiao et al. (2023) | ViT-B/16 | 82.7 | 83.5 | 92.5 | 93.8 | 94.1 | 89.3 |
| S-PCL (Ours) | ViT-B/16 | 80.3 | 95.4 | 80.4 | 94.1 | 95.6 | 89.1 |
| Model | Mean | Atelectasis | Cardiomegaly | Effusion | Infiltration | Mass | Nodule | Pneumonia | Pneumothorax | Consolidation | Edema | Emphysema | Fibrosis | Pleural Thicken | Hernia |
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| ConVIRT Zhang et al. (2022) | 56.0 | 45.1 | 44.3 | 63.2 | 65.1 | 61.6 | 57.2 | 63.6 | 54.1 | 63.7 | 70.2 | 41.9 | 47.4 | 56.0 | 51.2 |
| GLoRIA Huang et al. (2021) | 81.8 | 82.6 | 83.3 | 86.0 | 66.4 | 81.8 | 73.5 | 71.0 | 84.5 | 81.3 | 89.8 | 93.1 | 78.9 | 76.1 | 97.5 |
| MRM Zhou et al. (2023) | 85.9 | 84.2 | 93.0 | 89.6 | 71.8 | 88.2 | 78.5 | 77.3 | 90.2 | 82.2 | 91.0 | 94.3 | 86.7 | 81.4 | 94.4 |
| S-PCL (Ours) | 84.1 | 82.3 | 87.7 | 91.4 | 75.7 | 80.8 | 75.0 | 77.3 | 92.5 | 79.0 | 89.4 | 93.5 | 82.6 | 80.1 | 90.3 |
| Methods | Semantic Segmentation | Object Detection | ||||
|---|---|---|---|---|---|---|
| SIIM-ACR Pneumothorax | RSNA Pneumonia | |||||
| 1% | 10% | 100% | 1% | 10% | 100% | |
| GLoRIA Huang et al. (2021) | 37.4 | 57.1 | 64.0 | 11.6 | 16.1 | 24.8 |
| MGCA Wang et al. (2022) | 49.7 | 59.3 | 64.2 | 12.9 | 16.8 | 24.9 |
| MedKLIP Wu et al. (2023) | 50.2 | 60.8 | 63.9 | 8.9 | 16.3 | 24.5 |
| S-PCL (Ours) | 38.8 | 61.7 | 65.1 | 13.2 | 16.9 | 25.6 |
Medical Image Representation Learning
Medical image representation learning builds reusable visual features for diagnosis, segmentation, retrieval, and other clinical imaging tasks.
See all 260 papers