FEDetect: كشف وتصنيف البرمجيات الخبيثة المعتمد على التعلم الفيدرالي باستخدام خوارزميات الشبكات العصبية العميقة FEDetect: A Federated Learning-Based Malware Detection and Classification Using Deep Neural Network Algorithms

شارك:
المجلة: Arabian Journal for Science and Engineering، المجلد: 50، العدد: 19
DOI: https://doi.org/10.1007/s13369-025-10043-x
تاريخ النشر: 2025-03-17
المؤلف: Zeki Çıplak وآخرون
الموضوع الرئيسي: تقنيات الكشف المتقدمة عن البرمجيات الخبيثة

نظرة عامة

تتناول ورقة البحث التقاطع الحرج بين خصوصية البيانات واكتشاف البرمجيات الضارة في سياق التعلم الآلي، مع تسليط الضوء على قيود الأساليب التقليدية المركزية التي تشكل مخاطر كبيرة على تسرب البيانات. تدعو الدراسة إلى التعلم الفيدرالي كبديل واعد، مما يمكّن من اكتشاف وتصنيف البرمجيات الضارة بشكل فعال مع الحفاظ على خصوصية البيانات. باستخدام مجموعة بيانات CIC-MalMem-2022، طور المؤلفون 22 نموذجًا، بما في ذلك الأساليب الفيدرالية وغير الفيدرالية، مما يظهر أن التعلم الفيدرالي حقق دقة مثيرة للإعجاب تبلغ 0.999 في التصنيف الثنائي و0.845 في التصنيف متعدد الفئات، مما يؤكد إمكانيته كطريقة آمنة لمعالجة البيانات.

في الخاتمة، تؤكد الدراسة على الأداء القابل للمقارنة لنماذج التعلم الفيدرالي مقارنة بالطرق التقليدية مع الحفاظ على خصوصية البيانات. كما تناقش بيئة المحاكاة المستخدمة للاختبار، والتي سمحت بتحليل الأداء وقضايا القابلية للتوسع. تم الإشارة إلى إدخال آلية إيقاف مبكر كتحسين كبير، مما يحسن وقت التدريب ويقلل من عبء الاتصال. تشمل اتجاهات البحث المستقبلية استكشاف التعلم الفيدرالي في سيناريوهات العالم الحقيقي مع مصادر بيانات غير متجانسة واختبار هياكل التعلم العميق الأحدث لتعزيز قابلية تطبيق التعلم الفيدرالي. بشكل عام، توفر الدراسة إطارًا أساسيًا لتحقيق التوازن بين خصوصية البيانات وأداء النموذج في سياق التعلم الفيدرالي.

مقدمة

تتناول مقدمة هذه الورقة البحثية التهديد المتزايد للهجمات الإلكترونية التي تسهلها البرمجيات الضارة، والتي تعرض خصوصية المستخدم ووظائف الجهاز للخطر. مع زيادة استخدام الإنترنت، تزداد أيضًا شيوع هذه الهجمات، مما يبرز الحاجة الملحة إلى حلول فعالة لاكتشاف البرمجيات الضارة. تشكل الطرق التقليدية، التي تعتمد على تخزين البيانات المركزي، مخاطر كبيرة على خصوصية البيانات وتستهلك الموارد، مما يتطلب نقل بيانات المستخدم إلى خادم مركزي لتدريب النموذج. في المقابل، يقدم التعلم الفيدرالي (FL)، الذي قدمته جوجل في عام 2016، بديلاً واعدًا من خلال السماح للبيانات بالبقاء على الأجهزة المحلية بينما يتم مشاركة معلمات النموذج فقط مع خادم مركزي. لا تعزز هذه الطريقة أمان البيانات فحسب، بل تحسن أيضًا كفاءة تطوير النموذج.

تتمثل أصالة هذا البحث في استكشاف كيفية معالجة التعلم الفيدرالي بفعالية لقيود معالجة البيانات المركزية، مثل متطلبات النطاق الترددي العالية ومخاطر الخصوصية. تهدف الدراسة إلى سد الفجوات الموجودة في الأدبيات من خلال تفصيل الجوانب التقنية لمحاكاة التعلم الفيدرالي المحلي، لا سيما في سياق اكتشاف البرمجيات الضارة. يمكّن النظام المقترح، المسمى FEDetect، المستخدمين من اكتشاف التهديدات على أجهزتهم مع المساهمة في نموذج جماعي دون المساس بخصوصية بياناتهم. من خلال محاكاة النماذج المحلية ودمجها في خادم FL افتراضي، توضح الدراسة كيف يمكن للمستخدمين الاستفادة من المعرفة المشتركة بالتهديدات مع الحفاظ على سرية بياناتهم الفردية.

طرق

