arXiv:2510.22274cs.CRcs.LG2025-10

提出通用防御框架SecureLearn,保护多分类模型免受数据投毒攻击

SecureLearn -- An Attack-agnostic Defense for Multiclass Machine Learning Against Data Poisoning Attacks

  • 采用数据清洗与面向特征的对抗训练双层防御机制
  • 在10%-20%污染率下仍保持90%以上准确率,F1超75%
  • 适用于随机森林、决策树等传统模型,通用性强

数据投毒攻击会破坏机器学习模型性能,现有防御多针对特定攻击或算法,且主要聚焦深度神经网络或二分类任务。本文提出SecureLearn,一种针对多分类模型的两层攻击无关防御方法,包含数据净化和新型特征导向对抗训练。通过构建三维评估矩阵(攻击类型、数据净化、对抗训练),在随机森林(RF)、决策树(DT)、高斯朴素贝叶斯(GNB)和多层感知机(MLP)上,基于三个公开数据集,评估其在10%-20%污染率下的表现。对比三种投毒攻击与两种现有方法,结果表明SecureLearn显著提升模型鲁棒性,在传统模型上维持准确率>90%,召回率与F1分数>75%;对神经网络更达97%的召回率与F1分数,验证其跨算法泛化能力。

原文摘要 · Abstract (English)

Data poisoning attacks are a potential threat to machine learning (ML) models, aiming to manipulate training datasets to disrupt their performance. Existing defenses are mostly designed to mitigate specific poisoning attacks or are aligned with particular ML algorithms. Furthermore, most defenses are developed to secure deep neural networks or binary classifiers. However, traditional multiclass classifiers need attention to be secure from data poisoning attacks, as these models are significant in developing multi-modal applications. Therefore, this paper proposes SecureLearn, a two-layer attack-agnostic defense to defend multiclass models from poisoning attacks. It comprises two components of data sanitization and a new feature-oriented adversarial training. To ascertain the effectiveness of SecureLearn, we proposed a 3D evaluation matrix with three orthogonal dimensions: data poisoning attack, data sanitization and adversarial training. Benchmarking SecureLearn in a 3D matrix, a detailed analysis is conducted at different poisoning levels (10%-20%), particularly analysing accuracy, recall, F1-score, detection and correction rates, and false discovery rate. The experimentation is conducted for four ML algorithms, namely Random Forest (RF), Decision Tree (DT), Gaussian Naive Bayes (GNB) and Multilayer Perceptron (MLP), trained with three public datasets, against three poisoning attacks and compared with two existing mitigations. Our results highlight that SecureLearn is effective against the provided attacks. SecureLearn has strengthened resilience and adversarial robustness of traditional multiclass models and neural networks, confirming its generalization beyond algorithm-specific defenses. It consistently maintained accuracy above 90%, recall and F1-score above 75%. For neural networks, SecureLearn achieved 97% recall and F1-score against all selected poisoning attacks.

数据投毒多分类防御机制鲁棒性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。