arXiv:2601.09902cs.CRcs.AI2026-01被引 6

提出新对比损失,让模型同时识别已知和未知网络攻击。

A Novel Contrastive Loss for Zero-Day Network Intrusion Detection

  • 用良性与已知恶意流量联合训练,学习正常行为分布。
  • 在零日攻击检测上,AUROC提升0.060883,优于现有模型。
  • 适合需要防御未知攻击的网络安全系统使用。

机器学习在网络安全入侵检测中表现优异,但在面对新型攻击(零日攻击)时性能显著下降。传统方法依赖仅用良性数据训练的异常检测器,虽能泛化至未知攻击,但误报率过高。本文提出一种新型对比损失函数,结合良性与已知恶意样本训练,既保持对不平衡数据的鲁棒性,又能有效识别零日攻击。在Lycos2017数据集上,该方法在已知攻击检测中AUROC提升0.000065,在零日攻击检测中提升0.060883。进一步扩展至开放集识别任务,其OpenAUC较现有方法提升0.170883。

原文摘要 · Abstract (English)

Machine learning has achieved state-of-the-art results in network intrusion detection; however, its performance significantly degrades when confronted by a new attack class -- a zero-day attack. In simple terms, classical machine learning-based approaches are adept at identifying attack classes on which they have been previously trained, but struggle with those not included in their training data. One approach to addressing this shortcoming is to utilise anomaly detectors which train exclusively on benign data with the goal of generalising to all attack classes -- both known and zero-day. However, this comes at the expense of a prohibitively high false positive rate. This work proposes a novel contrastive loss function which is able to maintain the advantages of other contrastive learning-based approaches (robustness to imbalanced data) but can also generalise to zero-day attacks. Unlike anomaly detectors, this model learns the distributions of benign traffic using both benign and known malign samples, i.e. other well-known attack classes (not including the zero-day class), and consequently, achieves significant performance improvements. The proposed approach is experimentally verified on the Lycos2017 dataset where it achieves an AUROC improvement of .000065 and .060883 over previous models in known and zero-day attack detection, respectively. Finally, the proposed method is extended to open-set recognition achieving OpenAUC improvements of .170883 over existing approaches.

入侵检测零日攻击对比学习

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