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 条
  • [21] MalHAPGNN: An Enhanced Call Graph-Based Malware Detection Framework Using Hierarchical Attention Pooling Graph Neural Network
    Guo, Wenjie
    Du, Wenbiao
    Yang, Xiuqi
    Xue, Jingfeng
    Wang, Yong
    Han, Weijie
    Hu, Jingjing
    SENSORS, 2025, 25 (02)
  • [22] Multi-contrast multi-scale surface registration for improved alignment of cortical areas
    Tardif, Christine Lucas
    Schaefer, Andreas
    Waehnert, Miriam
    Dinse, Juliane
    Turner, Robert
    Bazin, Pierre-Louis
    NEUROIMAGE, 2015, 111 : 107 - 122
  • [23] A Graph-Based Temporal Attention Framework for Multi-Sensor Traffic Flow Forecasting
    Zhang, Shaokun
    Guo, Yao
    Zhao, Peize
    Zheng, Chuanpan
    Chen, Xiangqun
    IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2022, 23 (07) : 7743 - 7758
  • [24] A graph attention network-based learning framework for automatic detection of abnormal vessel behaviors
    Liang, Maohan
    Zhang, Yuanzhe
    Jin, Qiqiang
    Liu, Ryan Wen
    OCEAN ENGINEERING, 2025, 325
  • [25] A generic framework for the parcellation of the cortical surface into gyri using geodesic Voronoi diagrams
    Cachia, A
    Mangin, JF
    Rivière, D
    Papadopoulos-Orfanos, D
    Kherif, F
    Bloch, I
    Régis, J
    MEDICAL IMAGE ANALYSIS, 2003, 7 (04) : 403 - 416
  • [26] Multibranch Fusion: A Multibranch Attention Framework by Combining Graph Convolutional Network and CNN for Hyperspectral Image Classification
    Liu, Xun
    Ng, Alex Hay-Man
    Ge, Linlin
    Lei, Fangyuan
    Liao, Xuejiao
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2024, 62
  • [27] MOGAT: A Multi-Omics Integration Framework Using Graph Attention Networks for Cancer Subtype Prediction
    Tanvir, Raihanul Bari
    Islam, Md Mezbahul
    Sobhan, Masrur
    Luo, Dongsheng
    Mondal, Ananda Mohan
    INTERNATIONAL JOURNAL OF MOLECULAR SCIENCES, 2024, 25 (05)
  • [28] ncRPI-LGAT: Prediction of ncRNA-protein interactions with line graph attention network framework
    Han, Yong
    Zhang, Shao-Wu
    COMPUTATIONAL AND STRUCTURAL BIOTECHNOLOGY JOURNAL, 2023, 21 : 2286 - 2295
  • [29] SAGL: A self-attention-based graph learning framework for predicting survival of colorectal cancer patients
    Yang, Ping
    Qiu, Hang
    Yang, Xulin
    Wang, Liya
    Wang, Xiaodong
    COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE, 2024, 249
  • [30] A Framework for Metal Surface Energy Prediction Based on Crystal Graph Convolutional Neural Network
    Zhou L.
    Zhu G.
    Wu Y.
    Huang Y.
    Hong Z.
    Kuei Suan Jen Hsueh Pao/Journal of the Chinese Ceramic Society, 2023, 51 (02): : 389 - 396