0544

PhyCHarm : Physics-Constrained Deep Neural Networks for Multi-Scanner Harmonization
Gawon Lee1, Junhyeok Lee1, Dong Hye Ye2, and Se-Hong Oh1
1Biomedical Engineering, Hankuk University of Foreign Studies, Yongin-si, Korea, Republic of, 2Computer Science, Georgia State University, Atlanta, GA, United States

Synopsis

Keywords: Analysis/Processing, Machine Learning/Artificial Intelligence, Data Harmonization, Bloch equation

Motivation: The MR scanner effect in a multi-site dataset can affect bias in statistical analysis or reduce generality in deep neural networks.

Goal(s): We aim to suggest a MR physics-informed harmonization framework (PhyCHarm) that generates consistent quantitative maps and harmonized T1w images.

Approach: We introduce a Quantitative Maps Generator and a Harmonization Network to be trained with a constraint loss based on a signal equation.

Results: PhyCHarm shows the highest evaluation scores in both networks and consistent segmentation accuracy in the downstream task (FSL FAST GM and WM segmentation).

Impact: PhyCHarm works based on the Bloch equation. PhyCHarm enables us to reduce scanner effects efficiently in the dataset before conducting test/retest, longitudinal, or multi-site studies. It can be helpful to ensure deep neural networks' generality.

Introduction

Variations in the MR scanner, such as differences in gradient conditions, coil sensitivity, and the optimal values of each scan parameter, can impact MRI signals. These variations can introduce bias in results when dealing with large multi-site MRI datasets for statistical analysis or deep neural network training. To address this, harmonization methods have been suggested with the united acquisition tool or deep learning-based methods1-8. However, generative network-based methods have shown a risk of generating inaccurate anatomical features2,4,6. This study suggests an end-to-end Physics-Constrained deep neural network for multi-site MR Harmonization, PhyCHarm. (PhyCHarm is an improved version of the 2023 ISMRM abstract9). PhyCHarm ensures the high quality of harmonized T1w while reducing motion artifacts in generated quantitative maps by incorporating the Bloch equation as a training constraint.

Methods

