SECONDGRAM: Self-conditioned diffusion with gradient manipulation for longitudinal MRI imputation.

Radiology State Required CME US Health Policy
Journal: Patterns (New York, N.Y.)
Published Date:

Abstract

While individual MRI snapshots provide valuable insights, the longitudinal progression in repeated MRIs often holds more significant diagnostic and prognostic value. However, a scarcity of longitudinal datasets, comprising paired initial and follow-up scans, hinders the application of machine learning for crucial sequential tasks. We address this gap by proposing self-conditioned diffusion with gradient manipulation (SECONDGRAM) to generate absent follow-up imaging features, enabling predictions of MRI developments over time and enriching limited datasets through imputation. SECONDGRAM builds on neural diffusion models and introduces two key contributions: self-conditioned learning to leverage much larger, unlinked datasets and gradient manipulation to combat instability and overfitting in a low-data setting. We evaluate SECONDGRAM on the UK Biobank dataset and show that it not only models MRI patterns better than existing baselines but also enhances training datasets to achieve better downstream results over naive approaches.

Authors

  • Brandon Theodorou
    Department of Computer Science, University of Illinois at Urbana-Champaign, Urbana, IL, USA.
  • Anant Dadu
    Center for Alzheimer's and Related Dementias, National Institutes of Health, Bethesda, MD, USA.
  • Mike Nalls
    Center for Alzheimer's and Related Dementias (CARD), National Institute on Aging and National Institute of Neurological Disorders and Stroke, National Institutes of Health, Bethesda, Maryland, USA.
  • Faraz Faghri
    Center for Alzheimer's and Related Dementias, National Institutes of Health, Bethesda, MD, USA.
  • Jimeng Sun
    College of Computing Georgia Institute of Technology Atlanta, GA, USA.

Keywords

No keywords available for this article.