Benchmarking Machine Learning Models for IoT Malware Detection under Data Scarcity and Drift

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Lyon, Jake, Saeedizade, Ehsan, Sengupta, Shamik
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912849981341696
author Lyon, Jake
Saeedizade, Ehsan
Sengupta, Shamik
author_facet Lyon, Jake
Saeedizade, Ehsan
Sengupta, Shamik
contents The rapid expansion of the Internet of Things (IoT) in domains such as smart cities, transportation, and industrial systems has heightened the urgency of addressing their security vulnerabilities. IoT devices often operate under limited computational resources, lack robust physical safeguards, and are deployed in heterogeneous and dynamic networks, making them prime targets for cyberattacks and malware applications. Machine learning (ML) offers a promising approach to automated malware detection and classification, but practical deployment requires models that are both effective and lightweight. The goal of this study is to investigate the effectiveness of four supervised learning models (Random Forest, LightGBM, Logistic Regression, and a Multi-Layer Perceptron) for malware detection and classification using the IoT-23 dataset. We evaluate model performance in both binary and multiclass classification tasks, assess sensitivity to training data volume, and analyze temporal robustness to simulate deployment in evolving threat landscapes. Our results show that tree-based models achieve high accuracy and generalization, even with limited training data, while performance deteriorates over time as malware diversity increases. These findings underscore the importance of adaptive, resource-efficient ML models for securing IoT systems in real-world environments.
format Preprint
id arxiv_https___arxiv_org_abs_2601_18736
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Benchmarking Machine Learning Models for IoT Malware Detection under Data Scarcity and Drift
Lyon, Jake
Saeedizade, Ehsan
Sengupta, Shamik
Machine Learning
Networking and Internet Architecture
The rapid expansion of the Internet of Things (IoT) in domains such as smart cities, transportation, and industrial systems has heightened the urgency of addressing their security vulnerabilities. IoT devices often operate under limited computational resources, lack robust physical safeguards, and are deployed in heterogeneous and dynamic networks, making them prime targets for cyberattacks and malware applications. Machine learning (ML) offers a promising approach to automated malware detection and classification, but practical deployment requires models that are both effective and lightweight. The goal of this study is to investigate the effectiveness of four supervised learning models (Random Forest, LightGBM, Logistic Regression, and a Multi-Layer Perceptron) for malware detection and classification using the IoT-23 dataset. We evaluate model performance in both binary and multiclass classification tasks, assess sensitivity to training data volume, and analyze temporal robustness to simulate deployment in evolving threat landscapes. Our results show that tree-based models achieve high accuracy and generalization, even with limited training data, while performance deteriorates over time as malware diversity increases. These findings underscore the importance of adaptive, resource-efficient ML models for securing IoT systems in real-world environments.
title Benchmarking Machine Learning Models for IoT Malware Detection under Data Scarcity and Drift
topic Machine Learning
Networking and Internet Architecture
url https://arxiv.org/abs/2601.18736