Predicting suspended floc size in estuarine waters using self-adaptive parameterized physics-informed neural networks

Ya Wu; Leiping Ye; Jie Ren; Jiaxue Wu

doi:10.1016/j.watres.2025.124094

Predicting suspended floc size in estuarine waters using self-adaptive parameterized physics-informed neural networks

Water Res. 2025 Jun 24:285:124094. doi: 10.1016/j.watres.2025.124094. Online ahead of print.

Authors

Ya Wu¹, Leiping Ye², Jie Ren¹, Jiaxue Wu¹

Affiliations

¹ School of Marine Sciences, Sun Yat-sen University, and Southern Marine Science and Engineering Guangdong Laboratory (Zhuhai), Zhuhai, Guangdong 519082, PR China.
² School of Marine Sciences, Sun Yat-sen University, and Southern Marine Science and Engineering Guangdong Laboratory (Zhuhai), Zhuhai, Guangdong 519082, PR China. Electronic address: yeleiping@mail.sysu.edu.cn.

PMID: 40580835
DOI: 10.1016/j.watres.2025.124094

Abstract

The size dynamics of suspended flocs play fundamental roles in sediment transport processes and ecological functioning in estuarine waters. However, conventional flocculation dynamic models typically rely on fixed empirical parameters, limiting their adaptability under complex hydrodynamic conditions and reducing predictive accuracy. This study presents a new self-adaptive parameterized physics-informed neural networks (SAP-PINNs) model, dynamically optimizing aggregation coefficient, breakage coefficient and erosion parameters within the flocculation dynamic equation, enhancing the accuracy and physical consistency of floc size predictions by integrating data-driven methodologies with physical constraints. Laboratory experiments demonstrate that the model, calibrated under low shear stress conditions, accurately predicts mean floc size under high shear stress, confirming its robustness across variable hydrodynamic regimes. Field validations further indicate a significant improvement in predictive performance compared to traditional models, with an 88.31 % increase in accuracy, a coefficient of determination of 0.99, and a mean absolute error of 0.78. SHapley Additive exPlanations (SHAP) analysis reveals that shear stress and salinity are the dominant factors influencing flocculation, while suspended sediment concentration exhibits a facilitative effect within an optimal range. Temperature exerts a comparatively minor influence. This study demonstrates that the SAP-PINNs effectively combines physical laws with machine learning techniques, improving the modeling accuracy, interpretability, and generalizability of floc dynamics. It offers a new promising potential for application in complex hydrodynamic systems, supporting sediment transport predictions, ecological assessments, and water quality management in estuarine and coastal waters.

Keywords: Estuarine waters; Floc size; SHapley Additive exPlanations (SHAP) analysis; Sediment flocculation; Self-adaptive parameterized physics-informed neural networks (SAP-PINNs).