Kolmogorov-Arnold PointNet: Deep learning for prediction of fluid fields on irregular geometries

📅 2024-08-06
🏛️ arXiv.org
📈 Citations: 3
Influential: 0
📄 PDF

career value

208K/year
🤖 AI Summary
To address the challenge of predicting steady incompressible flow fields over irregular geometric domains, this paper proposes KA-PointNet: a novel framework integrating a learnable Kolmogorov–Arnold Network (KAN) into the segmentation branch of PointNet, enabling direct regression of flow variables from raw point clouds. Innovatively, a shared KAN is constructed using Jacobi polynomials; systematic ablation studies compare orthogonal bases—including Legendre, Chebyshev, and Gegenbauer—demonstrating superior expressivity and generalization. To the best of our knowledge, this is the first work to synergize KAN with PointNet for computational fluid dynamics modeling. Evaluated on the canonical cylinder flow benchmark, KA-PointNet achieves significantly higher accuracy than its MLP-based counterpart under comparable parameter counts: surface pressure and velocity predictions improve markedly, while lift and drag integral errors decrease by 37%.

Technology Category

Application Category

📝 Abstract
Kolmogorov-Arnold Networks (KANs) have emerged as a promising alternative to traditional Multilayer Perceptrons (MLPs) in deep learning. KANs have already been integrated into various architectures, such as convolutional neural networks, graph neural networks, and transformers, and their potential has been assessed for predicting physical quantities. However, the combination of KANs with point-cloud-based neural networks (e.g., PointNet) for computational physics has not yet been explored. To address this, we present Kolmogorov-Arnold PointNet (KA-PointNet) as a novel supervised deep learning framework for the prediction of incompressible steady-state fluid flow fields in irregular domains, where the predicted fields are a function of the geometry of the domains. In KA-PointNet, we implement shared KANs in the segmentation branch of the PointNet architecture. We utilize Jacobi polynomials to construct shared KANs. As a benchmark test case, we consider incompressible laminar steady-state flow over a cylinder, where the geometry of its cross-section varies over the data set. We investigate the performance of Jacobi polynomials with different degrees as well as special cases of Jacobi polynomials such as Legendre polynomials, Chebyshev polynomials of the first and second kinds, and Gegenbauer polynomials, in terms of the computational cost of training and accuracy of prediction of the test set. Additionally, we compare the performance of PointNet with shared KANs (i.e., KA-PointNet) and PointNet with shared MLPs. It is observed that when the number of trainable parameters is approximately equal, PointNet with shared KANs (i.e., KA-PointNet) outperforms PointNet with shared MLPs. Moreover, KA-PointNet predicts the pressure and velocity distributions along the surface of cylinders more accurately, resulting in more precise computations of lift and drag.
Problem

Research questions and friction points this paper is trying to address.

Predict fluid flow fields on irregular geometries using deep learning.
Combine Kolmogorov-Arnold Networks with PointNet for computational physics.
Evaluate performance of KA-PointNet versus traditional MLPs in fluid dynamics.
Innovation

Methods, ideas, or system contributions that make the work stand out.

Combines KANs with PointNet for fluid prediction
Uses Jacobi polynomials in shared KANs
Outperforms MLPs in accuracy and computational efficiency