Solving PDE related problems using deep-learning

31
Solving PDE related problems using deep-learning Adar Kahana Tel Aviv University Under the supervision of Eli Turkel (TAU) and Dan Givoli (Technion) Shai Dekel (TAU) Waves seminar, UC Merced February 4 th , 2021 1

Transcript of Solving PDE related problems using deep-learning

Page 1: Solving PDE related problems using deep-learning

Solving PDE related

problems using

deep-learning

Adar KahanaTel Aviv University

Under the supervision of

Eli Turkel (TAU) and

Dan Givoli (Technion)

Shai Dekel (TAU)

Waves seminar,

UC Merced

February 4th , 2021

1

Page 2: Solving PDE related problems using deep-learning

Agenda

Motivation

Data driven problems

Obstacle identification and deep-learning

Dealing with CFL instability using deep-learning

2

Page 3: Solving PDE related problems using deep-learning

Underwater acoustics

Sonar imaging

3

Page 4: Solving PDE related problems using deep-learning

The wave problem

4

ሷ𝑢 Ԧ𝑥, 𝑡 = 𝑑𝑖𝑣 𝑐 Ԧ𝑥 2𝛻𝑢 Ԧ𝑥, 𝑡 , Ԧ𝑥 ∈ Ω, t ∈ (0, 𝑇]

𝑢 Ԧ𝑥, 0 = 𝑢0 Ԧ𝑥 , Ԧ𝑥 ∈ Ω

ሶ𝑢 Ԧ𝑥, 0 = ሶ𝑢0 Ԧ𝑥 , Ԧ𝑥 ∈ Ω

𝑢 Ԧ𝑥, 𝑡 = 𝑓 Ԧ𝑥, 𝑡 , Ԧ𝑥 ∈ 𝜕Ω1, t ∈ 0, 𝑇

𝛻𝑢 Ԧ𝑥, 𝑡 = 𝑔 Ԧ𝑥, 𝑡 , Ԧ𝑥 ∈ 𝜕Ω2, t ∈ [0, 𝑇]

𝜕Ω1 ∪ 𝜕Ω2 = 𝜕Ω

where ሶ𝑢0 Ԧ𝑥 = 0 and 𝑓 Ԧ𝑥, 𝑡 = 𝑔 Ԧ𝑥, 𝑡 = 0

Page 5: Solving PDE related problems using deep-learning

Ill-posed problems

In an experiment, we store the pressure at a small

number of sensors for all time steps

We wish to find the properties of the source or

obstacle from the data stored at these sensors

where the number of sensors << mesh

This is an inverse problem which is highly ill-posed

Hence, one cannot usually reconstruct the initial

conditions perfectly

Can we solve these types of ill-posed problems

with learning?

5

Page 6: Solving PDE related problems using deep-learning

Partial information “Recording” the solution at a small set of

sensors placed in the domain Ԧ𝑥s𝑛 𝑛=1

𝐾∈ Ω

Data –

𝑢 Ԧ𝑥s1 , t

𝑢 Ԧ𝑥s2 , t

⋮𝑢 Ԧ𝑥sn , t

+ 𝒩 𝜇, 𝜎2

The ill-posedness raises sensitivity to noise at

the sensors

6

Page 7: Solving PDE related problems using deep-learning

Agenda

Motivation

Data driven problems

Obstacle identification and deep-learning

Dealing with CFL instability using deep-learning

7

Page 8: Solving PDE related problems using deep-learning

Data driven problems

Supervised learning

Input data

Output labels

Training

Prediction (testing)

Drawback - sensitivity

8

Page 9: Solving PDE related problems using deep-learning

Deep-learning

Training “weights” to learn connections in the data

Hidden multi-dimensional embeddings

Convolutions, Fully connected

“Deep” and non-linear

Loss

9

1 0 10 1 01 0 1

𝑤1 𝑤2 𝑤3

𝑤4 𝑤5 𝑤6

𝑤7 𝑤8 𝑤9

Page 10: Solving PDE related problems using deep-learning

Physically-informed NN

Input: set of points from the initial and boundary

conditions

Output: solution in the domain

Loss: the problem

10

M. Raissi, P. Perdikaris, G.E. Karniadakis, Journal of computational Physics, 2018

Page 11: Solving PDE related problems using deep-learning

Results

11

Page 12: Solving PDE related problems using deep-learning

Agenda

Motivation

Data driven problems

Obstacle identification and deep-learning

Dealing with CFL instability using deep-learning

12

Page 13: Solving PDE related problems using deep-learning

Problem definitionGiven the position of the source/s and data at a few

sensors but many time slices find the location, size and

shape of the unknown scatterers

Input: Sensors recordings (𝑁𝑠𝑎𝑚𝑝𝑙𝑒𝑠 × 𝑁𝑡𝑠𝑡𝑒𝑝𝑠 × 𝑁𝑠𝑒𝑛𝑠𝑜𝑟𝑠)

Output: Obstacle?

13

Page 14: Solving PDE related problems using deep-learning

Prior work

Location Ԧ𝑥

