Region-Aware Image Denoising by Exploring Parameter Preference

发表于 | 分类于

一种通过探索参数偏好的区域感知图片降噪（RAID）算法。

两个基本发现：

  • 使用较小的噪声水平作为参数，有时候 比直接使用噪声水平作为参数 能获得更好的效果
  • 复杂纹理图片区域的最优降噪参数通常比简单纹理的最优降噪参数要小

由此提出一种，通过寻找 preference parameter ，混合各区域的最佳降噪结果的图片降噪算法。

论文：Region-Aware Image Denoising by Exploring Parameter Preference

RAID

将图片按噪声水平分组，图片划分为 n×n 块处理。降噪算法采用 PGPD

  • 计算各个噪声水平的最佳改变率，获得最优降噪参数，并且定义理想偏好（ideal preference）
  • 训练参数偏好模型
  • 使用参数偏好模型对目标噪声图片进行偏好预测
  • 按参数偏好范围组合降噪结果

最佳改变率计算

将噪声图片按噪声水平分组，为不同噪声水平 α 计算最佳的改变率 ˜r 和最佳降噪参数 ˜α=˜r∗α

最佳的改变率 ˜r 选择 R={0.65,0.7,0.75,0.8,0.85,0.9,0.95}

  • 使用 α 和 α′∈α×R 作为降噪参数，采用 PGPD 对一张图片进行降噪，得到 8 张降噪结果 I1R *1 和 I2R *7
  • 将 8 个降噪结果按相同的方式划分为 n×n 个无重叠区域
  • 对于每个区域 j ，计算 α 和 α′（7种）降噪结果之间的偏好 Pj （7组，每组对应N张噪声图片）{S1j=∑p∈Ij|Ij(p)−I1Rj(p)|S2j=∑p∈Ij|Ij(p)−I2Rj(p)|
Pj={0,ifS1j≤S2j1,ifS1j≥S2j

其中 Ij(p) 为clean image 中第 j 区域的像素，I1Rj(p) 为使用 α 作降噪参数所得降噪结果的第 j 区域的像素，I2Rj(p) 为使用 α′ 作降噪参数所得降噪结果的第 j 区域的像素。S1j 和 S2j 表示像素差的绝对和。（Pj=0 偏好 I1R，Pj=1 则偏好 I2R）Pj 为区域的理想偏好

  • 根据偏好 Pj 组合 α′ 对应的组合降噪结果 Icj （7组，每组对应N张噪声图片的混合降噪结果），即为理想偏好图Icj={I1Rj,ifPj=0I2Rj,ifPj=1
  • 计算7张 Icj 的平均峰值信噪比 PSNRavg，选平均信噪比最大的 α′ 为最佳降噪参数 ˜α，其对应改变率为最佳改变率 ˜r

伪代码：

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
noiseImages = input()
iGroups = sortByNoiseLeve(noiseImages) 

R = [0.65, 0.7, 0.75, 0.8, 0.85, 0.9, 0.95] # len(R) = 7, 改变率集合

for g in iGroups: # 对每个噪声水平计算其最优改变率
  a = noiseLevel(g)
  a_ = [ a * r for r in R ]
  
  PSNRavgDic = {} # ai : PSNRavg
  iDenoisedComs = [] # 每个 a_ 会生成一个混合降噪结果
  for ai in a_:
  	iDenoisedComs = []
  	for img in g:
      iDenoised = PGPD(img, a)
      iDenoisedRegions = segment(iDenoised, n, n) # 分割为 n*n 个区域

      _iDenoised = PGPD(img, ai)
      _iDenoisedRegions = segment(iDenoised_, n, n) # 分割为 n*n 个区域
    
      for j in [region for r in n*n ]: # 选择每个区域用于组合
        P[j] = formulate_3(iDenoisedRegions[j], _iDenoisedRegions[j])
        I[j] = formulate_4(P[j])
      iDenoisedCom = combineRegion(I, n, n, iDenoisedRegions, _iDenoisedRegions) # 混合区域组成降噪图片
      iDenoisedComs.append(iDenoisedCom)
  
    PSNRavgDic[ai] = avg(PSNR(iDenoisedComs)) # 记录 ai 对应的平均 PSNR
    
  # 选去 PSNRavg 最大的 ai 作为最优改变率对应的 a
  PSNRavgDic = dict([val, key] for key, val in dic.items()) # 对调键值对 PSNRavg : ai
  bestR = PSNRavgDic[max(PSNRavgDic.keys())]

训练偏好参数模型

对与每张噪声图片 In，选择其两张降噪结果 I1R 和 I2R（来自 ˜α） ，两张差异图片 D=In−I1R 和 ˜D=In−I2R

将上述 4 张图划分为 n×n 个区域，对每个区域提出特征。

特征向量（含 3 部分）：

  1. I1R 和 I2R 的区域的所有像素 Fa={F1,F2,..,Fn∗n} 和 Fb={Fn∗n+1,Fn∗n+2,..,Fn∗n∗2}
  2. 两张差异图片的区域的所有像素 Fc={Fn∗n∗2+1,Fn∗n∗2+1,..,Fn∗n∗3} , Fd={Fn∗n∗3+1,Fn∗n∗3+1,..,Fn∗n∗4}
  3. 由于纹理简单区域的像素差异比纹理复杂的一般要小，抽取 Fa, Fb, Fc, Fd 之间的差异作为作为第三个特征，Fe={Fn∗n∗4+1,Fn∗n∗4+2,Fn∗n∗4+3,Fn∗n∗4+4}

故 FV={Fa,Fb,Fc,Fd,Fe}

标签 label ：对每个区域计算理想偏好 Pj 作为区域的 label

模型 M ：根据特征向量和标签，使用随机森林分类（RFC ）算法，对每个噪声水平进行训练模型

偏好预测

计算噪声图片 In 的噪声水平，根据噪声水平计算 I1R 和 I2R ，并且选择噪声水平对应的参数偏好模型 M。

将图片分化为 n×n 的区域，计算各个区域的特征向量，使用多个决策树计算参数偏好值，取对应多个决策树数量最多的值为最终的区域预测参数偏好 Pj

组合降噪结果

由于直接使用区域预测参数偏好进行区域拼接，生成的结果将具有明显的拼接痕迹，故不用区域预测参数偏好值，而采用区域预测参数偏好范围 P 进行区域拼接

P=f(AA+B)
f(x)={0x<1−λx+λ−12λ−11−λ≤x≤λ1x>λ

A 和 B 代表决策结果为 I1R 与 I2R 各自的决策树数量。则 x 即代表 I2R 的决策树在所有决策树中的比例。λ 为阈值（本文取值0.75）

使用 P 进行区域选择然后拼接所有区域，得到最后的降噪结果

实验

硬件设备： a 2.40GHz Intel Dual 6 Cores CPU and 16 GB memory.

降噪算法： PGPD, BM3D, NLM

数据集： BSD500， 400训练，68测试。和另一个具有14张常见图片的数据集作为测试数据集

噪声： AGWN，噪声水平 σ×255 ∈ {10, 20, …, 100}

评价指标：降噪结果的 PSNR 和 SSIM, visual effect

RAID 参数： n=7, N=400, λ=0.75, bestChangeRate（对于各个噪声水平和各个IDA是待实验计算的）

实验结果：

  • RAID-PGPD 的 PSNR 和 SSIM 均比 PGPD 高。BM3D, NLM, RAID-BM3D, 和 RAID-NLM 也具有相同结果。
  • 算法耗时：两倍的原 IDA 降噪时间+特征提取时间+特征预测时间+结果混合时间。RAID-BM3D, RAID-PGPD, and RAID-NLM 的平均额外时间分别为 0.408, 0.385, and 0.367 秒。

在不同噪声水平上，256 × 256大小的图片上的数据统计：

算法类型原算法 平均执行时间/s结合 RAID 平均执行时间为/s结合 RAID 平均模型训练时间/h
BM3D1.3283.0631.358
PGPD19.41939.2241.171
NLM33.01966.4051.354

（可见时间成本很大）

0%