Lightweight U-Net for Lesion Segmentation in Ultrasound Images - Inria - Institut national de recherche en sciences et technologies du numérique Access content directly
Conference Papers Year : 2021

Lightweight U-Net for Lesion Segmentation in Ultrasound Images

Abstract

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
final_submission_Lightweight U-net for lesion segmentation in ultrasound images.pdf (390.56 Ko) Télécharger le fichier
Origin Files produced by the author(s)

Dates and versions

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

Identifiers

  • HAL Id : hal-03140737 , version 2

Cite

Yingping Li, Emilie Chouzenoux, Benoit Charmettant, Baya Benatsou, Jean-Philippe Lamarque, et al.. Lightweight U-Net for Lesion Segmentation in Ultrasound Images. ISBI 2021 - IEEE International Symposium on Biomedical Imaging, Apr 2021, Nice / Virtual, France. ⟨hal-03140737v2⟩
172 View
684 Download

Share

Gmail Mastodon Facebook X LinkedIn More