Benchmarking learned non-Cartesian k-space trajectories and reconstruction networks

Chaithya G R,Philippe Ciuciu

DOI: https://doi.org/10.48550/arXiv.2201.11356

2022-01-27

Abstract:We benchmark the current existing methods to jointly learn non-Cartesian k-space trajectory and reconstruction: PILOT, BJORK, and compare them with those obtained from the recently developed generalized hybrid learning (HybLearn) framework. We present the advantages of using projected gradient descent to enforce MR scanner hardware constraints as compared to using added penalties in the cost function. Further, we use the novel HybLearn scheme to jointly learn and compare our results through a retrospective study on fastMRI validation dataset.

Machine Learning,Signal Processing,Optimization and Control

What problem does this paper attempt to address?

The problem that this paper attempts to solve is: in magnetic resonance imaging (MRI), optimize the joint learning method of non - Cartesian \(k\)-space trajectories and image reconstruction networks. Specifically, the author aims to compare and improve existing methods (such as PILOT and BJORK) to obtain better \(k\)-space trajectories and higher image reconstruction quality.

### Problem Background

In compressed - sensing MRI, optimizing the \(k\)-space sampling trajectory and reconstructing images from undersampled \(k\)-space data is a key issue. Existing methods such as PILOT and BJORK attempt to solve this problem by jointly learning the \(k\)-space trajectory and the reconstruction network. However, these methods have some limitations:

1. **PILOT**:
- Using automatic differentiation of the NUFFT operator may lead to sub - optimal local minima.
- Depending on using hardware constraints as penalty terms in the loss function, it requires adjusting hyper - parameters and may affect the overall gradient.

2. **BJORK**:
- Using a more accurate Jacobian approximation of the NUFFT operator, but still relying on penalty terms to enforce hardware constraints.
- Using B - spline curves to parameterize the trajectory, which limits the diversity of trajectory shapes.
- Not using a density compensation mechanism (DCp), which is crucial for obtaining clearer MR images in non - Cartesian deep - learning settings.

### Solution

To overcome these problems, the author proposes a general hybrid learning framework (HybLearn), which uses the projected gradient descent method to enforce MRI scanner hardware constraints instead of adding penalty terms to the loss function. The advantages of this method include:

- **More accurate \(k\)-space coverage**: Especially in the low - frequency region.
- **Higher image quality**: Compared with existing methods, the PSNR is improved by 3 - 4 dB, and the SSIM is improved by about 0.06.

### Experimental Verification

The author conducted experiments on the fastMRI validation dataset through a retrospective study. The results show that the HybLearn method performs well at multiple resolutions and significantly outperforms the PILOT and BJORK methods.

### Conclusion

Through benchmarking PILOT, BJORK, and HybLearn, the author demonstrates the superior performance of the hybrid learning scheme in optimizing \(k\)-space trajectories and improving image quality. Future work includes prospective implementation by modifying T1 and T2 - weighted imaging sequences.

### Formula Summary

The formulas involved in the paper mainly include:

- **Projection operation**: \(\Pi_{Q_N}\) is used to enforce hardware constraints.
- **Forward NUFFT operator**: \(F_S(K)\) is used to simulate the acquisition model.
- **Density compensator**: \(D_S(K)\) is used to estimate density compensation.
- **Reconstruction network**: \(R_K^\theta\) is used to generate the reconstructed image \(\hat{x}\).
- **Loss function**: Combining L1, L2, and MSSIM losses \(L\).

These formulas ensure the rigor and accuracy of the model training and evaluation processes.