/EDMSPRec

High-dimensional Embedding Network Derived Prior for CS-MRI

Primary LanguageMATLAB

EDMSPRec

The Code is created based on the method described in the following paper:
High-dimensional Embedding Network Derived Prior for Compressive Sensing MRI Reconstruction
M. Zhang, M. Li, J. Zhou, Y. Zhu, S. Wang, D. Liang, Q. Liu.
Medical Image Analysis, vol. 64, 101717, 2020.
https://www.sciencedirect.com/science/article/abs/pii/S1361841520300815?via%3Dihub

Motivation

Although recent deep learning methodologies have shown excellent performance in fast imaging, the network needs to be retrained for specific sampling patterns and ratios. Therefore, how to explore the network as a general prior and leverage it into the observation constraint flexibly is urgent. In this work, we present an enhanced Deep Mean-Shift Prior (EDMSP) to address the highly MRI under-sampling reconstruction problem. By extending the naive DMSP via integration of multi-model aggre-gation and multi-channel network learning, a high-dimensional embedding network derived prior is formed. Then, we apply the learned prior to single-channel image reconstruction via variable augmen-tation technique. The resulting model is tackled by proximal gradient descent and alternative iteration. Experimental results under various sampling trajectories and acceleration factors consistently demon-strated the superiority of the proposed prior.

Figs

repeat-MEDMSP Schematic illustration of the multi-channel network scheme at the training stage and the auxiliary variables technique used for single-channel intermediate image at the iterative reconstruction phase.

Experiments

Three test images from SIAT.

repeat-MEDMSP repeat-MEDMSP Reconstruction results at 90% radial undersampling. From left to right: DLMRI, PANO, FDLCP, NLR-CS, DC-CNN, DMSP-MWCNN and EDMSPRec.

Table

repeat-MEDMSP

Requirements and Dependencies

MATLAB R2016b
Cuda-9.0
MatConvNet
(https://pan.baidu.com/s/1ZsKlquIHqtgJYlq3iKNsdg Password:p130)
MEDMSPRec_model_v2
(https://pan.baidu.com/s/1voohJTmBSAF1XrHS7GpiRQ Password:8hnx)
Pretrained Model
(https://pan.baidu.com/s/1AdLa9kKiQ2F1_0jc-KS5EA Password:dvhs)

Testing Data

  • Three brain MRI images from SIAT [Dataset]
  • 50 proton-density weighted knee images from FastMRI [Dataset]

Other Related Projects