Classification, Regression and Segmentation directly from k-Space in Cardiac MRI

Ruochen Li,Jiazhen Pan,Youxiang Zhu,Juncheng Ni,Daniel Rueckert
2024-07-29
Abstract:Cardiac Magnetic Resonance Imaging (CMR) is the gold standard for diagnosing cardiovascular diseases. Clinical diagnoses predominantly rely on magnitude-only Digital Imaging and Communications in Medicine (DICOM) images, omitting crucial phase information that might provide additional diagnostic benefits. In contrast, k-space is complex-valued and encompasses both magnitude and phase information, while humans cannot directly perceive. In this work, we propose KMAE, a Transformer-based model specifically designed to process k-space data directly, eliminating conventional intermediary conversion steps to the image domain. KMAE can handle critical cardiac disease classification, relevant phenotype regression, and cardiac morphology segmentation tasks. We utilize this model to investigate the potential of k-space-based diagnosis in cardiac MRI. Notably, this model achieves competitive classification and regression performance compared to image-domain methods e.g. Masked Autoencoders (MAEs) and delivers satisfactory segmentation performance with a myocardium dice score of 0.884. Last but not least, our model exhibits robust performance with consistent results even when the k-space is 8* undersampled. We encourage the MR community to explore the untapped potential of k-space and pursue end-to-end, automated diagnosis with reduced human intervention.
Image and Video Processing,Artificial Intelligence,Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
The paper aims to address several key issues in the diagnosis of cardiovascular diseases, particularly in the use of Cardiac Magnetic Resonance Imaging (CMR) data. Specifically, the research objectives can be summarized as follows: 1. **Diagnosis directly from k-space**: Traditional CMR diagnosis mainly relies on Digital Imaging and Communications in Medicine (DICOM) images that only contain magnitude information, ignoring potentially valuable phase information. This paper proposes a new method that can directly process complex-valued k-space data containing both magnitude and phase information. 2. **Avoid intermediate conversion steps**: The proposed method (KMAE model) can directly handle k-space data without converting it to the image domain, thereby avoiding the intermediate conversion steps common in traditional methods. 3. **Perform multiple downstream tasks**: The model is not limited to image reconstruction but can also be used for cardiac disease classification, related phenotype regression, and cardiac morphology segmentation, among other downstream tasks. 4. **Handle undersampled data**: In accelerated MRI scans, k-space data is often undersampled, which can lead to degraded image quality and affect subsequent analysis. The model can maintain stable and high-quality performance even in the case of undersampled k-space data. 5. **Improve diagnostic efficiency**: By diagnosing directly from k-space data, the need for human intervention is reduced, and an end-to-end automated diagnostic process is expected to be achieved. In summary, the main goal of this paper is to demonstrate how a new model based on the Transformer architecture (KMAE) can directly perform tasks such as cardiac disease classification, phenotype regression, and cardiac segmentation from k-space data, and to validate the effectiveness and robustness of this approach, especially when dealing with undersampled data. Additionally, the study emphasizes the importance of exploring the untapped potential of k-space and encourages the medical imaging community to further investigate this direction.