Pyramid Attention Network for Medical Image Registration

Zhuoyuan Wang,Haiqiao Wang,Yi Wang
2024-02-14
Abstract:The advent of deep-learning-based registration networks has addressed the time-consuming challenge in traditional iterative methods.However, the potential of current registration networks for comprehensively capturing spatial relationships has not been fully explored, leading to inadequate performance in large-deformation image registration.The pure convolutional neural networks (CNNs) neglect feature enhancement, while current Transformer-based networks are susceptible to information
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
The main objective of this paper is to address the issue of large deformation registration in medical image registration. Specifically, although existing deep learning-based registration networks outperform traditional iterative methods (such as Elastix and SyN) in terms of speed, their ability to capture spatial relationships has not been fully utilized, resulting in insufficient performance when handling large deformation image registration. Pure convolutional neural networks (CNNs) neglect feature enhancement, while current Transformer-based networks are prone to information redundancy. To solve these problems, the authors propose a Pyramid Attention Network (PAN), which combines a dual-stream pyramid encoder and channel attention mechanism to enhance feature representation, and introduces a multi-head local attention Transformer as a decoder to analyze motion patterns and generate deformation fields. Experimental results show that this method achieves excellent registration performance on three public medical imaging datasets, outperforming various CNN and Transformer-based registration networks.