in

A machine learning-driven framework for optimizing disinfection in drinking water treatment

Abstract

Drinking water disinfection effectively prevents waterborne disease outbreaks but inevitably results in the formation of carcinogenic disinfection by-products (DBPs). Mitigating DBP formation without compromising microbial safety remains a critical challenge. Here we developed DISoptimizer, a water-quality-adaptive optimization framework designed to determine the optimal chlorine dose across heterogeneous water matrices, thereby maintaining a user-defined residual chlorine target as an operational surrogate for disinfection reliability while minimizing four trihalomethane (THM4) formation. Using machine learning models trained on experimental datasets covering diverse water quality conditions, DISoptimizer predicts residual chlorine and THM4 formation after 24 h, representing the point of delivery. Crucially, DISoptimizer incorporates a weighting factor to encode user-defined preferences, allowing utilities to adjust the operational emphasis on maintaining residual chlorine margins versus mitigating THM4 formation in response to fluctuating water quality and different management priorities. Computational simulations coupled with external validation demonstrated that DISoptimizer achieved a 5–35% reduction in THM4 formation compared to the empirical fixed-dosing strategies, while consistently maintaining residual chlorine margins. This work advances disinfection management from DBP concentration prediction towards proactive chlorine-dosing decisions, supporting the management of health risks associated with microbial contamination and DBP exposure in drinking water.

This is a preview of subscription content, access via your institution

Access options

Access through your institution

/* style specs start */

/* style specs end */

Buy this article

USD 39.95

Prices may be subject to local taxes which are calculated during checkout

