0810

Multidimensional MR Spatiospectral Reconstruction Integrating Subspace Modeling and a Plug&Play Denoiser with Recurrent Features
Ruiyang Zhao1,2, Zepeng Wang1,3, and Fan Lam1,2,3
1Beckman Institute for Advanced Science and Technology, University of illinois Urbana-Champaign, Champaign, IL, United States, 2Department of Electrical and Computer Engineering, University of illinois Urbana-Champaign, Champaign, IL, United States, 3Department of Bioengineering, University of illinois Urbana-Champaign, Champaign, IL, United States

Synopsis

Keywords: AI/ML Image Reconstruction, Machine Learning/Artificial Intelligence, Image reconstruction, High dimensional imaging

Motivation: Multidimensional MR spatiospectral imaging (MD-MRSI) has many applications but is challenging due to high dimensionality and limited SNR. Subspace and learning-based methods have both demonstrated success.

Goal(s): To develop a new MD-MRSI reconstruction method synergizing subspace modeling and a spatiospectral denoiser that can be ‘pre-learned’ without noisy/clean image pairs.

Approach: A self-supervised training strategy was proposed to learn a network-based denoiser combining convolutional, fully-connected, and recurrent features and effectively exploiting multidimensional “correlations”. A plug-and-play ADMM-based algorithm was used to integrate the denoising prior and subspace reconstruction.

Results: Impressive SNR-enhancing reconstruction was demonstrated using simulations and in vivo data from different MD-MRSI acquisitions.

Impact: A new approach is proposed for multidimensional MR spatiospectral image reconstruction integrating low-dimensional modeling and a prelearned denoiser trained via multidimensional interpolation using only noisy data. Potential impacts on quantitative molecular imaging are demonstrated using different MRSI acquisitions.

Introduction

Multiplexed, quantitative molecular imaging via multidimensional MR spatiospectral imaging (MD-MRSI) has many potential applications1-3. But the practical utility of MD-MRSI has been rather limited due to the high dimensionality challenge and intrinsic SNR limits. Learning-based reconstruction has shown potentials in addressing these challenges4-6. However, developing learning-based methods for MD-MRSI faces two major issues: 1) the need to consider multidimensional “correlations” in the data, e.g., space, FID/time and another parameter dimension (e.g., TE) and 2) difficulties in obtaining high-resolution, high-SNR training data unlike typical MRI problems. We present here a novel MD-MRSI reconstruction method that integrated explicit subspace modeling and a pre-learned “multidimensional” denoiser using the Plug-and-Play ADMM (PnP-ADMM) formulation7-8. Specifically, the algorithm alternates between a subspace-constrained update incorporating forward encoding model and a proximal update solved by a denoiser pretrained using the Noise2Void (N2V) concept9-11and noisy data only. Our denoising network was trained via multidimensional space-FID-parameter interpolation to effectively separate spectroscopic signals of interest from noise. A recurrent layer12 was included to exploit signal evolutions along the parameter dimension for better denoising. The effectiveness of our method was demonstrated using simulated and in vivo data from a multi-TE and a diffusion-weighted MRSI acquisitions as application examples.

Theory and Methods

Formulation and algorithm
The proposed MD-MRSI reconstruction formulation integrates a subspace model and an implicit spatiotemporal/spatiospectral regularizer $$$R(.)$$$, i.e.,
$$\hat{\mathbf{U}},\hat{\mathbf{H}},\hat{\mathbf{Z}}=\operatorname{arg}\underset{\mathbf{U},\mathbf{H},\mathbf{Z}}{\operatorname{min}}\left\|\mathcal{F}_{\mathrm{B}}(\mathbf{U}\hat{\mathbf{V}})-\mathbf{d}\right\|_{\mathbf{F}}^2+\lambda_1R(\mathbf{H})+\frac{\beta_1}{2}\left\|A(\mathbf{U} \hat{\mathbf{V}})-\mathbf{H}+\frac{\mathbf{Z}}{\beta_1}\right\|_{\mathbf{F}}^2, (1)$$
where $$$\mathcal{F}_{\mathrm{B}}, \hat{\mathbf{V}}, \mathbf{d}, \mathbf{U}$$$ represents the forward encoding operator, pre-learned MD subspace/basis13-14, noisy $$$(\mathrm{k}, \mathrm{t}, \theta)$$$-space data and the unknown spatial coefficients. $$$\mathbf{H}$$$ and $$$\mathbf{Z}$$$ are the introduced auxiliary variable and Lagrangian multiplier for the ADMM framework, allowing us to enforce the regularization using our prelearned denoiser. $$$A$$$ is an operator accounting for potential resolution difference between denoiser training data and the modeled image $$$( \mathbf{U} \hat{\mathbf{V}})$$$. The problem was solved by sequentially updating $$$\mathbf{U}$$$, $$$\mathbf{H}$$$ and $$$\mathbf{Z}$$$. While the subproblem for $$$\mathbf{U}$$$ is a quadratically regularized subspace reconstruction, the subproblem for $$$\mathbf{H}$$$:
$$\hat{\mathbf{H}}^{k+1}=\operatorname{arg}\underset{\mathbf{H}}{\operatorname{min}} \lambda_1 R(\mathbf{H})+\frac{\beta_1}{2}\left\|A\left(\hat{\mathbf{U}}^k \hat{\mathbf{V}}\right)-\mathbf{H}+\frac{\hat{\mathbf{Z}}^k}{\beta_1}\right\|_{\mathbf{F}}^2, (2)$$
can be solved via:
$$\hat{\mathbf{H}}=f_{\mathrm{W}}\left(A\left(\hat{\mathbf{U}}^k \hat{\mathbf{V}}\right)+\frac{\hat{\mathbf{Z}}^k}{\beta_1}\right), (3)$$
where $$$f_{\mathrm{W}}$$$ is the pretrained denoiser.

