LE-PDE++: Mamba for accelerating PDEs Simulations

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Liang, Aoming, Mu, Zhaoyang, liu, Qi, Li, Ruipeng, Ge, Mingming, Fan, Dixia
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917834368483328
author Liang, Aoming
Mu, Zhaoyang
liu, Qi
Li, Ruipeng
Ge, Mingming
Fan, Dixia
author_facet Liang, Aoming
Mu, Zhaoyang
liu, Qi
Li, Ruipeng
Ge, Mingming
Fan, Dixia
contents Partial Differential Equations are foundational in modeling science and natural systems such as fluid dynamics and weather forecasting. The Latent Evolution of PDEs method is designed to address the computational intensity of classical and deep learning-based PDE solvers by proposing a scalable and efficient alternative. To enhance the efficiency and accuracy of LE-PDE, we incorporate the Mamba model, an advanced machine learning model known for its predictive efficiency and robustness in handling complex dynamic systems with a progressive learning strategy. The LE-PDE was tested on several benchmark problems. The method demonstrated a marked reduction in computational time compared to traditional solvers and standalone deep learning models while maintaining high accuracy in predicting system behavior over time. Our method doubles the inference speed compared to the LE-PDE while retaining the same level of parameter efficiency, making it well-suited for scenarios requiring long-term predictions.
format Preprint
id arxiv_https___arxiv_org_abs_2411_01897
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle LE-PDE++: Mamba for accelerating PDEs Simulations
Liang, Aoming
Mu, Zhaoyang
liu, Qi
Li, Ruipeng
Ge, Mingming
Fan, Dixia
Machine Learning
Artificial Intelligence
Partial Differential Equations are foundational in modeling science and natural systems such as fluid dynamics and weather forecasting. The Latent Evolution of PDEs method is designed to address the computational intensity of classical and deep learning-based PDE solvers by proposing a scalable and efficient alternative. To enhance the efficiency and accuracy of LE-PDE, we incorporate the Mamba model, an advanced machine learning model known for its predictive efficiency and robustness in handling complex dynamic systems with a progressive learning strategy. The LE-PDE was tested on several benchmark problems. The method demonstrated a marked reduction in computational time compared to traditional solvers and standalone deep learning models while maintaining high accuracy in predicting system behavior over time. Our method doubles the inference speed compared to the LE-PDE while retaining the same level of parameter efficiency, making it well-suited for scenarios requiring long-term predictions.
title LE-PDE++: Mamba for accelerating PDEs Simulations
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2411.01897