/* style specs start */
style {
display: none !important;
}
.LiveAreaSection * {
align-content: stretch;
align-items: stretch;
align-self: auto;
animation-delay: 0s;
animation-direction: normal;
animation-duration: 0s;
animation-fill-mode: none;
animation-iteration-count: 1;
animation-name: none;
animation-play-state: running;
animation-timing-function: ease;
azimuth: center;
backface-visibility: visible;
background-attachment: scroll;
background-blend-mode: normal;
background-clip: borderBox;
background-color: transparent;
background-image: none;
background-origin: paddingBox;
background-position: 0 0;
background-repeat: repeat;
background-size: auto auto;
block-size: auto;
border-block-end-color: currentcolor;
border-block-end-style: none;
border-block-end-width: medium;
border-block-start-color: currentcolor;
border-block-start-style: none;
border-block-start-width: medium;
border-bottom-color: currentcolor;
border-bottom-left-radius: 0;
border-bottom-right-radius: 0;
border-bottom-style: none;
border-bottom-width: medium;
border-collapse: separate;
border-image-outset: 0s;
border-image-repeat: stretch;
border-image-slice: 100%;
border-image-source: none;
border-image-width: 1;
border-inline-end-color: currentcolor;
border-inline-end-style: none;
border-inline-end-width: medium;
border-inline-start-color: currentcolor;
border-inline-start-style: none;
border-inline-start-width: medium;
border-left-color: currentcolor;
border-left-style: none;
border-left-width: medium;
border-right-color: currentcolor;
border-right-style: none;
border-right-width: medium;
border-spacing: 0;
border-top-color: currentcolor;
border-top-left-radius: 0;
border-top-right-radius: 0;
border-top-style: none;
border-top-width: medium;
bottom: auto;
box-decoration-break: slice;
box-shadow: none;
box-sizing: border-box;
break-after: auto;
break-before: auto;
break-inside: auto;
caption-side: top;
caret-color: auto;
clear: none;
clip: auto;
clip-path: none;
color: initial;
column-count: auto;
column-fill: balance;
column-gap: normal;
column-rule-color: currentcolor;
column-rule-style: none;
column-rule-width: medium;
column-span: none;
column-width: auto;
content: normal;
counter-increment: none;
counter-reset: none;
cursor: auto;
display: inline;
empty-cells: show;
filter: none;
flex-basis: auto;
flex-direction: row;
flex-grow: 0;
flex-shrink: 1;
flex-wrap: nowrap;
float: none;
font-family: initial;
font-feature-settings: normal;
font-kerning: auto;
font-language-override: normal;
font-size: medium;
font-size-adjust: none;
font-stretch: normal;
font-style: normal;
font-synthesis: weight style;
font-variant: normal;
font-variant-alternates: normal;
font-variant-caps: normal;
font-variant-east-asian: normal;
font-variant-ligatures: normal;
font-variant-numeric: normal;
font-variant-position: normal;
font-weight: 400;
grid-auto-columns: auto;
grid-auto-flow: row;
grid-auto-rows: auto;
grid-column-end: auto;
grid-column-gap: 0;
grid-column-start: auto;
grid-row-end: auto;
grid-row-gap: 0;
grid-row-start: auto;
grid-template-areas: none;
grid-template-columns: none;
grid-template-rows: none;
height: auto;
hyphens: manual;
image-orientation: 0deg;
image-rendering: auto;
image-resolution: 1dppx;
ime-mode: auto;
inline-size: auto;
isolation: auto;
justify-content: flexStart;
left: auto;
letter-spacing: normal;
line-break: auto;
line-height: normal;
list-style-image: none;
list-style-position: outside;
list-style-type: disc;
margin-block-end: 0;
margin-block-start: 0;
margin-bottom: 0;
margin-inline-end: 0;
margin-inline-start: 0;
margin-left: 0;
margin-right: 0;
margin-top: 0;
mask-clip: borderBox;
mask-composite: add;
mask-image: none;
mask-mode: matchSource;
mask-origin: borderBox;
mask-position: 0 0;
mask-repeat: repeat;
mask-size: auto;
mask-type: luminance;
max-height: none;
max-width: none;
min-block-size: 0;
min-height: 0;
min-inline-size: 0;
min-width: 0;
mix-blend-mode: normal;
object-fit: fill;
object-position: 50% 50%;
offset-block-end: auto;
offset-block-start: auto;
offset-inline-end: auto;
offset-inline-start: auto;
opacity: 1;
order: 0;
orphans: 2;
outline-color: initial;
outline-offset: 0;
outline-style: none;
outline-width: medium;
overflow: visible;
overflow-wrap: normal;
overflow-x: visible;
overflow-y: visible;
padding-block-end: 0;
padding-block-start: 0;
padding-bottom: 0;
padding-inline-end: 0;
padding-inline-start: 0;
padding-left: 0;
padding-right: 0;
padding-top: 0;
page-break-after: auto;
page-break-before: auto;
page-break-inside: auto;
perspective: none;
perspective-origin: 50% 50%;
pointer-events: auto;
position: static;
quotes: initial;
resize: none;
right: auto;
ruby-align: spaceAround;
ruby-merge: separate;
ruby-position: over;
scroll-behavior: auto;
scroll-snap-coordinate: none;
scroll-snap-destination: 0 0;
scroll-snap-points-x: none;
scroll-snap-points-y: none;
scroll-snap-type: none;
shape-image-threshold: 0;
shape-margin: 0;
shape-outside: none;
tab-size: 8;
table-layout: auto;
text-align: initial;
text-align-last: auto;
text-combine-upright: none;
text-decoration-color: currentcolor;
text-decoration-line: none;
text-decoration-style: solid;
text-emphasis-color: currentcolor;
text-emphasis-position: over right;
text-emphasis-style: none;
text-indent: 0;
text-justify: auto;
text-orientation: mixed;
text-overflow: clip;
text-rendering: auto;
text-shadow: none;
text-transform: none;
text-underline-position: auto;
top: auto;
touch-action: auto;
transform: none;
transform-box: borderBox;
transform-origin: 50% 50%0;
transform-style: flat;
transition-delay: 0s;
transition-duration: 0s;
transition-property: all;
transition-timing-function: ease;
vertical-align: baseline;
visibility: visible;
white-space: normal;
widows: 2;
width: auto;
will-change: auto;
word-break: normal;
word-spacing: normal;
word-wrap: normal;
writing-mode: horizontalTb;
z-index: auto;
-webkit-appearance: none;
-moz-appearance: none;
-ms-appearance: none;
appearance: none;
margin: 0;
}
.LiveAreaSection {
width: 100%;
}
.LiveAreaSection .login-option-buybox {
display: block;
width: 100%;
font-size: 17px;
line-height: 30px;
color: #222;
padding-top: 30px;
font-family: Harding, Palatino, serif;
}
.LiveAreaSection .additional-access-options {
display: block;
font-weight: 700;
font-size: 17px;
line-height: 30px;
color: #222;
font-family: Harding, Palatino, serif;
}
.LiveAreaSection .additional-login > li:not(:first-child)::before {
transform: translateY(-50%);
content: “”;
height: 1rem;
position: absolute;
top: 50%;
left: 0;
border-left: 2px solid #999;
}
.LiveAreaSection .additional-login > li:not(:first-child) {
padding-left: 10px;
}
.LiveAreaSection .additional-login > li {
display: inline-block;
position: relative;
vertical-align: middle;
padding-right: 10px;
}
.BuyBoxSection {
display: flex;
flex-wrap: wrap;
flex: 1;
flex-direction: row-reverse;
margin: -30px -15px 0;
}
.BuyBoxSection .box-inner {
width: 100%;
height: 100%;
padding: 30px 5px;
display: flex;
flex-direction: column;
justify-content: space-between;
}
.BuyBoxSection p {
margin: 0;
}
.BuyBoxSection .readcube-buybox {
background-color: #f3f3f3;
flex-shrink: 1;
flex-grow: 1;
flex-basis: 255px;
background-clip: content-box;
padding: 0 15px;
margin-top: 30px;
}
.BuyBoxSection .subscribe-buybox {
background-color: #f3f3f3;
flex-shrink: 1;
flex-grow: 4;
flex-basis: 300px;
background-clip: content-box;
padding: 0 15px;
margin-top: 30px;
}
.BuyBoxSection .subscribe-buybox-nature-plus {
background-color: #f3f3f3;
flex-shrink: 1;
flex-grow: 4;
flex-basis: 100%;
background-clip: content-box;
padding: 0 15px;
margin-top: 30px;
}
.BuyBoxSection .title-readcube,
.BuyBoxSection .title-buybox {
display: block;
margin: 0;
margin-right: 10%;
margin-left: 10%;
font-size: 24px;
line-height: 32px;
color: #222;
text-align: center;
font-family: Harding, Palatino, serif;
}
.BuyBoxSection .title-asia-buybox {
display: block;
margin: 0;
margin-right: 5%;
margin-left: 5%;
font-size: 24px;
line-height: 32px;
color: #222;
text-align: center;
font-family: Harding, Palatino, serif;
}
.BuyBoxSection .asia-link,
.Link-328123652,
.Link-2926870917,
.Link-2291679238,
.Link-595459207 {
color: #069;
cursor: pointer;
text-decoration: none;
font-size: 1.05em;
font-family: -apple-system, BlinkMacSystemFont, “Segoe UI”, Roboto,
Oxygen-Sans, Ubuntu, Cantarell, “Helvetica Neue”, sans-serif;
line-height: 1.05em6;
}
.BuyBoxSection .access-readcube {
display: block;
margin: 0;
margin-right: 10%;
margin-left: 10%;
font-size: 14px;
color: #222;
padding-top: 10px;
text-align: center;
font-family: -apple-system, BlinkMacSystemFont, “Segoe UI”, Roboto,
Oxygen-Sans, Ubuntu, Cantarell, “Helvetica Neue”, sans-serif;
line-height: 20px;
}
.BuyBoxSection ul {
margin: 0;
}
.BuyBoxSection .link-usp {
display: list-item;
margin: 0;
margin-left: 20px;
padding-top: 6px;
list-style-position: inside;
}
.BuyBoxSection .link-usp span {
font-size: 14px;
color: #222;
font-family: -apple-system, BlinkMacSystemFont, “Segoe UI”, Roboto,
Oxygen-Sans, Ubuntu, Cantarell, “Helvetica Neue”, sans-serif;
line-height: 20px;
}
.BuyBoxSection .access-asia-buybox {
display: block;
margin: 0;
margin-right: 5%;
margin-left: 5%;
font-size: 14px;
color: #222;
padding-top: 10px;
text-align: center;
font-family: -apple-system, BlinkMacSystemFont, “Segoe UI”, Roboto,
Oxygen-Sans, Ubuntu, Cantarell, “Helvetica Neue”, sans-serif;
line-height: 20px;
}
.BuyBoxSection .access-buybox {
display: block;
margin: 0;
margin-right: 10%;
margin-left: 10%;
font-size: 14px;
color: #222;
opacity: 0.8px;
padding-top: 10px;
text-align: center;
font-family: -apple-system, BlinkMacSystemFont, “Segoe UI”, Roboto,
Oxygen-Sans, Ubuntu, Cantarell, “Helvetica Neue”, sans-serif;
line-height: 20px;
}
.BuyBoxSection .price-buybox {
display: block;
font-size: 30px;
color: #222;
font-family: -apple-system, BlinkMacSystemFont, “Segoe UI”, Roboto,
Oxygen-Sans, Ubuntu, Cantarell, “Helvetica Neue”, sans-serif;
padding-top: 30px;
text-align: center;
}
.BuyBoxSection .price-buybox-to {
display: block;
font-size: 30px;
color: #222;
font-family: -apple-system, BlinkMacSystemFont, “Segoe UI”, Roboto,
Oxygen-Sans, Ubuntu, Cantarell, “Helvetica Neue”, sans-serif;
text-align: center;
}
.BuyBoxSection .price-info-text {
font-size: 16px;
padding-right: 10px;
color: #222;
font-family: -apple-system, BlinkMacSystemFont, “Segoe UI”, Roboto,
Oxygen-Sans, Ubuntu, Cantarell, “Helvetica Neue”, sans-serif;
}
.BuyBoxSection .price-value {
font-size: 30px;
font-family: -apple-system, BlinkMacSystemFont, “Segoe UI”, Roboto,
Oxygen-Sans, Ubuntu, Cantarell, “Helvetica Neue”, sans-serif;
}
.BuyBoxSection .price-per-period {
font-family: -apple-system, BlinkMacSystemFont, “Segoe UI”, Roboto,
Oxygen-Sans, Ubuntu, Cantarell, “Helvetica Neue”, sans-serif;
}
.BuyBoxSection .price-from {
font-size: 14px;
padding-right: 10px;
color: #222;
font-family: -apple-system, BlinkMacSystemFont, “Segoe UI”, Roboto,
Oxygen-Sans, Ubuntu, Cantarell, “Helvetica Neue”, sans-serif;
line-height: 20px;
}
.BuyBoxSection .issue-buybox {
display: block;
font-size: 13px;
text-align: center;
color: #222;
font-family: -apple-system, BlinkMacSystemFont, “Segoe UI”, Roboto,
Oxygen-Sans, Ubuntu, Cantarell, “Helvetica Neue”, sans-serif;
line-height: 19px;
}
.BuyBoxSection .no-price-buybox {
display: block;
font-size: 13px;
line-height: 18px;
text-align: center;
padding-right: 10%;
padding-left: 10%;
padding-bottom: 20px;
padding-top: 30px;
color: #222;
font-family: -apple-system, BlinkMacSystemFont, “Segoe UI”, Roboto,
Oxygen-Sans, Ubuntu, Cantarell, “Helvetica Neue”, sans-serif;
}
.BuyBoxSection .vat-buybox {
display: block;
margin-top: 5px;
margin-right: 20%;
margin-left: 20%;
font-size: 11px;
color: #222;
padding-top: 10px;
padding-bottom: 15px;
text-align: center;
font-family: -apple-system, BlinkMacSystemFont, “Segoe UI”, Roboto,
Oxygen-Sans, Ubuntu, Cantarell, “Helvetica Neue”, sans-serif;
line-height: 17px;
}
.BuyBoxSection .tax-buybox {
display: block;
width: 100%;
color: #222;
padding: 20px 16px;
text-align: center;
font-family: -apple-system, BlinkMacSystemFont, “Segoe UI”, Roboto,
Oxygen-Sans, Ubuntu, Cantarell, “Helvetica Neue”, sans-serif;
line-height: NaNpx;
}
.BuyBoxSection .button-container {
display: flex;
padding-right: 20px;
padding-left: 20px;
justify-content: center;
}
.BuyBoxSection .button-container > * {
flex: 1px;
}
.BuyBoxSection .button-container > a:hover,
.Button-505204839:hover,
.Button-1078489254:hover,
.Button-2737859108:hover {
text-decoration: none;
}
.BuyBoxSection .btn-secondary {
background: #fff;
}
.BuyBoxSection .button-asia {
background: #069;
border: 1px solid #069;
border-radius: 0;
cursor: pointer;
display: block;
padding: 9px;
outline: 0;
text-align: center;
text-decoration: none;
min-width: 80px;
margin-top: 75px;
}
.BuyBoxSection .button-label-asia,
.ButtonLabel-3869432492,
.ButtonLabel-3296148077,
.ButtonLabel-1636778223 {
display: block;
color: #fff;
font-size: 17px;
line-height: 20px;
font-family: -apple-system, BlinkMacSystemFont, “Segoe UI”, Roboto,
Oxygen-Sans, Ubuntu, Cantarell, “Helvetica Neue”, sans-serif;
text-align: center;
text-decoration: none;
cursor: pointer;
}
.Button-505204839,
.Button-1078489254,
.Button-2737859108 {
background: #069;
border: 1px solid #069;
border-radius: 0;
cursor: pointer;
display: block;
padding: 9px;
outline: 0;
text-align: center;
text-decoration: none;
min-width: 80px;
max-width: 320px;
margin-top: 20px;
}
.Button-505204839 .btn-secondary-label,
.Button-1078489254 .btn-secondary-label,
.Button-2737859108 .btn-secondary-label {
color: #069;
}
.uList-2102244549 {
list-style: none;
padding: 0;
margin: 0;
}
.article-buy-button {
font-family: -apple-system, BlinkMacSystemFont, “Segoe UI”, Roboto, Oxygen-Sans, Ubuntu, Cantarell, “Helvetica Neue”, sans-serif;
color: #069;
}
/* style specs end */

