- Research Article
2
- 10.1109/les.2025.3567025
A Sample-Based, Multistage Machine Learning Pipeline for Scalable IoT Threat Detection
- Feb 01, 2026
- IEEE Embedded Systems Letters
- Marcelo V C Aragão + 3 more +3
The rapid growth of IoT devices demands scalable and efficient threat detection solutions. This paper introduces a sample-based, multi-stage machine learning (ML) pipeline for IoT threat detection using the CICIoT2023 dataset, integrating feature selection, data balancing, and hyperparameter optimization to improve detection accuracy while reducing the computational overhead associated with training. We evaluate, and across binary, multiclass, and fine-grained tasks, showing that with 10% sampling achieves the best trade-off between accuracy and efficiency. Compared to prior methods, our approach eliminates GPU dependence, maintains low latency, and preserves state-of-the-art performance while enabling scalable training for high generalization capacity. Additionally, we provide model selection guidelines based on dataset complexity and computational constraints. The results show that training with a sample-based approach enables effective threat detection on large datasets, producing models that generalize well to diverse IoT attack scenarios, thus ensuring practical applicability in real-world deployments.
Read more