Automated screening of privacy risks in medical AI
Authors: Konstantinos Vrettos, Eleftherios Tzanis, Haridimos Kondylakis, Evangelia E. Vassalou, Florence X. Doo, Michail E. Klontzas
Organizations: Artificial Intelligence and Translational Imaging (ATI) Lab, Department of Radiology, School of Medicine, University of Crete, Heraklion, Crete, Greece · Division of Radiology, Department of Clinical Science Intervention and Technology (CLINTEC), Karolinska Institute, Huddinge, Sweden · Computer Science Department, University of Crete, Crete, Greece · Institute of Computer Science, FORTH, Greece · Department of Medical Imaging, University Hospital of Heraklion, Crete, Greece · University of Maryland Institute for Health Computing, North Bethesda, MD, USA · Department of Diagnostic Radiology and Nuclear Medicine University of Maryland, Baltimore, MD, USA
Medical artificial intelligence (AI) models are often shared without their training data, assuming this protects the patients who contributed them. However, a model can reveal whether a patient's records were used in training, a privacy breach known as a membership inference attack. This risk is usually reported as a cohort average, which can appear negligible while individual patients are almost certainly exposed. Identifying potentially vulnerable records before developing a production model could make privacy assessment more practical. We measured each patient's vulnerability to membership inference attack, across 7,789 models trained on six clinical datasets in four modalities, including two adapted pretrained medical models. Training choices changed how many patients had high attack scores, but the patients who ranked highest by risk largely stayed the same across the configurations we tested. Stronger weight decay reduced the exposed cohort up to threefold, yet largely the same patients remained most at risk across training recipes, architectures and initializations, including foundation models. Therefore, we developed a lightweight screening tool that uses four small models to identify patients with high risk of exposure before production-model training. We also assessed if flagged patients could be protected without removing their data. Privacy screening should become incorporated into data preparation for clinical AI.
Universitat Rovira i Virgili, Department of Computer Engineering and Mathematics, CYBERCAT-Center for Cybersecurity Research of Catalonia, ComSCIAM-Center for Computational Science and Applied Mathematics Av. Països Catalans 26, 43007 Tarragona, Catalonia
Massachusetts Institute of Technology (MIT), Cambridge, Massachusetts, USA · Broad Institute of MIT and Harvard, Cambridge, Massachusetts, USA · Stanford University, Stanford, California, USA +5