用少量数据净化受控恶意软件分类器,1%数据即可让攻击成功率从100%降到接近0%
PBP: Post-training Backdoor Purification for Malware Classifiers
- 利用神经网络激活分布变化检测后门,不依赖特定触发机制
- 仅需1%训练数据,将攻击成功率从100%降至几乎为0
- 适合模型已部署后的后处理净化,适用于MLaaS场景
近年来,机器学习在网络安全中的应用带来了新挑战,尤其是针对恶意软件分类器的后门投毒攻击。攻击者可向公开恶意软件库注入恶意样本,污染训练数据,导致模型误分类。现有防御多依赖集成模型输出分歧检测中毒样本,但不适用于机器学习即服务(MLaaS)或模型训练后修复场景。为此,我们提出PBP,一种无需假设特定后门嵌入方式的后训练防御方法。该方法利用后门攻击对神经网络激活分布的影响——使各层激活变为混合分布。通过调节批量归一化层的统计量,可引导带毒模型表现如干净模型。实验在两个数据集、两种后门方法及多种攻击配置下验证,仅需1%训练数据即能将攻击成功率从100%降至接近0%,相比基线方法提升100倍。代码已开源。
原文摘要 · Abstract (English)
In recent years, the rise of machine learning (ML) in cybersecurity has brought new challenges, including the increasing threat of backdoor poisoning attacks on ML malware classifiers. For instance, adversaries could inject malicious samples into public malware repositories, contaminating the training data and potentially misclassifying malware by the ML model. Current countermeasures predominantly focus on detecting poisoned samples by leveraging disagreements within the outputs of a diverse set of ensemble models on training data points. However, these methods are not suitable for scenarios where Machine Learning-as-a-Service (MLaaS) is used or when users aim to remove backdoors from a model after it has been trained. Addressing this scenario, we introduce PBP, a post-training defense for malware classifiers that mitigates various types of backdoor embeddings without assuming any specific backdoor embedding mechanism. Our method exploits the influence of backdoor attacks on the activation distribution of neural networks, independent of the trigger-embedding method. In the presence of a backdoor attack, the activation distribution of each layer is distorted into a mixture of distributions. By regulating the statistics of the batch normalization layers, we can guide a backdoored model to perform similarly to a clean one. Our method demonstrates substantial advantages over several state-of-the-art methods, as evidenced by experiments on two datasets, two types of backdoor methods, and various attack configurations. Notably, our approach requires only a small portion of the training data -- only 1\% -- to purify the backdoor and reduce the attack success rate from 100\% to almost 0\%, a 100-fold improvement over the baseline methods. Our code is available at https://github.com/judydnguyen/pbp-backdoor-purification-official.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。