SUGAR: Spherical ultrafast graph attention framework for cortical surface registration

被引:4
|
作者
Ren, Jianxun [1 ]
An, Ning [1 ]
Zhang, Youjia [1 ]
Wang, Danyang [1 ]
Sun, Zhenyu [1 ]
Lin, Cong [1 ]
Cui, Weigang [2 ]
Wang, Weiwei [1 ]
Zhou, Ying [1 ]
Zhang, Wei [1 ,3 ]
Hu, Qingyu [1 ]
Zhang, Ping [1 ]
Hu, Dan [4 ]
Wang, Danhong [4 ]
Liu, Hesheng [1 ,5 ]
机构
[1] Changping Lab, Beijing, Peoples R China
[2] Beihang Univ, Sch Engn Med, Beijing, Peoples R China
[3] Peking Univ, Acad Adv Interdisciplinary Studies, Beijing, Peoples R China
[4] Massachusetts Gen Hosp, Athinoula A Martinos Ctr Biomed Imaging, Harvard Med Sch, Dept Radiol, Charlestown, MA USA
[5] Peking Univ, Biomed Pioneering Innovat Ctr BIOP, Beijing, Peoples R China
基金
中国博士后科学基金;
关键词
Cortical surface registration; Graph neural network; Attention mechanism; Registration distortion; HUMAN CEREBRAL-CORTEX; PARCELLATION; NETWORK;
D O I
10.1016/j.media.2024.103122
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Cortical surface registration plays a crucial role in aligning cortical functional and anatomical features across individuals. However, conventional registration algorithms are computationally inefficient. Recently, learningbased registration algorithms have emerged as a promising solution, significantly improving processing efficiency. Nonetheless, there remains a gap in the development of a learning -based method that exceeds the stateof-the-art conventional methods simultaneously in computational efficiency, registration accuracy, and distortion control, despite the theoretically greater representational capabilities of deep learning approaches. To address the challenge, we present SUGAR, a unified unsupervised deep -learning framework for both rigid and non -rigid registration. SUGAR incorporates a U -Net -based spherical graph attention network and leverages the Euler angle representation for deformation. In addition to the similarity loss, we introduce fold and multiple distortion losses to preserve topology and minimize various types of distortions. Furthermore, we propose a data augmentation strategy specifically tailored for spherical surface registration to enhance the registration performance. Through extensive evaluation involving over 10,000 scans from 7 diverse datasets, we showed that our framework exhibits comparable or superior registration performance in accuracy, distortion, and test -retest reliability compared to conventional and learning -based methods. Additionally, SUGAR achieves remarkable sub -second processing times, offering a notable speed-up of approximately 12,000 times in registering 9,000 subjects from the UK Biobank dataset in just 32 min. This combination of high registration performance and accelerated processing time may greatly benefit large-scale neuroimaging studies.
引用
收藏
页数:15
相关论文
共 50 条
  • [1] Hierarchical spherical deformation for cortical surface registration
    Lyu, Ilwoo
    Kang, Hakmook
    Woodward, Neil D.
    Styner, Martin A.
    Landman, Bennett A.
    MEDICAL IMAGE ANALYSIS, 2019, 57 : 72 - 88
  • [2] Unsupervised Learning of Cortical Surface Registration Using Spherical Harmonics
    Lee, Seungeun
    Ryu, Sunghwa
    Lee, Seunghwan
    Lyu, Ilwoo
    SHAPE IN MEDICAL IMAGING, SHAPEMI 2023, 2023, 14350 : 65 - 74
  • [3] A Deep-Discrete Learning Framework for Spherical Surface Registration
    Suliman, Mohamed A.
    Williams, Logan Z. J.
    Fawaz, Abdulah
    Robinson, Emma C.
    MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION, MICCAI 2022, PT VI, 2022, 13436 : 119 - 129
  • [4] Cortical Connectome Registration Using Spherical Demons
    Isaev, Dmitry
    Gutman, Boris A.
    Moyer, Daniel
    Faskowitz, Joshua
    Thompson, Paul M.
    12TH INTERNATIONAL SYMPOSIUM ON MEDICAL INFORMATION PROCESSING AND ANALYSIS, 2017, 10160
  • [5] Two-Stage Point Cloud Registration Framework Based on Graph Neural Network and Attention
    Zhang, Xiaoqian
    Li, Junlin
    Zhang, Wei
    Xu, Yansong
    Li, Feng
    ELECTRONICS, 2024, 13 (03)
  • [6] Spherical Demons: Fast Surface Registration
    Yeo, B. T. Thomas
    Sabuncu, Mert
    Vercauteren, Tom
    Ayache, Nicholas
    Fischl, Bruce
    Golland, Polina
    MEDICAL IMAGE COMPUTING AND COMPUTER-ASSISTED INTERVENTION - MICCAI 2008, PT I, PROCEEDINGS, 2008, 5241 : 745 - +
  • [7] Unsupervised Learning for Spherical Surface Registration
    Zhao, Fenqiang
    Wu, Zhengwang
    Wang, Li
    Lin, Weili
    Xia, Shunren
    Shen, Dinggang
    Li, Gang
    MACHINE LEARNING IN MEDICAL IMAGING, MLMI 2020, 2020, 12436 : 373 - 383
  • [8] Anatomically constrained squeeze-and-excitation graph attention network for cortical surface parcellation
    Li, Xinwei
    Tan, Jia
    Wang, Panyu
    Liu, Hong
    Li, Zhangyong
    Wang, Wei
    COMPUTERS IN BIOLOGY AND MEDICINE, 2022, 140
  • [9] ENCODING CORTICAL SURFACE BY SPHERICAL HARMONICS
    Chung, Moo K.
    Hartley, Richard
    Dalton, Kim M.
    Davidson, Richard J.
    STATISTICA SINICA, 2008, 18 (04) : 1269 - 1291
  • [10] A Combined Surface And VOlumetric Registration (SAVOR) Framework to Study Cortical Biomarkers and Volumetric Imaging Data
    Gibson, Eli
    Khan, Ali R.
    Beg, Mirza Faisal
    MEDICAL IMAGE COMPUTING AND COMPUTER-ASSISTED INTERVENTION - MICCAI 2009, PT I, PROCEEDINGS, 2009, 5761 : 713 - 720