في قسم الطرق، يقدم المؤلفون نظرة شاملة على هياكل النماذج المستخدمة في بحثهم، موضحين التكوينات والهياكل المحددة المستخدمة. كما يصفون عمليات المحاكاة، بما في ذلك المعلمات والظروف التي تم إجراء التجارب تحتها. علاوة على ذلك، يحدد القسم الأدوات والمكتبات المستخدمة للتنفيذ، جنبًا إلى جنب مع ميزات النظام التي تدعم المتطلبات الحسابية للدراسة. كما يتم مناقشة الخوارزميات المطبقة في التحليل، مع تسليط الضوء على أهميتها لأهداف البحث والنتائج المتوقعة.

نتائج

في هذه الدراسة، تم تطوير ما مجموعه 28 نموذجًا عبر أربع هياكل مختلفة للتعلم العميق، تشمل كل من الأساليب الفيدرالية وغير الفيدرالية. تم تحليل أداء هذه النماذج بدقة، حيث يوضح الجدول 5 الهيكل وعدد المستخدمين وحقب التدريب وأحجام الدفعات وكميات البيانات وأوقات التدريب ومعدلات الدقة. من الجدير بالذكر أن جميع النماذج الفيدرالية حققت مستويات دقة تعادل نظيراتها غير الفيدرالية، مما يظهر فعالية نموذج التعلم الفيدرالي، لا سيما في الحفاظ على خصوصية البيانات دون المساس بأداء النموذج. حافظت خوارزمية FedAvg على دقة النموذج من خلال تجميع المساهمات من النماذج المحلية، خاصة في السيناريوهات التي تحتوي على بيانات مستقلة وموزعة بشكل متطابق (IID).

أظهر التحليل أنه بينما أدى زيادة عدد المستخدمين عمومًا إلى زيادة أوقات التدريب وعدد الجولات، لم تكن العلاقة دائمًا متناسبة. على سبيل المثال، أظهرت بعض النماذج تقليل أوقات التدريب على الرغم من زيادة عدد المستخدمين، مما يبرز تعقيد ديناميات التقارب. بالإضافة إلى ذلك، أظهرت النماذج الفيدرالية أوقات تدريب أقل مقارنة بالنماذج غير الفيدرالية، مما يشير إلى كفاءة محسنة في التعامل مع مجموعات بيانات أكبر. استكشفت الدراسة أيضًا تأثير قيم الحقبة المحلية على كفاءة التدريب، مشيرة إلى أن زيادة الحقبات المحلية يمكن أن تقلل بشكل كبير من عدد الجولات ووقت التدريب، على الرغم من أنه يجب مراعاة القيود العملية المتعلقة بقدرات الأجهزة. بشكل عام، تؤكد النتائج على إمكانيات التعلم الفيدرالي في تحقيق دقة عالية مع ضمان خصوصية البيانات، لا سيما في مهام التصنيف الثنائي، بينما تعالج أيضًا التحديات المرتبطة بنماذج التصنيف متعدد الفئات.

نقاش

يؤكد قسم النقاش في ورقة البحث على أهمية محاكاة نظام التعلم الفيدرالي (FL) قبل تطبيقه في العالم الحقيقي. تستكشف هذه الدراسة بشكل مبتكر الحساب المحلي دون الاتصال بين المحطات، مما يوفر رؤى حول أداء وتحديات أنظمة التعلم الفيدرالي. يبرز أهمية تحسين هياكل نماذج التعلم العميق، مثل ضبط عدد الطبقات المخفية والخلايا العصبية، لتعزيز الكفاءة قبل النشر. يقدم تحليل الأعداد المتغيرة من أجهزة الكمبيوتر الافتراضية (8، 16، 32، 64، و128) منظورًا جديدًا حول قابلية التوسع والكفاءة الحسابية للتعلم الفيدرالي، مع التركيز على تحقيق قيم دقة مستهدفة من خلال تقنيات الإيقاف المبكر.

بالإضافة إلى ذلك، تسهم الورقة في الأدبيات المحدودة حول المحاكاة المحلية لأنظمة التعلم الفيدرالي من خلال تقديم حل كود زائف وتفصيل المنهجيات المستخدمة، بما في ذلك استخدام مجموعة بيانات CIC-MalMem-2022 لكل من اكتشاف البرمجيات الضارة الثنائي ومتعدد الفئات. تستعرض الدراسة أيضًا الأعمال ذات الصلة، مصنفة إياها إلى تلك التي تستخدم مجموعة بيانات CIC-MalMem-2022 بدون FL وتلك التي تطبق FL بدون هذه المجموعة. لا يقتصر هذا التحليل الشامل على تقييم المنهجيات المقترحة مقابل الدراسات الحالية، بل يحدد أيضًا الطرق المحتملة للبحث المستقبلي، مما يثري النقاش حول التعلم الفيدرالي في سياقات اكتشاف البرمجيات الضارة.

Journal: Arabian Journal for Science and Engineering, Volume: 50, Issue: 19
DOI: https://doi.org/10.1007/s13369-025-10043-x
Publication Date: 2025-03-17
Author(s): Zeki Çıplak et al.
Primary Topic: Advanced Malware Detection Techniques

