Lightweight CNN with Multi-Head Attention for Image-Based Malware Classification
A compact convolutional neural network architecture integrated with a multi-head attention mechanism to enhance feature discrimination in malware family classification that attains high weighted Precision, Recall, and F1 scores and enhanced macro-averaged performance, particularly enhancing classification for minority...