Where to Look Matters: Rethinking Sub-Volume Sampling in 3D Medical Self-supervised Learning
cs.CV
Submitted: 2026-10-09
Updated: 2026-10-09
Comments: This paper has been accepted by NeurIPS 2026 (The 40th Annual Conference on Neural Information Processing Systems)
License: http://creativecommons.org/publicdomain/zero/1.0/
The gist: Self-supervised learning on 3D medical images commonly relies on random sub-volume sampling during pretraining.
Terminology
Abstract
Self-supervised learning on 3D medical images commonly relies on random sub-volume sampling during pretraining. However, random cropping implicitly treats all spatial regions as equally worth observing, despite anatomical priors and spatial redundancy that make regions inherently unequal in their value for representation learning. To this end, we first show through controlled crop-quality manipulations that better sub-volumes lead to better downstream transfer, revealing the observation policy as an overlooked bottleneck in 3D medical SSL. Motivated by this finding, we revisit sub-volume sampling as a where-to-look problem. We propose, a plug-and-play framework that replaces random cropping with an offline coarse-to-fine probe bank guided by anatomy-awareness, informativeness, and diversity. We instantiate as a lightweight variant using geometric and image-statistical cues, and a diagnostic variant using frozen visual features, to examine whether lightweight volume-intrinsic priors suffice or richer semantic cues reliably provide additional value. Extensive experiments across multiple 3D medical SSL backbones and downstream tasks demonstrate that - consistently improves transfer performance over random sampling. Mechanistic analyses further attribute these gains to more anatomically plausible and informative observations, reduced spatial redundancy, more efficient learning dynamics, and more structured representations.
Sources
- Med3D: Transfer Learning for 3D Medical Image Analysis
- Learning Generalizable 3D Medical Image Representations from Mask-Guided Self-Supervision
- MeDUET: Disentangled Unified Pretraining for 3D Medical Image Synthesis and Analysis
- InvCoSS: Inversion-driven Continual Self-supervised Learning in Medical Multi-modal Image Pre-training
- Automatic Organ and Pan-cancer Segmentation in Abdomen CT: the FLARE 2023 Challenge
Related papers
- Loss Knows Best: Detecting Annotation Errors in Videos via Loss Trajectories
- AnchorWeave: World-Consistent Video Generation with Retrieved Local Spatial Memories
- Benchmarking the Robustness of Foundation Models for Mammography under Domain Shift
- MambaX-Net: Dual-Input Mamba-Enhanced Cross-Attention Network for Longitudinal MRI Segmentation
- TeleOCR: Navigating Document Parsing Across Digital and Camera-Captured Documents
- A Survey on Efficient Vision-Language-Action Models