REC_XPDNet_CC359_12_channel_poisson2d_5x_10x_NNEstimationCSM

Model Overview

XPDNet for 5x & 10x accelerated MRI Reconstruction on the CC359 dataset.

ATOMMIC: Training

To train, fine-tune, or test the model you will need to install ATOMMIC. We recommend you install it after you've installed latest Pytorch version.

pip install atommic['all']

How to Use this Model

The model is available for use in ATOMMIC, and can be used as a pre-trained checkpoint for inference or for fine-tuning on another dataset.

Corresponding configuration YAML files can be found here.

Automatically instantiate the model

pretrained: true
checkpoint: https://huggingface.co/wdika/REC_XPDNet_CC359_12_channel_poisson2d_5x_10x_NNEstimationCSM/blob/main/REC_XPDNet_CC359_12_channel_poisson2d_5x_10x_NNEstimationCSM.atommic
mode: test

Usage

You need to download the CC359 dataset to effectively use this model. Check the CC359 page for more information.

Model Architecture

model:
  model_name: XPDNet
  num_primal: 5
  num_dual: 1
  num_iter: 10
  use_primal_only: true
  kspace_model_architecture: CONV
  kspace_in_channels: 2
  kspace_out_channels: 2
  dual_conv_hidden_channels: 16
  dual_conv_num_dubs: 2
  dual_conv_batchnorm: false
  image_model_architecture: MWCNN
  imspace_in_channels: 2
  imspace_out_channels: 2
  mwcnn_hidden_channels: 16
  mwcnn_num_scales: 0
  mwcnn_bias: true
  mwcnn_batchnorm: false
  normalize_image: true
  dimensionality: 2
  reconstruction_loss:
    l1: 0.1
    ssim: 0.9
  estimate_coil_sensitivity_maps_with_nn: true

Training

  optim:
    name: adamw
    lr: 1e-4
    betas:
      - 0.9
      - 0.999
    weight_decay: 0.0
    sched:
        name: CosineAnnealing
        min_lr: 0.0
        last_epoch: -1
        warmup_ratio: 0.1

trainer:
  strategy: ddp_find_unused_parameters_false
  accelerator: gpu
  devices: 1
  num_nodes: 1
  max_epochs: 20
  precision: 16-mixed
  enable_checkpointing: false
  logger: false
  log_every_n_steps: 50
  check_val_every_n_epoch: -1
  max_steps: -1

Performance

To compute the targets using the raw k-space and the chosen coil combination method, accompanied with the chosen coil sensitivity maps estimation method, you can use targets configuration files.

Evaluation can be performed using the evaluation script for the reconstruction task, with --evaluation_type per_slice.

Results

Evaluation against RSS targets

5x: MSE = 0.004192 +/- 0.004255 NMSE = 0.06401 +/- 0.06475 PSNR = 24.27 +/- 4.135 SSIM = 0.7609 +/- 0.09962

10x: MSE = 0.00581 +/- 0.00445 NMSE = 0.08987 +/- 0.07376 PSNR = 22.65 +/- 3.225 SSIM = 0.6997 +/- 0.1119

Limitations

This model was trained on the CC359 using a UNet coil sensitivity maps estimation and might differ from the results reported on the challenge leaderboard.

References

[1] ATOMMIC

[2] Beauferris, Y., Teuwen, J., Karkalousos, D., Moriakov, N., Caan, M., Yiasemis, G., Rodrigues, L., Lopes, A., Pedrini, H., Rittner, L., Dannecker, M., Studenyak, V., Gröger, F., Vyas, D., Faghih-Roohi, S., Kumar Jethi, A., Chandra Raju, J., Sivaprakasam, M., Lasby, M., … Souza, R. (2022). Multi-Coil MRI Reconstruction Challenge—Assessing Brain MRI Reconstruction Models and Their Generalizability to Varying Coil Configurations. Frontiers in Neuroscience, 16. https://doi.org/10.3389/fnins.2022.919186