Hybrid K-means, Random Forest, and Simulated Annealing for Optimizing Underwater Image Segmentation

Authors

  • Mst Jannatul Kobra Information and Communication Engineering, Nanjing University of Information Science &Technology, China
  • Md Owahedur Rahman Information and Communication Engineering, Nanjing University of Information Science &Technology, China
  • Arman Mohammad Nakib School of Life Sciences (Applying Deep Learning), East China Normal University, China

DOI:

https://doi.org/10.64539/sjer.v1i4.2025.46

Keywords:

Underwater Image Segmentation, K-means Clustering, Random Forest, Simulated Annealing, Segmentation Accuracy

Abstract

The process of underwater image segmentation is also very difficult because the data collected by the underwater sensors and cameras is of very high complexity, and much data is generated and in that case, the data is not well seen, the color is distorted, and the features overlap. Current solutions, including K-means clustering and Random Forest classification, are unable to partition complex underwater images with high accuracy, or are unable to scale to large datasets, although the possibility of dynamically optimizing the number of clusters has not been fully explored. To fill these gaps, this paper advises a hybrid solution that combines K-means clustering, Random Forest classification and the Simulated Annealing optimization as a complete end to end system to maximize the efficiency and accuracy of segmentation. K-means clustering first divides images based on pixel intensity, Random Forest narrows its segmentation of images with features like texture, color and shape, and Simulated Annealing determines the desired number of clusters dynamically to segment images with minimal segmentation error. The segmentation error of the proposed method was 30 less than the baseline K-means segmentation accuracy of 65 percent and the proposed method segmentation accuracy was 95% with an optimal cluster number of 10 and a mean error of 7839.22. This hybrid system offers a large-scale, scalable system to underwater image processing that is robust and has applications in marine biology, environmental research, and autonomous underwater system exploration.

References

[1] L. Breiman, “Random Forests,” Mach Learn, vol. 45, no. 1, pp. 5–32, Oct. 2001, doi: 10.1023/A:1010933404324.

[2] S. Kirkpatrick, C. D. Gelatt, and M. P. Vecchi, “Optimization by Simulated Annealing,” Science (1979), vol. 220, no. 4598, pp. 671–680, May 1983, doi: 10.1126/science.220.4598.671.

[3] X. Zheng, Q. Lei, R. Yao, Y. Gong, and Q. Yin, “Image segmentation based on adaptive K-means algorithm,” EURASIP J Image Video Process, vol. 2018, no. 1, p. 68, Dec. 2018, doi: 10.1186/s13640-018-0309-3.

[4] M. Jian, X. Liu, H. Luo, X. Lu, H. Yu, and J. Dong, “Underwater image processing and analysis: A review,” Signal Process Image Commun, vol. 91, p. 116088, Feb. 2021, doi: 10.1016/j.image.2020.116088.

[5] Z. H. Munim, M. Dushenko, V. J. Jimenez, M. H. Shakil, and M. Imset, “Big data and artificial intelligence in the maritime industry: a bibliometric review and future research directions,” Maritime Policy & Management, vol. 47, no. 5, pp. 577–597, Jul. 2020, doi: 10.1080/03088839.2020.1788731.

[6] R. Schettini and S. Corchs, “Underwater Image Processing: State of the Art of Restoration and Image Enhancement Methods,” EURASIP J Adv Signal Process, vol. 2010, no. 1, p. 746052, Dec. 2010, doi: 10.1155/2010/746052.

[7] X. Zuo, J. Jiang, J. Shen, and W. Yang, “Improving underwater semantic segmentation with underwater image quality attention and muti-scale aggregation attention,” Pattern Analysis and Applications, vol. 28, no. 2, p. 80, Jun. 2025, doi: 10.1007/s10044-025-01460-7.

[8] M. J. Islam et al., “Semantic Segmentation of Underwater Imagery: Dataset and Benchmark,” in 2020 IEEE/RSJ International Conference on Intelligent Robots and Systems (IROS), IEEE, Oct. 2020, pp. 1769–1776. doi: 10.1109/IROS45743.2020.9340821.

[9] G. Pavoni et al., “TagLab: AI‐assisted annotation for the fast and accurate semantic segmentation of coral reef orthoimages,” J Field Robot, vol. 39, no. 3, pp. 246–262, May 2022, doi: 10.1002/rob.22049.

[10] H. Lu, Y. Li, Y. Zhang, M. Chen, S. Serikawa, and H. Kim, “Underwater Optical Image Processing: a Comprehensive Review,” Mobile Networks and Applications, vol. 22, no. 6, pp. 1204–1211, Dec. 2017, doi: 10.1007/s11036-017-0863-4.

[11] R. Lou, Z. Lv, S. Dang, T. Su, and X. Li, “Application of machine learning in ocean data,” Multimed Syst, vol. 29, no. 3, pp. 1815–1824, Jun. 2023, doi: 10.1007/s00530-020-00733-x.

