Adaptive geometric-attention multi-task framework with knowledge distillation for aortic dissection detection in non-contrast CT.

Journal: Physics in medicine and biology
Published Date:

Abstract

Aortic dissection (AD) is a life-threatening cardiovascular emergency. Non-contrast-enhanced CT (NCE-CT) could provide timely AD screening with fewer contraindications compared to contrast-enhanced CT angiography (CE-CT). However, NCE-CT examinations lack distinctive imaging characteristics of AD, leading to high rates of missed diagnoses and misdiagnoses, and increased radiologist workload. In this paper, we propose a novel end-to-end multi-task framework for automated aortic segmentation and AD detection using NCE-CT images. The framework comprises three main components: a deformable feature extractor enhancing aorta tubular-feature attention, an adaptive geometric information extraction module to optimize feature sharing between segmentation and classification tasks via the transformer cross-attention mechanism, and a knowledge distillation module transferring diagnostic information from the CE-CT-based teacher model to the NCE-CT-based student model. Multi-center tests across 3 internal and 2 external centers demonstrated that our model outperformed existing methods both for segmenting the aorta and detecting AD. Specifically, for segmenting the aorta, our framework achieved dice of 0.928 and 0.909, Jaccard index of 0.867 and 0.858, MIoU of 0.932 and 0.913, and FWIoU of 0.995 and 0.994, in internal and external testing datasets, respectively. For identifying AD patients from non-AD patients, our framework achieved accuracies of 0.911 and 0.840, sensitivities of 0.925 and 0.888, and F1-scores of 0.922 and 0.836, in internal and external testing datasets, respectively. Ablation experiment demonstrates the effectiveness of each module. The proposed model may serve as an effective diagnostic assistant for radiologists, acting as a "second pair of eyes" to assist in AD screening using NCE-CT images.

Authors

  • Rongli Zhang
    Department of Imaging and Interventional Radiology, The Chinese University of Hong Kong, Prince of Wales Hospital, Hong Kong, China.
  • Zhiquan Situ
    Guangzhou Medical University, Guangzhou Medical University (Panyu Campus), Guangzhou, 510182, CHINA.
  • Zhangbo Cheng
  • Yande Luo
    Guangzhou Medical University, Guangzhou Medical University (Panyu Campus), Guangzhou, 510182, CHINA.
  • Xiongfeng Qiu
    Guangzhou Medical University, Guangzhou Medical University (Panyu Campus), Guangzhou, 510182, CHINA.
  • Xin He
    Department of Nephrology, The Affiliated Hospital of Guizhou Medical, Guizhou, China.
  • Xinchen Yuan
    Guangzhou Medical University, Guangzhou, China.
  • Zijie Zhou
    Shanghai Ninth People's Hospital Affiliated to Shanghai Jiao Tong University School of Medicine, Shanghai, China.
  • Zhaowei Rong
    Guangzhou Medical University, Guangzhou Medical University (Panyu Campus), Guangzhou, 510182, CHINA.
  • Yunhai Lin
    Guangzhou Medical University, Guangzhou Medical University (Panyu Campus), Guangzhou, 510182, CHINA.
  • Qi Li
    The First Affiliated Hospital of Yangtze University, Jingzhou, Hubei, China.
  • Bin Sun
    Department of Urology, General Hospital of the Air Force, PLA, No. 30 Fucheng Road Haidian District, Beijing, 100142 China.
  • Huizhong Wu
    The Sixth People's Hospital of Huizhou, Huizhou, China.
  • Huilin Jiang
  • Wu Zhou
    School of Medical Information Engineering, Guangzhou University of Chinese Medicine, Guangzhou, China, 510006.
  • Guoxi Xie
    School of Basic Medical Sciences, Guangzhou Medical University, Guangzhou, China, 510182.

Keywords

No keywords available for this article.