Work

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

Year: 2025

Type: article

Abstract: ABSTRACT Permanganate index (CODMn), 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 (R2) 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.

Source: Water Science & Technology

DOI: 10.2166/wst.2025.032

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

Institution: Beihua University

Language: English

Topic: Water Quality Monitoring and Analysis

Subfield: Waste Management and Disposal

Field: Environmental Science

Domain: Physical Sciences

SDG: Clean water and sanitation

Open Access status: diamond

Funder: Jilin Scientific and Technological Development Program