Two-Stream Multiplicative Heavy-Tail Noise Despeckling Network With Truncation Loss

Abstract

In recent years, deep learning algorithms for speckle noise removal have attracted much attention. However, speckle noise is strongly heavy-tailed and signal dependent, which makes it difficult to remove. In this paper, we propose a two-stream convolutional neural network with hybrid truncation loss to eliminate multiplicative noise (HTNet). HTNet combines the major task of multiplicative noise removal and the auxiliary task of noise estimation to improve the despeckling effect while preserving texture details. The main branch of HTNet is composed of a feature extraction block and an improved U-Net that can extract multi-scale information, which is mainly used for speckle noise removal. The noise estimation auxiliary branch is designed to fit the speckle noise. A hybrid truncation loss function is to applied for robust estimation of heavy-tailed distribution instead of mean squared error. Extensive experimental results show that HTNet can effectively remove speckle noise and outperforms the state-of-the-art methods on both simulated and real SAR images. In addition, HTNet has advantages on textured images.