A Lightweight Ultrasound Image Denoiser Using Parallel Attention Modules and Capsule Generative Adversarial Network

Abstract

The quality of ultrasound (US) imaging has been constrained by its limited contrast and resolution, inherent speckle noise, and the presence of other artifacts. Existing traditional and deep learning-based US denoising approaches have many limitations, such as reliance on manual parameter configurations, poor performance for unknown noise levels, the requirement for a large number of training data, and high computational expense. To address these challenges, we propose a novel Generative Adversarial Network (GAN) based denoiser. Capsule networks are utilized in both the generator and discriminator of the proposed GAN to capture intricate sparse features with less complexity. In addition, bias components are removed from all neurons of the generator to handle the unknown noise levels. A parallel attention module is also included in the proposed model to further enhance denoising performance. The proposed approach is trained in a semi-supervised manner and can thus be trained with fewer labeled images. Experimental evaluation on publicly available HC18 and BUSI datasets showed that the proposed approach achieved state-of-the-art denoising performance, with PSNR values of 33.86 and 34.16, and SSIM indices of 0.91 and 0.90, respectively. Moreover, experiments showed that the proposed approach is lightweight and more than twice as fast as similar denoisers.

Keywords

• Capsule Network
• Generative Adversarial Network
• Image denoising
• Semi-supervision
• Ultrasound image