Skip to content

Repository of "A Physics-Informed Neural Network Approach for Nearfield Acoustic Holography"

License

Notifications You must be signed in to change notification settings

polimi-ispl/nah-khcnn

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

17 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

A Physics-Informed Neural Network Approach for Nearfield Acoustic Holography

Repository of "A Physics-Informed Neural Network Approach for Nearfield Acoustic Holography"

M. Olivieri, M. Pezzoli, F. Antonacci, A. Sarti

MDPI, Sensors, 2021
Special Issue in "Audio Signal Processing for Sensing Technologies"

In this manuscript we describe a novel methodology for Nearfield Acoustic Holography (NAH). The proposed technique is based on Convolutional Neural Networks, with an autoencoder archictecure, to reconstruct the pressure and velocity fields on the surface of the vibrating structure using the sampled pressure soundfield on the holographic plane as input. The loss function used for training the network is based on a combination of two components. The first component is the error in the reconstructed velocity. The second component is the error between the sound pressure on the holographic plane and its estimate obtained from forward propagating the pressure and velocity fields on the structure through the Kirchhoff-Helmholtz integral, thus bringing some knowledge about the physics of the process under study into the estimation algorithm. Due to the explicit presence of the Kirchhoff-Helmholtz integral in the loss function, we name the proposed technique as Kirchhoff-Helmholtz-based Convolutional Neural Network, KHCNN. KHCNN has been tested on two large datasets of rectangular plates (dataset released) and violin shells. Results show that it attains a very good accuracy, with a gain in the NMSE of the estimated velocity field that can top 10 dB with respect to state-of-the-art techniques. The same trend is observed if the Normalized Cross Correlation is used as a metric.

Proposed architecture

alt text Overall scheme of the proposed KHCNN model. The CNN architecture (yellow block) predicts the real and imaginary parts of Ps and V (stacked in two channels) from the input Ph. The two outputs are then propagated with the KH model in order to obtain the estimate of Ph. A proper loss function is built on top of the velocity ground truth and the pressure at the hologram.

About the code

The repo code is structured in the following folders:

├── src                         # code
│   ├── network_module              
│   │   ├── cnn_models.py
│   │   ├── kh_discretisation_tf.py
│   │   └── khcnn_network.py
│   ├── test_module              
│   │   ├── plot_results.py
│   │   └── test_kh_rec.py      # main
│   └── utils
│       ├── data_processing.py
│       ├── metrics.py
│       └── params.py           # initial parameters
├── data
│   ├── rectangular_plate              
│   │   └── test_rec.mat        # examples of rectangular plate data
│   └── weights
│       ├── khcnn_rec_1024.h5   # trained weights for 1024 P_h input points
│       └── khcnn_rec_64.h5     # trained weights for 64 P_h input points
└── images
    └── KHCNN_architecture.png                     

About

Repository of "A Physics-Informed Neural Network Approach for Nearfield Acoustic Holography"

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages