Improved hydro-epidemiological prediction of faecal indicator organisms using knowledge distillation inspired machine learning through dimensionality reduction

Amini, H., Lam, M.Y. ORCID: https://orcid.org/0000-0001-7259-968X and Ahmadian, R. ORCID: https://orcid.org/0000-0003-2665-4734 2026. Improved hydro-epidemiological prediction of faecal indicator organisms using knowledge distillation inspired machine learning through dimensionality reduction. Journal of Environmental Management 406 , 129792. 10.1016/j.jenvman.2026.129792

PDF - Published Version Available under License Creative Commons Attribution. Download (6MB)

Abstract

Reliable prediction of Faecal Indicator Organisms (FIOs) in coastal bathing waters remains a major challenge because FIO concentrations could vary rapidly with the flow, meteorological, and hydrological conditions. Mechanistic models require high computational resources, data and could be time consuming to set up. Machine learning (ML) models have been used as alternative solutions particularly for early warning systems and digital twins. However, conventional ML models often underperform when exposed to correlated environmental variables. In this study, we propose a hybrid, dimension-reduced ensemble framework that combines Principal Component Analysis (PCA) with stacked ML and knowledge-distillation-inspired teacher-student model to deliver accurate, interpretable forecasts from a reduced number of inputs. The hybrid approach was applied to the Swansea Bay, UK, where 23 different hydro-meteorological and E. coli and Enterococci data were collected during one bathing season at high-frequency, namely 30-min. The results showed that the first two principal components (PCs), highlighting the warm/sunny or cool/humid weather types (PC1) and the wind-driven or calmer and level of rain (PC2), were the two dominant axes that characterise the Bay. The PCs were fed into a stacked “teacher” ensemble (KNN-RF-XGBoost) model whose predictive signal was partially transferred to a lightweight neural network “student” via blended supervision inspired by knowledge distillation. The framework achieved R2 = 0.72, RMSE = 0.11 (∼259 cfu/100 ml) for E. coli, and R2 = 0.64, RMSE = 0.14 for Enterococci (∼334 cfu/100 ml), outperforming standalone ML or DL models by up to 30 percent. Beyond accuracy, the PCA layer reduced input by 65 percent, cutting requirements for sensors for early warning systems and digital twins, computational cost and revealing physically meaningful patterns driving contamination dynamics. This PCA-ensemble-teacher-student hybrid modelling provides an interpretable, scalable tool for real-time water quality forecasting in dynamic coastal environments.

Item Type:	Article
Date Type:	Publication
Status:	Published
Schools:	Schools > Engineering
Publisher:	Elsevier
ISSN:	0301-4797
Funders:	EPSRC
Projects:	EP/W027879/1
Date of First Compliant Deposit:	5 May 2026
Date of Acceptance:	23 April 2026
Last Modified:	17 Aug 2026 14:13
URI:	https://orca.cardiff.ac.uk/id/eprint/186797

Actions (repository staff only)

Edit Item