In variational-based image denoising, the regularizer derived from mixture distributions plays a crucial role in preserving image details. However, this type of mixture distribution priors has not been incorporated into deep learning-based denoising methods. In this paper, we propose a method for learning regularizers based on a learnable Laplacian mixture distribution for image denoising. Our approach is motivated by the assumption that deep image features follow a latent distribution with a mixture model. To address this assumption, we propose a regularizer with learnable weights by considering the dual problem of maximum likelihood estimation for the deep features. The dual variable in this problem represents an attention weight, which can be learned using a numerical scheme with an unrolling technique. Notably, our method establishes a connection between the mixture distribution prior and the popular attention mechanism in deep learning. Additionally, to capture multi-scale features, we introduce a multigrid solver for the optimization involved in our method. This enables the formulation of an encoder-decoder architecture based on a learnable mixture distribution network (LMDNet) for image denoising. We demonstrate the effectiveness of our proposed method by comparing it with several popular denoising methods. The results demonstrate the superior performance of our approach for image denoising. The code is publicly available at https://github.com/ylyslzx/LMDNet.
扫码关注我们
求助内容:
应助结果提醒方式:
