Full Waveform Inversion-Based Ultrasound Computed Tomography Acceleration Using 2D Convolutional Neural Networks

Abstract

Ultrasound computed tomography (USCT) shows great promise in nondestructive evaluation and medical imaging due to its ability to quickly scan and collect data from a region of interest. However, the processing of the collected data into a meaningful image requires both time and computational resources; existing approaches are a trade-off between the accuracy of the prediction and the speed at which the data can be analyzed. We propose to develop convolutional neural networks(CNNs) to accelerate and enhance the inversion results to reveal underlying structures or abnormalities that may be located within the region of interest. For training, the ultrasonic signals were first processed using the FWI technique for only a single iteration; the resulting image and the corresponding true model were used as the input and output, respectively. The proposed machine learning approach is based on implementing two-dimensional CNNs to find an approximate solution to the inverse problem of partial differential equation-based model reconstruction. To alleviate the time-consuming and computationally intensive data generation process, a high-performance computing (HPC)-based framework has been developed to generate the training data in parallel. At the inference stage, the acquired signals will be first processed by FWI for a single iteration; then the resulting image will be processed by a pre-trained CNN to instantaneously generate the final output image. The results showed that once trained, the CNN scan quickly generate the predicted wave speed distributions with significantly enhanced speed and accuracy.