IDEAS home Printed from https://ideas.repec.org/a/spr/waterr/v39y2025i7d10.1007_s11269-024-04067-5.html
   My bibliography  Save this article

Water Quality Prediction Method Coupling Mechanism Model and Machine Learning for Water Diversion Projects with a Lack of Data

Author

Listed:
  • Xiaochen Yang

    (China Institute of Water Resources and Hydropower Research
    Key Laboratory of Water Safety for Beijing–Tianjin–Hebei Region of Ministry of Water Resources)

  • Kai Liu

    (China South-to-North Water Diversion Middle Route Corporation Limited)

  • Xiaobo Liu

    (China Institute of Water Resources and Hydropower Research
    Key Laboratory of Water Safety for Beijing–Tianjin–Hebei Region of Ministry of Water Resources)

  • Fei Dong

    (China Institute of Water Resources and Hydropower Research
    Key Laboratory of Water Safety for Beijing–Tianjin–Hebei Region of Ministry of Water Resources)

  • Aiping Huang

    (China Institute of Water Resources and Hydropower Research
    Key Laboratory of Water Safety for Beijing–Tianjin–Hebei Region of Ministry of Water Resources)

  • Bing Ma

    (China Institute of Water Resources and Hydropower Research
    Key Laboratory of Water Safety for Beijing–Tianjin–Hebei Region of Ministry of Water Resources)

  • Yang Lei

    (China Institute of Water Resources and Hydropower Research
    Key Laboratory of Water Safety for Beijing–Tianjin–Hebei Region of Ministry of Water Resources
    Dezhou university)

  • Zhi Jiang

    (China Institute of Water Resources and Hydropower Research
    Key Laboratory of Water Safety for Beijing–Tianjin–Hebei Region of Ministry of Water Resources)

Abstract

Newly constructed water diversion projects and projects with inadequate monitoring facilities often lack water quality data, making it difficult to achieve accurate water quality predictions. Mechanism model and machine learning each have their own advantages and shortcomings in terms of water quality predictions, and coupling these two models may improve results; this is also a hot research topic. This study focuses on the water quality prediction task for water diversion projects that lack monitoring data. Using the Xihe and Zhaohe River section of the Yangtze–to–Huaihe Water Diversion Project, which is a typical water diversion project in China, as the study area, we have constructed a mechanism water quality prediction model (MIKE11) and a machine learning support vector regression model (SVR), then proposed a coupled mechanism model–machine learning water quality prediction model to explore the impacts of different input features on the model’s performance. The coupled model is also adopted to predict the water quality variation process under typical water diversion scenarios of the Yangtze–to–Huaihe Water Diversion Project. The study shows that the coupled model with both the flow rate and water quality as input features have an average relative error of 0.03% and 0.21% in predicting COD and NH3-N concentrations, respectively, and the prediction performance of it is good. It successfully overcomes the problem of poor prediction performance faced by the SVR model when there are insufficient sample data, and it can be used to predict water quality for water diversion projects that lack monitoring data. This paper proposes a new method to predict water quality for water diversion projects that lack monitoring data, expanding the applicability of machine learning in this field, providing a theoretical basis for water diversion project-related water quality prediction.