Denoiser learning via multidimensional interpolation
Learning denoiser using standard supervised learning is challenging for MD-MRSI. Self-supervised denoising via training an interpolation network to suppress the spatially independent noise has been proposed 9-11. A unique challenge here is to design the network to effectively exploit the signal characteristics across different dimensions. For example, in multi-TE MRSI, the signals can be considered locally correlated in space and globally correlated along the FID and TE dimensions. Therefore, our network was trained to interpolate locally in space but globally across FID and TEs. Specifically, we partitioned the high-dimensional data into spatial patches each containing entire FIDs and all TEs, and randomly masked out voxels at different time points and TEs for training the network to interpolate missing voxels (Fig. 1).
More specifically, our proposed network integrated spatial convolution, temporal fully-connected combinations (along the FID), and an RNN-based feature extraction along TE. The information flow in the first layer can be expressed as:
$$\overrightarrow{\boldsymbol{H}}_{TE_i}=\operatorname{Relu}\left(\boldsymbol{W}_{xh} * \boldsymbol{X}_{TE_i}+\boldsymbol{W}_{hh} * \overrightarrow{\boldsymbol{H}}_{TE_{i-1}}+\boldsymbol{B}\right), (4) $$
$$\overleftarrow{\boldsymbol{H}}_{TE_i}=\operatorname{Relu}\left(\boldsymbol{W}_{xh} * \boldsymbol{X}_{TE_i}+\boldsymbol{W}_{hh} * \overleftarrow{\boldsymbol{H}}_{TE_{i+1}}+\boldsymbol{B}\right),(5)$$
$$\boldsymbol{O}_{TE_i}=\operatorname{Concat}\left(\overrightarrow{\boldsymbol{H}}_{TE_i}, \overleftarrow{\boldsymbol{H}}_{TE_i}\right),(6)$$
where $$$\boldsymbol{X}_{TE_i}$$$ denotes the input patches at $$$i_{th}$$$ TE, $$$\overrightarrow{\boldsymbol{H}}_{TE_i}$$$ the latent space propagated forward, and $$$\overleftarrow{\boldsymbol{H}}_{TE_i}$$$ the one propagated backward. $$$\boldsymbol{W}_{xh}, \boldsymbol{W}_{hh}$$$ and $$$\boldsymbol{B}$$$ represent shared convolution kernels and biases. Multi-TE features were concatenated in the output as $$$\boldsymbol{O}_{TE_i}$$$ and then further taken by a U-net for multidimensional interpolation (Fig. 1). This feature effectively mined information across TEs for improved denoising.

Experiments and Results

Numerical multi-TE 1H-MRSI phantoms with different spatiospectral variations were created (TE=[35,80,160] ms, matrix size=64×64 and 256 FID points) for validation. The network was trained with 6,300 patches from the simulated noisy data, with patches of 8×8×256×3 (TEs). As shown in Fig. 2, the proposed N2V-based joint-TE denoiser demonstrated superior SNR enhancement over the denoisers trained for each TE separately. Improved reconstruction was achieved when using the denoiser as a plug-in prior integrated with the subspace constraint, as shown in Fig. 3.
Both in vivo multi-TE and diffusion-weighted 1H-MRSI data were acquired from healthy volunteers (IRB approved) to evaluate the utility of the proposed method, using sequence previously described in [15,16]. Quantified metabolite $$$T_2$$$ and concentration maps from the multi-TE data in Fig. 4 show apparent improvement for the proposed over competing methods (less noisy and artifact). Spatiospectral reconstructions from the DW-MRSI data were shown in Fig. 5. The proposed method again produced stronger SNR enhancement than competing methods.

Conclusion

We presented a novel multidimensional MRSI reconstruction method. Simulations and in vivo results from different MRSI acquisition scenarios demonstrated the effectiveness and potential wide applicability of our method.

Acknowledgements

This work was supported in part by NSF-CBET 1944249 and NIH R35GM142969.

References

1. Tang, Lihong, et al. "Accelerated J‐resolved 1H‐MRSI with limited and sparse sampling of (k, t1, t2)‐space." MRM. 2021; 85: 30-41.

2. Kirov, Ivan I., and Assaf Tal. "Potential clinical impact of multiparametric quantitative MR spectroscopy in neurological disorders: A review and analysis." MRM. 2020; 83:22-44.

