An LLM-Assisted AutoML Framework for Intrusion Detection in IoT Networks

📅 2026-09-19
📈 Citations: 0
Influential: 0
📄 PDF
🤖 AI Summary
本文提出了一种基于大语言模型辅助的AutoML框架,用于物联网入侵检测,通过自动生成优化策略来提高检测效率和准确性。
📝 Abstract
Internet of Things (IoT) systems are increasingly deployed in smart homes, transportation, energy systems, and critical infrastructure. This broad connectivity improves service intelligence, but also enlarges the attack surface of IoT networks. Machine Learning (ML)-based Intrusion Detection Systems (IDSs) are widely used to identify malicious network threats and protect IoT systems, but developing effective ML-based IDS models often requires human expertise and repeated manual decisions on many procedures, including data pre-processing, feature selection, model selection, and hyperparameter tuning. Automated Machine Learning (AutoML) reduces this burden by automating steps of the ML pipeline using optimization techniques, but conventional AutoML methods can consume substantial optimization time because they explore broad candidate model families and large hyperparameter spaces. This paper proposes a Large Language Model (LLM)-assisted AutoML framework for IoT intrusion detection. The proposed framework uses an LLM as a policy generator that converts dataset profiles into bounded and validated AutoML policies for automated data balancing, automated feature engineering, and Combined Algorithm Selection and Hyperparameter Optimization (CASH). Under an equal 10-trial budget, the proposed LLM-assisted policy achieves higher weighted test F1-score than traditional AutoML using the Tree-structured Parzen Estimator (TPE) on both datasets, reaching 99.680% on CICIDS2017 and 99.186% on IoTID20. Relative to the broader 30-trial Traditional AutoML-TPE baseline, the 10-trial proposed method reduces optimizer time by 63.7% and 49.9%, respectively, while achieving slightly higher F1-score. These results show that a bounded LLM policy can improve the quality of a low-budget AutoML search while retaining a clear efficiency advantage relative to a larger conventional search budget.
Problem

Research questions and friction points this paper is trying to address.

Intrusion Detection
IoT Networks
AutoML
Machine Learning
Data Preprocessing
Innovation

Methods, ideas, or system contributions that make the work stand out.

Large Language Model (LLM)
Automated Machine Learning (AutoML)
Intrusion Detection
Feature Engineering
Hyperparameter Optimization
🔎 Similar Papers
No similar papers found.
L
Li Yang
Faculty of Business and Information Technology, Ontario Tech University, Oshawa, ON L1G 0C5, Canada; and also with the Department of Electrical and Computer Engineering, Western University, London, ON N6A 3K7, Canada