6 February 2021 Investigating efficient CNN architecture for multiple sclerosis lesion segmentation
Alexandre Fenneteau, Pascal Bourdon, David Helbert, Christine Fernandez-Maloigne, Christophe Habas, Rémy Guillevin
Author Affiliations +
Abstract

Purpose: The automatic segmentation of multiple sclerosis lesions in magnetic resonance imaging has the potential to reduce radiologists’ efforts on a daily time-consuming task and to bring more reproducibility. Almost all new segmentation techniques make use of convolutional neural networks with their own different architecture. Architectural choices are rarely explained. We aimed at presenting the relevance of a U-net-like architecture for our specific task and at building an efficient and simple model.

Approach: An experimental study was performed by observing the impact of applying different mutations and deletions to a simple U-net-like architecture.

Results: The power of the U-net architecture is explained by the joint benefits of using an encoder–decoder architecture and by linking them with long skip connections. Augmenting the number of convolutional layers and decreasing the number of feature maps allowed us to build an exceptionally light and competitive architecture, the minimally parameterized U-net (MPU-net), with only ∼30,000 parameters.

Conclusion: The empirical study of the U-net has led to a better understanding of its architecture. It has guided the building of the MPU-net, a model far less parameterized than others (at least by a factor of seven). This neural network achieves a human-level segmentation of multiple sclerosis lesions on fluid-attenuated inversion recovery images only. It shows that this segmentation task does not necessitate overly complicated models to be achieved. This gives the opportunity to build more explainable models that can help such methods to be adopted in a clinical environment.

© 2021 Society of Photo-Optical Instrumentation Engineers (SPIE) 2329-4302/2021/$28.00 © 2021 SPIE
Alexandre Fenneteau, Pascal Bourdon, David Helbert, Christine Fernandez-Maloigne, Christophe Habas, and Rémy Guillevin "Investigating efficient CNN architecture for multiple sclerosis lesion segmentation," Journal of Medical Imaging 8(1), 014504 (6 February 2021). https://doi.org/10.1117/1.JMI.8.1.014504
Received: 19 June 2020; Accepted: 11 January 2021; Published: 6 February 2021
Lens.org Logo
CITATIONS
Cited by 8 scholarly publications.
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Image segmentation

Convolution

Neural networks

Brain

Computer programming

Magnetic resonance imaging

Medical imaging

Back to Top