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 条
  • [21] A unified structure learning framework for graph attention networks
    Yuan, Jinliang
    Cao, Meng
    Cheng, Hao
    Yu, Hualei
    Xie, Junyuan
    Wang, Chongjun
    NEUROCOMPUTING, 2022, 495 : 194 - 204
  • [22] ROBUST REGISTRATION OF MULTISPECTRAL IMAGES OF THE CORTICAL SURFACE IN NEUROSURGERY
    Stoyanov, Danail
    Rayshubskiy, Aleksandr
    Hillman, Elizabeth
    2012 9TH IEEE INTERNATIONAL SYMPOSIUM ON BIOMEDICAL IMAGING (ISBI), 2012, : 1643 - 1646
  • [23] Brain Cortical Surface Registration with Anatomical Atlas Constraints
    Zeng, Wei
    Chang, Xuebin
    Yang, Liqun
    Razib, Muhammad
    Lu, Zhong-Lin
    Yang, Yi-Jun
    ADVANCES IN VISUAL COMPUTING, ISVC 2023, PT I, 2023, 14361 : 357 - 369
  • [24] Generalized surface flows for deformable registration and cortical matching
    Eckstein, I.
    Joshi, A. A.
    Kuo, C. -C. J.
    Leahy, R.
    Desbrun, M.
    MEDICAL IMAGE COMPUTING AND COMPUTER-ASSISTED INTERVENTION - MICCAI 2007, PT 1, PROCEEDINGS, 2007, 4791 : 692 - +
  • [25] GeoMorph: Geometric Deep Learning for Cortical Surface Registration
    Suliman, Mohamed A.
    Williams, Logan Z. J.
    Fawaz, Abdulah
    Robinson, Emma C.
    GEOMETRIC DEEP LEARNING IN MEDICAL IMAGE ANALYSIS, VOL 194, 2022, 194 : 118 - 129
  • [26] A joint parcellation and boundary network with multi-rate-shared dilated graph attention for cortical surface parcellation
    Liu, Siqi
    Ye, Hailiang
    Yang, Bing
    Li, Ming
    Cao, Feilong
    MEDICAL & BIOLOGICAL ENGINEERING & COMPUTING, 2024, 62 (02) : 537 - 549
  • [27] A joint parcellation and boundary network with multi-rate-shared dilated graph attention for cortical surface parcellation
    Siqi Liu
    Hailiang Ye
    Bing Yang
    Ming Li
    Feilong Cao
    Medical & Biological Engineering & Computing, 2024, 62 : 537 - 549
  • [28] High Performance Groupwise Cortical Surface Registration with Multimodal Surface Matching
    Besenczi, Renato
    Guo, Yourong
    Robinson, Emma C.
    BIOMEDICAL IMAGE REGISTRATION, WBIR 2024, 2025, 15249 : 321 - 332
  • [29] A hybrid framework for surface registration and deformable models
    Montagnat, J
    Delingette, H
    1997 IEEE COMPUTER SOCIETY CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, PROCEEDINGS, 1997, : 1041 - 1046
  • [30] Two-stage graph matching point cloud registration method based on graph attention network
    Guo, Jiacheng
    Liu, Xuejun
    Zhang, Shuo
    Yan, Yong
    Sha, Yun
    Jiang, Yinan
    JOURNAL OF APPLIED REMOTE SENSING, 2024, 18 (03)