A novel framework to improve the consistency of water quality attribution from natural and anthropogenic factors.

One critical question for water security and sustainable development is how water quality responses to the changes in natural factors and human activities, especially in light of the expected exacerbation in water scarcity. Although machine learning models have shown noticeable advances in water quality attribution analysis, they have limited interpretability in explaining the feature importance with theoretical guarantees of consistency. To fill this gap, this study built a modelling framework that employed the inverse distance weighting method and the extreme gradient boosting model to simulate the water quality at grid scale, and adapted the Shapley additive explanation to interpret the contributions of the drivers to water quality over the Yangtze River basin. Different from previous studies, we calculated the contribution of features to water quality at each grid within river basin and aggregated the contribution from all the grids as the feature importance. Our analysis revealed dramatic changes in response magnitudes of water quality to drivers within river basin. Air temperature had high importance in the variability of key water quality indicators (i.e. ammonia-nitrogen, total phosphorus, and chemical oxygen demand), and dominated the changes of water quality in Yangtze River basin, especially in the upstream region. In the mid- and downstream regions, water quality was mainly affected by human activities. This study provided a modelling framework applicable to robustly identify the feature importance by explaining the contribution of features to water quality at each grid.