Neural Stream Functions

Abstract

We present a neural network approach to compute stream functions, which are scalar functions with gradients orthogonal to a given vector field. As a result, isosurfaces of the stream function extract stream surfaces, which can be visualized to analyze flow features. Our approach takes a vector field as input and trains an implicit neural representation to learn a stream function for that vector field. The network learns to map input coordinates to a stream function value by minimizing the inner product of the gradient of the neural network's output and the vector field. Since stream function solutions may not be unique, we give optional constraints for the network to learn particular stream functions of interest. Specifically, we introduce regularizing loss functions that can optionally be used to generate stream function solutions whose stream surfaces follow the flow field's curvature, or that can learn a stream function that includes a stream surface passing through a seeding rake. We also discuss considerations for properly visualizing the trained implicit network and extracting artifact-free surfaces. We compare our results with other implicit solutions and present qualitative and quantitative results for several synthetic and simulated vector fields.

arXiv commentAccepted and presented at PVIS2023 in Seoul, South Korea

References in corpus (9)

• Adam: A Method for Stochastic Optimization
• Instant Neural Graphics Primitives with a Multiresolution Hash Encoding
• DGM: A deep learning algorithm for solving partial differential equations
• Physics Informed Deep Learning (Part I): Data-driven Solutions of Nonlinear Partial Differential Equations
• A unified deep artificial neural network approach to partial differential equations in complex geometries
• A public turbulence database cluster and applications to study Lagrangian evolution of velocity increments in turbulence
• Physics Informed Deep Learning (Part II): Data-driven Discovery of Nonlinear Partial Differential Equations
• InSituNet: Deep Image Synthesis for Parameter Space Exploration of Ensemble Simulations
• NNVA: Neural Network Assisted Visual Analysis of Yeast Cell Polarization Simulation