Physics-informed neural networks for transcranial ultrasound wave propagation.

Journal: Ultrasonics
PMID:

Abstract

Transcranial ultrasound imaging has been playing an increasingly important role in the non-invasive treatment of brain disorders. However, the conventional mesh-based numerical wave solvers, which are an integral part of imaging algorithms, suffer from limitations such as high computational cost and discretization error in predicting the wavefield passing through the skull. In this paper, we explore the use of physics-informed neural networks (PINNs) for predicting the transcranial ultrasound wave propagation. The wave equation, two sets of time snapshots data and a boundary condition (BC) are embedded as physical constraints in the loss function during training. The proposed approach has been validated by solving the two-dimensional (2D) acoustic wave equation under three increasingly complex spatially varying velocity models. Our cases demonstrate that due to the meshless nature of PINNs, they can be flexibly applied to different wave equations and types of BCs. By adding physics constraints to the loss function, PINNs can predict wavefields far outside the training data, providing ideas for improving the generalization capability of existing deep learning methods. The proposed approach offers exciting perspectives because of the powerful framework and simple implementation. We conclude with a summary of the strengths, limitations and further research directions of this work.

Authors

  • Linfeng Wang
    State Key Laboratory of Precision Measuring Technology and Instruments, Tianjin University, Tianjin 300072, China.
  • Hao Wang
    Department of Cardiology, Second Medical Center, Chinese PLA General Hospital, Beijing, China.
  • Lin Liang
    Department of Orthopedics, Jinjiang Municipal Hospital/Clinical Research Center for Orthopaedic Trauma and Reconstruction of Fujian Province, Jinjiang, Quanzhou, Fujian, China.
  • Jian Li
    Fujian Key Laboratory of Traditional Chinese Veterinary Medicine and Animal Health, College of Animal Science, Fujian Agriculture and Forestry University, Fuzhou, China.
  • Zhoumo Zeng
    State Key Laboratory of Precision Measuring Technology and Instruments, Tianjin University, Tianjin 300072, China.
  • Yang Liu
    Department of Computer Science, Hong Kong Baptist University, Hong Kong, China.