Bayesian Enhancement Models for One-to-Many Mapping in Image Enhancement

Code

• github.com/Anonymous1563/Bayesian-Enhancement-Model
  Officialpytorch★ 76

Abstract

Image enhancement is considered an ill-posed inverse problem due to its tendency to have multiple solutions. The loss of information makes accurately reconstructing the original image from observed data challenging. Also, the quality of the result is often subjective to individual preferences. This obviously poses a one-to-many mapping challenge. To address this, we propose a Bayesian Enhancement Model (BEM) that leverages Bayesian estimation to capture inherent uncertainty and accommodate diverse outputs. Our approach, integrated within a two-stage framework, first employs a Bayesian Neural Network (BNN) to model reduced-dimensional image representations, followed by a deterministic network for refinement. We further introduce a dynamic Momentum Prior to overcome convergence issues typically faced by BNNs in high-dimensional spaces. Extensive experiments across multiple low-light and underwater image enhancement benchmarks demonstrate the superiority of our method over traditional deterministic models, particularly in real-world applications lacking reference images, highlighting the potential of Bayesian models in handling one-to-many mapping problems.

Tasks

Reproductions