Deep‐Learning‐Based Neural Tissue Segmentation of MRI in Multiple Sclerosis: Effect of Training Set Size

Abstract

BACKGROUND: The dependence of deep-learning (DL)-based segmentation accuracy of brain MRI on the training size is not known. PURPOSE: To determine the required training size for a desired accuracy in brain MRI segmentation in multiple sclerosis (MS) using DL. STUDY TYPE: Retrospective analysis of MRI data acquired as part of a multicenter clinical trial. STUDY POPULATION: In all, 1008 patients with clinically definite MS. FIELD STRENGTH/SEQUENCE: -weighted turbo spin echo sequences. ASSESSMENT: Segmentation results using an automated analysis pipeline and validated by two neuroimaging experts served as the ground truth. A DL model, based on a fully convolutional neural network, was trained separately using 16 different training sizes. The segmentation accuracy as a function of the training size was determined. These data were fitted to the learning curve for estimating the required training size for desired accuracy. STATISTICAL TESTS: The performance of the network was evaluated by calculating the Dice similarity coefficient (DSC), and lesion true-positive and false-positive rates. RESULTS: lesions, 0.87 ± 009 to 0.94 ± 0.004 for GM, 0.86 ± 0.08 to 0.94 ± 0.005 for WM, and 0.91 ± 0.009 to 0.96 ± 0.003 for CSF. DATA CONCLUSION: Excellent segmentation was achieved with a training size as small as 10 image volumes for GM, WM, and CSF. In contrast, a training size of at least 50 image volumes was necessary for adequate lesion segmentation. LEVEL OF EVIDENCE: 1 Technical Efficacy Stage: 1 J. Magn. Reson. Imaging 2020;51:1487-1496.

Ähnliche Arbeiten