机构:[1]School of Computer Science, Sichuan University, Chengdu, P. R. China.[2]Department of Radiation Oncology, Cancer Center, West China Hospital, Sichuan University, Chengdu, P. R. China.四川大学华西医院[3]Agile and Intelligent Computing Key Laboratory, Southwest China Institute of Electronic Technology, Chengdu, P. R. China.[4]School of Computer Science, Chengdu University of Information Technology, P. R. China.[5]Department of Biotherapy, Cancer Center, West China Hospital, Sichuan University, Chengdu, P. R. China.四川大学华西医院
Radiotherapy is one of the leading treatments for cancer. To accelerate the implementation of radiotherapy in clinic, various deep learning-based methods have been developed for automatic dose prediction. However, the effectiveness of these methods heavily relies on the availability of a substantial amount of data with labels, i.e. the dose distribution maps, which cost dosimetrists considerable time and effort to acquire. For cancers of low-incidence, such as cervical cancer, it is often a luxury to collect an adequate amount of labeled data to train a well-performing deep learning (DL) model. To mitigate this problem, in this paper, we resort to the unsupervised domain adaptation (UDA) strategy to achieve accurate dose prediction for cervical cancer (target domain) by leveraging the well-labeled high-incidence rectal cancer (source domain). Specifically, we introduce the cross-attention mechanism to learn the domain-invariant features and develop a cross-attention transformer-based encoder to align the two different cancer domains. Meanwhile, to preserve the target-specific knowledge, we employ multiple domain classifiers to enforce the network to extract more discriminative target features. In addition, we employ two independent convolutional neural network (CNN) decoders to compensate for the lack of spatial inductive bias in the pure transformer and generate accurate dose maps for both domains. Furthermore, to enhance the performance, two additional losses, i.e. a knowledge distillation loss (KDL) and a domain classification loss (DCL), are incorporated to transfer the domain-invariant features while preserving domain-specific information. Experimental results on a rectal cancer dataset and a cervical cancer dataset have demonstrated that our method achieves the best quantitative results with [Formula: see text], [Formula: see text], and HI of 1.446, 1.231, and 0.082, respectively, and outperforms other methods in terms of qualitative assessment.
基金:
This work is supported by the National
Natural Science Foundation of China (NSFC
62371325, 62071314), Sichuan Science and Technology
Program 2023YFG0263, 2023YFG0025,
2023NSFSC0497, and Opening Foundation of Agile
and Intelligent Computing Key Laboratory of Sichuan
Province.
语种:
外文
PubmedID:
中科院(CAS)分区:
出版当年[2023]版:
大类|2 区计算机科学
小类|2 区计算机:人工智能
最新[2023]版:
大类|2 区计算机科学
小类|2 区计算机:人工智能
第一作者:
第一作者机构:[1]School of Computer Science, Sichuan University, Chengdu, P. R. China.
共同第一作者:
通讯作者:
推荐引用方式(GB/T 7714):
Cui Jiaqi,Xiao Jianghong,Hou Yun,et al.Unsupervised Domain Adaptive Dose Prediction Via Cross-Attention Transformer and Target-Specific Knowledge Preservation[J].International journal of neural systems.2023,2350057.doi:10.1142/S0129065723500570.
APA:
Cui Jiaqi,Xiao Jianghong,Hou Yun,Wu Xi,Zhou Jiliu...&Wang Yan.(2023).Unsupervised Domain Adaptive Dose Prediction Via Cross-Attention Transformer and Target-Specific Knowledge Preservation.International journal of neural systems,,
MLA:
Cui Jiaqi,et al."Unsupervised Domain Adaptive Dose Prediction Via Cross-Attention Transformer and Target-Specific Knowledge Preservation".International journal of neural systems .(2023):2350057