Fig. 1: Workflow of DISoptimizer and the multi-objective optimization function J(d).
Fig. 2: Predictive performance of machine learning models.
Fig. 3: Model interpretation.
Fig. 4: Optimization behaviour of DISoptimizer.
Fig. 5: DISoptimizer performance.

Similar content being viewed by others

High-molecular-weight by-products of chlorine disinfection

Insights to estimate exposure to regulated and non-regulated disinfection by-products in drinking water

Assessment of water quality and microbial contamination in institutional water resources: a necessity to understand health risks

Data availability

The dataset used for model development in this study is available via GitHub at https://github.com/pinfish94/DISoptimizer.git. Source data supporting the findings of this study are provided with this paper.

Code availability

All code created in this work is available via GitHub at https://github.com/pinfish94/DISoptimizer.git.

References

  1. United Nations General Assembly. Transforming Our World: The 2030 Agenda for Sustainable Development Resolution A/RES/70/1 (United Nations, 2015).

  2. McGuire, M. J. Eight revolutions in the history of US drinking water disinfection. J. AWWA 98, 123–149 (2006).

    Article 
    CAS 

    Google Scholar 

  3. Plewa, M. J. & Richardson, S. D. Disinfection by-products in drinking water, recycled water and wastewater: formation, detection, toxicity and health effects: preface. J. Environ. Sci. 58, 1 (2017).

    Article 

    Google Scholar 

  4. Rosario-Ortiz, F., Rose, J., Speight, V., Gunten, U. von. & Schnoor, J. How do you like your tap water? Science 351, 912–914 (2016).

    Article 
    CAS 
    PubMed 

    Google Scholar 

  5. Khor, Y., Aziz, A. R. A. & Chong, S. S. Recent developments and sustainability in monitoring chlorine residuals for water quality control: a critical review. RSC Sustain. 2, 2468–2485 (2024).

    Article 
    CAS 

    Google Scholar 

  6. Sikder, M., Daraz, U., Lantagne, D. & Saltori, R. Effectiveness of multilevel risk management emergency response activities to ensure free chlorine residual in household drinking water in southern Syria. Environ. Sci. Technol. 52, 14402–14410 (2018).

    Article 
    CAS 
    PubMed 

    Google Scholar 

  7. Wagner, E. D. & Plewa, M. J. CHO cell cytotoxicity and genotoxicity analyses of disinfection by-products: an updated review. J. Environ. Sci. 58, 64–76 (2017).

    Article 
    CAS 

    Google Scholar 

  8. Evlampidou, I. et al. Trihalomethanes in drinking water and bladder cancer burden in the European Union. Environ. Health Perspect. 128, 17001 (2020).

    Article 
    PubMed 
    PubMed Central 

    Google Scholar 

  9. Liu, H., Wang, R., Hu, C., Plewa, M. J. & Liu, C. Unveiling the mammalian cell cytotoxicity of tyre-impacted water in disinfection. Nat. Water 3, 902–912 (2025).

    Article 
    CAS 

    Google Scholar 

  10. Liu, M. et al. Spatial assessment of tap-water safety in China. Nat. Sustain. 5, 689–698 (2022).

    Article 

    Google Scholar 

  11. Guidelines for Drinking-Water Quality 4th edn incorporating the first and second addendum (WHO, 2017).

  12. Stage 2 Disinfectants and Disinfection Byproducts Rules (US EPA, 2008).

  13. Stage 1 Disinfectant and Disinfection Byproduct Rule (US EPA, 1998).

  14. Li, X. F. & Mitch, W. A. Drinking water disinfection byproducts (DBPs) and human health effects: multidisciplinary challenges and opportunities. Environ. Sci. Technol. 52, 1681–1689 (2018).

    Article 
    CAS 
    PubMed 

    Google Scholar 

  15. Han, J., Li, W. & Zhang, X. An effective and rapidly degradable disinfectant from disinfection byproducts. Nat. Commun. 15, 4888 (2024).

    Article 
    CAS 
    PubMed 
    PubMed Central 

    Google Scholar 

  16. Cooper, W. J., Mitch, W. A., Richardson, S. D. & Gonsior, M. Mitigating disinfection byproducts: chloramination or granular activated carbon treatment? Environ. Sci. Technol. 59, 17387–17389 (2025).

    Article 
    CAS 
    PubMed 

    Google Scholar 

  17. Chen, Y., Li, S. & Hu, J. Photoelectrocatalytic degradation of organics and formation of disinfection byproducts in reverse osmosis concentrate. Water Res. 168, 115105 (2020).

    Article 
    CAS 
    PubMed 

    Google Scholar 

  18. Wang, P. et al. Removal of disinfection by-product precursors by Al-based coagulants: a comparative study on coagulation performance. J. Hazard. Mater. 420, 126558 (2021).

    Article 
    CAS 
    PubMed 

    Google Scholar 

  19. Yang, M. & Zhang, X. Current trends in the analysis and identification of emerging disinfection byproducts. Trends Environ. Anal. Chem. 10, 24–34 (2016).

    Article 
    CAS 

    Google Scholar 

  20. Hua, L. C., Tsia, S. R., Wang, G.-S., Dong, C. D. & Huang, C. Increasing bromine in intracellular organic matter of freshwater algae growing in bromide-elevated environments and its impacts on characteristics of DBP precursors. Environ. Sci. Technol. Lett. 8, 307–312 (2021).

    Article 
    CAS 

    Google Scholar 

  21. Chowdhury, S. Effects of seawater intrusion on the formation of disinfection byproducts in drinking water. Sci. Total Environ. 827, 154398 (2022).

    Article 
    CAS 
    PubMed 

    Google Scholar 

  22. Du, Z. et al. Does snowfall introduce disinfection by-product precursors to surface water? Environ. Sci. Technol. 56, 14487–14497 (2022).

    Article 
    CAS 
    PubMed 

    Google Scholar 

  23. Du, Z. et al. Disinfection by-product precursors introduced by sandstorm events: composition, formation characteristics and potential risks. Water Res. 244, 120429 (2023).

    Article 
    CAS 
    PubMed 

    Google Scholar 

  24. Zhao, C. et al. Extreme precipitation amplified the cumulative effects of DOM availability on organic-sourced DIC in the Yangtze River. Water Res. 287, 124312 (2025).

    Article 
    CAS 
    PubMed 

    Google Scholar 

  25. Chen, X. et al. Novel insights into impacts of the “7.20” extreme rainstorm event on water supply security of Henan Province, China: levels and health risks of tap water disinfection by-products. J. Hazard. Mater. 452, 131323 (2023).

    Article 
    CAS 
    PubMed 

    Google Scholar 

  26. He, J. et al. Removal of CX3R-type disinfection by-product precursors from rainwater with conventional drinking water treatment processes. Water Res. 185, 116099 (2020).

    Article 
    CAS 
    PubMed 

    Google Scholar 

  27. Zhang, W., Wang, L., Yang, Y., Gaskin, P. & Teng, K. S. Recent advances on electrochemical sensors for the detection of organic disinfection byproducts in water. ACS Sens. 4, 1138–1150 (2019).

    Article 
    CAS 
    PubMed 

    Google Scholar 

  28. Wujcik, E. K., Duirk, S. E., Chase, G. G. & Monty, C. N. A visible colorimetric sensor based on nanoporous polypropylene fiber membranes for the determination of trihalomethanes in treated drinking water. Sens. Actuators B. 223, 1–8 (2016).

    Article 
    CAS 

    Google Scholar 

  29. Liang, Y. et al. Machine learning-guided prediction of chlorinated/chloraminated disinfection by-product formation in drinking water treatment. Water Res. 283, 123849 (2025).

    Article 
    CAS 
    PubMed 

    Google Scholar 

  30. Hossain, M. M., Sikder, R., Hua, G. & Ye, T. From model development to mitigation: machine learning for predicting and minimizing iodinated trihalomethanes in water treatment. Environ. Sci. Technol. 59, 11638–11652 (2025).

    Article 
    CAS 
    PubMed 

    Google Scholar 

  31. Sikder, R., Zhang, T. & Ye, T. Predicting THM formation and revealing its contributors in drinking water treatment using machine learning. ACS ES and T Water 4, 899–912 (2024).

    Article 
    CAS 

    Google Scholar 

  32. Maliwan, T., Do, Q. T. T., Nguyen, C. M., Teo, W. K. & Hu, J. Exploring the co-occurrence of microplastics, DOM and DBPs inside PVC pipes undergoing chlorination by correlation analysis and unsupervised learning. Chemosphere 373, 144171 (2025).

    Article 
    CAS 
    PubMed 

    Google Scholar 

  33. Zhang, C. & Lu, J. Optimizing disinfectant residual dosage in engineered water systems to minimize the overall health risks of opportunistic pathogens and disinfection by-products. Sci. Total Environ. 770, 145356 (2021).

    Article 
    CAS 
    PubMed 
    PubMed Central 

    Google Scholar 

  34. Hu, G. et al. Appraisal of machine learning techniques for predicting emerging disinfection byproducts in small water distribution networks. J. Hazard. Mater. 446, 130633 (2023).

    Article 
    CAS 
    PubMed 

    Google Scholar 

  35. Zhang, J. et al. The combination of multiple linear regression and adaptive neuro-fuzzy inference system can accurately predict trihalomethane levels in tap water with fewer water quality parameters. Sci. Total Environ. 896, 165269 (2023).

    Article 
    CAS 
    PubMed 

    Google Scholar 

  36. Wang, H., Liang, Q., Hancock, J. T. & Khoshgoftaar, T. M. Feature selection strategies: a comparative analysis of SHAP-value and importance-based methods. J. Big Data 11, 44 (2024).

    Article 

    Google Scholar 

  37. Marcilio, W. E. & Eler, D. M. From explanations to feature selection: assessing SHAP values as feature selection mechanism. In Proc. 2020 33rd SIBGRAPI Conference on Graphics, Patterns and Images 340–347 (IEEE, 2020).

  38. Wang, P. et al. Enhanced coagulation for mitigation of disinfection by-product precursors: a review. Adv. Colloid Interface Sci. 296, 102518 (2021).

    Article 
    CAS 
    PubMed 

    Google Scholar 

  39. Krasner, S. W. et al. Occurrence of a new generation of disinfection byproducts. Environ. Sci. Technol. 40, 7175–7185 (2006).

    Article 
    CAS 
    PubMed 

    Google Scholar 

  40. Hua, G., Reckhow, D. A. & Abusallout, I. Correlation between SUVA and DBP formation during chlorination and chloramination of NOM fractions from different sources. Chemosphere 130, 82–89 (2015).

    Article 
    CAS 
    PubMed 

    Google Scholar 

  41. Hua, G. & Reckhow, D. A. Evaluation of bromine substitution factors of DBPs during chlorination and chloramination. Water Res. 46, 4208–4216 (2012).

    Article 
    CAS 
    PubMed 

    Google Scholar 

  42. Deborde, M. & von Gunten, U. Reactions of chlorine with inorganic and organic compounds during water treatment-Kinetics and mechanisms: a critical review. Water Res. 42, 13–51 (2008).

    Article 
    CAS 
    PubMed 

    Google Scholar 

  43. Criquet, J. et al. Reaction of bromine and chlorine with phenolic compounds and natural organic matter extracts – electrophilic aromatic substitution and oxidation. Water Res. 85, 476–486 (2015).

    Article 
    CAS 
    PubMed 

    Google Scholar 

  44. Heeb, M. B., Criquet, J., Zimmermann-Steffens, S. G. & Von Gunten, U. Oxidative treatment of bromide-containing waters: formation of bromine and its reactions with inorganic and organic compounds – a critical review. Water Res. 48, 15–42 (2014).

    Article 
    CAS 
    PubMed 

    Google Scholar 

  45. Dickenson, E. R. V., Summers, R. S., Croué, J. P. & Gallard, H. Haloacetic acid and trihalomethane formation from the chlorination and bromination of aliphatic β-Dicarbonyl acid model compounds. Environ. Sci. Technol. 42, 3226–3233 (2008).

    Article 
    CAS 
    PubMed 

    Google Scholar 

  46. Langsa, M., Heitz, A., Joll, C. A., Von Gunten, U. & Allard, S. Mechanistic aspects of the formation of adsorbable organic bromine during chlorination of bromide-containing synthetic waters. Environ. Sci. Technol. 51, 5146–5155 (2017).

    Article 
    CAS 
    PubMed 

    Google Scholar 

  47. Forster, A. L. B., Wiskur, S. L. & Richardson, S. D. Formation of eight classes of DBPs from chlorine, chloramine, and ozone: mechanisms and formation pathways. Environ. Sci. Technol. 59, 15594–15611 (2025).

    Article 
    CAS 
    PubMed 

    Google Scholar 

  48. Gallard, H. & Von, U. Chlorination of natural organic matter: kinetics of chlorination and of THM formation. Water Res. 36, 65–74 (2002).

    Article 
    CAS 
    PubMed 

    Google Scholar 

  49. Helbling, D. E. & VanBriesen, J. M. Free chlorine demand and cell survival of microbial suspensions. Water Res. 41, 4424–4434 (2007).

    Article 
    CAS 
    PubMed 

    Google Scholar 

  50. Chen, L., Shi, H., Medema, G., van der Meer, W. & Liu, G. Long-term impacts of free chlorine and monochloramine on the development of drinking water biofilm. Water Res. 281, 123566 (2025).

    Article 
    CAS 
    PubMed 

    Google Scholar 

  51. Australian Drinking Water Guidelines Paper 6 National Water Quality Management Strategy (National Health and Medical Research Council, 2011).

  52. Standards for Drinking Water Quality (Ministry of Public Health of China, 2022).

  53. Initial Distribution System Evaluation Guidance Manual for the Final Stage 2 Disinfectants and Disinfection Byproducts Rule (US EPA, 2006).

  54. Chen, W., Westerhoff, P., Leenheer, J. A. & Booksh, K. Fluorescence excitation-emission matrix regional integration to quantify spectra for dissolved organic matter. Environ. Sci. Technol. 37, 5701–5710 (2003).

    Article 
    CAS 
    PubMed 

    Google Scholar 

