Abstract:Medical image registration, which is the process of aligning two medical images to ensure spatial consistency, is extensively utilized in surgical navigation, lesion localization, and other fields. In recent years, Transformer-based models have achieved state-of-the-art image registration results. However, these models often lose significant information during image downsampling, and the window division operation in the window attention module disrupts the integrity of the anatomical structure, which limits the model's ability to capture global features. To address these problems, this study introduces an image registration model based on the wavelet transform and linear attention. Specifically, the discrete wavelet transform is leveraged to decompose the image, to provide more features to the model and harness the rich high-frequency information achieved from the wavelet transform, to effectively mitigate the information loss arising from downsampling. Wavelet transform and linear attention are combined to propose a linear wavelet self-attention (LWSA) module. Compared with other attention modules, this unique linear attention calculation method in LWSA reduces the computational complexity and significantly expands the receptive field of the module, ensuring the integrity of the anatomical structure. Additionally, the introduction of the wavelet transform allows the model to pay differential attention to different areas, thereby improving the model's performance. The efficacy of the proposed model in registration tasks was validated through comparative experiments with other models on three publicly available brain imaging datasets.

Image Registration Using a New scheme of Wavelet Decomposition

A Novel Method of Medical Image Registration Based on DTCWT and NPSO

An Medical Image Registration Approach Using Improved Hausdorff Distance Combined with Particle Swarm Optimization

Multimodal Registration Using the Discrete Wavelet Frame Transform

Novel Image Registration Method Based on Wavelet Transform

Application of wavelet transform and phase correlation in sub-pixel registration

Ant Colony Optimization Image Registration Algorithm Based on Wavelet Transform and Mutual Information

A novel method for rotated images registration based on wavelet transform]

A Novel Image Registration Algorithm Using Wavelet Transform and Matrix-Multiply Discrete Fourier Transform

Robust Image Registration Based on Wavelet and Increment Sign Correlation

Image Registration Based on Shape Feature

Image Registration Based on Complex Wavelet Pyramid and Hvs Modeling

Automatic Registration Algorithm Based on Wavelet Transform

A novel medical image registration algorithm based on contour features and multiwavelet transform

A Remote Sensing Image Registration Method Based on Wavelet Decomposition and the Improved Hausdorff Distance

[The Multi-Resolution Image Registration Algorithm for Digital Subtraction Angiography].

Medical Image Registration Methods Based on Wavelet Transformation and Probability Estimate

Medical Image Registration-Based on Methodologies of Mutual Information, Wavelet Decomposition, and Hybrid Optimization

Digital Medical Image Enhanced by Stationary Wavelet Transform

Deformable medical image registration based on wavelet transform and linear attention

An improved algorithm of multi-source remote sensing image registration based on SIFT and Wavelet Transform