English

A Calibrated Memorization Index (MI) for Detecting Training Data Leakage in Generative MRI Models

Computer Vision and Pattern Recognition 2026-02-16 v1

Abstract

Image generative models are known to duplicate images from the training data as part of their outputs, which can lead to privacy concerns when used for medical image generation. We propose a calibrated per-sample metric for detecting memorization and duplication of training data. Our metric uses image features extracted using an MRI foundation model, aggregates multi-layer whitened nearest-neighbor similarities, and maps them to a bounded \emph{Overfit/Novelty Index} (ONI) and \emph{Memorization Index} (MI) scores. Across three MRI datasets with controlled duplication percentages and typical image augmentations, our metric robustly detects duplication and provides more consistent metric values across datasets. At the sample level, our metric achieves near-perfect detection of duplicates.

Keywords

Cite

@article{arxiv.2602.13066,
  title  = {A Calibrated Memorization Index (MI) for Detecting Training Data Leakage in Generative MRI Models},
  author = {Yash Deo and Yan Jia and Toni Lassila and Victoria J Hodge and Alejandro F Frang and Chenghao Qian and Siyuan Kang and Ibrahim Habli},
  journal= {arXiv preprint arXiv:2602.13066},
  year   = {2026}
}

Comments

Accepted in ISBI 2026

R2 v1 2026-07-01T10:35:31.905Z