English

On the Efficacy of Differentially Private Few-shot Image Classification

Machine Learning 2023-12-20 v3 Cryptography and Security Machine Learning

Abstract

There has been significant recent progress in training differentially private (DP) models which achieve accuracy that approaches the best non-private models. These DP models are typically pretrained on large public datasets and then fine-tuned on private downstream datasets that are relatively large and similar in distribution to the pretraining data. However, in many applications including personalization and federated learning, it is crucial to perform well (i) in the few-shot setting, as obtaining large amounts of labeled data may be problematic; and (ii) on datasets from a wide variety of domains for use in various specialist settings. To understand under which conditions few-shot DP can be effective, we perform an exhaustive set of experiments that reveals how the accuracy and vulnerability to attack of few-shot DP image classification models are affected as the number of shots per class, privacy level, model architecture, downstream dataset, and subset of learnable parameters in the model vary. We show that to achieve DP accuracy on par with non-private models, the shots per class must be increased as the privacy level increases. We also show that learning parameter-efficient FiLM adapters under DP is competitive with learning just the final classifier layer or learning all of the network parameters. Finally, we evaluate DP federated learning systems and establish state-of-the-art performance on the challenging FLAIR benchmark.

Keywords

Cite

@article{arxiv.2302.01190,
  title  = {On the Efficacy of Differentially Private Few-shot Image Classification},
  author = {Marlon Tobaben and Aliaksandra Shysheya and John Bronskill and Andrew Paverd and Shruti Tople and Santiago Zanella-Beguelin and Richard E Turner and Antti Honkela},
  journal= {arXiv preprint arXiv:2302.01190},
  year   = {2023}
}

Comments

49 pages, 24 figures; published in TMLR 12/2023 https://openreview.net/forum?id=hFsr59Imzm