1
|
He J, Zhang M, Li W, Peng Y, Fu B, Liu C, Wang J, Wang R. SaB-Net: Self-attention backward network for gastric tumor segmentation in CT images. Comput Biol Med 2024; 169:107866. [PMID: 38134751 DOI: 10.1016/j.compbiomed.2023.107866] [Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Abstract] [Key Words] [MESH Headings] [Track Full Text] [Journal Information] [Subscribe] [Scholar Register] [Received: 08/31/2023] [Revised: 11/04/2023] [Accepted: 12/17/2023] [Indexed: 12/24/2023]
Abstract
Gastric cancer is a significant contributor to cancer-related fatalities globally. The automated segmentation of gastric tumors has the potential to analyze the medical condition of patients and enhance the likelihood of surgical treatment success. However, the development of an automatic solution is challenged by the heterogeneous intensity distribution of gastric tumors in computed tomography (CT) images, the low-intensity contrast between organs, and the high variability in the stomach shapes and gastric tumors in different patients. To address these challenges, we propose a self-attention backward network (SaB-Net) for gastric tumor segmentation (GTS) in CT images by introducing a self-attention backward layer (SaB-Layer) to feed the self-attention information learned at the deep layer back to the shallow layers. The SaB-Layer efficiently extracts tumor information from CT images and integrates the information into the network, thereby enhancing the network's tumor segmentation ability. We employed datasets from two centers, one for model training and testing and the other for external validation. The model achieved dice scores of 0.8456 on the test set and 0.8068 on the external verification set. Moreover, we validated the model's transfer learning ability on a publicly available liver cancer dataset, achieving results comparable to state-of-the-art liver cancer segmentation models recently developed. SaB-Net has strong potential for assisting in the clinical diagnosis of and therapy for gastric cancer. Our implementation is available at https://github.com/TyrionJ/SaB-Net.
Collapse
Affiliation(s)
- Junjie He
- Engineering Research Center of Text Computing & Cognitive Intelligence, Ministry of Education, Key Laboratory of Intelligent Medical Image Analysis and Precise Diagnosis of Guizhou Province, State Key Laboratory of Public Big Data, College of Computer Science and Technology, Guizhou University, No. 2288, Huaxi Avenue, Guiyang, 550025, Guizhou, China; Department of Radiology, International Exemplary Cooperation Base of Precision Imaging for Diagnosis and Treatment, Guizhou Provincial People's Hospital, No. 83, Zhongshan Dong Road, Guiyang, 550002, Guizhou, China
| | - Mudan Zhang
- Department of Radiology, International Exemplary Cooperation Base of Precision Imaging for Diagnosis and Treatment, Guizhou Provincial People's Hospital, No. 83, Zhongshan Dong Road, Guiyang, 550002, Guizhou, China
| | - Wuchao Li
- Department of Radiology, International Exemplary Cooperation Base of Precision Imaging for Diagnosis and Treatment, Guizhou Provincial People's Hospital, No. 83, Zhongshan Dong Road, Guiyang, 550002, Guizhou, China
| | - Yunsong Peng
- Department of Radiology, International Exemplary Cooperation Base of Precision Imaging for Diagnosis and Treatment, Guizhou Provincial People's Hospital, No. 83, Zhongshan Dong Road, Guiyang, 550002, Guizhou, China
| | - Bangkang Fu
- Department of Radiology, International Exemplary Cooperation Base of Precision Imaging for Diagnosis and Treatment, Guizhou Provincial People's Hospital, No. 83, Zhongshan Dong Road, Guiyang, 550002, Guizhou, China
| | - Chen Liu
- Department of Radiology, Southwest Hospital, Army Medical University (Third Military Medical University), No. 30, Gao Tan Yan Street, 400038, Chongqing, China
| | - Jian Wang
- Department of Radiology, Southwest Hospital, Army Medical University (Third Military Medical University), No. 30, Gao Tan Yan Street, 400038, Chongqing, China
| | - Rongpin Wang
- Department of Radiology, International Exemplary Cooperation Base of Precision Imaging for Diagnosis and Treatment, Guizhou Provincial People's Hospital, No. 83, Zhongshan Dong Road, Guiyang, 550002, Guizhou, China.
| |
Collapse
|