PRISM: Privacy-preserving Inter-Site MRI Harmonization via Disentangled Representation Learning

Sarang Galada, Tanurima Halder, Kunal Deo, Ram P Krish, Kshitij Jadhav
11/10/2024
eess.IVcs.CV

Official code: saranggalada/PRISM (2 stars, Apache-2.0, last commit 2025-02-18).

Abstract

Multi-site MRI studies often suffer from site-specific variations arising from differences in methodology, hardware, and acquisition protocols, thereby compromising accuracy and reliability in clinical AI/ML tasks. We present PRISM (Privacy-preserving Inter-Site MRI Harmonization), a novel Deep Learning framework for harmonizing structural brain MRI across multiple sites while preserving data privacy. PRISM employs a dual-branch autoencoder with contrastive learning and variational inference to disentangle anatomical features from style and site-specific variations, enabling unpaired image translation without traveling subjects or multiple MRI modalities. Our modular design allows harmonization to any target site and seamless integration of new sites without the need for retraining or fine-tuning. Using multi-site structural MRI data, we demonstrate PRISM's effectiveness in downstream tasks such as brain tissue segmentation and validate its harmonization performance through multiple experiments. Our framework addresses key challenges in medical AI/ML, including data privacy, distribution shifts, model generalizability and interpretability. Code is available at https://github.com/saranggalada/PRISM

View on arXivView PDF

Code Implementations(1)

saranggalada/PRISMOfficial100%
Apache-2.0

PRISM: Privacy-preserving Inter-Site MRI Harmonization via Disentangled Representation Learning (Accepted for Oral Presentation at ISBI 2025!)

20Python, Jupyter NotebookSep 12, 20241 years agoApache-2.0
contrastive-learningfederated-learningprivacy-preserving-machine-learningrepresentation-learningstyle-transfer

Similar Papers

Cite this paper

@article{galada2024prism,
  title  = {PRISM: Privacy-preserving Inter-Site MRI Harmonization via Disentangled Representation Learning},
  author = {Sarang Galada and Tanurima Halder and Kunal Deo and Ram P Krish and Kshitij Jadhav},
  year   = {2024},
  eprint = {2411.06513},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2411.06513v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2411.06513.svg)](https://codeofpaper.com/paper/2411.06513)

Discussion