CrossRAFT: Cross-Domain Complex-Valued Feature Extraction for Ultrasound Motion Estimation

Abstract

Accurate multi-dimensional motion estimation is fundamental to broad biomedical ultrasound applications, primarily performed using real-valued radio-frequency (RF), analytic, and in-phase and quadrature (IQ) signals. Real-valued RF and analytic signals offer high subsample accuracy but are computationally intensive. Regardless of the signal domain used, traditional algorithms suffer from interframe decorrelation, resolution tradeoffs, and a lack of a carrier and low sampling in the lateral direction. Emerging deep learning (DL) models demonstrate improved performance and faster inference. However, existing architectures are real-valued and disrupt inherent phase coupling crucial for accurate motion estimation by treating complex components as independent channels. To address this, we propose CrossRAFT, an end-to-end trainable network designed for subsample motion estimation directly from complex-valued ultrasound signals. Built on the RAFT architecture, CrossRAFT features complex-valued encoders and a custom correlation module to explicitly extract and use phase information. We evaluated CrossRAFT across real RF, analytic, and IQ signal presentations on simulated phantoms, synthetic echocardiographic, and private in vivo echocardiographic datasets. Experimental results show that CrossRAFT estimates displacements accurately under complex tissue motion and in the challenging lateral direction. Compared to real RF-based RAFT, analytic-based CrossRAFT reduces vertical and horizontal displacement errors by 69% and 51% on the synthetic echocardiographic data, respectively. These findings demonstrate that leveraging phase information via complex-valued neural networks provides a powerful framework for ultrasound motion estimation, showing potential for clinical elastography and functional imaging.