The Noise Plane: A Diffusion Bridge Across Exponent and Dispersion
Organizations: The Hong Kong University of Science and Technology (Guangzhou), Guangzhou 511453, China · KTH Royal Institute of Technology, SE-100 44 Stockholm, Sweden · University of Montenegro, 81000 Podgorica, Montenegro
Abstract
Diffusion-based denoising commonly uses Gaussian perturbations, whereas sensor noise follows diverse signal-dependent statistical laws. In this paper, we propose -Bridge, a unified measurement-indexed diffusion framework for Tweedie noise. The noise exponent describes the variance--intensity relation, while accumulated measurement controls the noise level. Together, they define a noise plane that includes Gaussian, Poisson, and Gamma models. The stochastic bridge connects measurement distributions through independent increments, giving denoising a physical interpretation as progressive measurement accumulation. Both the starting and stopping indices can be specified, allowing restoration from an observed measurement level to a prescribed higher level. Measurement superposition yields a shared reverse-mean update, paired with a deviance-based objective for learning the required conditional mean. A latent compound Poisson--Gamma construction enables sampling at intermediate exponents. A single exponent-conditioned network adapts to different Tweedie noise models without separate training for each exponent. Experiments show that exponent conditioning improves cross-exponent robustness over fixed-exponent models. Using estimated exponents, the shared network achieves robustness comparable to an oracle-selected set of five specialist models. Experiments on synthetic images, FMD fluorescence microscopy, and multi-sensor SAR images further demonstrate its effectiveness across different signal-dependent noise regimes.
Figures & tables
| index | ||||
| frames | ||||
| exposure | ||||
| exposure | series ∗ | Binom. Beta ∗ | ||
| looks |
| fixed specialist, worst case over | conditional | envelope | |||||||
|---|---|---|---|---|---|---|---|---|---|
| severity | told | blind | |||||||
| severe | 18.997 | 20.150 | 21.339 | 20.036 | 18.007 | 22.260 | 22.260 | 22.254 | |
| moderate | 20.921 | 21.899 | 23.112 | 22.903 | 21.747 | 24.635 | 24.637 | 24.631 | |
| mild | 24.142 | 25.052 | 26.022 | 26.336 | 25.551 | 27.436 | 27.435 | 27.431 | |
| method | params (M) | severe, | moderate, | mild, | |||||||
|---|---|---|---|---|---|---|---|---|---|---|---|
| Classical, transform-based, and Gaussian-bridge baselines | |||||||||||
| BM3D [ 1 ] | — | 1 | 18.63/.493 | 17.59/.479 | 16.87/.467 | 23.19/.632 | 22.70/.622 | 22.08/.609 | 26.58/.763 | 26.21/.747 | 25.63/.724 |
| VST + BM3D [ 17 , 43 ] | — | 1 | 21.21/.485 | 20.12/.434 | 19.86/.462 | 23.87/.634 | 23.77/.638 | 23.62/.641 | 26.89/.778 | 26.76/.778 | 26.59/.776 |
| Noise2VST [ 20 ] | 32.64 | 1 | 21.99/.543 | 21.53/.521 | 21.05/.499 | 24.71/.691 | 24.56/.689 | 24.40/.685 | 27.66 / .812 | 27.55/.812 | 27.38 /.811 |
| -Bridge, | 15.19 | 5 | 18.67/.407 | 17.51/.374 | 16.97/.357 | 21.39/.541 | 20.04/.503 | 19.13/.480 | 24.85/.701 | 23.52/.666 | 22.42/.642 |
| method | CF-FISH | CF-MICE | TP-MICE | mean |
|---|---|---|---|---|
| BM3D [ 1 ] ∗ | 27.01/.826 | 33.18/.942 | 31.63/.886 | 30.61/.885 |
| VST + BM3D [ 17 , 43 ] ∗ | 25.73/.809 | 33.62/.946 | 31.55/.889 | 30.30/.881 |
| Noise2VST [ 20 ] ∗ | 28.58/.879 | 34.02/.949 | 32.14/.895 | 31.58/.908 |
| -Bridge, ∗ | 24.68/.811 | 31.18/.932 | 30.45/.887 | 28.77/.877 |
| NAFNet [ 7 ] | 28.23/.878 | 34.06/ .951 | 32.61 / .898 | 31.63/.909 |
| Restormer [ 8 ] | 28.00/.873 | 34.19 / .951 | 32.39/ .898 | 31.52/.908 |
| method | / , lower better | |||
|---|---|---|---|---|
| miniSAR | FARAD-X | TerraSAR-X | mean | |
| BM3D [ 1 ] | 32.5/55.1 | 41.0/78.1 | 49.2/109.5 | 40.9/80.9 |
| VST + BM3D [ 17 , 43 ] | 34.7/46.4 | 35.8/64.2 | 45.4/77.9 | 38.6/62.8 |
| Noise2VST [ 20 ] | 25.5/37.9 | 30.5/50.3 | 39.6/77.5 | 31.9/55.2 |
| -Bridge, | 36.3/63.8 | 39.9/73.6 | 46.7/100.9 | 40.9/79.4 |
| NAFNet [ 7 ] | 12.0 / 17.4 | 21.8 / 34.7 | 32.2 /60.8 | 22.0 / 37.7 |
| component | full model | ablation | min. PSNR | |
| Full conditional model | – | |||
| Reconstruction weight | ||||
| Exponent-dependent weight | ||||
| Moment regularization | ||||
| Consistency regularization | ||||
| Index schedule | Geometric | Linear |
| vs. fixed | ||||
|---|---|---|---|---|
| severe | ||||
| moderate | ||||
| mild |
| Gaussian | Poisson | Poisson–Gamma | Gamma | |
|---|---|---|---|---|
| severe | |||
|---|---|---|---|
| diagonal mean PSNR | 22.326 | 22.330 | 22.225 |
| worst over | 20.739 | 21.339 | 21.594 |
| mean off-diagonal penalty | |||
| diagonal dark log-RMSE | 0.2841 | 0.2831 | 0.2854 |
| diagonal res. var. | 0.926 | 0.949 | 0.968 |
| sensor | nominal | at ENL band | implied | ||||
|---|---|---|---|---|---|---|---|
| all | 5–95 | 10–95 | 25–95 | 40–95 | |||
| miniSAR | 1.00 | 2.150 | 2.042 | 2.014 | 2.005 | 2.002 | 1.06 |
| FARAD-X | 1.00 | 2.103 | 2.071 | 2.058 | 2.030 | 2.013 | 0.92 |
| TerraSAR-X | 1.16 | 2.106 | 2.067 | 2.044 | 2.014 | 2.005 | 1.37 |