Lightweight U-Net for Lesion Segmentation in Ultrasound Images - Archive ouverte HAL
Communication Dans Un Congrès Année : 2021

Lightweight U-Net for Lesion Segmentation in Ultrasound Images

Résumé

Acquiring ultrasound images of suspected lesion areas allows radiologists to monitor the cancer development of patients. The goal of this paper is to provide an automatic lesion segmentation tool for assisting them on the analysis of ultrasound images, by relying on recent neural network methods. Specifically, we perform a comparative study for the segmentation of 348 ultrasound image pairs acquired in 19 centers across France, displaying different tumor types. We show that, with a careful hyperparameter tuning, U-net outperforms other state-of-the-art networks, reaching a Dice coefficient of 0.929. We then propose to introduce group convolution into U-net architecture. This leads to a lightweight network named Lighter U-net @128 that achieves comparable segmentation performance with obviously reduced model size, hence paving the way for an embedded integration within hospital environment. We made our code publicly available for reproducibility purpose.
Fichier principal
Vignette du fichier
1_Ultrasound_Image_Segmentation_ISBI2021.pdf (390.47 Ko) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-03140737 , version 1 (13-02-2021)
hal-03140737 , version 2 (02-03-2021)

Identifiants

  • HAL Id : hal-03140737 , version 1

Citer

Yingping Li, Emilie Chouzenoux, Benoit Charmettant, Baya Benatsou, Jean-Philippe Lamarque, et al.. Lightweight U-Net for Lesion Segmentation in Ultrasound Images. IEEE International Symposium on Biomedical Imaging (ISBI 2021), 2021, Nice (virtual), France. ⟨hal-03140737v1⟩
201 Consultations
763 Téléchargements

Partager

More