Are You Stealing My Model? Sample Correlation for Fingerprinting Deep Neural Networks
This addresses the threat of model stealing for commercial AI service providers, offering a robust detection method that is incremental but improves upon existing fingerprinting techniques.
The paper tackles the problem of detecting stolen deep neural network models by proposing a method based on sample correlation (SAC), which achieves state-of-the-art performance with high AUC scores across various datasets and model architectures, even under adversarial training or transfer learning scenarios.
An off-the-shelf model as a commercial service could be stolen by model stealing attacks, posing great threats to the rights of the model owner. Model fingerprinting aims to verify whether a suspect model is stolen from the victim model, which gains more and more attention nowadays. Previous methods always leverage the transferable adversarial examples as the model fingerprint, which is sensitive to adversarial defense or transfer learning scenarios. To address this issue, we consider the pairwise relationship between samples instead and propose a novel yet simple model stealing detection method based on SAmple Correlation (SAC). Specifically, we present SAC-w that selects wrongly classified normal samples as model inputs and calculates the mean correlation among their model outputs. To reduce the training time, we further develop SAC-m that selects CutMix Augmented samples as model inputs, without the need for training the surrogate models or generating adversarial examples. Extensive results validate that SAC successfully defends against various model stealing attacks, even including adversarial training or transfer learning, and detects the stolen models with the best performance in terms of AUC across different datasets and model architectures. The codes are available at https://github.com/guanjiyang/SAC.