Segmenting Hepatic Lesions Using Residual Attention U-Net with an Adaptive Weighted Dice Loss

Yu Cheng Liu, Daniel Stanley Tan, Jyh Cheng Chen, Wen Huang Cheng, Kai Lung Hua

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

25 Scopus citations

Abstract

We propose a novel network architecture called Residual Attention U-Net (ResAttU-Net) for segmenting hepatic lesions. Our model incorporates residual blocks that can extract more complex features as compared with traditional convolutional layers combined with a skip-connection attention module that learns to focus on the relevant features for the task of hepatic lesions segmentation. Moreover, we train our model using an adaptive weighted dice loss that prioritizes the pixels of the tumor class over the pixels of the background class. We evaluate our model on the MICCAI Liver Tumor Segmentation (LiTS) benchmark dataset. Our experimental results show that our method significantly improves upon several state-of-the-art baselines for hepatic lesion or liver tumor segmentation.

Original languageEnglish
Title of host publication2019 IEEE International Conference on Image Processing, ICIP 2019 - Proceedings
PublisherIEEE Computer Society
Pages3322-3326
Number of pages5
ISBN (Electronic)9781538662496
DOIs
StatePublished - Sep 2019
Event26th IEEE International Conference on Image Processing, ICIP 2019 - Taipei, Taiwan
Duration: 22 Sep 201925 Sep 2019

Publication series

NameProceedings - International Conference on Image Processing, ICIP
Volume2019-September
ISSN (Print)1522-4880

Conference

Conference26th IEEE International Conference on Image Processing, ICIP 2019
Country/TerritoryTaiwan
CityTaipei
Period22/09/1925/09/19

Keywords

  • attention module
  • CT image segmentation
  • hepatic lesion factor
  • residual block

Fingerprint

Dive into the research topics of 'Segmenting Hepatic Lesions Using Residual Attention U-Net with an Adaptive Weighted Dice Loss'. Together they form a unique fingerprint.

Cite this