A Two-Stage Interpretable Machine Learning Framework for Accurate Prediction of Trace Pollutants: With an Application to Microcystin

Abstract

Trace pollutants are widely observed in aquatic ecosystems and can significantly impact human health and the environment. Accurate prediction of trace pollutants and understanding their response to environmental drivers are key to environmental management, yet these tasks remain challenging. An important reason for this challenge is that monitoring data for trace pollutants are often left-censored, leading to biased estimation and inaccurate response-driver relationships. Here we propose a novel two-stage interpretable machine learning framework applicable to left-censored trace pollutant data. The two stages include (1) a classifier to predict the presence of the pollutant and (2) a regressor to predict the pollutant concentration if present. The two stages were followed by a model interpretation to understand the contribution of drivers to the presence and concentration of the pollutant accordingly. We take the prediction of microcystin (MICX) in lakes across the United States as a case study. Applying this framework to MICX consistently improved prediction accuracy, including prediction of its occurrence and concentration regardless of the algorithms and performance metrics used. The best-performing algorithm using the two-stage framework, compared with the baseline model, improves classification performance by 48% to 290% and the regression performance by 11% to 33%, depending on the metric used to evaluate the performance. The interpretable machine learning model also successfully revealed the impacts of the most important drivers on the presence of MICX and its concentration. Our results showed the advantages of this framework, including its interpretability to understand the driver-response relationship, ability to handle nonlinearity, better prediction performance, differentiation between the underlying processes, and potential to be generalized to other pollutants. As such, we anticipate that the framework we propose will be a starting point for using state-of-the-art interpretable machine learning models for predicting trace pollutants.

Applications of Artificial Intelligence, Machine Learning, and Data Analytics in Water Environments

“…The special issue includes several review articles encompassing a wide spectrum, ranging from a historical perspective of water data to computational modeling in wastewater treatment to ML modeling of environmental chemical reactions, environmental toxicology, heavy metal removal, and cyanobacterial harmful algal blooms (HABs) . One significant application of these innovative tools is ML-assisted environmental monitoring, which can address diverse problems, such as predicting effluent nutrients or influent flow rates and nutrient loads at wastewater treatment plants, , formation of disinfection byproducts, drivers of the accumulation of potentially toxic elements in sediments, greenhouse gas emissions, , occurrence of PFAS, water quality assessment, microplastics, microcystins, and differentiation of landfill leachate and domestic sludge . ML has also been extensively employed to model environmental chemical reactions and processes, including adsorption onto various materials, , biodegradation, photodegradation, and the physicochemical and meteorological variables that affect the seasonal growth and decline of HABs .…”