Preoperative prediction of lymph node (LN) metastasis based on computed tomography (CT) scans is an important task in gastric cancer, but few machine learning-based techniques have been proposed. While multi-center datasets increase sample size and representation ability, they suffer from inter-center heterogeneity. To tackle the above issue, we propose a novel multi-source domain adaptation framework for this diagnosis task, which not only considers domain-invariant and domain-specific features, but also achieves the imbalanced knowledge transfer and class-aware feature alignment across domains. First, we develop a 3D improved feature pyramidal network (i.e., 3D IFPN) to extract common multi-level features from the high-resolution 3D CT images, where a feature dynamic transfer (FDT) module can promote the network's ability to recognize the small target (i.e., LN). Then, we design an unsupervised domain selective graph convolutional network (i.e., UDS-GCN), which mainly includes three types of components: domain-specific feature extractor, domain selector and class-aware GCN classifier. Specifically, multiple domain-specific feature extractors are employed for learning domain-specific features from the common multi-level features generated by the 3D IFPN. A domain selector via the optimal transport (OT) theory is designed for controlling the amount of knowledge transferred from source domains to the target domain. A class-aware GCN classifier is developed to explicitly enhance/weaken the intra-class/inter-class similarity of all sample pairs across domains. To optimize UDS-GCN, the domain selector and the class-aware GCN classifier provide reliable target pseudo-labels to each other in the iterative process by collaborative learning. The extensive experiments are conducted on an in-house CT image dataset collected from four medical centers to demonstrate the efficacy of our proposed method. Experimental results verify that the proposed method boosts LN metastasis diagnosis performance and outperforms state-of-the-art methods. Our code is publically available at https://github.com/infinite-tao/LN _ MSDA .(c) 2022 Elsevier B.V. All rights reserved.
基金:
National Natural Science Foundation of China [62071309, 61871274, 81971585, 62006160]; National Natural Science Foundation of Guangdong Province [2019B1515120029, 2019A1515111205]; Shenzhen Key Basic Research Project [SGDX20201103095802007, KCXFZ20201221173213036, JCYJ20180507184647636, JCYJ20190808155618806, GJHZ20190822095414576, JCYJ20190808145011259]; National Key Research and Development Program of China [2020YFC2003903, 2020YFC2007301]; Beijing Municipal Science and Technology Project [Z211100003521009]; Guangzhou Science and technology planning project [202103010001]
语种:
外文
WOS:
中科院(CAS)分区:
出版当年[2021]版:
大类|1 区工程技术
小类|1 区计算机:人工智能1 区计算机:跨学科应用1 区工程:生物医学1 区核医学
最新[2025]版:
大类|1 区医学
小类|1 区计算机:人工智能1 区计算机:跨学科应用1 区工程:生物医学1 区核医学
JCR分区:
出版当年[2020]版:
Q1RADIOLOGY, NUCLEAR MEDICINE & MEDICAL IMAGINGQ1COMPUTER SCIENCE, INTERDISCIPLINARY APPLICATIONSQ1ENGINEERING, BIOMEDICALQ1COMPUTER SCIENCE, ARTIFICIAL INTELLIGENCE
最新[2023]版:
Q1COMPUTER SCIENCE, ARTIFICIAL INTELLIGENCEQ1COMPUTER SCIENCE, INTERDISCIPLINARY APPLICATIONSQ1ENGINEERING, BIOMEDICALQ1RADIOLOGY, NUCLEAR MEDICINE & MEDICAL IMAGING