Sustainable Sewage Treatment Prediction Using Integrated KAN-LSTM with Multi-Head Attention

Jiaming Zheng, Genki Suzuki and Hiroyuki Shioya ()
Additional contact information
Jiaming Zheng: Division of Information and Electronic Engineering, Muroran Institute of Technology, Muroran 050-8585, Japan
Genki Suzuki: Division of Information and Electronic Engineering, Muroran Institute of Technology, Muroran 050-8585, Japan
Hiroyuki Shioya: Division of Information and Electronic Engineering, Muroran Institute of Technology, Muroran 050-8585, Japan

Sustainability, 2025, vol. 17, issue 10, 1-17

Abstract: The accurate prediction of sewage treatment indicators is crucial for optimizing management and supporting sustainable water use. This study proposes the KAN-LSTM model, a hybrid deep learning model combining Long short-term memory (LSTM) networks, Kolmogorov-Arnold Network (KAN) layers, and multi-head attention. The model effectively captures complex temporal dynamics and nonlinear relationships in sewage data, outperforming conventional methods. We applied correlation analysis with time-lag consideration to select key indicators. The KAN-LSTM model then processes them through LSTM layers for sequential dependencies, KAN layers for enhanced nonlinear modeling via learnable B-spline transformations, and multi-head attention for dynamic weighting of temporal features. This combination handles short-term patterns and long-range dependencies effectively. Experiments showed the model’s superior performance, achieving 95.13% R-squared score for FOss (final sedimentation basin outflow suspended solid, one indicator of our research predictions)and significantly improving prediction accuracy. These advancements in intelligent sewage treatment prediction modeling not only enhance water sustainability but also demonstrate the transformative potential of hybrid deep learning approaches. This methodology could be extended to optimize predictive tasks in sustainable aquaponic systems and other smart aquaculture applications.

Keywords: sustainable sewage treatment; Kolmogorov-Arnold Network; long short-term memory; attention; aquaculture (search for similar items in EconPapers)
JEL-codes: O13 Q Q0 Q2 Q3 Q5 Q56 (search for similar items in EconPapers)
Date: 2025
References: Add references at CitEc
Citations:

Downloads: (external link)
https://www.mdpi.com/2071-1050/17/10/4417/pdf (application/pdf)
https://www.mdpi.com/2071-1050/17/10/4417/ (text/html)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:gam:jsusta:v:17:y:2025:i:10:p:4417-:d:1654476

Access Statistics for this article

Sustainability is currently edited by Ms. Alexandra Wu

More articles in Sustainability from MDPI
Bibliographic data for series maintained by MDPI Indexing Manager ().