arrow
Return

Automatic Classification Method for Software Vulnerability Based on Deep Neural Network

delete2019-01-01
delete42
delete
OA
AI
G
Guoyan Huang
Y
Yazhou Li
Q
Qian Wang *
任
任家东 (Jiadong Ren)
Y
Yongqiang Cheng
X
Xiaolin Zhao
DOI:10.1109/ACCESS.2019.2900462delete
deleteOriginal
deleteShare
deleteSave
View PDF
Abstract

Abstract

En 中文
Software vulnerabilities are the root causes of various security risks. Once a vulnerability is exploited by malicious attacks, it will greatly compromise the safety of the system, and may even cause catastrophic losses. Hence automatic classification methods are desirable to effectively manage the vulnerability in software, improve the security performance of the system, and reduce the risk of the system being attacked and damaged. In this paper, a new automatic vulnerability classification model (TFI-DNN) has been proposed. The model is built upon term frequency-inverse document frequency (TF-IDF), information gain (IG), and deep neural network (DNN): the TF-IDF is used to calculate the frequency and weight of each word from vulnerability description; the IG is used for feature selection to obtain an optimal set of feature word, and; the DNN neural network model is used to construct an automatic vulnerability classifier to achieve effective vulnerability classification. The National Vulnerability Database of the United States has been used to validate the effectiveness of the proposed model. Compared to SVM, Naive Bayes, and KNN, the TFI-DNN model has achieved better performance in multi-dimensional evaluation indexes including accuracy, recall rate, precision, and Fl-score.
Keywords:
Deep neural network
information gain
software security
vulnerability classification

Journal

IEEE Access cover
IEEE Access
IF:
3.6
Papers:
9.8W
Citations:
29.4W

Organization

Y
Yanshan University
Scholars:
1.7W
Papers: 1.1W
Citations: 1.3W
U
University of Hull
Scholars:
7.3K
Papers: 7.0K
Citations: 6.8K
B
beijing institute of technology
Scholars:
5.5W
Papers: 4.0W
Citations: 63
researcher View more organizations
Cited Papers

Cited Papers

errShare
errSave
ImageNet Large Scale Visual Recognition Challenge
err2015-04-11
err2.7W
PREAI
errRussakovsky, Olga; Deng, Jia; Su, Hao; Krause, Jonathan; Satheesh, Sanjeev; Ma, Sean; Huang, Zhiheng; Karpathy, Andrej; Khosla, Aditya; Bernstein, Michael; Berg, Alexander C.; Fei-Fei, Li
errShare
errSave
Factors associated with influenza vaccine receipt in community dwelling adults and their children
err2014-04-01
err0
errOAAI
errRyan Malosh; Suzanne E. Ohmit; Joshua G. Petrie; Mark G. Thompson; Allison E. Aiello; Arnold S. Monto
errShare
errSave
Toward Human Parity in Conversational Speech Recognition
err2017-12-01
err254
PREAI
errXiong, Wayne; Droppo, Jasha; Huang, Xuedong; Seide, Frank; Seltzer, Michael L.; Stolcke, Andreas; Yu, Dong; Zweig, Geoffrey
errShare
errSave
ICT for informal workers in Sub-Saharan Africa: Systematic review and analysis
err2017-09-01
err0
PREAI
errNasibu Mramba; Joel Rumanyika; Mikko Apiola; Jarkko Suhonen
errShare
errSave
Single-Atom Transistor as a Precise Magnetic Field Sensor
err2018-01-03
err0
errOAAI
errKrzysztof Jachymski; Tomasz Wasak; Zbigniew Idziaszek; Paul S. Julienne; Antonio Negretti; Tommaso Calarco
errShare
errSave
researcher View more