3. Fotso, Kevin, et al. "Diffusion tensor spectroscopic imaging of the human brain in children and adults." MRM. 2017; 78: 1246-1256.

4. Lam, Fan, et al. "High-dimensional MR spatiospectral imaging by integrating physics-based modeling and data-driven machine learning: Current progress and future directions." IEEE SPM. 2023; 40:101-115.

5. Li, Yudu, et al. "Machine learning-enabled high-resolution dynamic deuterium MR spectroscopic imaging." IEEE TMI. 2021; 40:3879-3890.

6. Zhao, R., et al. "MR spatiospectral reconstruction using plug&play denoiser with self-supervised training." In Proc. of ISMRM. 2023, p.0955.

7. Venkatakrishnan, Singanallur V., et al. "Plug-and-play priors for model based reconstruction." In Proc. of IEEE GlobSIP. 2013: 945-948

8. Kamilov, Ulugbek S., et al. "Plug-and-play methods for integrating physical and learned models in computational imaging: Theory, algorithms, and applications." IEEE SPM. 2023; 40: 85-97.

9. Lecoq, Jérôme, et al. "Removing independent noise in systems neuroscience data using DeepInterpolation." Nature Methods. 2021;18:1401-1408.

10. Akçakaya, Mehmet, et al. "Unsupervised deep learning methods for biological image reconstruction and enhancement: An overview from a signal processing perspective." IEEE SPM. 2022; 39:28-44.

11. Krull, Alexander, et al. "Noise2void-learning denoising from single noisy images." In Proc. of Intl. IEEE CVPR. 2019.

12. Schuster, Mike, and Kuldip K. Paliwal. "Bidirectional recurrent neural networks." IEEE TSP. 1997; 45 : 2673-2681.

13. Lam, Fan, et al. "A subspace approach to high‐resolution spectroscopic imaging." MRM. 2014; 71:1349-1357.

14. Lam, Fan, et al. "Ultrafast magnetic resonance spectroscopic imaging using SPICE with learned subspaces." MRM. 2020; 83:377-390.

15. Wang, Zepeng, et al. "High-Resolution brain metabolite T2 mapping using optimized multi-TE MRSI." In Proc. of ISMRM. 2022, p.4998.

16. Wang, Zepeng, and Fan Lam. "High resolution volumetric diffusion-weighted MRSI using a subspace approach." In Proc. of ISMRM. 2021, p.0337

Figures

The proposed N2V denoising network and self-supervised training strategy (using multi-TE MRSI as an example): Spatially local and temporally global patches were extracted. About 6% voxels were randomly masked out at different time points and TEs for each patch input to the network, with output being the original unmasked patches. The network first treated the FID dimension as channels to exploit the global signal dependence. The extracted features at each TE were then integrated by a bidirectional RNN module and further fed to a U-net for multidimensional interpolation.

Simulation results: denoising using just the pretrained network with TE-by-TE training without considering correlation across TEs (as proposed in Ref. 6) versus multi-TE joint training with the proposed recurrent module. Significant SNR improvement was achieved by both strategies, but the proposed joint denoising outperformed the TE-by-TE strategy, as shown in the denoised images, localized spectra (most right column) and overall relative $$$l_2$$$ errors (on the bottom-left of each image).

Spatiospectral reconstructions from simulation: Integrating pretrained denoisers and the subspace model outperformed using subspace or the denoising network alone (Fig. 2), as shown by the reconstructed images (top, columns 2,4 and 6) and relative $$$l_2$$$ error maps (top, column 3,5 and 7). The overall errors are shown in white letters. Smaller errors were obtained using the multi-TE jointly trained denoiser (proposed method). The difference is bigger when the subspace is less accurate (not shown here due to space limit).

In vivo results from a multi-TE MRSI acquisition (TR/TEs=1100/[35,200,245,275] ms, FOV = 220×220×64 mm3, and matrix size=32×32×8): metabolite $$$T_2$$$ maps from different methods are shown in Rows 2-4 and relative concentration maps in Rows 5-7. Both methods incorporating the pretrained denoisers via PnP-ADMM, either trained TE by TE (N2V TE by TE) or jointly (proposed) produced better results than subspace reconstruction (visually noisier). The proposed method yielded better maps, especially $$$T_2$$$, which benefited more from improved long-TE reconstruction.

Results from an in vivo DW-MRSI acquisition (b-values= [0,1500, 3000] s/mm2, FOV = 220×220×56 mm3, and matrix size=32×32×8): Reconstructed b-value-dependent images from each method are shown in Columns 1-3 and localized spectra in the last two columns. Subspace reconstruction with edge-preserving regularization is shown in (a), subspace reconstruction integrating denoisers trained independently for each b-value in (b), and the proposed reconstruction in (c), which produced less noisy images and better lineshapes (indicated by arrows).

Proc. Intl. Soc. Mag. Reson. Med. 32 (2024)
0810
DOI: https://doi.org/10.58530/2024/0810