CNN for multiple sclerosis lesion segmentation: How many patients for a fully supervised method?

Alexandre Fenneteau (1, 2, 3, 4) , Pascal Bourdon (1, 2, 3) , David Helbert (1, 2, 3) , Christine Fernandez-Maloigne (1, 2, 3) , Christophe N Habas (3, 5, 6) , Rémy Guillevin (3, 7, 8)

1 XLIM-ASALI - Synthèse et analyse d'images
2 UP - Université de Poitiers = University of Poitiers
3 I3M - Laboratoire commun Imagerie Métabolique Multi-Noyaux Multi-Organes
4 Siemens Healthcare
5 CHNO - Centre Hospitalier National d'Ophtalmologie des Quinze-Vingts
6 UVSQ - Université de Versailles Saint-Quentin-en-Yvelines
7 CHU de Poitiers [La Milétrie] - Centre hospitalier universitaire de Poitiers = Poitiers University Hospital
8 DACTIM-MIS - Data Analysis and Computations Through Imaging Modeling-Mathématiques, Imagerie, Santé

In this study we propose to improve an existing artificial neural network architecture, the MPU-net, which is designed for having very few parameters for multiple sclerosis lesion segmentation on magnetic resonance images. With this improved architecture we conducted a study to assess the influence of the number of training examples on the model performance and generalization. The question behind this study is: "With an appropriate architecture, how many patients do we need?". We evaluated 9 different adaptations of the MPU-net architecture. Then, after the selection of the best architecture we learned the model multiple times with different numbers of patients and assessed its performances. The addition of deep supervision, the reduction of number of convolutional layers and the addition of regularization layers produced a more stable and performant architecture. Learnings of selected model with only 10 exams delivered performances equivalent to learnings with 23 exams. So, in our experimental setup, it is possible to learn a performant model with only 10 fully annotated examples.

Domaines: Traitement du signal et de l'image [eess.SP]

Alexandre Fenneteau, Pascal Bourdon, David Helbert, Christine Fernandez-Maloigne, Christophe N Habas, et al.. CNN for multiple sclerosis lesion segmentation: How many patients for a fully supervised method?. International Conference on Advances in Biomedical Engineering, Oct 2021, Wardanyeh, Lebanon. pp.30-33, ⟨10.1109/ICABME53305.2021.9604859⟩. ⟨hal-03335735⟩