mlsanityData to 7 Oct 2026

Data-Driven Open-Loop Simulation for Digital-Twin Operator Decision Support in Wastewater Treatment

arXiv 2604.20935 ↗First posted 22 Apr 2026cs.LG cs.AIcited by 2

Not on the map. Why?

Abstract (arXiv)

Wastewater treatment plants (WWTPs) need digital-twin-style decision support tools that can simulate plant response under prescribed control plans, tolerate irregular and missing sensing, and remain informative over 12-36 h planning horizons. Meeting these requirements with full-scale plant data remains an open engineering-AI challenge. We present CCSS-RS, a controlled continuous-time state-space model that separates historical state inference from future control and exogenous rollout. The model combines typed context encoding, gain-weighted forcing of prescribed and forecast drivers, semigroup-consistent rollouts, and Student-t plus hurdle outputs for heavy-tailed and zero-inflated WWTP sensor data. On the public Avedore full-scale benchmark, with 906,815 timesteps, 43% missingness, and 1-20 min irregular sampling, CCSS-RS achieves RMSE 0.696 and CRPS 0.349 at H=1000 across 10,000 test windows. This reduces RMSE by 40-46% relative to Neural CDE baselines and by 31-35% relative to simplified internal variants. Four case studies using a frozen checkpoint on test data demonstrate operational value: oxygen-setpoint perturbations shift predicted ammonium by -2.3 to +1.4 over horizons 300-1000; a smoothed setpoint plan ranks first in multi-criterion screening; context-only sensor outages raise monitored-variable RMSE by at most 10%; and ammonium, nitrate, and oxygen remain more accurate than persistence throughout the rollout. These results establish CCSS-RS as a practical learned simulator for offline scenario screening in industrial wastewater treatment, complementary to mechanistic models.

Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 25 of its 46 references that are in the map. First posted on arXiv 22 Apr 2026.

Builds on

25 of 46 references.

References · 25

  1. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
  2. 2021Efficiently Modeling Long Sequences with Structured State Spacescited by 2,424
  3. 2020Informer: Beyond Efficient Transformer for Long Sequence Time-Series Forecastingcited by 2,141
  4. 2022A Time Series is Worth 64 Words: Long-term Forecasting with Transformerscited by 1,981
  5. 2021Autoformer: Decomposition Transformers with Auto-Correlation for Long-Term Series Forecastingcited by 1,654
  6. 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,598
  7. 2018An Empirical Evaluation of Generic Convolutional and Recurrent Networks for Sequence Modelingcited by 1,447
  8. 2018World Modelscited by 1,444
  9. 2018Learning Latent Dynamics for Planning from Pixelscited by 1,221
  10. 2023Mastering Diverse Domains through World Modelscited by 1,193
  11. 2017DeepAR: Probabilistic Forecasting with Autoregressive Recurrent Networkscited by 1,004
  12. 2020Mastering Atari with Discrete World Modelscited by 990
13 more
  1. 2022Simplified State Space Layers for Sequence Modelingcited by 776
  2. 2019Temporal Fusion Transformers for Interpretable Multi-horizon Time Series Forecastingcited by 748
  3. 2021CSDI: Conditional Score-based Diffusion Models for Probabilistic Time Series Imputationcited by 577
  4. 2016Recurrent Neural Networks for Multivariate Time Series with Missing Valuescited by 548
  5. 2020Neural Controlled Differential Equations for Irregular Time Seriescited by 501
  6. 2021Autoregressive Denoising Diffusion Models for Multivariate Probabilistic Time Series Forecastingcited by 332
  7. 2018BRITS: Bidirectional Recurrent Imputation for Time Seriescited by 288
  8. 2024TimeXer: Empowering Transformers for Time Series Forecasting with Exogenous Variablescited by 241
  9. 2019GRU-ODE-Bayes: Continuous modeling of sporadically-observed time seriescited by 230
  10. 2022SAITS: Self-Attention-based Imputation for Time Seriescited by 167
  11. 2024Log Neural Controlled Differential Equations: The Lie Brackets Make a Differencecited by 34
  12. 2023Deep-OSG: Deep Learning of Operators in Semigroupcited by 9
  13. 2024Time Series Dataset for Modeling and Forecasting of N2O in Wastewater Treatmentcited by 3

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 2.

2 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026HydroSphere: A Framework for Governed, Self-Healing Wastewater Infrastructure
  2. 2026WaterAdmin: Orchestrating Community Water Distribution Optimization via AI Agents
  3. 2025Leveraging LLM Agents and Digital Twins for Fault Handling in Process Plants
  4. 2026Mimir: Physics-Grounded LLM Agents for Long-Horizon Irrigation Control
  5. 2022The Pump Scheduling Problem: A Real-World Scenario for Reinforcement Learning
  6. 2025Large Language Models for Water Distribution Systems Modeling and Decision-Making
  7. 2025A Review of Generative AI in Aquaculture: Foundations, Applications, and Future Directions for Smart and Sustainable Farming
  8. 2026AI-Driven Framework for Adaptive Water Network Management with Proof-of-Concept Implementation: Addressing Non-Revenue Water in Jordan