Metaheuristic-optimized interaction-aware deep learning with large language model assistance for data-driven water quality prediction
摘要
Accurate prediction of water-quality indicators remains challenging in small tabular environmental datasets because physicochemical variables can exhibit nonlinear interdependencies and modern deep-learning models are sensitive to hyperparameter configuration. In this study, the supervised regression task is defined as predicting dissolved oxygen (mg/L) from the remaining measured physicochemical variables using a modest public dataset of 200 samples. To address this task, the Automatic Feature Interaction Network (AutoInt) is coupled with the Ninja Optimization Algorithm (NiOA) for wrapper-based hyperparameter optimization under a controlled and reproducible experimental protocol. All evaluations use a fixed train/validation/test split (70%/15%/15%) with leakage-safe preprocessing based only on training-set statistics, and results are aggregated over