ILKOMNIKA: Journal of Computer Science and Applied Informatics
Vol 7 No 1 (2025): Volume 7, Number 1, April 2025

Optimization of LightGBM Model with Bayesian Optimization for Malware Detection

Kasim, Afrianto Pratama (Unknown)
Nasib, Salmun K. (Unknown)
Hasan, Isran K. (Unknown)
Wungguli, Djihad (Unknown)
Yahya, Nisky Imansyah (Unknown)



Article Info

Publish Date
30 Apr 2025

Abstract

Cyberattacks through malware on Android devices continue to rise, making accurate detection crucial. This research optimizes the LightGBM model using Bayesian Optimization to enhance accuracy and efficiency in detecting Android malware. A feature selection mechanism based on Attention Mechanism is applied to select the most relevant features for classification. The dataset used comes from the Canadian Institute for Cybersecurity (CIC) and consists of 17,804 Android applications, with a balanced distribution between malware and normal applications. The dataset is split into ratios of 80%:20%, 75%:25%, and 70%:30%. Feature selection reduces the number of features from 9503 to 300, 500, and 1000. The LightGBM model is then optimized with Bayesian Optimization to fine-tune parameters such as learning rate, number of iterations, and maximum number of leaves. The model's performance is evaluated using accuracy, precision, and recall metrics. Experimental results show that the model achieves 96,99% accuracy, 97,30% precision, and 96,70% recall with an 80%:20% dataset split and 1000 features. The combination of Attention Mechanism and Bayesian Optimization effectively improves processing efficiency without compromising performance.

Copyrights © 2025






Journal Info

Abbrev

ilkomnika

Publisher

Subject

Computer Science & IT Control & Systems Engineering Decision Sciences, Operations Research & Management

Description

ILKOMNIKA: Journal of Computer and Applied Informatics is is a peer reviewed open-access journal. The journal invites scientists and engineers throughout the world to exchange and disseminate theoretical and practice-oriented topics of computer science and applied informatics which covers five (5) ...