Memory-efficient neural network for non-linear ultrasound computed tomography reconstruction

Abstract

Deep neural networks have proven to excel classical medical image reconstruction techniques. Some networks are based on fully connected (FC) layers to achieve domain transformation such as from the data acquisition domain to the image domain. However, FC layers result in huge numbers of parameters which take a lot of GPU memory. Hence, they do not scale well, and the overall performance is limited. For ultrasound computed tomography (USCT) application, we propose a memory-efficient convolutional network that reconstructs images from the frequency domain to image domain with much less parameters compared with multilayer perceptron, by using data-driven learning. Extensive experiments demonstrate that our method achieves high reconstruction quality. It improves the structural similarity measure (SSIM) from 0.73 to 0.99 when compared with state-of-the-art reconstruction methods in this field while reduces 2/3 parameters when compared with deep neural network with FC layers to reconstruct images from frequency domain to image domain.