[12] M. Xu, J. Su, and Y. Liu, “AquaSAM: Underwater Image Foreground Segmentation,” in Communications in Computer and Information Science, vol. 2066, 2024, pp. 3–14. doi: 10.1007/978-981-97-3623-2_1.

[13] M. Chen, S. Mao, and Y. Liu, “Big Data: A Survey,” Mobile Networks and Applications, vol. 19, no. 2, pp. 171–209, Apr. 2014, doi: 10.1007/s11036-013-0489-0.

[14] A. Gandomi and M. Haider, “Beyond the hype: Big data concepts, methods, and analytics,” Int J Inf Manage, vol. 35, no. 2, pp. 137–144, Apr. 2015, doi: 10.1016/j.ijinfomgt.2014.10.007.

[15] C.-Y. Li, J.-C. Guo, R.-M. Cong, Y.-W. Pang, and B. Wang, “Underwater Image Enhancement by Dehazing With Minimum Information Loss and Histogram Distribution Prior,” IEEE Transactions on Image Processing, vol. 25, no. 12, pp. 5664–5677, Dec. 2016, doi: 10.1109/TIP.2016.2612882.

[16] A. K. Jain, “Data clustering: 50 years beyond K-means,” Pattern Recognit Lett, vol. 31, no. 8, pp. 651–666, Jun. 2010, doi: 10.1016/j.patrec.2009.09.011.

[17] M. J. Kobra, M. O. Rahman, and A. M. Nakib, “A Novel Hybrid Framework for Noise Estimation in High-Texture Images using Markov, MLE, and CNN Approaches,” Scientific Journal of Engineering Research, vol. 1, no. 2, pp. 54–63, 2025, doi: 10.64539/sjer.v1i2.2025.25.

[18] M. Pal, “Random forest classifier for remote sensing classification,” Int J Remote Sens, vol. 26, no. 1, pp. 217–222, Jan. 2005, doi: 10.1080/01431160412331269698.

[19] M. J. Kobra, A. M. Nakib, and M. O. Rahman, “Advanced underwater image restoration: A comparative study of white balance, dehazing, and contrast enhancement techniques,” Engineering Science & Technology Journal, vol. 6, no. 4, pp. 201–215, May 2025, doi: 10.51594/estj.v6i4.1930.

[20] C. Li et al., “An Underwater Image Enhancement Benchmark Dataset and Beyond,” IEEE Transactions on Image Processing, vol. 29, pp. 4376–4389, 2020, doi: 10.1109/TIP.2019.2955241.

[21] G. Hou, X. Zhao, Z. Pan, H. Yang, L. Tan, and J. Li, “Benchmarking Underwater Image Enhancement and Restoration, and Beyond,” IEEE Access, vol. 8, pp. 122078–122091, 2020, doi: 10.1109/ACCESS.2020.3006359.

[22] K. Iqbal, M. Odetayo, A. James, Rosalina Abdul Salam, and Abdullah Zawawi Hj Talib, “Enhancing the low quality images using Unsupervised Colour Correction Method,” in 2010 IEEE International Conference on Systems, Man and Cybernetics, IEEE, Oct. 2010, pp. 1703–1709. doi: 10.1109/ICSMC.2010.5642311.

[23] Richard. S. Sutton and Andrew. G. Barto, Reinforcement learning: An introduction, vol. 1. Cambridge: MIT Press, 1999. doi: 10.1017/S0263574799271172.

[24] C. M. Bishop, “Pattern Recognition and Machine Learning,” in Information Science and Statistics, 1st ed., Springer New York, 2006, pp. 1–778. Accessed: Oct. 16, 2025. [Online]. Available: https://link.springer.com/book/9780387310732

[25] R. O. Duda, P. E. Hart, and D. G. Stork, Pattern classification. 2nd edn Wiley, 2nd ed., vol. 153. New York, 2000.

[26] K. T. Vojjila, “Image Segmentation and Classification of Marine Organisms,” San Jose State University, San Jose, CA, USA, 2018. doi: 10.31979/etd.d4ga-mamb.

[27] G. Padmavathi, M. Muthukumar, and Suresh Kumar Thakur, “Implementation and Comparison of different segmentation algorithms used for underwater images based on nonlinear objective assessments,” in 2010 3rd International Conference on Advanced Computer Theory and Engineering(ICACTE), IEEE, Aug. 2010, pp. V2-393-V2-397. doi: 10.1109/ICACTE.2010.5579301.

Downloads

Published

2025-10-16

How to Cite

Kobra, M. J., Rahman, M. O., & Nakib, A. M. (2025). Hybrid K-means, Random Forest, and Simulated Annealing for Optimizing Underwater Image Segmentation. Scientific Journal of Engineering Research, 1(4), 153–163. https://doi.org/10.64539/sjer.v1i4.2025.46

Issue

Section

Articles

Similar Articles

1 2 > >> 

You may also start an advanced similarity search for this article.