A Blockchain and Federated Learning Framework for Image-Based IoT Malware Detection and Prevention
Open Access
Abstract
Internet of Things (IoT) devices are increasingly targeted by rapidly evolving malware, yet collaborative detection remains challenged by privacy leakage, noisy and imbalanced training data, and weak integrity guarantees when sharing model updates. This paper presents Mal-Fedchain, a secure and privacy-preserving framework for image-based IoT malware detection and prevention that couples federated learning with blockchain and honeypot-assisted behavioral monitoring, targeting Linux-capable IoT gateway devices. Portable Executable (PE) binaries are transformed into grayscale images using a corrected fixed-width byte-mapping pipeline stabilized by an information-maximizing GAN (IMGAN). A bi-level preprocessing pipeline applies two-sided weighted sparse representation (T-WSR) denoising—designed to selectively suppress zero-padding artifacts, high-entropy packed regions, and sparse opcode noise while preserving discriminative section-boundary texture—followed by geometric augmentation to mitigate class imbalance. Malware detection and family attribution are performed using a residual capsule-based network (RBCN) that fuses discriminative visual representations with PE-header features via concatenation, improving robustness against polymorphism and obfuscation. A formal threat model governs three adversary classes: a semi-honest aggregation server, a bounded fraction of malicious clients (up to 30%), and a passive eavesdropper. To enable collaboration without exposing raw data, clients train locally and share only MemCbar-encrypted updates; a permissioned Hyperledger Fabric blockchain ledger records hashed updates and security events to provide integrity, traceability, and tamper resistance. A file-system-integrated honeypot captures evasive behaviors and logs auditable evidence to strengthen prevention. Experiments on the Malimg dataset across five ablation configurations demonstrate that the corrected RBCN pipeline achieves accuracy, precision, recall, F-measure, MCC of , and AUC of in its centralized configuration, and accuracy with AUC of in the full federated configuration with five clients and eight communication rounds, substantially outperforming all baselines across all reported metrics.
Keywords
This publication has 32 references indexed in Scilit:
- Federated learning for malware detection in IoT devicesComputer Networks, 2022
- Adaptive online learning for IoT botnet detectionInformation Sciences, 2021
- Malware detection employed by visualization and deep neural networkComputers & Security, 2021
- Towards an interpretable deep learning model for mobile malware detection and family identificationComputers & Security, 2021
- Similarity hash based scoring of portable executable files for efficient malware detection in IoTFuture Generation Computer Systems, 2020
- Cognitive and Scalable Technique for Securing IoT Networks Against Malware EpidemicsIEEE Access, 2020
- A Digital DNA Sequencing Engine for Ransomware Detection Using Machine LearningIEEE Access, 2020
- Dynamic Analysis for IoT Malware Detection With Convolution Neural Network ModelIEEE Access, 2020
- Image-Based malware classification using ensemble of CNN architectures (IMCEC)Computers & Security, 2020
- IMCFN: Image-based malware classification using fine-tuned convolutional neural network architectureComputer Networks, 2020