L2GSSL: Label-Efficient Local-to-Global Self-Supervised Learning for Volumetric Healthcare Data
Organizations: Computer Science and Information Engineering, National Central University
Abstract
Healthcare institutions routinely accumulate high-dimensional computed tomography (CT) volumes, but only a fraction can be annotated by experts. We present L2GSSL, a local-to-global self-supervised learning framework that leverages unlabeled volumes by decomposing pretraining into intra-slice and inter-slice representation spaces. A slice encoder learns view-invariant local features, and a global encoder aggregates the resulting tokens to model volumetric context. The stop-gradient isolates the local space from the global objective, while slice-wise execution and a three-phase schedule support training under constrained GPU memory. On LIDC-IDRI lung nodule malignancy classification, L2GSSL reaches AUROC with only of the labels and with , exceeding the volume-level VICReg by and , respectively. With all labels, the strongest supervised baseline exceeds L2GSSL by only AUROC, consistent with self-supervised pretraining providing its greatest value when annotations are limited. Without adaptation to the external LNDb cohort, a model fine-tuned on of LIDC-IDRI obtains AUROC and AUPRC, improving over the strongest evaluated self-supervised baseline by and . Experiments across data sources, thoracic and abdominal anatomy, downstream tasks, and two backbones examine healthcare-data variety, while lesion-slice masking shows that the predictions depend on lesion-bearing content.
Figures & tables
| Method | Pretraining input | Alignment target | Explicit negatives | Auxiliary network state | Local/global split |
|---|---|---|---|---|---|
| MoCo v2 [ 17 ] | Slice-aggregated volume | Final volume embedding | Yes | Momentum encoder and queue | No |
| SimSiam [ 16 ] | Slice-aggregated volume | Final volume embedding | No | Predictor | No |
| Volume VICReg [ 4 ] | Slice-aggregated volume | Final volume embedding | No | None | No |
| VoCo [ 3 ] | Overlapping 3D crops | Volumetric spatial relation | Spatially defined | EMA teacher | No |
| 2D-VoCo [ 15 ] | Individual CT slices | Slice-level spatial-semantic relation | Spatially defined | None | No |
| L2GSSL | Slices and ordered tokens | Slice and volume embeddings | No | None | Yes |
| Dataset | Anatomy/modality | Available scale | Use in this study | Label use | Evaluation role |
|---|---|---|---|---|---|
| LIDC-IDRI [ 5 ] | Thoracic CT | cases; processed nodules | SSL and downstream training | , , or training nodules | Internal discrimination and reader analysis |
| LNDb [ 6 ] | Thoracic CT | scans | External testing only | No target-domain training | Direct cross-dataset transfer |
| RSNA 2023 [ 7 ] | Abdominal CT | patients | Downstream trauma pipeline | Competition task labels | Multi-organ task transfer |
| AbdomenCT-1K [ 8 ] | Abdominal CT | Auxiliary volumetric collection | Additional SSL source | Labels not used by SSL | Anatomically aligned pretraining |
| Protocol | Encoder update | Supervision | Controlled factor | Primary question |
|---|---|---|---|---|
| Fixed-batch linear probe | Frozen | LIDC-IDRI labels | Local batch ; two accumulation steps | Are pretrained features linearly separable? |
| VRAM-capped linear probe | Frozen | LIDC-IDRI labels | Peak VRAM constrained to – GB | How effectively is a fixed memory budget used? |
| Label-fraction fine-tuning | Full network | , , or of LIDC-IDRI | Shared split and downstream pipeline | Does pretraining reduce dependence on labels? |
| Direct LNDb evaluation | Frozen at inference | No LNDb training labels | LIDC-IDRI model evaluated without adaptation | Does performance transfer across data sources? |
| RSNA pipeline integration | Full downstream pipeline | RSNA task labels | Original multi-expert inference design retained | Does initialization transfer to abdominal trauma CT? |
| labels ( ) | labels ( ) | labels ( ) | ||||
|---|---|---|---|---|---|---|
| Method | AUROC | AUPRC | AUROC | AUPRC | AUROC | AUPRC |
| Supervised MST-DINOv2 | ||||||
| MoCo v2 | ||||||
| SimSiam | ||||||
| Volume VICReg | ||||||
| L2GSSL | ||||||
| Method | AUROC | AUPRC |
|---|---|---|
| MoCo v2 | ||
| SimSiam | ||
| Volume VICReg | ||
| L2GSSL |
| Method | AUROC | AUPRC |
|---|---|---|
| Supervised | ||
| MoCo v2 | ||
| SimSiam | ||
| Volume VICReg | ||
| L2GSSL |
| Method | Peak VRAM (GB) | AUROC | AUPRC |
|---|---|---|---|
| MoCo v2 | – | ||
| SimSiam | – | ||
| Volume VICReg | – | ||
| L2GSSL | – |
| LP | FT, labels | FT, labels | FT, labels | ||||||
|---|---|---|---|---|---|---|---|---|---|
| Method | Peak VRAM (GB) | AUROC | AUPRC | AUROC | AUPRC | AUROC | AUPRC | AUROC | AUPRC |
| VoCo (scratch) | |||||||||
| VoCo (MONAI) | |||||||||
| L2GSSL | – | ||||||||
| Method | AUROC | AUPRC |
|---|---|---|
| MoCo v2 | ||
| SimSiam | ||
| Volume VICReg | ||
| L2GSSL |
| Condition | Malignant TPR | AUROC | AUPRC |
|---|---|---|---|
| No mask | |||
| Tumor-slice mask | |||
| Change |
| Initialization | Configuration | AUROC | AUPRC |
|---|---|---|---|
| Random | Full L2GSSL | ||
| Random | Without stop-gradient | ||
| Random | Without phased schedule | ||
| DINOv2 | Full L2GSSL | ||
| DINOv2 | Without stop-gradient | ||
| DINOv2 | Without phased schedule |