Residential College | false |
Status | 已發表Published |
Uncertainty-aware Hierarchical Aggregation Network for Medical Image Segmentation | |
Zhou, Tao1; Zhou, Yi2; Li, Guangyu1; Chen, Geng3; Shen, Jianbing4 | |
2024-02 | |
Source Publication | IEEE Transactions on Circuits and Systems for Video Technology |
ISSN | 1051-8215 |
Volume | 34Issue:8Pages:7440 - 7453 |
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. |
Keyword | Biomedical Imaging Feature Extraction Fuses Hierarchical Feature Fusion Image Segmentation Lesions Medical Diagnostic Imaging Medical Image Segmentation Scale Aggregation Module Semantics Uncertainty-induced Cross-level Fusion |
DOI | 10.1109/TCSVT.2024.3370685 |
URL | View the original |
Indexed By | SCIE |
Language | 英語English |
WOS Research Area | Engineering |
WOS Subject | Engineering, Electrical & Electronic |
WOS ID | WOS:001327614800062 |
Publisher | IEEE-INST ELECTRICAL ELECTRONICS ENGINEERS INC445 HOES LANE, PISCATAWAY, NJ 08855-4141 |
Scopus ID | 2-s2.0-85186995244 |
Fulltext Access | |
Citation statistics | |
Document Type | Journal article |
Collection | Faculty of Science and Technology THE STATE KEY LABORATORY OF INTERNET OF THINGS FOR SMART CITY (UNIVERSITY OF MACAU) DEPARTMENT OF COMPUTER AND INFORMATION SCIENCE |
Corresponding Author | Li, Guangyu |
Affiliation | 1.PCA Lab, the Key Laboratory of Intelligent Perception and Systems for High-Dimensional Information of Ministry of Education, and the School of Computer Science and Engineering, Nanjing University of Science and Technology, Nanjing, China 2.School of Computer Science and Engineering, Southeast University, Nanjing, China 3.School of Computer Science and Engineering, National Engineering Laboratory for Integrated Aero-Space-Ground-Ocean Big Data Application Technology, Northwestern Polytechnical University, Xi‘an, China 4.Department of Computer and Information Science, State Key Laboratory of Internet of Things for Smart City, University of Macau, Macau, Macau, China |
Recommended Citation GB/T 7714 | Zhou, Tao,Zhou, Yi,Li, Guangyu,et al. Uncertainty-aware Hierarchical Aggregation Network for Medical Image Segmentation[J]. IEEE Transactions on Circuits and Systems for Video Technology, 2024, 34(8), 7440 - 7453. |
APA | Zhou, Tao., Zhou, Yi., Li, Guangyu., Chen, Geng., & Shen, Jianbing (2024). Uncertainty-aware Hierarchical Aggregation Network for Medical Image Segmentation. IEEE Transactions on Circuits and Systems for Video Technology, 34(8), 7440 - 7453. |
MLA | Zhou, Tao,et al."Uncertainty-aware Hierarchical Aggregation Network for Medical Image Segmentation".IEEE Transactions on Circuits and Systems for Video Technology 34.8(2024):7440 - 7453. |
Files in This Item: | There are no files associated with this item. |
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment