AttNS: Attention-Inspired Numerical Solving For Limited Data Scenarios

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Huang, Zhongzhan, Liang, Mingfu, Zhong, Shanshan, Lin, Liang
Formato: Preprint
Publicado: 2023
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866912221067476992
author Huang, Zhongzhan
Liang, Mingfu
Zhong, Shanshan
Lin, Liang
author_facet Huang, Zhongzhan
Liang, Mingfu
Zhong, Shanshan
Lin, Liang
contents We propose the attention-inspired numerical solver (AttNS), a concise method that helps the generalization and robustness issues faced by the AI-Hybrid numerical solver in solving differential equations due to limited data. AttNS is inspired by the effectiveness of attention modules in Residual Neural Networks (ResNet) in enhancing model generalization and robustness for conventional deep learning tasks. Drawing from the dynamical system perspective of ResNet, we seamlessly incorporate attention mechanisms into the design of numerical methods tailored for the characteristics of solving differential equations. Our results on benchmarks, ranging from high-dimensional problems to chaotic systems, showcases AttNS consistently enhancing various numerical solvers without any intricate model crafting. Finally, we analyze AttNS experimentally and theoretically, demonstrating its ability to achieve strong generalization and robustness while ensuring the convergence of the solver. This includes requiring less data compared to other advanced methods to achieve comparable generalization errors and better prevention of numerical explosion issues when solving differential equations.
format Preprint
id arxiv_https___arxiv_org_abs_2302_10184
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle AttNS: Attention-Inspired Numerical Solving For Limited Data Scenarios
Huang, Zhongzhan
Liang, Mingfu
Zhong, Shanshan
Lin, Liang
Machine Learning
Artificial Intelligence
Numerical Analysis
We propose the attention-inspired numerical solver (AttNS), a concise method that helps the generalization and robustness issues faced by the AI-Hybrid numerical solver in solving differential equations due to limited data. AttNS is inspired by the effectiveness of attention modules in Residual Neural Networks (ResNet) in enhancing model generalization and robustness for conventional deep learning tasks. Drawing from the dynamical system perspective of ResNet, we seamlessly incorporate attention mechanisms into the design of numerical methods tailored for the characteristics of solving differential equations. Our results on benchmarks, ranging from high-dimensional problems to chaotic systems, showcases AttNS consistently enhancing various numerical solvers without any intricate model crafting. Finally, we analyze AttNS experimentally and theoretically, demonstrating its ability to achieve strong generalization and robustness while ensuring the convergence of the solver. This includes requiring less data compared to other advanced methods to achieve comparable generalization errors and better prevention of numerical explosion issues when solving differential equations.
title AttNS: Attention-Inspired Numerical Solving For Limited Data Scenarios
topic Machine Learning
Artificial Intelligence
Numerical Analysis
url https://arxiv.org/abs/2302.10184