用增强数据作负样本,提升网络入侵检测的准确率与效率
Contrastive Self-Supervised Network Intrusion Detection using Augmented Negative Pairs
- 将增强后的数据视为负样本,利用良性流量样本作为正样本进行对比学习
- 在Lycos2017数据集上,二分类准确率超越现有自监督与异常检测方法
- 预训练后微调少量标注数据即达更好多分类性能,适合标注稀缺场景
网络入侵检测仍是网络安全的关键挑战。尽管监督学习模型表现优异,但依赖大量标注数据使其难以在真实场景应用。仅基于正常流量训练的异常检测方法误报率高,实用性受限。近期自监督学习通过学习正常流量的判别性表征,在降低误报率方面取得进展,其核心是使相似(正)样本更接近、不相似(负)样本更远离。现有方法通过数据增强生成正样本,将其他样本视为负样本。本文提出一种新范式:对比学习增强负样本(CLAN),将增强后的样本视为负样本(代表潜在恶意分布),而其他良性样本作为正样本。该方法在良性流量预训练后,显著提升分类准确率与推理效率。在Lycos2017数据集上的实验表明,该方法在二分类任务中优于现有自监督与异常检测技术。此外,在有限标注数据上微调后,其多分类性能也优于现有自监督模型。
原文摘要 · Abstract (English)
Network intrusion detection remains a critical challenge in cybersecurity. While supervised machine learning models achieve state-of-the-art performance, their reliance on large labelled datasets makes them impractical for many real-world applications. Anomaly detection methods, which train exclusively on benign traffic to identify malicious activity, suffer from high false positive rates, limiting their usability. Recently, self-supervised learning techniques have demonstrated improved performance with lower false positive rates by learning discriminative latent representations of benign traffic. In particular, contrastive self-supervised models achieve this by minimizing the distance between similar (positive) views of benign traffic while maximizing it between dissimilar (negative) views. Existing approaches generate positive views through data augmentation and treat other samples as negative. In contrast, this work introduces Contrastive Learning using Augmented Negative pairs (CLAN), a novel paradigm for network intrusion detection where augmented samples are treated as negative views - representing potentially malicious distributions - while other benign samples serve as positive views. This approach enhances both classification accuracy and inference efficiency after pretraining on benign traffic. Experimental evaluation on the Lycos2017 dataset demonstrates that the proposed method surpasses existing self-supervised and anomaly detection techniques in a binary classification task. Furthermore, when fine-tuned on a limited labelled dataset, the proposed approach achieves superior multi-class classification performance compared to existing self-supervised models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。