W-STFNet: A Wavelet Transform-Based Regularized Hybrid Recursive Spatiotemporal Fusion Registration Network.

purposeDeformable image registration (DIR) is a crucial technique in medical image analysis and is particularly important for 4D-CT-guided lung radiotherapy, where accurate spatiotemporal alignment and deformation plausibility are required for downstream tasks such as dose accumulation. However, many existing learning-based methods are limited in modeling global spatiotemporal dependencies and in preserving fine anatomical structures under large respiratory motion.   

methodsTo address these issues, this paper proposes a wavelet transform-based regularized hybrid recursive spatiotemporal fusion registration network (W-STFNet). The proposed method incorporates SwinLSTM to effectively capture global spatiotemporal dependencies. To achieve better semantic integration of features across scales and time steps, a multi-scale spatiotemporal attention fusion (MSTAF) module is proposed, which improves the network's robustness and stability. Additionally, we design a novel frequency-domain loss function based on Discrete Wavelet Transform (DWT), which optimizes fine-grained structural matching by aligning high-frequency sub-bands, effectively improving the accuracy of high-frequency detail registration. The method is optimized in an unsupervised, patient-specific one-shot setting without anatomical annotations or multi-patient pretraining.   

resultsExperiments on two public 4D-CT datasets (DIR-Lab and POPI-model) show that W-STFNet achieves competitive registration accuracy and stable performance across cases with varying deformation amplitudes. On DIR-Lab, W-STFNet attains a mean TRE of