Deep Learning-Based Enhanced Cluster Head Selection for Underwater Wireless Sensor Networks
Corressponding author's email:
thept@hcmute.edu.vnDOI:
https://doi.org/10.54644/jte.2025.1837Keywords:
Underwater Wireless Sensor Networks, Deep Learning, Cluster Head Selection, Energy Efficiency, Graph Attention NetworksAbstract
Underwater wireless sensor networks (UWSNs) are subject to unique operational challenges, including constrained energy availability, dynamic topological structures, and unreliable acoustic communication. These factors significantly impact the efficiency and stability of data collection processes, particularly the selection of cluster heads (CHs), which plays a vital role in prolonging network functionality. This paper presents GAT-CHS, a cluster head selection algorithm that integrates graph-based attention mechanisms with deep reinforcement learning to adaptively optimize clustering decisions in underwater environments. The proposed approach encodes critical node attributes into a spatial–topological representation, applies multi-head attention to quantify inter-node relevance, and utilizes a Deep Q-Network (DQN) to determine CH roles based on long-term network performance. The algorithm is evaluated across a range of simulated UWSN scenarios reflecting varying node distributions and environmental conditions. Results show that GAT-CHS reduces energy usage by 24%, improves network longevity by 36%, and achieves a packet delivery ratio of 98.6%. These findings underscore the model’s robustness and scalability, establishing GAT-CHS as a promising direction for next-generation clustering in complex underwater sensor deployments.
Downloads: 0
References
J. Yick, B. Mukherjee, and D. Ghosal, “Wireless sensor network survey,” Comput. Netw., vol. 52, no. 12, pp. 2292–2330, Dec. 2020. DOI: https://doi.org/10.1016/j.comnet.2008.04.002
S. Gupta and N. P. Singh, “Underwater wireless sensor networks: A review of routing protocols, taxonomy, and future directions,” The Journal of Supercomputing, vol. 80, no. 4, pp. 5163–5196, 2024, doi: DOI: https://doi.org/10.1007/s11227-023-05646-w
1007/s11227-023-05012-3
I. F. Akyildiz, D. Pompili, and T. Melodia, “Challenges for efficient communication in underwater sensor networks,” ACM SIGCOMM Comput. Commun. Rev., vol. 35, no. 2, pp. 112–119, Apr. 2020.
J. Heidemann, W. Ye, J. Wills, A. Syed, and Y. Li, “Research challenges and applications for underwater sensor networking,” IEEE Wireless Commun., vol. 16, no. 2, pp. 12–18, Apr. 2021.
P. Goyal, R. Tripathi, and H. M. Pandey, “A comprehensive review of underwater wireless sensor networks,” J. Ambient Intell. Humaniz. Comput., vol. 13, no. 4, pp. 4999–5020, Jul. 2022.
R. Bhar and S. Misra, “Advances in clustering techniques for wireless sensor networks: A survey,” IEEE Trans. Netw. Serv. Manag., vol. 19, no. 3, pp. 2545–2561, Sep. 2022.
X. Xu, Y. Lu, and Y. Wang, “Energy-efficient clustering in underwater wireless sensor networks: Recent advances and research challenges,” Sensors, vol. 21, no. 5, pp. 1892, Mar. 2021.
C. Liu and W. Zhao, “Deep learning-based clustering models for wireless sensor networks,” ACM Comput. Surv., vol. 55, no. 7, pp. 150–178, Sep. 2022.
P. Nazareth and B. R. Chandavarkar, “Cluster-based multi-attribute routing protocol for underwater acoustic sensor networks,”
Wireless Personal Communications, vol. 134, no. 2, pp. 781–808, 2024. DOI: https://doi.org/10.1007/s11277-024-10926-6
X. Zhao, Y. Lin, and M. Chen, “Graph neural networks for energy-efficient underwater communication,” IEEE Access, vol. 12, pp. 20567–20580, 2025.
L. Nguyen, H. Vo, and T. H. Tran, “Reinforcement learning-driven topology control in underwater WSNs,” Sensors, vol. 25, no. 1, p. 148, 2025.
N. Sharma and P. Gupta, “Reinforcement learning for underwater sensor networks: A survey,” IEEE Access, vol. 9, pp. 105432–105452, Dec. 2021.
M. S. Khan, S. H. Ahmed, and D. H. Kim, “Machine learning-based clustering in UWSNs,” IEEE Sensors J., vol. 22, no. 7, pp. 5845–5854, Apr. 2022.
T. Melodia, H. Kulhandjian, L. Kuo, and E. Demirors, “Advances in underwater acoustic networking,” IEEE Commun. Mag., vol. 58, no. 11, pp. 22–28, Nov. 2020.
M. A. Azad and M. Hassan, “A survey on graph neural networks for wireless networks,” IEEE Trans. Wireless Commun., vol. 20, no. 8, pp. 5035–5051, Aug. 2021.
X. Guo, X. Liu, and S. Zhao, “Energy-efficient routing in underwater sensor networks using deep reinforcement learning,” IEEE Trans. Veh. Technol., vol. 72, no. 1, pp. 452–465, Jan. 2023.
H. Cheng and H. Yu, “A hybrid deep learning model for clustering in underwater wireless sensor networks,” IEEE Internet Things J., vol. 9, no. 3, pp. 2411–2422, Feb. 2022.
Y. Feng, Z. Zhang, and L. Xie, “Graph-based anomaly detection in underwater networks,” Sensors, vol. 21, no. 6, pp. 2099, Apr. 2021.
H. Lu, P. Li, and M. Wu, “Adaptive clustering for UWSNs using AI-based optimization,” IEEE Trans. Mobile Comput., vol. 21, no. 4, pp. 1810–1825, Apr. 2022.
J. Tan and L. Xu, “Intelligent underwater communication protocols based on deep learning,” IEEE Commun. Surv. Tutor., vol. 24, no. 1, pp. 500–525, Mar. 2022.
F. Javed and R. Rehman, “Clustering algorithms in UWSNs: A comparative study,” Comput. Electr. Eng., vol. 103, pp. 108323, Jun. 2023.
A. Kumar and S. Yadav, “An overview of graph attention networks for wireless communication,” IEEE Access, vol. 9, pp. 55678–55693, Sep. 2021.
Y. Wei and H. Sun, “Graph neural networks for underwater sensor networks,” IEEE Internet Things J., vol. 9, no. 9, pp. 6752–6765, May 2022.
S. Yang and R. Chang, “Reinforcement learning-based clustering in UWSNs,” Sensors, vol. 23, no. 4, pp. 1804, Jan. 2023.
L. Zhao and T. Zhang, “Hybrid AI-driven clustering models for UWSNs,” IEEE Trans. Emerg. Top. Comput. Intell., vol. 5, no. 3, pp. 429–442, Jul. 2021. DOI: https://doi.org/10.1109/TETCI.2019.2961190
J. Dai and M. Zhang, “Deep Q-learning for cluster-based communication in UWSNs,” IEEE Trans. Cogn. Commun. Netw., vol. 9, no. 2, pp. 323–340, May 2023.
J. Wu and P. Lin, “Machine learning approaches to underwater clustering,” J. Netw. Comput. Appl., vol. 200, pp. 103224, Aug. 2022.
X. Luo and H. Cao, “Comparative analysis of clustering in UWSNs using AI,” Wireless Netw., vol. 28, no. 5, pp. 3223–3241, Nov. 2022.
R. Chen and P. Yang, “Adaptive reinforcement learning for clustering in UWSNs,” IEEE Trans. Green Commun. Netw., vol. 7, no. 2, pp. 1234–1250, Jun. 2023.
L. He and Y. Wang, “Hybrid AI and heuristic-based clustering,” Comput. Electr. Eng., vol. 99, pp. 107534, Oct. 2021.
M. Li and J. Sun, “AI-empowered clustering approaches in UWSNs,” IEEE Access, vol. 10, pp. 123456–123478, Dec. 2022.
R. Rao and P. Kumar, “Advances in deep reinforcement learning for clustering,” Neural Netw., vol. 164, pp. 199–214, Feb. 2023.
T. Xiong and X. Zhang, “Machine learning-driven clustering in UWSNs,” J. Ambient Intell. Smart Environ., vol. 15, no. 2, pp. 345–361, Jul. 2023.
Downloads
Published
How to Cite
Issue
Section
Categories
License
Copyright (c) 2025 Journal of Technical Education Science

This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.
Copyright © JTE.


