.png){kind=icon}
.png){kind=icon}
Return
SRViT-MCNet: An IoT Malware Classification Model
.png){kind=icon}
PRE
AI
DOI：10.1007/978-981-95-3061-8_16.png)
.png){kind=icon}
Original request for help.png){kind=icon}
Share.png){kind=icon}
Save
Abstract
En 中文
In the Internet of Things (IoT) domain, the heterogeneity of devices and the widespread deployment of resources significantly impact traditional malware classification methods. To effectively extract both local and global features of IoT malware and improve classification accuracy, a hybrid model, named SRViT-MCNet, is proposed. The model first visualizes IoT malware by applying the Occlusion Sensitivity method to capture regions where the model's prediction confidence significantly drops. Topological Data Analysis (TDA) is then employed to generate Persistent Images (PI). The model utilizes a parallel feature extraction strategy, combining the SRViT module with the MCNet module to capture multi-dimensional features. The MC module integrated into the CNN enhances the model's ability to capture fine-grained local features. Additionally, the SCFN design replaces the FFN in RepViT, further improving the model's sensitivity and representation power for local features while retaining global context. Experimental results show that SRViT-MCNet achieves classification accuracies of 99.54% and 99.32% on the BIG2015 and Malimg datasets, respectively.
Keywords:
IoT
Malware classification
Visual interpretability
ViT
Deep learning
Journal
K
IF:
0
Papers:
29
Citations:
0