An Efficient Membership Inference Attack for the Diffusion Model by Proximal Initialization

Fei Kong, Jinhao Duan, RuiPeng Ma, Hengtao Shen, Xiaofeng Zhu +2 more
5/26/2023
cs.SDcs.AIcs.LG

Official code: kong13661/PIA (16 stars, last commit 2024-03-08).

Abstract

Recently, diffusion models have achieved remarkable success in generating tasks, including image and audio generation. However, like other generative models, diffusion models are prone to privacy issues. In this paper, we propose an efficient query-based membership inference attack (MIA), namely Proximal Initialization Attack (PIA), which utilizes groundtruth trajectory obtained by $ε$ initialized in $t=0$ and predicted point to infer memberships. Experimental results indicate that the proposed method can achieve competitive performance with only two queries on both discrete-time and continuous-time diffusion models. Moreover, previous works on the privacy of diffusion models have focused on vision tasks without considering audio tasks. Therefore, we also explore the robustness of diffusion models to MIA in the text-to-speech (TTS) task, which is an audio generation task. To the best of our knowledge, this work is the first to study the robustness of diffusion models to MIA in the TTS task. Experimental results indicate that models with mel-spectrogram (image-like) output are vulnerable to MIA, while models with audio output are relatively robust to MIA. {Code is available at \url{https://github.com/kong13661/PIA}}.

View on arXivView PDF

Code Implementations(1)

kong13661/PIAOfficial100%

Official repo for An Efficient Membership Inference Attack for the Diffusion Model by Proximal Initialization

164Cython, PythonMay 24, 20232 years ago

Cite this paper

@article{kong2023efficient,
  title  = {An Efficient Membership Inference Attack for the Diffusion Model by Proximal Initialization},
  author = {Fei Kong and Jinhao Duan and RuiPeng Ma and Hengtao Shen and Xiaofeng Zhu and Xiaoshuang Shi and Kaidi Xu},
  year   = {2023},
  eprint = {2305.18355},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2305.18355v2},
  journal = {ICLR 2023 2023}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2305.18355.svg)](https://codeofpaper.com/paper/2305.18355)

Discussion