Prediction of surface water pollution using wavelet transform and 1D-CNN.

Authors

Gaofeng Wang, Hao Zhang, Man Gao, Tao Zhou, Yun Qian

Published in

Water science and technology : a journal of the International Association on Water Pollution Research. Volume 91. Issue 6. Pages 684-697. Epub Mar 04, 2025.

Abstract

Permanganate index (COD_Mn), total nitrogen, and ammonia nitrogen are important indicators that represent the degree of pollution of surface water. This study combined ultraviolet-visible (UV-vis) spectroscopy with a one-dimensional convolutional neural network (1D-CNN) to spectrally analyze 708 samples with different concentrations. The wavelet transform was used to preprocess the spectra to improve the model's accuracy. The results show the best prediction results using a fixed threshold (sqtwolog) of wavelets in combination with 1D-CNN, and the coefficient of determination (R²) values of the models on the test dataset all reached more than 0.98. A comparison between the backpropagation neural network model and the extreme learning machine model reveals that the 1D-CNN model has better prediction accuracy and robustness. The experimental results show the strong practical value of using 1D-CNN to predict the levels of different compounds in surface water.

PMID:
40156446
Bibliographic data and abstract were imported from PubMed on 30 Mar 2025.

Read full publication at:
Please sign in to see all details.

Stats

1-terrible, 9-excellent. How would you rate this publication? Sign in in to submit your rating.

Post a comment

You need to be signed in to post comments. You can sign in here.

Comments

There are no comments yet.