Shape and size –

Circles: Radius

Rectangles: Height and Width

Complex shapes: need to be parametrized

“Soft” obstacles –

Semi-penetrable

Multiphysics

14

Page 15: Solving PDE related problems using deep-learning

Labels solution - segments

15

Labels are 𝑚 × 𝑛binary matrices

Predictions will be

𝑚 × 𝑛 probability

matrices

Loss: NLL

Page 16: Solving PDE related problems using deep-learning

Spatio-temporal architecture

16

Page 17: Solving PDE related problems using deep-learning

Loss diagram

17

Probability

map

Page 18: Solving PDE related problems using deep-learning

Physically informed loss

Using the segmentation network and output ෨𝑂

Define a loss component based on:

Solve: 𝑢𝑡𝑡 = 1 − ෨𝑂 𝑥 𝑐2(𝑥) Δ𝑢

Get sensor data: 𝑢𝑘 𝑥𝑠𝑖 𝑖=1

#𝑠𝑒𝑛𝑠𝑜𝑟𝑠for each sample

Calculate MSE between ground truth

𝑢𝑘 𝑥𝑠𝑖 𝑖=1

#𝑠𝑒𝑛𝑠𝑜𝑟𝑠and the prediction as component 𝑙2

Define the loss function for our network as:

𝛼 ⋅ 𝑙1 + 1 − 𝛼 ⋅ 𝑙2such that 𝑙1 is the NLL loss described earlier

18

Page 19: Solving PDE related problems using deep-learning

Numerical experiments Dirichlet BC

Compact Gaussian initial condition

Arbitrary polygonal obstacles –

Generate number of edges

Generate edge length and angle

Generate location (𝑥0, 𝑦0, 𝑧0)

❖ Enormous samples space

Generated only 25,000 samples

19

Page 20: Solving PDE related problems using deep-learning

Probability images

20

Page 21: Solving PDE related problems using deep-learning

Neural network – results Intersection over union: 0 ≤ 𝐼𝑂𝑈 𝐴, 𝐵 =

𝐴∩𝐵

𝐴∪𝐵≤ 1

Up to 66% IOU score

21

A.K., E. Turkel, D. Givoli, S. Dekel, Journal of computational Physics, 2020

Page 22: Solving PDE related problems using deep-learning

Agenda

Motivation

Data driven problems

Obstacle identification and deep-learning

Dealing with CFL instability using deep-learning

22

Page 23: Solving PDE related problems using deep-learning

Explicit schemes and CFL One-dimensional wave equation

CFL condition for stability: 𝛼 =𝑐Δ𝑡

Δ𝑥≤ 1

FDCD: 𝑢𝑖𝑛+1 = 2𝑢𝑖

𝑛 − 𝑢𝑖𝑛−1 + 𝛼2 𝑢𝑖+1

𝑛 − 2𝑢𝑖𝑛 + 𝑢𝑖−1

𝑛

23

Page 24: Solving PDE related problems using deep-learning

Network architecture

Input: 𝑢 𝑛−1 𝑚, 𝑢𝑛𝑚

Output: 𝑢 𝑛+1 𝑚

Spatio-temporal architecture

Non-linear activation (PReLU)

Loss: MSE between 𝑢 𝑛+1 𝑚 and ො𝑢 𝑛+1 𝑚

24

Page 25: Solving PDE related problems using deep-learning

Network diagram

25

Page 26: Solving PDE related problems using deep-learning

Physics informed loss

Use 𝑢 𝑛−1 𝑚, 𝑢𝑛𝑚 to predict ො𝑢 𝑛+1 𝑚

Inside the loss:

Use 𝑢 𝑛+1 𝑚, 𝑢 𝑛+1 𝑚+1 to calculate 𝑢 𝑛+1 𝑚+𝑗

Use ො𝑢 𝑛+1 𝑚, 𝑢 𝑛+1 𝑚+1 to predict ො𝑢 𝑛+1 𝑚+𝑗

Calculate the MSE between 𝑢 𝑛+1 𝑚+𝑗 and ො𝑢 𝑛+1 𝑚+𝑗

Network loss is the linear combination of the two

MSE losses

26

Page 27: Solving PDE related problems using deep-learning

Numerical experiments Dirichlet BC

Data:

Linear combinations with random coefficients

created from the basis sin 𝜋𝑘𝑥 𝑘=120

1250 different initial condition and 397 time-steps for

each one, total of 496,250 samples

Samples created with CFL = 0.875 and only each

10th sample was taken to get CFL = 8.75

27

Page 28: Solving PDE related problems using deep-learning

Results

28

Page 29: Solving PDE related problems using deep-learning

Results29

O. Ovadia, A. K, E. Turkel, S. Dekel, Journal of computational physics, submitted

Page 30: Solving PDE related problems using deep-learning

Summary and future work

Obstacle location and identification

Investigating source location

High measurement noise

Stability

Extending to 2,3 dimensions

Dispersion relation problem – optimized kernels

Experimental data

30

Page 31: Solving PDE related problems using deep-learning

Thanks!

31