Malware detection using federated learning and incremental learning

Articles in Press

Document Type : Original Article

Authors

1 Master's student, Boali Sina Hamedan University, Hamedan, Iran

2 Assistant Professor, Boali Sina University, Hamedan, Hamedan, Iran

3 Associate Professor, Boali Sina University, Hamedan, Hamedan, Iran

Abstract

Android-based mobile devices are widely used due to their ease of use among users. Individuals perform various tasks on their mobile phones, such as banking activities, social networking, and diverse business systems, thereby exposing considerable personal information to risks due to the vulnerabilities of the Android operating system. The rapid development of Android malware has rendered many traditional malware detection methods less accurate over time. Research indicates that machine learning is an effective approach for detecting malware. The rapid evolution of malware contributes to the degradation of accuracy in trained models over time. Moreover, the collection of malware-related data from Android devices jeopardizes users' privacy. To address these issue, this paper employs federated and incremental learning. Recently, federated learning has been introduced for training machine learning models on decentralized devices with the aim of preserving privacy. This study utilizes a Multi-Layer Perceptron (MLP) within the framework of federated learning. Stacking, a type of ensemble learning, is employed for incremental learning. The CICMalDroid 2020 dataset is utilized in this research, using static data to develop the final model. The outcome of this study is a model with an accuracy of 96.49%, demonstrating significant improvement in computational time complexity along with maintaining the quality of learning and model accuracy compared to existing methods.

Keywords

Main Subjects

Electronic and Cyber Defense

Articles in Press, Accepted Manuscript
Available Online from 04 November 2024

Files

History

  • Receive Date: 01 May 2024
  • Revise Date: 26 September 2024
  • Accept Date: 30 September 2024
  • Publish Date: 04 November 2024

Share

How to cite

Statistics