ALL-Net: Anatomical information lesion-wise loss function integrated into neural network for multiple sclerosis lesion segmentation

•A new algorithm, ALL-Net, is introduced to improve MS lesion segmentation.•ALL-Net integrates anatomical coordinate information into the neural network.•ALL-Net integrates lesion-wise loss function to improve small lesion detection.•ALL-Net is robust to both small- and large-scale MS lesion dataset...

Celý popis

Uloženo v:
Podrobná bibliografie
Vydáno v:NeuroImage clinical Ročník 32; s. 102854
Hlavní autoři: Zhang, Hang, Zhang, Jinwei, Li, Chao, Sweeney, Elizabeth M., Spincemaille, Pascal, Nguyen, Thanh D., Gauthier, Susan A., Wang, Yi, Marcille, Melanie
Médium: Journal Article
Jazyk:angličtina
Vydáno: Netherlands Elsevier Inc 01.01.2021
Elsevier
Témata:
ISSN:2213-1582, 2213-1582
On-line přístup:Získat plný text
Tagy: Přidat tag
Žádné tagy, Buďte první, kdo vytvoří štítek k tomuto záznamu!
Popis
Shrnutí:•A new algorithm, ALL-Net, is introduced to improve MS lesion segmentation.•ALL-Net integrates anatomical coordinate information into the neural network.•ALL-Net integrates lesion-wise loss function to improve small lesion detection.•ALL-Net is robust to both small- and large-scale MS lesion datasets. Accurate detection and segmentation of multiple sclerosis (MS) brain lesions on magnetic resonance images are important for disease diagnosis and treatment. This is a challenging task as lesions vary greatly in size, shape, location, and image contrast. The objective of our study was to develop an algorithm based on deep convolutional neural network integrated with anatomic information and lesion-wise loss function (ALL-Net) for fast and accurate automated segmentation of MS lesions. Distance transformation mapping was used to construct a convolutional module that encoded lesion-specific anatomical information. To overcome the lesion size imbalance during network training and improve the detection of small lesions, a lesion-wise loss function was developed in which individual lesions were modeled as spheres of equal size. On the ISBI-2015 longitudinal MS lesion segmentation challenge dataset (19 subjects in total), ALL-Net achieved an overall score of 93.32 and was amongst the top performing methods. On the larger Cornell MS dataset (176 subjects in total), ALL-Net significantly improved both voxel-wise metrics (Dice improvement of 3.9% to 35.3% with p-values ranging from p < 0.01 to p < 0.0001, and AUC of voxel-wise precision-recall curve improvement of 2.1% to 29.8%) and lesion-wise metrics (lesion-wise F1 score improvement of 12.6% to 29.8% with all p-values p < 0.0001, and AUC of lesion-wise ROC curve improvement of 1.4% to 20.0%) compared to leading publicly available MS lesion segmentation tools.
Bibliografie:ObjectType-Article-1
SourceType-Scholarly Journals-1
ObjectType-Feature-2
content type line 23
ISSN:2213-1582
2213-1582
DOI:10.1016/j.nicl.2021.102854