Boosting multiple sclerosis lesion segmentation through attention mechanism

Abstract

Magnetic resonance imaging is a fundamental tool to reach a diagnosis of multiple sclerosis and monitoring its progression. Although several attempts have been made to segment multiple sclerosis lesions using artificial intelligence, fully automated analysis is not yet available. State-of-the-art methods rely on slight variations in segmentation architectures (e.g. U-Net, etc.). However, recent research has demonstrated how exploiting temporal-aware features and attention mechanisms can provide a significant boost to traditional architectures. This paper proposes a framework that exploits an augmented U-Net architecture with a convolutional long short-term memory layer and attention mechanism which is able to segment and quantify multiple sclerosis lesions detected in magnetic resonance images. Quantitative and qualitative evaluation on challenging examples demonstrated how the method outperforms previous state-of-the-art approaches, reporting an overall Dice score of 89% and also demonstrating robustness and generalization ability on never seen new test samples of a new dedicated under construction dataset.

References in corpus (7)

• Convolutional LSTM Network: A Machine Learning Approach for Precipitation Nowcasting
• An overview of gradient descent optimization algorithms
• Multi-branch Convolutional Neural Network for Multiple Sclerosis Lesion Segmentation
• Applications of Deep Learning Techniques for Automated Multiple Sclerosis Detection Using Magnetic Resonance Imaging: A Review
• DeepLesionBrain: Towards a broader deep-learning generalization for multiple sclerosis lesion segmentation
• Multiple Sclerosis Lesion Segmentation from Brain MRI via Fully Convolutional Neural Networks
• Ensemble CNN and Uncertainty Modeling to Improve Automatic Identification/Segmentation of Multiple Sclerosis Lesions in Magnetic Resonance Imaging

Cited by in corpus (3)

• Early detection of hip periprosthetic joint infections through CNN on Computed Tomography images
• UNISELF: A Unified Network with Instance Normalization and Self-Ensembled Lesion Fusion for Multiple Sclerosis Lesion Segmentation
• Attention-Based Segmentation of WMHs and Differentiation of Vascular vs. Demyelinating Lesions