File size: 1,201 Bytes
8218b31
 
 
009aa6f
 
8218b31
205b5f5
d99ce82
205b5f5
 
009aa6f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
---
license: mit
---
Bayesian Enhancement Model (BEM)

AAAI 2026 โ€” Bayesian Neural Networks for One-to-Many Mapping in Image Enhancement

Paper: [arxiv:2501.14265](https://arxiv.org/abs/2501.14265)

Github: https://github.com/BinCVER/BEM

๐ŸŒŸ Overview

Image enhancement is inherently one-to-many โ€” a single degraded image often corresponds to multiple plausible enhanced outputs, especially in low-light and underwater environments.

However, existing deterministic models can only produce one prediction, failing to capture this ambiguity.

BEM (Bayesian Enhancement Model) solves this by:

Modeling uncertainty using a Bayesian Neural Network (BNN)

Generating multiple valid enhancement candidates

Selecting or aggregating candidates via Ranking or Monte Carlo

Refining details using a second-stage DNN

Achieving DNN-level inference speed with BNN-level diversity

This makes BEM the first practical Bayesian solution for large-scale image enhancement tasks.

๐Ÿ” Key Features
โœ“ One-to-Many Enhancement via Bayesian Inference

BEM samples weights from a variational posterior learned by a Bayesian UNet.
Each sample produces a distinct enhancement, capturing the ambiguity of the task