CapsRule: Explainable Deep Learning for Classifying Network Attacks

被引:8
作者
Mahdavifar, Samaneh [1 ]
Ghorbani, Ali A. A. [2 ]
机构
[1] McGill Univ, Data Min & Secur Lab, Montreal, PQ H3A 1X1, Canada
[2] Univ New Brunswick, Canadian Inst Cybersecur, Fac Comp Sci, Fredericton, NB E3B 5A3, Canada
基金
加拿大自然科学与工程研究理事会;
关键词
Feature extraction; Security; Neural networks; Intrusion detection; Denial-of-service attack; Data mining; Couplings; Capsule network; Distributed Denial-of-Service (DDoS) attacks; explainability; explainable deep neural network (DNN); rule extraction; RULE EXTRACTION; NEURAL-NETWORKS; GENERATION; ALGORITHM;
D O I
10.1109/TNNLS.2023.3262981
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Despite the potential deep learning (DL) algorithms have shown, their lack of transparency hinders their widespread application. Extracting if-then rules from deep neural networks is a powerful explanation method to capture nonlinear local behaviors. However, existing rule extraction methods suffer from inefficiency, incomprehensibility, infidelity, and not scaling well. Concerning security applications, they are not optimized regarding the decision boundary, data types and ranges, classification tasks, and dataset size. In this article, we propose CapsRule, an effective and efficient rule-based DL explanation method dedicated to classifying network attacks. It extracts high-fidelity rules from the feed-forward capsule network that explains how an input sample is classified. Using precomputed coupling coefficients, the training phase overlaps the rule extraction process to increase efficiency. The activation vector of a capsule can represent semantic intelligence about the attributes of the input sample. The rules extracted from CapsRule address the major concerns of network attack detection. The rules: 1) approximate the nonlinear decision boundary of the underlying data; 2) reduce the number of false positives significantly; 3) increase transparency; and 4) help find errors and noise in the data. We evaluate CapsRule on the CICDDoS2019 dataset that contains over a million of the most advanced Distributed Denial-of-Service (DDoS) attacks. The extensive evaluation shows that it generates accurate, high-fidelity, and comprehensible rules. CapsRule achieves an average accuracy of 99.0% and a false positive rate of 0.70% for reflection-and exploitation-based attacks. We verify that the learned features from the rulesets match our domain-specific knowledge. They also help find flaws in the dataset generation process and erroneous patterns caused by attack simulators.
引用
收藏
页码:12434 / 12448
页数:15
相关论文
共 46 条
[1]  
Andrews R., 1995, Proceedings of the Sixth Australian Conference on Neural Networks (ACNN'95), P213
[2]   Survey and critique of techniques for extracting rules from trained artificial neural networks [J].
Andrews, R ;
Diederich, J ;
Tickle, AB .
KNOWLEDGE-BASED SYSTEMS, 1995, 8 (06) :373-389
[3]  
[Anonymous], 2020, scikit-learn Machine Learning in Python
[4]   Reverse Engineering the Neural Networks for Rule Extraction in Classification Problems [J].
Augasta, M. Gethsiyal ;
Kathirvalavakumar, T. .
NEURAL PROCESSING LETTERS, 2012, 35 (02) :131-150
[5]   A rule generation algorithm from neural network using classified and misclassified data [J].
Biswas, Saroj Kr. ;
Chakraborty, Manomita ;
Purkayastha, Biswajit .
INTERNATIONAL JOURNAL OF BIO-INSPIRED COMPUTATION, 2018, 11 (01) :60-70
[6]   A Two-Step Rule-Extraction Technique for a CNN [J].
Bologna, Guido ;
Fossati, Silvio .
ELECTRONICS, 2020, 9 (06)
[7]   Rule extraction from neural network trained using deep belief network and back propagation [J].
Chakraborty, Manomita ;
Biswas, Saroj Kumar ;
Purkayastha, Biswajit .
KNOWLEDGE AND INFORMATION SYSTEMS, 2020, 62 (09) :3753-3781
[8]   Recursive Rule Extraction from NN using Reverse Engineering Technique [J].
Chakraborty, Manomita ;
Biswas, Saroj Kr. ;
Purkayastha, Biswajit .
NEW GENERATION COMPUTING, 2018, 36 (02) :119-142
[9]  
Chan VKH, 2020, PETROLEUM-PRC, V6, P329, DOI [10.1016/j.petlm.2019.11.005, 10.1016/j.petlm.2019.11.005, DOI 10.1016/J.PETLM.2019.11.005]
[10]   Intelligent approach to build a Deep Neural Network based IDS for cloud environment using combination of machine learning algorithms [J].
Chiba, Zouhair ;
Abghour, Noreddine ;
Moussaid, Khalid ;
El Omri, Amina ;
Rida, Mohamed .
COMPUTERS & SECURITY, 2019, 86 :291-317