Contrast-Informed Augmentation and Domain-Adversarial Training for Adult-to-Neonatal MR Reconstruction Generalization

· Source: Artificial Intelligence · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Computer Vision & Pattern Recognition · Depth: Expert, quick

Summary

Contrast-Informed Augmentation and Domain-Adversarial Training for Adult-to-Neonatal MR Reconstruction Generalization investigates methods to improve the E2E-VarNet's ability to generalize from adult to neonatal magnetic resonance (MR) data. The study compared three training regimes: adult-only, mixed with contrast-informed augmented adult data, and mixed with a domain-adversarial objective. Models were trained on retrospectively undersampled multi-coil adult T2-weighted brain MR data and evaluated on neonatal and adult test data at acceleration factors R=4 and R=8. Results showed that mixed training (Mixed) and mixed domain-adversarial training (Mixed-DAT) significantly outperformed unaugmented adult-only training (Unaug-Only) on neonatal data. At R=4, Mixed-DAT achieved the best performance (SSIM = 0.924 +/- 0.027, PSNR = 33.98 +/- 1.15 dB). At R=8, Mixed-DAT led in SSIM (0.848 +/- 0.031), while Mixed performed best in PSNR (29.56 +/- 0.83 dB). Qualitative analysis indicated Mixed-DAT increased latent representation overlap across data types.

Key takeaway

For Machine Learning Engineers developing medical image reconstruction models, if you face domain shift challenges, consider integrating contrast-informed data augmentation and domain-adversarial training. This approach significantly improves generalization from adult to neonatal MR data, as demonstrated by superior SSIM and PSNR scores. You should evaluate these techniques to enhance model robustness and reduce the need for extensive target domain data.

Key insights

Contrast-informed augmentation and domain-adversarial training enhance deep learning MR reconstruction generalization from adult to neonatal data.

Principles

Method

The E2E-VarNet was trained using retrospectively undersampled adult T2-weighted brain MR data, comparing adult-only, mixed with contrast-informed augmentation, and mixed with domain-adversarial objective.

In practice

Topics

Best for: AI Scientist, Research Scientist, Machine Learning Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.