Method of Malware Family Classification Based on Attention-DenseNet-BC Model Mechanism

Malware is one of the most serious threats to the Internet.The existing malware has huge data size and various features.Convolutional Neural Network has the features of autonomous learning,which can be used to solve the problems that the feature extraction of malware is complex and the feature selec...

Full description

Bibliographic Details
Main Author: LI Yi-meng, LI Cheng-hai, SONG Ya-fei, WANG Jian
Format: Article
Language:zho
Published: Editorial office of Computer Science 2021-10-01
Series:Jisuanji kexue
Subjects:
Online Access:http://www.jsjkx.com/fileup/1002-137X/PDF/1002-137X-2021-10-308.pdf
_version_ 1818355841432027136
author LI Yi-meng, LI Cheng-hai, SONG Ya-fei, WANG Jian
author_facet LI Yi-meng, LI Cheng-hai, SONG Ya-fei, WANG Jian
author_sort LI Yi-meng, LI Cheng-hai, SONG Ya-fei, WANG Jian
collection DOAJ
description Malware is one of the most serious threats to the Internet.The existing malware has huge data size and various features.Convolutional Neural Network has the features of autonomous learning,which can be used to solve the problems that the feature extraction of malware is complex and the feature selection is difficult.However,in convolutional neural network,conti-nuously increasing the network layers will cause a disappear of the gradient,leading to a degradation of network performance and low accuracy.To solve this problem,an Attention-DenseNet-BC model that is suitable for malware image detection is proposed.First,the Attention-DenseNet-BC model is constructed by combining the DenseNet-BC network and the attention mechanism.Then,the malware images are used as the input of the model,and the detection results are obtained by training and testing the model.The experimental results indicate that compared with other deep learning models,the Attention-DenseNet-BC model can achieve better classification results.A high classification accuracy can be attained with the model based on the malimg public dataset.
first_indexed 2024-12-13T19:47:44Z
format Article
id doaj.art-b7b5d192a5364d42a8a2d9c185cf2243
institution Directory Open Access Journal
issn 1002-137X
language zho
last_indexed 2024-12-13T19:47:44Z
publishDate 2021-10-01
publisher Editorial office of Computer Science
record_format Article
series Jisuanji kexue
spelling doaj.art-b7b5d192a5364d42a8a2d9c185cf22432022-12-21T23:33:31ZzhoEditorial office of Computer ScienceJisuanji kexue1002-137X2021-10-01481030831410.11896/jsjkx.210200166Method of Malware Family Classification Based on Attention-DenseNet-BC Model MechanismLI Yi-meng, LI Cheng-hai, SONG Ya-fei, WANG Jian0Air and Missile Defense College,Air Force Engineering University,Xi'an 710051,ChinaMalware is one of the most serious threats to the Internet.The existing malware has huge data size and various features.Convolutional Neural Network has the features of autonomous learning,which can be used to solve the problems that the feature extraction of malware is complex and the feature selection is difficult.However,in convolutional neural network,conti-nuously increasing the network layers will cause a disappear of the gradient,leading to a degradation of network performance and low accuracy.To solve this problem,an Attention-DenseNet-BC model that is suitable for malware image detection is proposed.First,the Attention-DenseNet-BC model is constructed by combining the DenseNet-BC network and the attention mechanism.Then,the malware images are used as the input of the model,and the detection results are obtained by training and testing the model.The experimental results indicate that compared with other deep learning models,the Attention-DenseNet-BC model can achieve better classification results.A high classification accuracy can be attained with the model based on the malimg public dataset.http://www.jsjkx.com/fileup/1002-137X/PDF/1002-137X-2021-10-308.pdfmalware|densenet-bc network|attention mechanism
spellingShingle LI Yi-meng, LI Cheng-hai, SONG Ya-fei, WANG Jian
Method of Malware Family Classification Based on Attention-DenseNet-BC Model Mechanism
Jisuanji kexue
malware|densenet-bc network|attention mechanism
title Method of Malware Family Classification Based on Attention-DenseNet-BC Model Mechanism
title_full Method of Malware Family Classification Based on Attention-DenseNet-BC Model Mechanism
title_fullStr Method of Malware Family Classification Based on Attention-DenseNet-BC Model Mechanism
title_full_unstemmed Method of Malware Family Classification Based on Attention-DenseNet-BC Model Mechanism
title_short Method of Malware Family Classification Based on Attention-DenseNet-BC Model Mechanism
title_sort method of malware family classification based on attention densenet bc model mechanism
topic malware|densenet-bc network|attention mechanism
url http://www.jsjkx.com/fileup/1002-137X/PDF/1002-137X-2021-10-308.pdf
work_keys_str_mv AT liyimenglichenghaisongyafeiwangjian methodofmalwarefamilyclassificationbasedonattentiondensenetbcmodelmechanism