Models Genesis: Generic Autodidactic Models for 3D Medical Image Analysis

Transfer learning from natural image to medical image has established as one of the most practical paradigms in deep learning for medical image analysis. However, to fit this paradigm, 3D imaging tasks in the most prominent imaging modalities (e.g., CT and MRI) have to be reformulated and solved in 2D, losing rich 3D anatomical information and inevitably compromising the performance. To overcome this limitation, we have built a set of models, called Generic Autodidactic Models, nicknamed Models Genesis, because they are created ex nihilo (with no manual labeling), self-taught (learned by self-supervision), and generic (served as source models for generating application-specific target models). Our extensive experiments demonstrate that our Models Genesis significantly outperform learning from scratch in all five target 3D applications covering both segmentation and classification. More importantly, learning a model from scratch simply in 3D may not necessarily yield performance better than transfer learning from ImageNet in 2D, but our Models Genesis consistently top any 2D approaches including fine-tuning the models pre-trained from ImageNet as well as fine-tuning the 2D versions of our Models Genesis, confirming the importance of 3D anatomical information and significance of our Models Genesis for 3D medical imaging. This performance is attributed to our unified self-supervised learning framework, built on a simple yet powerful observation: the sophisticated yet recurrent anatomy in medical images can serve as strong supervision signals for deep models to learn common anatomical representation automatically via self-supervision. As open science, all pre-trained Models Genesis are available at https://github.com/MrGiovanni/ModelsGenesis.

PDF Abstract
Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Brain Tumor Segmentation BRATS-2013 ModelGenesis Dice Score 0.9258 # 2
Lung Nodule Segmentation LIDC-IDRI ModelGenesis IoU 77.62 # 1
Dice 75.86 # 1
Liver Segmentation LiTS2017 ModelGenesis IoU 79.52 # 4
Dice 91.13 # 4
Lung Nodule Detection LUNA2016 FPRED ModelGenesis AUC 98.2 # 2
Medical Image Segmentation Medical Segmentation Decathlon Models Genesis Dice (Average) 76.97 # 4
NSD 87.19 # 5
Pulmonary Embolism Detection PE-CAD FPRED ModelGenesis AUC 88.04 # 1

Methods


No methods listed for this paper. Add relevant methods here