In this paper, we present a discontinuity and cusp capturing physics-informed neural network (PINN) to solve Stokes equations with a piecewise-constant viscosity and singular force along an interface. We first reformulate the governing equations in each fluid domain separately and replace the singular force effect with the traction balance equation between solutions in two sides along the interface. Since the pressure is discontinuous and the velocity has discontinuous derivatives across the interface, we hereby use a network consisting of two fully-connected sub-networks that approximate the pressure and velocity, respectively. The two sub-networks share the same primary coordinate input arguments but with different augmented feature inputs. These two augmented inputs provide the interface information, so we assume that a level set function is given and its zero level set indicates the position of the interface. The pressure sub-network uses an indicator function as an augmented input to capture the function discontinuity, while the velocity sub-network uses a cusp-enforced level set function to capture the derivative discontinuities via the traction balance equation. We perform a series of numerical experiments to solve two- and three-dimensional Stokes interface problems and perform an accuracy comparison with the augmented immersed interface methods in literature. Our results indicate that even a shallow network with a moderate number of neurons and sufficient training data points can achieve prediction accuracy comparable to that of immersed interface methods.

翻译：本文提出了一种间断与尖点捕捉物理信息神经网络（PINN），用于求解沿界面具有分段常数黏性与奇异力的Stokes方程。我们首先在各流体域内分别重新表述控制方程，并用界面两侧解之间的牵引平衡方程替代奇异力效应。由于压力在界面上不连续且速度具有不连续导数，因此我们采用由两个全连接子网络组成的网络结构，分别逼近压力与速度。这两个子网络共享相同的主坐标输入参数，但采用不同的增强特征输入。这些增强输入提供界面信息，因此我们假设给定一个水平集函数，其零水平集表示界面位置。压力子网络使用指示函数作为增强输入以捕捉函数间断性，而速度子网络则通过牵引平衡方程使用尖点增强水平集函数来捕捉导数间断性。我们进行了一系列数值实验求解二维与三维Stokes界面问题，并与文献中的增强浸入界面方法进行精度比较。结果表明，即使采用中等数量神经元和充足训练数据点的浅层网络，也能达到与浸入界面方法相当的预测精度。