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
相关论文
共 47 条
  • [31] A lightweight surface defect detection framework combined with dual-domain attention mechanism
    Tang, Jun
    Wang, Zidong
    Zhang, Hongyi
    Li, Han
    Wu, Peishu
    Zeng, Nianyin
    EXPERT SYSTEMS WITH APPLICATIONS, 2024, 238
  • [32] A Joint Framework for Predicting Disease-Gene Interactions Based on Pre-trained Models and Graph Attention Networks
    Deng, Qiwen
    Han, Yuexia
    Sun, Jianfei
    PROCEEDINGS OF 2024 4TH INTERNATIONAL CONFERENCE ON BIOINFORMATICS AND INTELLIGENT COMPUTING, BIC 2024, 2024, : 474 - 478
  • [33] FedBFGCN: A Graph Federated Learning Framework Based on Balanced Channel Attention and Cross-Layer Feature Fusion Convolution
    Wang, Hefei
    Gu, Ruichun
    Wang, Jingyu
    Zhang, Xiaolin
    Wei, Hui
    IEEE ACCESS, 2025, 13 : 21980 - 21991
  • [34] T-GAN: A deep learning framework for prediction of temporal complex networks with adaptive graph convolution and attention mechanism
    Huang, Ru
    Ma, Lei
    He, Jianhua
    Chu, Xiaoli
    DISPLAYS, 2021, 68
  • [35] GC-Loc: A Graph Attention Based Framework for Collaborative Indoor Localization Using Infrastructure-free Signals
    He, Tao
    Niu, Qun
    Liu, Ning
    PROCEEDINGS OF THE ACM ON INTERACTIVE MOBILE WEARABLE AND UBIQUITOUS TECHNOLOGIES-IMWUT, 2022, 6 (04):
  • [36] A Deep Reinforcement Learning Framework Based on an Attention Mechanism and Disjunctive Graph Embedding for the Job-Shop Scheduling Problem
    Chen, Ruiqi
    Li, Wenxin
    Yang, Hongbing
    IEEE TRANSACTIONS ON INDUSTRIAL INFORMATICS, 2023, 19 (02) : 1322 - 1331
  • [37] Construction of 4D infant cortical surface atlases with sharp folding patterns via spherical patch-based group-wise sparse representation
    Wu, Zhengwang
    Wang, Li
    Lin, Weili
    Gilmore, John H.
    Li, Gang
    Shen, Dinggang
    HUMAN BRAIN MAPPING, 2019, 40 (13) : 3860 - 3880
  • [38] Graph Attention U-Net for Retinal Layer Surface Detection and Choroid Neovascularization Segmentation in OCT Images
    Shen, Yuhe
    Li, Jiang
    Zhu, Weifang
    Yu, Kai
    Wang, Meng
    Peng, Yuanyuan
    Zhou, Yi
    Guan, Liling
    Chen, Xinjian
    IEEE TRANSACTIONS ON MEDICAL IMAGING, 2023, 42 (11) : 3140 - 3154
  • [39] An Adaptively Attention-Driven Cascade Part-Based Graph Embedding Framework for UAV Object Re-Identification
    Shen, Bo
    Zhang, Rui
    Chen, Hao
    REMOTE SENSING, 2022, 14 (06)
  • [40] STM2CN: A Multi-graph Attention-based Framework for Sensor Data Prediction in Smart Cities
    Jin, Zhiling
    Xu, Jing
    Huang, Ruiqi
    Shao, Wei
    Xiao, Xiao
    2022 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2022,