UAV Multispectral Image-Based Urban River Water Quality Monitoring Using Stacked Ensemble Machine Learning Algorithms—A Case Study of the Zhanghe River, China

Abstract

Timely monitoring of inland water quality using unmanned aerial vehicle (UAV) remote sensing is critical for water environmental conservation and management. In this study, two UAV flights were conducted (one in February and the other in December 2021) to acquire images of the Zhanghe River (China), and a total of 45 water samples were collected concurrently with the image acquisition. Machine learning (ML) methods comprising Multiple Linear Regression, the Least Absolute Shrinkage and Selection Operator, a Backpropagation Neural Network (BP), Random Forest (RF), and eXtreme Gradient Boosting (XGBoost) were applied to retrieve four water quality parameters: chlorophyll-a (Chl-a), total nitrogen (TN), total phosphors (TP), and permanganate index (CODMn). Then, ML models based on the stacking approach were developed. Results show that stacked ML models could achieve higher accuracy than a single ML model; the optimal methods for Chl-a, TN, TP, and CODMn were RF-XGB, BP-RF, RF, and BP-RF, respectively. For the testing dataset, the R2 values of the best inversion models for Chl-a, TN, TP, and CODMn were 0.504, 0.839, 0.432, and 0.272, the root mean square errors were 1.770 μg L−1, 0.189 mg L−1, 0.053 mg L−1, and 0.767 mg L−1, and the mean absolute errors were 1.272 μg L−1, 0.632 mg L−1, 0.045 mg L−1, and 0.674 mg L−1, respectively. This study demonstrated the great potential of combined UAV remote sensing and stacked ML algorithms for water quality monitoring.