arXiv · 2208.07585
Neural network fragile watermarking with no model performance degradation
Abstract
Deep neural networks are vulnerable to malicious fine-tuning attacks such as data poisoning and backdoor attacks. Therefore, in recent research, it is proposed how to detect malicious fine-tuning of neural network models. However, it usually negatively affects the performance of the protected model. Thus, we propose a novel neural network fragile watermarking with no model performance degradation. In the process of watermarking, we train a generative model with the specific loss function and secret key to generate triggers that are sensitive to the fine-tuning of the target classifier. In the process of verifying, we adopt the watermarked classifier to get labels of each fragile trigger. Then, malicious fine-tuning can be detected by comparing secret keys and labels. Experiments on classic datasets and classifiers show that the proposed method can effectively detect model malicious fine-tuning with no model performance degradation.
Explore related subjects
Keep this discovery
Zhaoxia Yin, Heng Yin, Xinpeng Zhang. 2022-08-16. Neural network fragile watermarking with no model performance degradation. https://doi.org/10.1109/icip46576.2022.9897413
Cite the original work for its findings. Save a collection to share your selection of sources.