用少量恶意样本实现精准网络入侵检测,解决传统方法数据不足难题。
Few-Shot Network Intrusion Detection Using Online Triplet Mining
- 采用在线三元组挖掘的孪生网络,结合KNN分类器进行少样本学习。
- 仅需每类10个恶意样本,即可在多分类任务中达到先进水平性能。
- 适合新网络或新型攻击场景,对低资源环境下的安全防护有实际价值。
网络入侵检测系统在保护网络免受恶意流量威胁方面至关重要,可由网络安全运营中心进一步分析。当前主流方法依赖监督学习训练分类模型识别已知攻击,但需要大量标注数据,小样本下表现差。异常检测方法虽无需恶意样本,却因误报率高而难以实用。尤其在新网络或未知攻击出现时,缺乏足够标注样本导致检测能力薄弱。本文提出一种基于在线三元组挖掘的孪生网络与KNN分类器结合的方法,支持少样本分类,在仅10个恶意样本/类的情况下仍能有效检测。通过消融实验对比不同三元组挖掘算法、推理策略及距离度量,最终模型在少样本二分类与多分类任务中均优于或媲美现有先进方法。
原文摘要 · Abstract (English)
Network intrusion detection systems play a vital role in protecting networks by detecting malicious network traffic which can then be investigated by a cybersecurity operations centre. State-of-the-art approaches utilise supervised machine learning methods to train a classification model to recognise known cyberattacks; however, these models require a large labelled dataset to train and show poor performance when trained on smaller datasets. In an attempt to address this shortcoming, anomaly detection models learn the distribution of benign traffic and flag non-conforming traffic as malicious. While these methods do not require malicious examples to train, they suffer from high false-positive rates rendering them impractical. As a result, networks may be particularly vulnerable when there are insufficient labelled instances of a specific attack class to train an effective classifier. This often occurs in newly established networks or when previously unseen types of attacks emerge. To address this challenge, this work proposes the use of a triplet network, utilising online triplet mining and a KNN classifier, which is able to perform few-shot classification, enabling effective intrusion detection after being trained on a limited number of malicious examples. Various online triplet mining algorithms were explored and model design choices, such as the inference algorithm and optimised distance metrics, were compared and evaluated through a series of ablation studies. The final model was compared against other state-of-the-art approaches in few-shot binary and multiclass classification, where the proposed approach was found to be competitive with existing methods when trained on as little as 10 malicious samples of each class.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。