Suggested Citation

  • Xiaochen Yang & Kai Liu & Xiaobo Liu & Fei Dong & Aiping Huang & Bing Ma & Yang Lei & Zhi Jiang, 2025. "Water Quality Prediction Method Coupling Mechanism Model and Machine Learning for Water Diversion Projects with a Lack of Data," Water Resources Management: An International Journal, Published for the European Water Resources Association (EWRA), Springer;European Water Resources Association (EWRA), vol. 39(7), pages 3015-3030, May.
    • Handle: RePEc:spr:waterr:v:39:y:2025:i:7:d:10.1007_s11269-024-04067-5
    DOI: 10.1007/s11269-024-04067-5
    as

    Download full text from publisher

    File URL: http://link.springer.com/10.1007/s11269-024-04067-5
    File Function: Abstract
    Download Restriction: Access to the full text of the articles in this series is restricted.
    File URL: https://libkey.io/10.1007/s11269-024-04067-5?utm_source=ideas
    LibKey link: if access is restricted and if your library uses this service, LibKey will redirect you to where you can use your library subscription to access this item
    ---><---

    As the access to this document is restricted, you may want to search for a different version of it.

    References listed on IDEAS

    as
    1. Elhabashy, Ahmed & Li, Jing & Sokolova, Ekaterina, 2023. "Water quality modeling of a eutrophic drinking water source: Impact of future climate on Cyanobacterial blooms," Ecological Modelling, Elsevier, vol. 477(C).
    2. Mahdi Valikhan Anaraki & Saeed Farzin & Sayed-Farhad Mousavi & Hojat Karami, 2021. "Uncertainty Analysis of Climate Change Impacts on Flood Frequency by Using Hybrid Machine Learning Methods," Water Resources Management: An International Journal, Published for the European Water Resources Association (EWRA), Springer;European Water Resources Association (EWRA), vol. 35(1), pages 199-223, January.
    3. Mohammadi, Babak & Mehdizadeh, Saeid, 2020. "Modeling daily reference evapotranspiration via a novel approach based on support vector regression coupled with whale optimization algorithm," Agricultural Water Management, Elsevier, vol. 237(C).
    Full references (including those not matched with items on IDEAS)

    Most related items

    These are the items that most often cite the same works as this one and are cited by the same works as this one.
    1. Pezhman Allahbakhshian-Farsani & Mehdi Vafakhah & Hadi Khosravi-Farsani & Elke Hertig, 2025. "Projection of Mean Annual and Maximum 24-h Precipitation under Future Climatic Scenarios in Semi-Arid Regions," Water Resources Management: An International Journal, Published for the European Water Resources Association (EWRA), Springer;European Water Resources Association (EWRA), vol. 39(6), pages 2785-2817, April.
    2. Zhou, Hanmi & Ma, Linshuang & Niu, Xiaoli & Xiang, Youzhen & Chen, Jiageng & Su, Yumin & Li, Jichen & Lu, Sibo & Chen, Cheng & Wu, Qi, 2024. "A novel hybrid model combined with ensemble embedded feature selection method for estimating reference evapotranspiration in the North China Plain," Agricultural Water Management, Elsevier, vol. 296(C).
    3. Manish Kumar & Anuradha Kumari & Daniel Prakash Kushwaha & Pravendra Kumar & Anurag Malik & Rawshan Ali & Alban Kuriqi, 2020. "Estimation of Daily Stage–Discharge Relationship by Using Data-Driven Techniques of a Perennial River, India," Sustainability, MDPI, vol. 12(19), pages 1-21, September.
    4. Babak Mohammadi & Farshad Ahmadi & Saeid Mehdizadeh & Yiqing Guan & Quoc Bao Pham & Nguyen Thi Thuy Linh & Doan Quang Tri, 2020. "Developing Novel Robust Models to Improve the Accuracy of Daily Streamflow Modeling," Water Resources Management: An International Journal, Published for the European Water Resources Association (EWRA), Springer;European Water Resources Association (EWRA), vol. 34(10), pages 3387-3409, August.
    5. Yan, Zheping & Zhang, Jinzhong & Zeng, Jia & Tang, Jialing, 2021. "Nature-inspired approach: An enhanced whale optimization algorithm for global optimization," Mathematics and Computers in Simulation (MATCOM), Elsevier, vol. 185(C), pages 17-46.
    6. Hadeel E. Khairan & Salah L. Zubaidi & Mustafa Al-Mukhtar & Anmar Dulaimi & Hussein Al-Bugharbee & Furat A. Al-Faraj & Hussein Mohammed Ridha, 2023. "Assessing the Potential of Hybrid-Based Metaheuristic Algorithms Integrated with ANNs for Accurate Reference Evapotranspiration Forecasting," Sustainability, MDPI, vol. 15(19), pages 1-19, September.
    7. Elbeltagi, Ahmed & Deng, Jinsong & Wang, Ke & Malik, Anurag & Maroufpoor, Saman, 2020. "Modeling long-term dynamics of crop evapotranspiration using deep learning in a semi-arid environment," Agricultural Water Management, Elsevier, vol. 241(C).
    8. Farshad Ahmadi & Saeid Mehdizadeh & Babak Mohammadi, 2021. "Development of Bio-Inspired- and Wavelet-Based Hybrid Models for Reconnaissance Drought Index Modeling," Water Resources Management: An International Journal, Published for the European Water Resources Association (EWRA), Springer;European Water Resources Association (EWRA), vol. 35(12), pages 4127-4147, September.
    9. Dong-Her Shih & Ching-Hsien Liao & Ting-Wei Wu & Huan-Shuo Chang & Ming-Hung Shih, 2022. "WSI: A New Early Warning Water Survival Index for the Domestic Water Demand," Mathematics, MDPI, vol. 10(23), pages 1-19, November.
    10. Mona Nemati & Mahmoud Mohammad Rezapour Tabari & Seyed Abbas Hosseini & Saman Javadi, 2021. "A Novel Approach Using Hybrid Fuzzy Vertex Method-MATLAB Framework Based on GMS Model for Quantifying Predictive Uncertainty Associated with Groundwater Flow and Transport Models," Water Resources Management: An International Journal, Published for the European Water Resources Association (EWRA), Springer;European Water Resources Association (EWRA), vol. 35(12), pages 4189-4215, September.
    11. Mojtaba Kadkhodazadeh & Saeed Farzin, 2022. "Introducing a Novel Hybrid Machine Learning Model and Developing its Performance in Estimating Water Quality Parameters," Water Resources Management: An International Journal, Published for the European Water Resources Association (EWRA), Springer;European Water Resources Association (EWRA), vol. 36(10), pages 3901-3927, August.
    12. Srishti Gaur & Arnab Bandyopadhyay & Rajendra Singh, 2021. "From Changing Environment to Changing Extremes: Exploring the Future Streamflow and Associated Uncertainties Through Integrated Modelling System," Water Resources Management: An International Journal, Published for the European Water Resources Association (EWRA), Springer;European Water Resources Association (EWRA), vol. 35(6), pages 1889-1911, April.
    13. Isa Ebtehaj & Hossein Bonakdari, 2024. "Generalized Structure of Group Method of Data Handling: Novel Technique for Flash Flood Forecasting," Water Resources Management: An International Journal, Published for the European Water Resources Association (EWRA), Springer;European Water Resources Association (EWRA), vol. 38(9), pages 3235-3253, July.
    14. Milan Gocić & Mohammad Arab Amiri, 2021. "Reference Evapotranspiration Prediction Using Neural Networks and Optimum Time Lags," Water Resources Management: An International Journal, Published for the European Water Resources Association (EWRA), Springer;European Water Resources Association (EWRA), vol. 35(6), pages 1913-1926, April.
    15. Zhang, Lei & Zhao, Xin & Zhu, Ge & He, Jun & Chen, Jian & Chen, Zhicheng & Traore, Seydou & Liu, Junguo & Singh, Vijay P., 2023. "Short-term daily reference evapotranspiration forecasting using temperature-based deep learning models in different climate zones in China," Agricultural Water Management, Elsevier, vol. 289(C).
    16. Zahra khoramipoor & Saeed Farzin, 2025. "A Methodology to Improving the Performance of MOAHA Optimization Algorithm using Chaos Theory; Principle and Application in Optimal Reservoir Operation," Water Resources Management: An International Journal, Published for the European Water Resources Association (EWRA), Springer;European Water Resources Association (EWRA), vol. 39(6), pages 2819-2840, April.
    17. Mojtaba Kadkhodazadeh & Saeed Farzin, 2021. "A Novel LSSVM Model Integrated with GBO Algorithm to Assessment of Water Quality Parameters," Water Resources Management: An International Journal, Published for the European Water Resources Association (EWRA), Springer;European Water Resources Association (EWRA), vol. 35(12), pages 3939-3968, September.
    18. Dong, Juan & Xing, Liwen & Cui, Ningbo & Zhao, Lu & Guo, Li & Wang, Zhihui & Du, Taisheng & Tan, Mingdong & Gong, Daozhi, 2024. "Estimating reference crop evapotranspiration using improved convolutional bidirectional long short-term memory network by multi-head attention mechanism in the four climatic zones of China," Agricultural Water Management, Elsevier, vol. 292(C).
    19. Valipour, Mohammad & Khoshkam, Helaleh & Bateni, Sayed M. & Jun, Changhyun & Band, Shahab S., 2023. "Hybrid machine learning and deep learning models for multi-step-ahead daily reference evapotranspiration forecasting in different climate regions across the contiguous United States," Agricultural Water Management, Elsevier, vol. 283(C).
    20. Ahmadi, Farshad & Mehdizadeh, Saeid & Mohammadi, Babak & Pham, Quoc Bao & DOAN, Thi Ngoc Canh & Vo, Ngoc Duong, 2021. "Application of an artificial intelligence technique enhanced with intelligent water drops for monthly reference evapotranspiration estimation," Agricultural Water Management, Elsevier, vol. 244(C).

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:spr:waterr:v:39:y:2025:i:7:d:10.1007_s11269-024-04067-5. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    If CitEc recognized a bibliographic reference but did not link an item in RePEc to it, you can help with this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: Sonal Shukla or Springer Nature Abstracting and Indexing (email available below). General contact details of provider: http://www.springer.com .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.