Abstract:Recent self-supervised advances in medical computer vision exploit global and local anatomical self-similarity for pretraining prior to downstream tasks such as segmentation. However, current methods assume i.i.d. image acquisition, which is invalid in clinical study designs where follow-up longitudinal scans track subject-specific temporal changes. Further, existing self-supervised methods for medically-relevant image-to-image architectures exploit only spatial or temporal self-similarity and only do so via a loss applied at a single image-scale, with naive multi-scale spatiotemporal extensions collapsing to degenerate solutions. To these ends, this paper makes two contributions: (1) It presents a local and multi-scale spatiotemporal representation learning method for image-to-image architectures trained on longitudinal images. It exploits the spatiotemporal self-similarity of learned multi-scale intra-subject features for pretraining and develops several feature-wise regularizations that avoid collapsed identity representations; (2) During finetuning, it proposes a surprisingly simple self-supervised segmentation consistency regularization to exploit intra-subject correlation. Benchmarked in the one-shot segmentation setting, the proposed framework outperforms both well-tuned randomly-initialized baselines and current self-supervised techniques designed for both i.i.d. and longitudinal datasets. These improvements are demonstrated across both longitudinal neurodegenerative adult MRI and developing infant brain MRI and yield both higher performance and longitudinal consistency.

What problem does this paper attempt to address?

The main problems that this paper attempts to solve are the two major bottlenecks existing in the current self - supervised medical image analysis methods when dealing with longitudinal neuroimaging data: 1. **Learning with a small amount of labeled data**: Modern imaging studies may scan hundreds to thousands of individuals, but the cost of manually delineating the volume - of - interest structures of multiple individuals for supervised segmentation training is extremely high. Therefore, the current research focus is to use a large number of unlabeled images to pre - train image - to - image architectures (such as U - Net), and then efficiently fine - tune these models in a single - or few - shot setting. However, the existing self - supervised methods usually assume that the data are independently and identically distributed (i.i.d.), which is not valid in clinical research design because longitudinal scans track the temporal changes of specific objects. 2. **Violation of the i.i.d. assumption**: Most of the existing self - supervised frameworks assume that the data are independently and identically distributed, but in longitudinal studies, the temporal images of the same object are highly correlated. The existing longitudinal representation learning methods mainly focus on imposing temporal consistency at the encoder bottleneck, enabling the encoder to learn representations that are sensitive to the acquisition order and the overall trajectory. However, these methods have uncertainties when extended to pixel - level tasks (such as segmentation). To overcome these bottlenecks, this paper proposes a local multi - scale spatio - temporal representation learning framework, aiming to improve the existing methods in the following ways: - **Local multi - scale spatio - temporal representation learning**: Utilize the spatial and temporal self - similarity of local multi - scale deep features to pre - train the image - to - image architectures, and develop multiple feature regularization methods to avoid degenerate solutions. - **Self - supervised segmentation consistency regularization**: In the fine - tuning stage, a simple self - supervised segmentation consistency regularization method is proposed, which utilizes the internal correlations within the same object to improve the segmentation performance and longitudinal consistency. This framework has been benchmarked on multiple segmentation tasks, including the longitudinal segmentation tasks of elderly and infant brain MRI. The results show that this framework not only improves the segmentation performance but also enhances the longitudinal consistency.

Local Spatiotemporal Representation Learning for Longitudinally-consistent Neuroimage Analysis

MsVRL: Self-Supervised Multiscale Visual Representation Learning Via Cross-Level Consistency for Medical Image Segmentation

Spatiotemporal Representation Learning for Short and Long Medical Image Time Series

Dual Meta-Learning with Longitudinally Generalized Regularization for One-Shot Brain Tissue Segmentation Across the Human Lifespan

Self-Supervised Alignment Learning for Medical Image Segmentation

Self-supervised Learning of Dense Hierarchical Representations for Medical Image Segmentation

Learning unbiased group-wise registration (LUGR) and joint segmentation: evaluation on longitudinal diffusion MRI

Self-Supervised Learning for Non-Rigid Registration Between Near-Isometric 3D Surfaces in Medical Imaging.

Self-supervised Lesion Change Detection and Localisation in Longitudinal Multiple Sclerosis Brain Imaging

Longitudinal Segmentation of MS Lesions via Temporal Difference Weighting

LSOR: Longitudinally-Consistent Self-Organized Representation Learning

Localized Region Contrast for Enhancing Self-Supervised Learning in Medical Image Segmentation

Unsupervised Local Discrimination for Medical Images

Patch-Based Techniques in Medical Imaging

A hybrid deep learning framework for integrated segmentation and registration: evaluation on longitudinal white matter tract changes

SSL^2: Self-Supervised Learning meets Semi-Supervised Learning: Multiple Sclerosis Segmentation in 7T-MRI from large-scale 3T-MRI

COLLATOR: Consistent spatial-temporal longitudinal atlas construction via implicit neural representation

Contrastive learning of global and local features for medical image segmentation with limited annotations

3D Self-Supervised Methods for Medical Imaging

Keypoint-Augmented Self-Supervised Learning for Medical Image Segmentation with Limited Annotation

Spatial Context-Aware Self-Attention Model For Multi-Organ Segmentation