Backdoor Secrets Unveiled: Identifying Backdoor Data with Optimized Scaled Prediction Consistency

Soumyadeep Pal,Yuguang Yao,Ren Wang,Bingquan Shen,Sijia Liu
2024-03-16
Abstract:Modern machine learning (ML) systems demand substantial training data, often resorting to external sources. Nevertheless, this practice renders them vulnerable to backdoor poisoning attacks. Prior backdoor defense strategies have primarily focused on the identification of backdoored models or poisoned data characteristics, typically operating under the assumption of access to clean data. In this work, we delve into a relatively underexplored challenge: the automatic identification of backdoor data within a poisoned dataset, all under realistic conditions, i.e., without the need for additional clean data or without manually defining a threshold for backdoor detection. We draw an inspiration from the scaled prediction consistency (SPC) technique, which exploits the prediction invariance of poisoned data to an input scaling factor. Based on this, we pose the backdoor data identification problem as a hierarchical data splitting optimization problem, leveraging a novel SPC-based loss function as the primary optimization objective. Our innovation unfolds in several key aspects. First, we revisit the vanilla SPC method, unveiling its limitations in addressing the proposed backdoor identification problem. Subsequently, we develop a bi-level optimization-based approach to precisely identify backdoor data by minimizing the advanced SPC loss. Finally, we demonstrate the efficacy of our proposal against a spectrum of backdoor attacks, encompassing basic label-corrupted attacks as well as more sophisticated clean-label attacks, evaluated across various benchmark datasets. Experiment results show that our approach often surpasses the performance of current baselines in identifying backdoor data points, resulting in about 4%-36% improvement in average AUROC. Codes are available at
Machine Learning,Artificial Intelligence,Cryptography and Security
What problem does this paper attempt to address?
The paper attempts to address the problem of automatically identifying backdoor samples in datasets that have been injected with backdoor data under real-world conditions. Specifically, the goal of the paper is to accurately identify backdoor data in contaminated datasets without the need for additional clean data (i.e., without access to an extra clean dataset) and without manually defining detection thresholds. This is a more practical and challenging problem compared to existing methods, as current approaches typically require clean data or rely on manually set thresholds to identify backdoor data. The paper achieves this goal by introducing a novel method called Mask-Aware Scaled Prediction Consistency (MSPC), which leverages the prediction consistency characteristics of backdoor data and combines it with bi-level optimization techniques for hierarchical data segmentation. Experimental results show that this method outperforms existing baseline methods on various benchmark datasets for different types of backdoor attacks.