A hybrid inference system for improved curvature estimation in the level-set method using machine learning

Abstract

We present a novel hybrid strategy based on machine learning to improve curvature estimation in the level-set method. The proposed inference system couples enhanced neural networks with standard numerical schemes to compute curvature more accurately. The core of our hybrid framework is a switching mechanism that relies on well established numerical techniques to gauge curvature. If the curvature magnitude is larger than a resolution-dependent threshold, it uses a neural network to yield a better approximation. Our networks are multilayer perceptrons fitted to synthetic data sets composed of sinusoidal- and circular-interface samples at various configurations. To reduce data set size and training complexity, we leverage the problem's characteristic symmetry and build our models on just half of the curvature spectrum. These savings lead to a powerful inference system able to outperform any of its numerical or neural component alone. Experiments with stationary, smooth interfaces show that our hybrid solver is notably superior to conventional numerical methods in coarse grids and along steep interface regions. Compared to prior research, we have observed outstanding gains in precision after training the regression model with data pairs from more than a single interface type and transforming data with specialized input preprocessing. In particular, our findings confirm that machine learning is a promising venue for reducing or removing mass loss in the level-set method.

arXiv commentSubmitted

References in corpus (13)

• Adam: A Method for Stochastic Optimization
• Scikit-learn: Machine Learning in Python
• Distributed Representations of Words and Phrases and their Compositionality
• Artificial Neural Networks for Solving Ordinary and Partial Differential Equations
• The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networks
• Physics Informed Deep Learning (Part I): Data-driven Solutions of Nonlinear Partial Differential Equations
• A high-bias, low-variance introduction to Machine Learning for physicists
• To prune, or not to prune: exploring the efficacy of pruning for model compression
• Deep Hidden Physics Models: Deep Learning of Nonlinear Partial Differential Equations
• Computing Curvature for Volume of Fluid Methods using Machine Learning
• A robust method for calculating interface curvature and normal vectors using an extracted local level set
• A hybrid particle volume-of-fluid method for curvature estimation in multiphase flows
• A deep learning approach for the computation of curvature in the level-set method

Cited by in corpus (3)

• Deep learning of interfacial curvature: a symmetry-preserving approach for the volume of fluid method
• Error-correcting neural networks for semi-Lagrangian advection in the level-set method
• Error-Correcting Neural Networks for Two-Dimensional Curvature Computation in the Level-Set Method