On-device learning based vulnerability detection in IoT environment

Li, Hongping, Li, Shan Cang and Min, Geyong 2025. On-device learning based vulnerability detection in IoT environment. Journal of Industrial Information Integration 47 , 100900. 10.1016/j.jii.2025.100900

Full text not available from this repository.

Abstract

Pre-trained machine learning models have demonstrated significant potential for enhancing vulnerability detection in Internet of Things (IoT) systems. By applying quantization techniques, this work constrained model weights and activations to binary values to significantly reduce model size and computational cost, making them deployable on IoT devices. A novel Binary Neural Networks (BNN) scheme is proposed to binarize vulnerability detection models, optimizing memory usage and computational costs on resource-constrained IoT devices. A vulnerability detection BNN was implemented with optimized parameters, including the number of activation layers, kernel size, and the size of the fully connected layer. The BNN model was evaluated on a Raspberry Pi using the IoT23 and NSL-KDD datasets, demonstrating promising performance in vulnerability detection.

Item Type:	Article
Date Type:	Publication
Status:	Published
Schools:	Schools > Computer Science & Informatics
Publisher:	Elsevier
ISSN:	2452-414X
Date of Acceptance:	23 June 2025
Last Modified:	12 Dec 2025 11:15
URI:	https://orca.cardiff.ac.uk/id/eprint/183163

Actions (repository staff only)

Edit Item