Download references

Acknowledgements

We thank C. Ye, Z. Du, R. Zhang, R. Xiao, Y. Kong, F. Ao and Y. Li for assistance with water sample collection.

Funding

This work was supported by the National Natural Science Foundation of China (52325001 to W.C. and 52500010 to P.W.) and the Fundamental and Interdisciplinary Disciplines Breakthrough Plan of the Ministry of Education of China (JYB2025XDXM908 to W.C.). Jing-Jin-Ji Regional Integrated Environmental Improvement-National Science and Technology Major Project (2026ZD1211901 to W.C.).

Author information

Authors and Affiliations

Authors

Contributions

W.C. and P.W. conceived the work. P.W., Y.Y. and Z.W. conducted the experiments. P.W. and Y.Y. developed and tested the machine learning framework. C.W. and P.W. analysed the results. P.W. wrote and revised the paper. W.C. and S.D. provided constructive advice on result interpretation and paper preparation. All authors discussed and reviewed the final paper.

Corresponding author

Correspondence to
Wenhai Chu.

Ethics declarations

Competing interests

The authors declare no competing interests.

Peer review

Peer review information

Nature Water thanks Xiaoliu Huangfu and the other, anonymous, reviewers for their contribution to the peer review of this work.

Additional information

Publisher’s note Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Supplementary information

Supplementary Information (download PDF )

Supplementary Figs. 1–13, Text 1 and 2 and Tables 1–7.

Supplementary Data 1 (download XLSX )

Source data for Supplementary figures containing 11 sheets.

Source data

Source Data Fig. 2 (download XLSX )

This file contains the statistical source data underlying Fig. 2.

Source Data Fig. 3 (download XLSX )

This file contains the statistical source data underlying Fig. 3.

Source Data Fig. 4 (download XLSX )

This file contains the statistical source data underlying Fig. 4.

Source Data Fig. 5 (download XLSX )

This file contains the statistical source data underlying Fig. 5.

Rights and permissions

Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.

Reprints and permissions

About this article

Cite this article

Wang, P., Wu, Z., Yang, Y. et al. A machine learning-driven framework for optimizing disinfection in drinking water treatment.
Nat Water (2026). https://doi.org/10.1038/s44221-026-00702-0

Download citation

  • Received:

  • Accepted:

  • Published:

  • Version of record:

  • DOI: https://doi.org/10.1038/s44221-026-00702-0


Source: Resources - nature.com

Carbon emissions exacerbating the Western US water crisis

Hierarchical net-zero-liquid discharge approach for sustainable salt recovery and carbon footprint reduction in steel wastewater treatment