Attention-Guided TransMorph for Real-Time Tumor Tracking in Cine-MRI.
Authors
Affiliations (2)
Affiliations (2)
- Biomedical Engineering Lab (BEL), School of Electrical and Computer Engineering, National Technical University of Athens, Athens, 15780, Greece. [email protected].
- Biomedical Engineering Lab (BEL), School of Electrical and Computer Engineering, National Technical University of Athens, Athens, 15780, Greece.
Abstract
Magnetic Resonance Imaging (MRI) is a key modality in cancer treatment, providing high soft tissue contrast for the visualization of tumors and internal anatomy. Radiotherapy, which is widely used in treatments, requires precise tumor segmentation to ensure targeting the true tumor and minimizing radiation exposure to healthy tissues. In this regard, real-time automatic tumor tracking from cine-MRI can provide accurate tumor localization supporting adaptive radiotherapy. Conventional image registration techniques exhibit limitations when handling large misalignments and high computational demands, unlike deep learning methods, with high learning capabilities and fast inference times. A real-time tumor tracking approach for 2D cine-MRI using deep learning-based deformable image registration, based on an improved TransMorph architecture, is presented. The approach adheres to a two-step training paradigm: (1) unsupervised pretraining on unlabeled patient image pairs, and (2) supervised fine-tuning with segmentation labels. Attention gates are integrated into skip connections to enhance spatial selectivity regarding the most relevant regions for alignment. A composite loss function is utilized, synthesizing boundary-weighted Dice, adaptive MSE, L1 and smooth diffusion and edge-based regularization. Both overlap and distance-based metrics were computed to assess the model's segmentation accuracy in the registration of various frames within the patients. The proposed model achieved DSC <math xmlns="http://www.w3.org/1998/Math/MathML"><mrow><mn>93.42</mn> <mo>%</mo> <mo>±</mo> <mn>5</mn> <mo>%</mo></mrow> </math> , 95HD <math xmlns="http://www.w3.org/1998/Math/MathML"><mrow><mn>2.75</mn> <mo>±</mo> <mn>3.24</mn></mrow> </math> mm, 50HD <math xmlns="http://www.w3.org/1998/Math/MathML"><mrow><mn>0.88</mn> <mo>±</mo> <mn>0.40</mn></mrow> </math> mm. Extensive benchmarking demonstrated that the proposed framework consistently achieved superior performance compared to the TransMorph model, its variants, and other existing state-of-the-art image registration approaches. Experimental results indicate that this technique may serve as a critical tool for the advancement of MRI-guided radiotherapy.