LSTM-Based Proactive Congestion Management for Internet of Vehicle Networks

Aly Sabri Abdalla,Ahmad Al-Kabbany,Ehab F. Badran,Vuk Marojevic
2024-10-13
Abstract:Vehicle-to-everything (V2X) networks support a variety of safety, entertainment, and commercial applications. This is realized by applying the principles of the Internet of Vehicles (IoV) to facilitate connectivity among vehicles and between vehicles and roadside units (RSUs). Network congestion management is essential for IoVs and it represents a significant concern due to its impact on improving the efficiency of transportation systems and providing reliable communication among vehicles for the timely delivery of safety-critical packets. This paper introduces a framework for proactive congestion management for IoV networks. We generate congestion scenarios and a data set to predict the congestion using LSTM. We present the framework and the packet congestion dataset. Simulation results using SUMO with NS3 demonstrate the effectiveness of the framework for forecasting IoV network congestion and clustering/prioritizing packets employing recurrent neural networks.
Networking and Internet Architecture,Artificial Intelligence,Systems and Control
What problem does this paper attempt to address?
The paper attempts to address the issue of network congestion management in Internet of Vehicles (IoV) networks. Specifically, the authors focus on how to reduce packet loss and ensure the reliable transmission of critical safety messages by predicting network congestion and taking proactive measures. Network congestion management is crucial for improving the efficiency of traffic systems, providing reliable inter-vehicle communication, and timely delivery of critical safety packets. ### Main Contributions: 1. **Design and implementation of a real-time framework**: For proactive network congestion management, by predicting congestion events and scheduling packets before they occur, thereby increasing the transmission rate of important packets. 2. **Generation of large-scale datasets**: Using the network simulator NS3 to generate datasets containing congestion channel parameters, which serve as historical records for the prediction phase. 3. **Design of an LSTM architecture**: To maximize prediction accuracy, capturing hierarchical traffic patterns through a multi-layer LSTM model. 4. **Introduction of K-means clustering technique**: To classify packets into two categories, ensuring that high-priority safety messages are transmitted in a timely manner even under congestion conditions. ### Method Overview: - **Congestion dataset generation**: Generating datasets containing the number of packet losses and the time of occurrence by capturing congestion parameters under different conditions. - **Real-time congestion prediction**: Using the LSTM model to train on historical congestion data to predict future congestion events. - **Packet prioritization and classification**: Using the K-means algorithm to classify packets based on their priority and TTL fields, ensuring high-priority packets are transmitted first. ### Performance Evaluation: - **LSTM model structure and results**: Optimizing the performance of the LSTM model by adjusting batch size, training epochs, and hidden layer size. Results show that a two-layer stacked LSTM model performs best in terms of prediction accuracy and stability. - **Packet prioritization and classification model**: Using the K-means algorithm to classify packets, effectively distinguishing between high-priority and low-priority packets. ### Conclusion: This study proposes a novel pipeline that effectively addresses the congestion management issue in IoV networks by using LSTM for proactive congestion prediction and K-means clustering for packet classification. Experimental results indicate that the proposed framework significantly improves prediction accuracy and packet transmission performance.