Residential College | false |
Status | 已發表Published |
Shift-channel attention and weighted-region loss function for liver and dense tumor segmentation | |
Li, Jiajian1; Huang, Guoheng1; He, Junlin1; Chen, Ziyang1; Pun, Chi Man2; Yu, Zhiwen3; Ling, Wing Kuen4; Liu, Lizhi5; Zhou, Jian5; Huang, Jinhua6 | |
2022-06 | |
Source Publication | Medical Physics |
ISSN | 0094-2405 |
Volume | 49Issue:11Pages:7193-7206 |
Abstract | Purpose: To assist physicians in the diagnosis and treatment planning of tumor, a robust and automatic liver and tumor segmentation method is highly demanded in the clinical practice. Recently, numerous researchers have improved the segmentation accuracy of liver and tumor by introducing multiscale contextual information and attention mechanism. However, this tends to introduce more training parameters and suffer from a heavier computational burden. In addition, the tumor has various sizes, shapes, locations, and numbers, which is the main reason for the poor accuracy of automatic segmentation. Although current loss functions can improve the learning ability of the model for hard samples to a certain extent, these loss functions are difficult to optimize the segmentation effect of small tumor regions when the large tumor regions in the sample are in the majority. Methods: We propose a Liver and Tumor Segmentation Network (LiTS-Net) framework. First, the Shift-Channel Attention Module (S-CAM) is designed to model the feature interdependencies in adjacent channels and does not require additional training parameters. Second, the Weighted-Region (WR) loss function is proposed to emphasize the weight of small tumors in dense tumor regions and reduce the weight of easily segmented samples. Moreover, the Multiple 3D Inception Encoder Units (MEU) is adopted to capture the multiscale contextual information for better segmentation of liver and tumor. Results: Efficacy of the LiTS-Net is demonstrated through the public dataset of MICCAI 2017 Liver Tumor Segmentation (LiTS) challenge, with Dice per case of 96.9% and 75.1%, respectively. For the 3D Image Reconstruction for Comparison of Algorithm and DataBase (3Dircadb), Dices are 96.47% for the liver and 74.54% for tumor segmentation. The proposed LiTS-Net outperforms existing state-of-the-art networks. Conclusions: We demonstrated the effectiveness of LiTS-Net and its core components for liver and tumor segmentation. The S-CAM is designed to model the feature interdependencies in the adjacent channels, which is characterized by no need to add additional training parameters. Meanwhile, we conduct an in-depth study of the feature shift proportion of adjacent channels to determine the optimal shift proportion. In addition, the WR loss function can implicitly learn the weights among regions without the need to manually specify the weights. In dense tumor segmentation tasks, WR aims to enhance the weights of small tumor regions and alleviate the problem that small tumor segmentation is difficult to optimize further when large tumor regions occupy the majority. Last but not least, our proposed method outperforms other state-of-the-art methods on both the LiTS dataset and the 3Dircadb dataset. |
Keyword | Class Imbalance Learning Multiscale Contextual Information Shift-channel Attention Weighted-region Loss Function |
DOI | 10.1002/mp.15816 |
URL | View the original |
Indexed By | SCIE |
Language | 英語English |
WOS Research Area | Radiology, Nuclear Medicine & Medical Imaging |
WOS Subject | Radiology, Nuclear Medicine & Medical Imaging |
WOS ID | WOS:000821189300001 |
Scopus ID | 2-s2.0-85134346707 |
Fulltext Access | |
Citation statistics | |
Document Type | Journal article |
Collection | DEPARTMENT OF COMPUTER AND INFORMATION SCIENCE |
Corresponding Author | Huang, Guoheng; Liu, Lizhi; Zhou, Jian; Huang, Jinhua |
Affiliation | 1.School of Computer Science and Technology, Guangdong University of Technology, Guangzhou, China 2.Department of Computer and Information Science, University of Macau, Macao 3.School of Computer Science and Engineering, South China University of Technology, Guangzhou, China 4.School of Information Engineering, Guangdong University of Technology, Guangzhou, China 5.Department of Medical Imaging, Sun Yat-sen University Cancer Center, State Key Laboratory of Oncology in South China, Collaborative Innovation Center for Cancer Medicine, Guangzhou, China 6.Department of Minimal Invasive Interventional Therapy, Sun Yat-sen University Cancer Center, State Key Laboratory of Oncology in South China, Collaborative Innovation Center for Cancer Medicine, Guangzhou, China |
Recommended Citation GB/T 7714 | Li, Jiajian,Huang, Guoheng,He, Junlin,et al. Shift-channel attention and weighted-region loss function for liver and dense tumor segmentation[J]. Medical Physics, 2022, 49(11), 7193-7206. |
APA | Li, Jiajian., Huang, Guoheng., He, Junlin., Chen, Ziyang., Pun, Chi Man., Yu, Zhiwen., Ling, Wing Kuen., Liu, Lizhi., Zhou, Jian., & Huang, Jinhua (2022). Shift-channel attention and weighted-region loss function for liver and dense tumor segmentation. Medical Physics, 49(11), 7193-7206. |
MLA | Li, Jiajian,et al."Shift-channel attention and weighted-region loss function for liver and dense tumor segmentation".Medical Physics 49.11(2022):7193-7206. |
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