Abstract

IoT networks now handle traffic from billions of devices, and edge nodes are under constant pressure to classify that traffic and dispatch tasks within tight latency deadlines. Most existing systems treat classification and scheduling as two separate steps that run one after the other. This sequence adds unnecessary delay and breaks the feedback between the two tasks: the scheduler never sees the traffic type, and the classifier never sees the queue state. We propose ParallelEdge-AI, a system built around a shared flow encoder that feeds two task-specific heads in parallel, one for multi-class traffic classification and one for task-urgency scoring. Both heads are trained end-to-end using a joint loss that combines cross-entropy and pairwise ranking. A load-balance controller then reads the urgency scores alongside live queue lengths to decide, every 200 ms, whether a task stays local or moves to a less-loaded edge node. No global synchronisation is needed. We test the system on three real IoT datasets: RT-IoT2022, N-BaIoT, and CICIoT2023. ParallelEdge-AI reaches 97.63% accuracy and an F1-score of 97.34%, which is 3.16 percentage points above the best baseline. Inference latency is 19.62 ms per batch, the deadline-miss rate is 2.34%, and the load-imbalance index is 0.083, all three are the best results in our comparison. These numbers show that running classification and scheduling together on a shared representation is both faster and more accurate than treating them as separate problems.

IPC Classification

Keywords