Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Model Inversion Robustness: Can Transfer Learning Help?

About

Model Inversion (MI) attacks aim to reconstruct private training data by abusing access to machine learning models. Contemporary MI attacks have achieved impressive attack performance, posing serious threats to privacy. Meanwhile, all existing MI defense methods rely on regularization that is in direct conflict with the training objective, resulting in noticeable degradation in model utility. In this work, we take a different perspective, and propose a novel and simple Transfer Learning-based Defense against Model Inversion (TL-DMI) to render MI-robust models. Particularly, by leveraging TL, we limit the number of layers encoding sensitive information from private training dataset, thereby degrading the performance of MI attack. We conduct an analysis using Fisher Information to justify our method. Our defense is remarkably simple to implement. Without bells and whistles, we show in extensive experiments that TL-DMI achieves state-of-the-art (SOTA) MI robustness. Our code, pre-trained models, demo and inverted data are available at: https://hosytuyen.github.io/projects/TL-DMI

Sy-Tuyen Ho, Koh Jun Hao, Keshigeyan Chandrasegaran, Ngoc-Bao Nguyen, Ngai-Man Cheung• 2024

Related benchmarks

TaskDatasetResultRank
Model Inversion DefenseCelebA
Accuracy91.12
64
Model Inversion DefenseCelebA 64x64
Accuracy83.41
41
PPA Model Inversion AttackFaceScrub 224x224 Dpriv = Facescrub, Dpub = FFHQ (test)
Accuracy91.12
27
Model InversionCelebA (private)
Accuracy86.7
24
Model Inversion DefenseFacescrub 224x224 (test)
Accuracy93.01
21
Image ClassificationCelebA Private Public
Accuracy86.7
12
Image ClassificationCelebA (Private) FFHQ (Public)
Accuracy86.7
12
Model Inversion AttackCelebA 64x64
Accuracy86.7
12
Model Inversion DefenseFACESCRUB (test)
Accuracy93.01
6
User study on image reconstruction similarityFaceScrub 530 classes
Number of Similar Selections537
4
Showing 10 of 15 rows

Other info

Code

Follow for update