Recursive deformable image registration network with mutual attention

Conference item

Deformable image registration, estimating the spatial transformation between different images, is an important task in medical imaging. Many previous studies have used learning-based methods for multi-stage registration to perform 3D image registration to improve performance. The performance of the multi-stage approach, however, is limited by the size of the receptive field where complex motion does not occur at a single spatial scale. We propose a new registration network combining recursive network architecture and mutual attention mechanism to overcome these limitations. Compared with the state-of-the-art deep learning methods, our network based on the recursive structure achieves the highest accuracy in lung Computed Tomography (CT) data set (Dice score of 92% and average surface distance of 3.8mm for lungs) and one of the most accurate results in abdominal CT data set with 9 organs of various sizes (Dice score of 55% and average surface distance of 7.8mm). We also showed that adding 3 recursive networks is sufficient to achieve the state-of-the-art results without a significant increase in the inference time.

Authors: Zheng, J-Q; Wang, Z; Huang, B; Vincent, T; Lim, NH

• Publication status: Published
• Peer review status: Peer reviewed
• Publisher: Springer
• Host title: Medical Image Understanding and Analysis
• Pages: 75-86
• Series: Lecture Notes in Computer Science
• Series number: 13413
• Publication date: 2022-07-25
• Event series: 26th Medical Image Understanding and Analysis (MIUA 2022)
• Event location: Cambridge, UK
• Event start date: 2022-07-27
• Event end date: 2022-07-29
• DOI: 10.1007/978-3-031-12053-4_6
• EISSN: 1611-3349
• ISSN: 0302-9743
• EISBN: 9783031120534
• ISBN: 9783031120527
• Language: English
• Keywords: deformable image registration; recursive network; mutual attention