Graph Neural Networks for IoT Data Aggregation Scheduling

Van Vi Vo, Syed M. Raza, Duc Tai Le, Moonseong Kim, Hyunseung Choo

Abstract: Data aggregation is an important approach in IoT sensor networks since it reduces data transfer while also preserving energy and bandwidth. This research investigates the challenge of time-efficient data aggregation in wireless sensor networks, which is critical in military, civilian, and industrial applications. Effective data aggregation algorithm design and optimization are required for quick and interference-free data collection. Machine learning has received attention for outperforming classical heuristic techniques. The research presents the first Graph Neural Network (GNN) model for data aggregation in IoT sensor networks, which incorporates Graph Attention Networks (GATs) and fully connected layers. The GNN-based model learns network topology and node attributes, creating node embeddings and correcting sensor node transmitting time slots. With a centralized training procedure and adapted execution for network size change, the proposed approach achieves satisfactory performance compared to the heuristic algorithm.