Uncertainty-Aware Hierarchical Aggregation Network for Medical Image Segmentation

Tao Zhou, Yi Zhou, Guangyu Li, Geng Chen, Jianbing Shen

Research output: Contribution to journalArticlepeer-review

17 Scopus citations

Abstract

Medical image segmentation is an essential process to assist clinics with computer-aided diagnosis and treatment. Recently, a large amount of convolutional neural network (CNN)-based methods have been rapidly developed and achieved remarkable performances in several different medical image segmentation tasks. However, the same type of infected region or lesions often has a diversity of scales, making it a challenging task to achieve accurate medical image segmentation. In this paper, we present a novel Uncertainty-aware Hierarchical Aggregation Network, namely UHA-Net, for medical image segmentation, which can fully make utilization of cross-level and multi-scale features to handle scale variations. Specifically, we propose a hierarchical feature fusion (HFF) module to aggregate high-level features, which is used to produce a global map for the coarse localization of the segmented target. Then, we propose an uncertainty-induced cross-level fusion (UCF) module to fully fuse features from the adjacent levels, which can learn knowledge guidance to capture the contextual information from adjacent resolutions. Further, a scale aggregation module (SAM) is presented to learn multi-scale features by using different convolution kernels, to effectively deal with scale variations. At last, we formulate a unified framework to simultaneously fuse inter-layer convolutional features and learn the discriminability of multi-scale representations from the intra-layer features, leading to accurate segmentation results. We carry out experiments on three different medical image segmentation tasks, and the results demonstrate that our UHA-Net outperforms state-of-the-art segmentation methods. Our implementation code and segmentation maps will be publicly at https://github.com/taozh2017/UHANet.

Original languageEnglish
Pages (from-to)7440-7453
Number of pages14
JournalIEEE Transactions on Circuits and Systems for Video Technology
Volume34
Issue number8
DOIs
StatePublished - 2024

Keywords

  • hierarchical feature fusion
  • Medical image segmentation
  • scale aggregation module
  • uncertainty-induced cross-level fusion

Fingerprint

Dive into the research topics of 'Uncertainty-Aware Hierarchical Aggregation Network for Medical Image Segmentation'. Together they form a unique fingerprint.

Cite this