Automatic Classification for Vulnerability Based on Machine Learning

被引:0
|
作者
Shuai, Bo [1 ]
Li, Haifeng [1 ]
Li, Mengjun [1 ]
Zhang, Quan [1 ]
Tang, Chaojing [1 ]
机构
[1] Natl Univ Def Technol, Sch Elect Sci & Engn, Changsha 410073, Hunan, Peoples R China
关键词
Vulnerability classification; Words location; Vulnerability distribution; LDA model; SVM;
D O I
暂无
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
In order to solve the problems of traditional machine learning methods for automatic classification of vulnerability, this paper presents a novel machine learning method based on LDA model and SVM. Firstly, word location information is introduced into LDA model called WL-LDA (Weighted Location LDA), which could acquire better effect through generating vector space on themes other than on words. Secondly, a multi-class classifier called HT-SVM (Huffman Tree SVM) is constructed, which could make a faster and more stable classification by making good use of the prior knowledge about distribution of the number of vulnerabilities. Experiments show that the method could obtain higher classification accuracy and efficiency.
引用
收藏
页码:312 / 318
页数:7
相关论文
共 50 条
  • [1] Automatic Vulnerability Classification Using Machine Learning
    Gawron, Marian
    Cheng, Feng
    Meinel, Christoph
    [J]. RISKS AND SECURITY OF INTERNET AND SYSTEMS, CRISIS 2017, 2018, 10694 : 3 - 17
  • [2] A Machine Learning Based Automatic Tomato Classification System
    Chen, Xin
    Sun, Zhan-Li
    Chen, Xia
    [J]. PROCEEDINGS OF THE 33RD CHINESE CONTROL AND DECISION CONFERENCE (CCDC 2021), 2021, : 5105 - 5108
  • [3] Review of Automatic Citation Classification Based on Machine Learning
    Zhou, Zhichao
    [J]. Data Analysis and Knowledge Discovery, 2021, 5 (12) : 14 - 24
  • [4] Machine Learning Based Automatic Classification of Customer Sentiment
    Hasan, Tonmoy
    Matin, Abdul
    Joy, M. Shakif Rahman
    [J]. 2020 23RD INTERNATIONAL CONFERENCE ON COMPUTER AND INFORMATION TECHNOLOGY (ICCIT 2020), 2020,
  • [5] Vulnerability Assessment of Machine Learning Based Malware Classification Models
    Raju, Godwin
    Zavarsky, Pavol
    Makanju, Adetokunbo
    Malik, Yasir
    [J]. PROCEEDINGS OF THE 2019 GENETIC AND EVOLUTIONARY COMPUTATION CONFERENCE COMPANION (GECCCO'19 COMPANION), 2019, : 1615 - 1618
  • [6] Automatic Classification of Gastrointestinal Diseases Based on Machine Learning Techniques
    Rabi, Bahaa
    Attallah, Omneya
    Zaghlool, Mohamed S.
    Sharkas, Maha A.
    [J]. 29TH INTERNATIONAL CONFERENCE ON COMPUTER THEORY AND APPLICATIONS (ICCTA 2019), 2019, : 85 - 89
  • [7] Automatic classification and recognition of geomagnetic interference events based on machine learning
    Liu, Gaochuan
    Shan, Weifeng
    Chen, Jun
    Che, Mengqi
    Teng, Yuntian
    Huang, Yongming
    [J]. JOURNAL OF COMPUTATIONAL METHODS IN SCIENCES AND ENGINEERING, 2022, 22 (04) : 1157 - 1170
  • [8] A Machine Learning Based Ensemble Method for Automatic Multiclass Classification of Decisions
    Fu, Liming
    Liang, Peng
    Li, Xueying
    Yang, Chen
    [J]. PROCEEDINGS OF EVALUATION AND ASSESSMENT IN SOFTWARE ENGINEERING (EASE 2021), 2021, : 40 - 49
  • [9] Automatic Scene Classification Based on Gist Feature and Extreme Learning Machine
    Liang, Ying
    Wang, Lu Ping
    Zhang, Lu Ping
    [J]. FRONTIERS OF MANUFACTURING SCIENCE AND MEASURING TECHNOLOGY V, 2015, : 923 - 930
  • [10] Automatic classification of magnetocardiograms with the machine learning approach
    Fenici, R
    Brisinda, D
    Meloni, AM
    Fenici, P
    [J]. EUROPEAN HEART JOURNAL, 2004, 25 : 560 - 560