Robust Alzheimer's Progression Modeling using Cross-Domain Self-Supervised Deep Learning

Published: 05 Aug 2023, Last Modified: 05 Aug 2023Accepted by TMLREveryoneRevisionsBibTeX
Abstract: Developing successful artificial intelligence systems in practice depends on both robust deep learning models and large, high-quality data. However, acquiring and labeling data can be prohibitively expensive and time-consuming in many real-world applications, such as clinical disease models. Self-supervised learning has demonstrated great potential in increasing model accuracy and robustness in small data regimes. In addition, many clinical imaging and disease modeling applications rely heavily on regression of continuous quantities. However, the applicability of self-supervised learning for these medical-imaging regression tasks has not been extensively studied. In this study, we develop a cross-domain self-supervised learning approach for disease prognostic modeling as a regression problem using medical images as input. We demonstrate that self-supervised pretraining can improve the prediction of Alzheimer's Disease progression from brain MRI. We also show that pretraining on extended (but not labeled) brain MRI data outperforms pretraining on natural images. We further observe that the highest performance is achieved when both natural images and extended brain-MRI data are used for pretraining.
License: Creative Commons Attribution 4.0 International (CC BY 4.0)
Submission Length: Regular submission (no more than 12 pages of main content)
Changes Since Last Submission: - Methodology, Introduction and Result sections have been revised per reviewer's comments. - Two extra tables were added to the manuscript per reviewer's comments. - Figure 1 and Figure 2 were modified per reviewer's comments. - More details added to the Discussion section per reviewer's comments.
Assigned Action Editor: ~Nicolas_THOME2
Submission Number: 834