Correlation-Guided Recursive Pyramid Network for Deformable Brain MRI Registration.
Authors
Abstract
As a key preprocessing technique in medical image analysis, deformable image registration has remained a research focus over the past decade. Recently, deep learning-based registration methods have become mainstream. Nevertheless, simultaneously handling large-scale deformations and accurate feature matching remains a persistent challenge. While pyramid architectures are widely employed to mitigate large-scale deformations, existing methods often exhibit an unbalanced focus. One group emphasizes iterative refinement to handle large deformations but relies on implicit, coarse feature interactions. Conversely, the other group concentrates on explicit matching techniques, but such static matching is often unreliable in regions with significant anatomical discrepancies. To bridge this gap, we propose a novel Correlation-Guided Recursive Pyramid Network (CRPNet). Unlike previous approaches, CRPNet addresses these challenges in a unified manner by embedding explicit correlation modeling directly into the recursive optimization. Specifically, we propose a Correlation-Guided Intra-layer Recursive Strategy (CGIRS), which enables the network to continuously refine matching accuracy through recursive feedback while preventing cross-scale error propagation. To facilitate this, we design a Spatial Correlation Module (SPCM) for accurate spatial correspondence and a Semantic Correlation Module (SECM) for high-level semantic alignment. Extensive experiments on three brain imaging datasets demonstrate that our method achieves state-of-the-art performance, particularly exhibiting exceptional robustness under extreme deformations, proving the efficacy of our method for deformable brain MRI registration. The code is available at https://github.com/ZhangWH0129/CRPNet.