The pretraining domain outweighs the training objective in setting the privacy-utility trade-off of differentially private medical image analysis

· Source: cs.CV updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Computer Vision & Pattern Recognition · Depth: Expert, quick

Summary

A study on differentially private medical image analysis demonstrates that the pretraining domain significantly outweighs the training objective in determining the privacy-utility trade-off. Researchers trained ConvNeXt classifiers using differentially private stochastic gradient descent from five initializations, varying both objective and domain independently. Evaluated on over 590,000 chest radiographs from five external datasets, supervised pretraining on chest radiographs consistently ranked first in 24 of 25 dataset and budget combinations. Its lead over ImageNet pretraining grew from 2.5 to 14.6 points of macro-averaged area under the receiver operating characteristic curve as privacy budgets tightened, with the domain effect exceeding the objective effect by a factor of 2.2 to 3.4. Pretraining the corpus privately cost about 5 points but still outperformed every public initialization under privacy. Low-Rank Adaptation (LoRA) removed about half the residual gap, and in-domain pretraining improved the worst-performing demographic subgroup.

Key takeaway

For AI Scientists developing differentially private medical imaging models, your strategy for pretraining data is paramount. Prioritize supervised pretraining on in-domain medical datasets, such as chest radiographs, even if it requires private pretraining. This approach significantly improves diagnostic accuracy under strict privacy budgets and enhances fairness for demographic subgroups, outperforming models pretrained on generic public datasets like ImageNet. Consider integrating Low-Rank Adaptation (LoRA) to further optimize utility.

Key insights

Pretraining domain is more critical than the training objective for utility in differentially private medical imaging.

Principles

Method

ConvNeXt classifiers were trained with differentially private stochastic gradient descent from five initializations, varying pretraining objective and domain, then evaluated on 590,000 chest radiographs.

In practice

Topics

Best for: Computer Vision Engineer, AI Scientist, Research Scientist, AI Security Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CV updates on arXiv.org.