DL-HEED: A Deep Learning Approach to Energy-Efficient Clustering in Heterogeneous Wireless Sensor Networks

Wireless sensor networks (WSNs) are widely used in environmental monitoring, industrial automation, and smart cities. The Hybrid Energy-Efficient Distributed (HEED) protocol is a popular clustering algorithm designed to prolong network lifetime by balancing energy consumption among sensor nodes. However, HEED relies on simple heuristics for cluster-head (CH) selection, which may not fully exploit the complex spatiotemporal patterns in node energy and topology. This paper introduces a novel protocol, Deep Learning–Hybrid Energy-Efficient Distributed (DL-HEED), which, for the first time, integrates a Graph Neural Network (GNN) into the clustering process. By leveraging the relational structure of WSNs and a comprehensive set of node and network features—including residual energy, node degree, spatial position, and signal strength—DL-HEED enables intelligent, context-aware, and adaptive CH selection. DL-HEED leverages the relational structure of WSNs through deep learning, enabling more adaptive and energy-efficient cluster head selection than traditional heuristic-based protocols. Extensive simulations demonstrate that DL-HEED significantly outperforms classic HEED achieving up to 60% improvement in the network lifetime and energy efficiency as the network size increases. This work establishes DL-HEED as a robust, scalable, and practical solution for next-generation WSN deployments, marking a substantial advancement in the application of deep learning to resource-constrained IoT environments.