Variable splitting network for accelerated parallel MRI reconstruction
The code in this repository implements VS-Net (Fig.1), a model-driven neural network for accelerated parallel MRI reconstruction (see our presentation slides and poster). Specifically, we formulate the generalized parallel compressed sensing reconstruction as an energy minimization problem, for which a variable splitting optimization method is derived. Based on this formulation we propose a novel, end-to-end trainable deep neural network architecture by unrolling the resulting iterative process of such variable splitting scheme. We evaluated VS-Net on complex valued multi-coil knee images for 4-fold and 6-fold acceleration factors showed improved performance (Fig.2).
:--: Fig.1: VS-Net overall architecture (left) and each block in VS-net (right). DB, DCB and WAB stand for Denoiser Block, Data Consistency Block and Weighted Average Block, respectively.
:--: Fig.2: Visual comparison using Cartesian undersampling with AF 4 (top) and 6 (bottom). From left to right: zero-filling, l1-SPIRiT, Variational Network, VS-Net and ground truth. Click here for more visual comparison.
The files in this repository are organized into 5 directories and 1 root directory:
csv
files where the quantitative metrics (PSNR, SSIM and NMSE) over each iteration are savedvs-200.mat
, zero_filling.mat
and reference.mat
. On top of the three mat files, running save_png.py will produce png files.To start the training process with vs_net.py, please follow the following four steps:
Download the data we used for our experiments at GLOBUS.
pip install visdom torch==1.2.1 matplotlib h5py scipy scikit-image
python -m visdom.server
you need to change the path in this python script to where you save the knee data downloaded above. For visualization during training, open your browser and enter http://localhost:8097
If you find this software useful for your project or research. Please give some credits to authors who developed it by citing the following paper. We really appreciate that. If you encounter any problem during installation, please feel free to contact me via [email protected]
[1] Duan J, Schlemper J, Qin C, Ouyang C, Bai W, Biffi C, Bello G, Statton B, O'Regan DP, Rueckert D. VS-Net: Variable splitting network for accelerated parallel MRI reconstruction. arXiv preprint arXiv:1907.10033. MICCAI (2019).