Malware traffic classification (MTC) is a very important component of cyber security, and a number of the MTC techniques are based on deep learning (DL) with a strong capability of feature mining and classification. However, these DL-based MTC methods are heavily dependent on a large amount of network traffic samples. In the few-shot scenarios, these methods usually overfit and have poor classification performance. Considering that the update cycle of malware is faster and faster, and there are more and more types of malware, collecting enough training samples for all malware is very challenging, if not impossible. In this paper, a novel few-shot MTC(FS-MTC) method is proposed based on convolutional neural network (CNN) and model-agnostic meta-learning (MAML) algorithm. Specifically, the CNN is trained on samples from normal softwares by MAML rather than the conventional optimization methods, then the CNN is finetuned by a few samples from malware for MTC. Simulation results show that our proposed MAML-based FS-MTC can outperform the traditional MTC methods. The performance of our proposed method can reach up to 95.69%.


    Access

    Check access

    Check availability in my library

    Order at Subito €


    Export, share and cite



    Title :

    Few-Shot Malware Traffic Classification Method Using Network Traffic and Meta Transfer Learning


    Contributors:
    Guo, Hanyi (author) / Zhang, Xixi (author) / Wang, Yu (author) / Adebisi, Bamidele (author) / Gacanin, Haris (author) / Gui, Guan (author)


    Publication date :

    2022-09-01


    Size :

    2163474 byte





    Type of media :

    Conference paper


    Type of material :

    Electronic Resource


    Language :

    English



    A review of few-shot network traffic classification based on deep learning

    Yang, Kai / Wang, Ruo Nan / Meng, Qi et al. | British Library Conference Proceedings | 2023


    A Novel Malware Traffic Classification Method Based on Differentiable Architecture Search

    Shi, Yunxiao / Zhang, Xixi / He, Zhengran et al. | IEEE | 2022


    Understanding Network Traffic States using Transfer Learning

    Krishnakumari, Panchamy / Perotti, Alan / Pinto, Viviana et al. | IEEE | 2018


    Traffic Sign Recognition Framework Using Zero-Shot Learning

    Shah, Prachi / Patel, Parmanand / Kothadiya, Deep | Springer Verlag | 2024


    SeqNet: Sequential Networks for One-Shot Traffic Sign Recognition With Transfer Learning

    Abdi, Nariman / Parvaresh, Farzad / Farzan Sabahi, Mohamad | IEEE | 2024