机构:[1]Glasgow College, University of Electronic Science and Technology of China, Chengdu, 611731, China.[2]Department of Radiology, Peking University People's Hospital, Beijing, 100044, China.[3]School of Mechanical and Electrical Engineering, University of Electronic Science and Technology of China, Chengdu, 611731, China.[4]School of Automation Engineering, University of Electronic Science and Technology of China, Chengdu, 611731, China.[5]Department of Radiation Oncology, Sichuan Cancer Hospital and Institute, University of Electronic Science and Technology of China, Chengdu, 610042, China.四川省肿瘤医院[6]SenseTime Research, Shanghai, 200233, China.
Radiotherapy is one of the main treatments of nasopharyngeal cancer (NPC) and lung cancer. Accurate segmentation of organs at risks (OARs) in CT images is a key step in radiotherapy planning for NPC and lung cancer. However, the segmentation of OARs is influenced by highly imbalanced size of organs, which often results in very poor segmentation results for small and difficult-to-segment organs. In addition, the complex morphological changes and fuzzy boundaries of OARs also pose great challenges to the segmentation task. In this paper, we propose a cross-layer attention fusion network (CLAF-CNN) to solve the problem of accurately segmenting OARs.In CLAF-CNN, we integrate the spatial attention maps of the adjacent spatial attention modules to make the segmentation targets more accurately focused, so that the network can capture more target-related features. In this way, the spatial attention modules in the network can be learned and optimized together. In addition, we introduce a new Top-K exponential logarithmic Dice loss (TELD-Loss) to solve the imbalance problem in OAR segmentation. The TELD-Loss further introduces Top-K optimization mechanism based on Dice loss and exponential logarithmic loss, which makes the network pay more attention to small organs and difficult-to-segment organs, so as to enhance the overall performance of the segmentation model.We validated our framework on the OAR segmentation datasets of the head & neck and lung CT images in the StructSeg 2019 challenge. Experiments show that the CLAF-CNN outperforms the state-of-the-arts attention-based segmentation methods in the OAR segmentation task with average Dice coefficient of 79.65% for head & neck OARs and 88.39% for lung OARs.This work provides a new network named CLAF-CNN which contains cross-layer spatial attention map fusion architecture and TELD-Loss for OAR segmentation. Results demonstrated that the proposed method could obtain accurate segmentation results for OARs, which has a potential of improving the efficiency of radiotherapy planning for nasopharynx cancer and lung cancer.This article is protected by copyright. All rights reserved.
第一作者机构:[1]Glasgow College, University of Electronic Science and Technology of China, Chengdu, 611731, China.
共同第一作者:
通讯作者:
推荐引用方式(GB/T 7714):
Liu Zuhao,Sun Chao,Wang Huan,et al.Automatic segmentation of organs-at-risks of nasopharynx cancer and lung cancer by cross-layer attention fusion network with TELD-Loss.[J].MEDICAL PHYSICS.2021,48(11):6987-7002.doi:10.1002/mp.15260.
APA:
Liu Zuhao,Sun Chao,Wang Huan,Li Zhiqi,Gao Yibo...&Zhang Shaoting.(2021).Automatic segmentation of organs-at-risks of nasopharynx cancer and lung cancer by cross-layer attention fusion network with TELD-Loss..MEDICAL PHYSICS,48,(11)
MLA:
Liu Zuhao,et al."Automatic segmentation of organs-at-risks of nasopharynx cancer and lung cancer by cross-layer attention fusion network with TELD-Loss.".MEDICAL PHYSICS 48..11(2021):6987-7002