Title
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(1), for reproducibility purpose.
Year
DOI
Venue
2021
10.1109/ISBI48211.2021.9434086
2021 IEEE 18TH INTERNATIONAL SYMPOSIUM ON BIOMEDICAL IMAGING (ISBI)
Keywords
DocType
ISSN
Ultrasound images, lesion segmentation, U-net, lightweight network
Conference
1945-7928
Citations 
PageRank 
References 
0
0.34
0
Authors
6
Name
Order
Citations
PageRank
Yingping Li100.34
Emilie Chouzenoux220226.37
Benoit Charmettant300.34
Baya Benatsou400.34
Jean-Philippe Lamarque500.68
Nathalie Lassau600.34