Search papers, labs, and topics across Lattice.
This paper tackles the challenge of domain shift in pulmonary embolism (PE) detection in CTPA images by proposing an unsupervised domain adaptation (UDA) framework based on a Transformer backbone and Mean-Teacher architecture. The core innovation lies in enhancing pseudo-label quality through the integration of Prototype Alignment (PA), Global and Local Contrastive Learning (GLCL), and an Attention-based Auxiliary Local Prediction (AALP) module. Experiments on cross-center datasets (FUMPE and CAD-PE) demonstrate substantial IoU improvements in both directions, and cross-modality experiments on MMWHS further validate the method's robustness.
Achieve a 30% absolute IoU improvement in cross-center pulmonary embolism detection by learning structural information in feature space, without target-domain labels.
While deep learning has demonstrated considerable promise in computer-aided diagnosis for pulmonary embolism (PE), practical deployment in Computed Tomography Pulmonary Angiography (CTPA) is often hindered by "domain shift" and the prohibitive cost of expert annotations. To address these challenges, an unsupervised domain adaptation (UDA) framework is proposed, utilizing a Transformer backbone and a Mean-Teacher architecture for cross-center semantic segmentation. The primary focus is placed on enhancing pseudo-label reliability by learning deep structural information within the feature space. Specifically, three modules are integrated and designed for this task: (1) a Prototype Alignment (PA) mechanism to reduce category-level distribution discrepancies; (2) Global and Local Contrastive Learning (GLCL) to capture both pixel-level topological relationships and global semantic representations; and (3) an Attention-based Auxiliary Local Prediction (AALP) module designed to reinforce sensitivity to small PE lesions by automatically extracting high-information slices from Transformer attention maps. Experimental validation conducted on cross-center datasets (FUMPE and CAD-PE) demonstrates significant performance gains. In the FUMPE -> CAD-PE task, the IoU increased from 0.1152 to 0.4153, while the CAD-PE -> FUMPE task saw an improvement from 0.1705 to 0.4302. Furthermore, the proposed method achieved a 69.9% Dice score in the CT -> MRI cross-modality task on the MMWHS dataset without utilizing any target-domain labels for model selection, confirming its robustness and generalizability for diverse clinical environments.
