Federated Defense for Malware Detection and Resilience Against Adversarial Attacks

Farhan Ullah,Gautam Srivastava
DOI: https://doi.org/10.1109/mce.2024.3418129
2024-01-01
IEEE Consumer Electronics Magazine
Abstract:Consumer electronics are substantially compromised by malware, which can traverse numerous operating systems and file formats. Considerable effort has been devoted to developing malware detection systems that employ Machine Learning (ML) and Deep Learning (DL). However, these models are susceptible to adversarial attacks, where maliciously crafted inputs can bypass detection mechanisms. In this paper, we present Fed-Adversarial, a novel technique for malware detection against adversarial attacks that employ intermittent clients-based Federated Learning (FL). This method can improve adversarial attack detection while preserving data privacy for each client. The raw malware images are first normalized and converted to color to extract features efficiently. Additionally, a wide range of adversarial examples is generated using normalized images to maximize evasion opportunities and reduce perturbations. Following this, adversarial examples are employed by Deep Convolutional Neural Networks (CNNs) during local training, resulting in Local Model Updates (LMUs). After combining these LMUs, the global server produces Global Model Updates (GMU), delivered to distant clients. The proposed approach is evaluated on standard datasets, including dumpware10 , malimg , and MaleVis , and it obtains high detection accuracy of 99.18%, 98.12%, and 98.38%, respectively.
What problem does this paper attempt to address?