New Framework Boosts Operator Learning for Large-Scale PDEs

Christian Munoz, Alexandre Tartakovsky· June 30, 2026 View original

Summary

A new scalable operator-learning framework, KL-DNN, combines Karhunen-Loeve expansions with Deep Neural Networks to efficiently model large-scale partial differential equations (PDEs). It achieves high accuracy and significant speedups for applications like geological carbon storage.

Training operator-learning models for complex, large-scale problems governed by partial differential equations (PDEs) presents significant challenges, including high dimensionality, memory constraints, and limited data. These issues are prevalent in critical scientific and engineering fields such as climate modeling and geological carbon storage. Researchers have introduced a scalable framework called Karhunen-Loeve Deep Neural Network (KL-DNN) to address these limitations. This method constructs latent spaces using low-rank singular value decomposition and a nested Karhunen-Loeve expansion, enabling full-resolution predictions without data subsampling. Applied to geological carbon storage, KL-DNN demonstrated superior performance, achieving lower errors for pressure and CO2 saturation while offering a two-order-of-magnitude speedup compared to DeepONet. Its rapid training and inference times make it a practical solution for real-time decision support and uncertainty quantification in large-scale PDE problems.

Why it matters

This framework offers a powerful solution for modeling complex physical systems, enabling faster and more accurate simulations critical for scientific research, engineering design, and environmental management. Professionals can achieve significant computational savings and improved predictive capabilities.

How to implement this in your domain

  1. 1Evaluate KL-DNN for existing large-scale PDE modeling tasks in your organization, especially those with memory or data constraints.
  2. 2Integrate this framework into simulation pipelines for applications like subsurface flow, climate modeling, or material science.
  3. 3Explore how the 'trainable-by-parts' approach can be adapted to other deep learning architectures for scalability.
  4. 4Develop real-time decision support systems leveraging the rapid inference capabilities of KL-DNN for complex engineering problems.

Who benefits

EnergyEnvironmental ScienceAerospaceManufacturing

Key takeaways

  • KL-DNN provides a scalable and accurate operator-learning framework for large-scale PDEs.
  • It significantly reduces training time and improves prediction accuracy compared to existing methods.
  • The framework enables full-resolution predictions without needing data subsampling.
  • It is highly effective for applications like geological carbon storage, offering real-time decision support.

Original post by Christian Munoz, Alexandre Tartakovsky

"arXiv:2606.28519v1 Announce Type: new Abstract: Training operator-learning models for large-scale problems governed by partial differential equations (PDEs) is challenging due to the curse of dimensionality, memory constraints, and limited training data. These challenges arise in…"

View on X

Originally posted by Christian Munoz, Alexandre Tartakovsky on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses