$$\alpha $$ -UNet++: A Data-Driven Neural Network Architecture for Medical Image Segmentation

Yaxin Chen, Benteng Ma, Yong Xia

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

4 Scopus citations

Abstract

UNet++, an encoder-decoder architecture constructed based on the famous UNet, has achieved state-of-the-art results on many medical image segmentation tasks. Despite improved performance, UNet++ introduces densely connected decoding blocks, some of which, however, are redundant for a specific task. In this paper, we propose-UNet++ that allows us to automatically identify and discard redundant decoding blocks without the loss of precision. To this end, we design an auxiliary indicator function layer to compress the network architecture via removing a decoding block, in which all individual responses are less than a given threshold. We evaluated the segmentation architecture obtained respectively for liver segmentation and nuclei segmentation, denoted by UNet++, against UNet and UNet++. Comparing to UNet++, our UNet++ reduces the parameters by 18.89% in liver segmentation and 34.17% in nuclei segmentation, yielding an average improvement of IoU by 0.27% and 0.11% on two tasks. Our results suggest that the UNet++ produced by the proposed-UNet++ not only improves the segmentation accuracy slightly but also reduces the model complexity considerably.

Original languageEnglish
Title of host publicationDomain Adaptation and Representation Transfer, and Distributed and Collaborative Learning - 2nd MICCAI Workshop, DART 2020, and 1st MICCAI Workshop, DCL 2020, Held in Conjunction with MICCAI 2020, Proceedings
EditorsShadi Albarqouni, Spyridon Bakas, Konstantinos Kamnitsas, M. Jorge Cardoso, Bennett Landman, Wenqi Li, Fausto Milletari, Nicola Rieke, Holger Roth, Daguang Xu, Ziyue Xu
PublisherSpringer Science and Business Media Deutschland GmbH
Pages3-12
Number of pages10
ISBN (Print)9783030605476
DOIs
StatePublished - 2020
Event2nd MICCAI Workshop on Domain Adaptation and Representation Transfer, DART 2020, and the 1st MICCAI Workshop on Distributed and Collaborative Learning, DCL 2020, held in conjunction with the Medical Image Computing and Computer Assisted Intervention, MICCAI 2020 - Lima, Peru
Duration: 4 Oct 20208 Oct 2020

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume12444 LNCS
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

Conference2nd MICCAI Workshop on Domain Adaptation and Representation Transfer, DART 2020, and the 1st MICCAI Workshop on Distributed and Collaborative Learning, DCL 2020, held in conjunction with the Medical Image Computing and Computer Assisted Intervention, MICCAI 2020
Country/TerritoryPeru
CityLima
Period4/10/208/10/20

Keywords

  • Auxiliary indicator function
  • Medical image segmentation
  • Network compression
  • UNet++

Fingerprint

Dive into the research topics of '$$\alpha $$ -UNet++: A Data-Driven Neural Network Architecture for Medical Image Segmentation'. Together they form a unique fingerprint.

Cite this