2022 | OriginalPaper | Chapter
Diffusion MRI Specific Pretraining by Self-supervision on an Auxiliary Dataset
Authors : Leon Weninger, Jarek Ecke, Chuh-Hyoun Na, Kerstin Jütten, Dorit Merhof
Published in: Bildverarbeitung für die Medizin 2022
Publisher: Springer Fachmedien Wiesbaden
Activate our intelligent search to find suitable subject content or patents.
Select sections of text to find matching patents with Artificial Intelligence. powered by
Select sections of text to find additional relevant content using AI-assisted search. powered by
Training deep learning networks is very data intensive. Especially in fields with a very limited number of annotated datasets, such as diffusion MRI, it is of great importance to develop approaches that can cope with a limited amount of data. It was previously shown that transfer learning can lead to better results and more stable training in various medical applications. However, the use of off-the-shelf transfer learning tools in high angular resolution diffusion MRI is not straightforward, as such 3D approaches are commonly designed for scalar data. Here, an extension of self-supervised pretraining to diffusion MRI data is presented, and enhanced with a modality-specific procedure, where artifacts encountered in diffusion MRI need to be removed. We pretrained on publicly available data from the Human Connectome Project and evaluated the success on data from a local hospital with three modality-related experiments: segmentation of brain microstructure, detection of fiber crossings, and regression of nerve fiber spatial orientation. The results were compared against a setting without pretraining, and against classical autoencoder pretraining. We find that it is possible to achieve both improved metrics and a more stable training with the proposed diffusion MRI specific pretraining procedure.