The PhyCHarm has two networks: (1) Quantitative Maps Generator and (2) Harmonization Network. Figure 1 shows the inference pipeline. The Quantitative Maps Generator generates T1-map and M0-map, and then the Bloch equation is utilized to calculate constrained T1w images. The Harmonization Network works to harmonize T1w images across Siemens, GE, and Philips.
Quantitative Maps Generator
1) Dataset
We used an open dataset, MICA-MICs10, consisting of INV1, INV2, T1w, and T1-map acquired from MP2RAGE sequence11. M0-map was generated from INV2 and T1-map through the inversion recovery signal equation. We used 20 subjects for training, 5 for validation, and 25 for inference.
2) Preprocessing
Brain reorientation was conducted to align axial slices along the z-axis using FSL12, FOV crop was carried out using ANTsPy13, and skull stripping was completed using HD-BET14.
3) Training Quantitative Maps Generator
Figure 2(A) illustrates the training pipeline of the Quantitative Maps Generator. The network was trained using 2D-U-Net15, MSE loss, and the ADAM16 optimizer with a learning rate 0.001. The total loss ($$$\mathcal{L}_{total}$$$) is defined as a weighted sum of the reconstruction loss ($$$\mathcal{L}_{T_{1}}$$$, $$$\mathcal{L}_{M_{0}}$$$) and the consistency loss ($$$\mathcal{L}_{cons}$$$).
$$\mathcal{L}_{total} = \lambda_{1}\times\mathcal{L}_{T_{1}} + \lambda_{2}\times\mathcal{L}_{M_{0}}+\lambda_{3}\times{L}_{cons}$$
where λ1=1, λ2=1, λ3=1e-6. These values were defined experimentally.
The reconstruction loss was calculated between the predicted maps and the ground truth. The consistency loss was computed between the constrained T1w (ConsT1w) and the input T1w. The ConsT1w is generated as below:
$$T1w = M0 \times (1-2\times exp^{-\frac{TI}{T1}}+exp^{-\frac{TR}{T1}})\times exp^{-\frac{TE}{T2}}$$
$$consT1w' = \hat{M0}\times(1-2\times exp^{-\frac{TI}{\hat{T1}}}+exp^{-\frac{TR}{\hat{T1}}})$$
$$\hat{T2}term =\frac{T1w}{consT1w'}
=exp^{-\frac{TE}{T2}}$$
$$consT1w =\hat{M0}\times(1-2\times exp^{-\frac{TI}{\hat{T1}}}+exp^{-\frac{TR}{\hat{T1}}})\times\hat{T2}term $$
By using the inversion-recovery signal equation for T1w, we generated without the T2 term ($$$exp^{-\frac{TE}{T2}}$$$), employing a predicted M0map ($$$\hat{M0}$$$), a predicted T1map ($$$\hat{T1}$$$), TI / TR = 2830 / 5000 ms. To generate consT1w with $$$\hat{T2}term$$$, we calculated $$$\hat{T2}term$$$ by dividing the input T1w by consT1w'. Then, consT1w is used as a training constraint.
Harmonization Network
1) Dataset
T1w images (IRB-approved) of four healthy traveling subjects from three scanners at different sites were used: Siemens Trio (3T), GE SIGNA (3T), and Philips Ingenia CS (3T).
2) Preprocessing
The preprocessing method used for the Quantitative Maps Generator was utilized for the Harmonization Network dataset. To standardize the voxel size to 0.8 iso-voxel, we applied resampling to the GE dataset using spline interpolation. Subsequently, an affine registration was utilized to align the spatial orientation of the Siemens and Philips datasets to match that of the GE dataset. Finally, N4 bias correction17 was applied.
3) Training Harmonization Network
Figure 2(B) displays the training pipeline of the Harmonization Network. It was trained to generate a harmonized T1w image from the constrained T1w image for each pair of the traveling dataset: (a) GE and Siemens, (b) Siemens and Philips, and (c) Philips and GE. The pre-trained Quantitative Maps Generator was utilized to generate T1-map and M0-map of each source scanner’s T1w images while its parameters were all fixed. The ConsT1w was generated using the following equations:
$$consT1w_{source}'= \hat{M0}_{source}\times(1-2\times exp^{-\frac{TI_{source}}{\hat{T1}_{source}}}+exp^{-\frac{TR_{source}}{\hat{T1}_{source}}})$$
$$\hat{T2}term=\frac{T1w_{source}}{consT1w_{source}'} = exp^{-\frac{TE}{T2}}$$
$$\hat{T2} = -\frac{TE_{source}}{\log{\hat{T2}term}}$$
$$consT1w_{target} = \hat{M0}_{source}\times(1-2\times exp^{-\frac{TI_{target}}{\hat{T1_{source}}}}+exp^{-\frac{TR_{target}}{\hat{T1}_{source}}})\times exp^{-\frac{TE_{target}}{\hat{T2}}}$$
where consT1wtarget is the input of the Harmonization Network.
The Harmonization Network was trained using 2D U-Net, the reconstruction loss based on the MSE loss, and the ADAM optimizer with a learning rate of 0.001. To avoid over-fitting, we applied 4-fold cross-validation and dropout with 0.1 of the probability.

Results

Figure 3 shows the evaluation scores and the impact of motion artifact minimization achieved using the Quantitative Maps Generator. Figure 4 represents the comparison results of PhyCHarm with U-Net and Pix2Pix18. Figure 5 compares segmentation consistency for GM and WM through FSL FAST.

Discussion and Conclusion

In this study, we provide evidence that incorporating the Bloch equation during the training of deep neural networks results in improved quality for both harmonized T1w images and quantitative maps, notwithstanding the restricted size of our training dataset.

Acknowledgements

This work was supported by the National Research Foundation of Korea (NRF) grant funded by the Korea government (MSIT) (NRF-2023R1A2C1007292)

References

1. Dewey BE, Zhao C, Reinhold JC, et al. DeepHarmony: A deep learning approach to contrast harmonization across scanner changes. Magnetic resonance imaging. 2019;64:160-70.
2. Zuo L, Dewey BE, Liu Y, et al. Unsupervised MR harmonization by learning disentangled representations using information bottleneck theory. NeuroImage. 2021;243:118569.
3. Moyer D, Ver Steeg G, Tax CM, Thompson PM. Scanner invariant representations for diffusion MRI harmonization. Magnetic resonance in medicine. 2020;84(4):2174-89.
4. Torbati ME, Minhas DS, Laymon CM, et al. MISPEL: A supervised deep learning harmonization method for multi-scanner neuroimaging data. Medical image analysis. 2023;89:102926.
5. Ren M, Dey N, Fishbaugh J, Gerig G. Segmentation-renormalized deep feature modulation for unpaired image harmonization. IEEE transactions on medical imaging. 2021;40(6):1519-30.
6. Guan H, Liu Y, Yang E, et al. Multi-site MRI harmonization via attention-guided deep domain adaptation for brain disorder identification. Medical image analysis. 2021;71:102076.
7. Bashyam VM, Doshi J, Erus G, et al. Deep generative medical image harmonization for improving cross‐site generalization in deep learning predictors. Journal of Magnetic Resonance Imaging. 2022;55(3):908-16.
8. Karakuzu A, Biswas L, Cohen‐Adad J, Stikov N. Vendor‐neutral sequences and fully transparent workflows improve inter‐vendor reproducibility of quantitative MRI. Magnetic Resonance in Medicine. 2022;88(3):1212-28.
9. Lee G, Lee J, Ye D-H, Oh S-H. Bloch-informed neural network for multi-scanner MR Harmonization. International Society for Magnetic Resonance in Medicine, 2023.
10. Royer J, Rodríguez-Cruces R, Tavakol S, et al. An open MRI dataset for multiscale neuroscience. Scientific Data. 2022;9(1):569.
11. Marques JP, Kober T, Krueger G, et al. MP2RAGE, a self bias-field corrected sequence for improved segmentation and T1-mapping at high field. Neuroimage. 2010;49(2):1271-81.
12. Smith SM, Jenkinson M, Woolrich MW, et al. Advances in functional and structural MR image analysis and implementation as FSL. Neuroimage. 2004;23:S208-S19.
13. Tustison NJ, Cook PA, Holbrook AJ, et al. The ANTsX ecosystem for quantitative biological and medical imaging. Scientific reports. 2021;11(1):9068.
14. Schell M, Tursunova I, Fabian I, et al. Automated brain extraction of multi-sequence MRI using artificial neural networks. 2019. European Congress of Radiology-ECR 2019.
15. Ronneberger O, Fischer P, Brox T. U-net: Convolutional networks for biomedical image segmentation. Medical Image Computing and Computer-Assisted Intervention–MICCAI 2015: 18th International Conference, Munich, Germany, October 5-9, 2015, Proceedings, Part III 18, 2015. Springer: 234-41.
16. Kingma DP, Ba J. Adam: A method for stochastic optimization. arXiv preprint arXiv:1412.6980. 2014.
17. Tustison NJ, Avants BB, Cook PA, et al. N4ITK: improved N3 bias correction. IEEE transactions on medical imaging. 2010;29(6):1310-20.
18. Isola P, Zhu J-Y, Zhou T, Efros AA. Image-to-image translation with conditional adversarial networks. Proceedings of the IEEE conference on computer vision and pattern recognition, 2017. 1125-34.

Figures

Figure 1 Inference pipeline. The T1w image from Scanner A is translated into the domain of the T1w image from Scanner B through the following steps: generating quantitative maps, calculating the Bloch equation for a constrained T1w image, and harmonizing the constrained T1w image to the target scanner's T1w image.

Figure 2 Training pipeline. For (A), the Quantitative Maps Generator, T1-map, and M0-map are generated from the T1w image. It was trained by minimizing the reconstruction loss between the predicted quantitative maps and the ground truth, as well as the consistency loss between the input T1w image and the constrained T1w image. For (B) the Harmonization Network, we used the pre-trained Quantitative Maps Generator while keeping all parameters fixed. The Harmonization Network was trained to refine the quality of the constrained T1w image by reducing the reconstruction loss .

Figure 3 Quantitative and qualitative evaluation of the Quantitative Maps Generator. (1) shows the mean evaluation scores for 25 subjects depending on the weight value of each loss. When each weight value ratio is 1, 1, and $$$1e^{-6}$$$, SSIM, PSNR, and NRMSE are highest for T1-map. For M0-map, a weight value of $$$1e^{-8}$$$ for the consistency loss outperforms $$$1e^{-6}$$$ in SSIM, PSNR, and NRMSE but introduces more motion artifacts, as shown in (2). The weight value of $$$1e^{-6}$$$ contributes to motion artifacts compensation compared to $$$1e^{-8}$$$.

Figure 4 Quantitative and qualitative evaluation of the Harmonization Network. (1) is the comparison of SSIM and PSNR for the harmonization results by U-Net, Pix2Pix, and PhyCHarm. These scores are computed based on the average results through 4-fold cross-validation. PhyCHarm shows the highest SSIM and PSNR across all harmonization cases. (2) are the qualitative comparison results. PhyCHarm preserves well small structures with hyper/hypo intensity compared to U-Net and Pix2Pix, as highlighted in red arrows.

Figure 5 Dice score comparison of FSL FAST GM and WM segmentation. This figure shows the mean DICE score across all subjects. PhyCHarm shows an improved DICE score than U-Net or Pix2Pix.

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