Overview

The research paper addresses the critical intersection of data privacy and malware detection within the context of machine learning, highlighting the limitations of traditional centralized approaches that pose significant risks of data breaches. The study advocates for federated learning as a promising alternative, which enables effective malware detection and classification while preserving data privacy. Utilizing the CIC-MalMem-2022 dataset, the authors developed 22 models, including both federated and non-federated approaches, demonstrating that federated learning achieved an impressive accuracy of 0.999 in binary classification and 0.845 in multiclass classification, thereby affirming its potential as a secure data processing method.

In the conclusion, the study emphasizes the comparable performance of federated learning models to traditional methods while maintaining data privacy. It also discusses the simulation environment used for testing, which allowed for the analysis of performance and scalability issues. The introduction of an early round stopping mechanism was noted as a significant advancement, optimizing training time and reducing communication overhead. Future research directions include exploring federated learning in real-world scenarios with heterogeneous data sources and testing newer deep learning architectures to enhance the applicability of federated learning. Overall, the study provides a foundational framework for balancing data privacy with model performance in the context of federated learning.

Introduction

The introduction of this research paper addresses the growing threat of cyberattacks facilitated by malware, which compromises user privacy and device functionality. As internet usage increases, so does the prevalence of these attacks, highlighting the urgent need for effective malware detection solutions. Traditional methods, reliant on centralized data storage, pose significant risks to data privacy and are resource-intensive, necessitating the transfer of user data to a central server for model training. In contrast, federated learning (FL), introduced by Google in 2016, presents a promising alternative by allowing data to remain on local devices while only model parameters are shared with a central server. This approach not only enhances data security but also improves the efficiency of model development.

The originality of this research lies in its exploration of how FL can effectively address the limitations of centralized data processing, such as high bandwidth requirements and privacy risks. The study aims to fill existing gaps in the literature by detailing the technical aspects of local FL simulations, particularly in the context of malware detection. The proposed system, named FEDetect, enables users to detect threats on their devices while contributing to a collective model without compromising their data privacy. By simulating local models and integrating them at a virtual FL server, the research demonstrates how users can benefit from shared knowledge of threats while maintaining individual data confidentiality.

Methods

In the Methods section, the authors provide a comprehensive overview of the model architectures employed in their research, detailing the specific configurations and structures utilized. They also describe the simulation processes, including the parameters and conditions under which the experiments were conducted. Furthermore, the section outlines the tools and libraries leveraged for implementation, alongside the system features that support the computational requirements of the study. The algorithms applied in the analysis are also discussed, highlighting their relevance to the research objectives and the expected outcomes.

Results

In this study, a total of 28 models were developed across four distinct deep learning architectures, incorporating both federated and non-federated learning approaches. The performance of these models was meticulously analyzed, with Table 5 detailing the architecture, user count, training epochs, batch sizes, data amounts, training times, and accuracy rates. Notably, all federated models achieved accuracy levels equivalent to their non-federated counterparts, demonstrating the efficacy of the federated learning paradigm, particularly in preserving data privacy without compromising model performance. The FedAvg algorithm effectively maintained model accuracy by aggregating contributions from local models, especially in scenarios with independent and identically distributed (IID) data.

The analysis revealed that while increasing the number of users generally led to longer training times and more rounds, the relationship was not always proportional. For instance, certain models exhibited reduced training times despite an increase in user count, highlighting the complexity of convergence dynamics. Additionally, federated models demonstrated lower training times compared to non-federated models, suggesting enhanced efficiency in handling larger datasets. The study also explored the impact of local epoch values on training efficiency, indicating that increasing local epochs could significantly reduce the number of rounds and training time, although practical limitations related to device capabilities must be considered. Overall, the findings underscore the potential of federated learning in achieving high accuracy while ensuring data privacy, particularly in binary classification tasks, while also addressing the challenges associated with multiclass classification models.

Discussion

The discussion section of the research paper emphasizes the significance of simulating a federated learning (FL) system prior to its real-world application. This study innovatively explores local computation without inter-station communication, providing insights into the performance and challenges of FL systems. It highlights the importance of optimizing deep learning model architectures, such as adjusting the number of hidden layers and neurons, to enhance efficiency before deployment. The analysis of varying numbers of virtual computers (8, 16, 32, 64, and 128) offers a fresh perspective on the scalability and computational efficiency of federated learning, with a focus on achieving target accuracy values through early round stopping techniques.

Additionally, the paper contributes to the limited literature on local simulations of FL systems by presenting a pseudo code solution and detailing the methodologies employed, including the use of the CIC-MalMem-2022 dataset for both binary and multiclass malware detection. The study also reviews related works, categorizing them into those utilizing the CIC-MalMem-2022 dataset without FL and those applying FL without this dataset. This comprehensive analysis not only benchmarks the proposed methodologies against existing studies but also identifies potential avenues for future research, thereby enriching the discourse on federated learning in malware detection contexts.

شارك: