Analysis of N-Way K-Shot Malware Detection Using Few-Shot Learning

Kwok Tai Chui, Brij B. Gupta, Lap Kei Lee, Miguel Torres-Ruiz

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

Solving machine learning problems with small-scale training datasets becomes an emergent research area to fill the opposite end of big data applications. Attention is drawn to malware detection using few-shot learning, which is typically formulated as N-way K-shot problems. The aims are to reduce the effort in data collection, learn the rare cases, reduce the model complexity, and increase the accuracy of the detection model. The performance of the malware detection model is analyzed with the variation in the number of ways and the number of shots. This facilitates the understanding on the design and formulation of three algorithms namely relation network, prototypical network, and relation network for N-way K-shot problems. Two benchmark datasets are selected for the performance evaluation. Results reveal the general characteristics of the performance of malware detection model with fixed ways and varying shots, and varying ways and fixed shots based on the trends of results of 30 scenarios.

Original languageEnglish
Title of host publicationInternational Conference on Cyber Security, Privacy and Networking, ICSPN 2022
EditorsNadia Nedjah, Gregorio Martínez Pérez, B.B. Gupta
PublisherSpringer Science and Business Media Deutschland GmbH
Pages33-44
Number of pages12
ISBN (Print)9783031220173
DOIs
StatePublished - 2023
EventInternational Conference on Cyber Security, Privacy and Networking, ICSPN 2022 - Virtual, Online
Duration: 9 Sep 202111 Sep 2021

Publication series

NameLecture Notes in Networks and Systems
Volume599 LNNS
ISSN (Print)2367-3370
ISSN (Electronic)2367-3389

Conference

ConferenceInternational Conference on Cyber Security, Privacy and Networking, ICSPN 2022
CityVirtual, Online
Period9/09/2111/09/21

Keywords

  • Few-shot learning
  • Imbalanced dataset
  • Malware detection
  • N-way K-shot
  • Small-scale dataset

Fingerprint

Dive into the research topics of 'Analysis of N-Way K-Shot Malware Detection Using Few-Shot Learning'. Together they form a unique fingerprint.

Cite this