原文:High-Quality Hyperspectral Reconstruction Using a Spectral Prior
作者:Inchang Choi, Daniel S. Jeon, and Giljoo Nam
单位:Universidad de Zaragoza
– Abstract:
本文提出了一种高光谱重构成像的方法,以缓解基于压缩感知成像技术的光谱分辨率和空间分辨率的相互制约问题。所提出的方法包括两步:1)构建了一个卷积自动编码器(convolutional autoencoder)来学习non-linear的光谱表征;2)构建了一个稀疏(sparsity)和空间梯度(spatial gradients)联合约束的优化问题。我们的方法在光谱成像、光谱插值和去马赛克等任务上均表现出较好的结果。此外,我们还发布了一个新的高光谱数据集。
– Introduction
高光谱图像具有丰富的光谱信息,在许多领域具有重要作用。早期的成像技术,例如推扫式和滤光式均存在缺点,例如空间和光谱分辨率相互制约。基于压缩感知的成像技术可缓解该问题,受到了研究人员的广泛关注,然而目前的重构算法所需时间比较长。
本文提出的算法可很好地解决以上的问题。算法主要包括了两部分:(1)通过卷积自动编码器学习non-linear的光谱信息表征;(2)构建了一个稀疏(sparsity)和空间梯度(spatial gradients)联合约束的优化问题。
– Hyperspectral Image Reconstruction
算法整体流程如下:
自动编码器网络结构如下:
构建重构优化问题:
将测量信号
i
=
Φ
h
i = \Phi h
i=Φh 替换为自动编码器的输出(
i
=
Φ
D
(
α
)
i = \Phi D(\alpha)
i=ΦD(α))。因此,将原始问题转换为:
m
i
n
i
m
i
z
e
α
∣
∣
i
−
Φ
D
(
α
)
∣
∣
2
2
+
λ
1
∣
∣
α
−
E
(
D
(
α
)
)
∣
∣
2
2
+
λ
2
∣
∣
∇
x
y
D
(
α
)
∣
∣
1
,
{\rm minimize}_\alpha \ ||i - \Phi D(\alpha) ||^2_2 + \lambda_1 ||\alpha - E(D(\alpha))||_2^2 + \lambda_2|| \nabla_{xy} D(\alpha)||_1,
minimizeα ∣∣i−ΦD(α)∣∣22+λ1∣∣α−E(D(α))∣∣22+λ2∣∣∇xyD(α)∣∣1,其中E为自动编码器的encoder,D为decoder。
优化过程:
因为第二个约束项(
∣
∣
∇
x
y
D
(
α
)
∣
∣
1
|| \nabla_{xy} D(\alpha)||_1
∣∣∇xyD(α)∣∣1)不可求导,我们将上述问题拆解成两个子问题:
α
k
+
1
=
a
r
g
m
i
n
α
∣
∣
i
−
Φ
D
(
α
)
∣
∣
2
2
+
λ
1
∣
∣
α
−
E
(
D
(
α
)
)
∣
∣
2
2
,
\alpha^{k+1} \ = \ {\rm argmin}_\alpha \ ||i - \Phi D(\alpha) ||^2_2 + \lambda_1 ||\alpha - E(D(\alpha))||_2^2,
αk+1 = argminα ∣∣i−ΦD(α)∣∣22+λ1∣∣α−E(D(α))∣∣22,
z
=
a
k
+
1
=
a
r
g
m
i
n
z
λ
2
∣
∣
z
∣
∣
1
,
z = a^{k+1} \ = \ {\rm argmin}_z \ \lambda_2|| z ||_1,
z=ak+1 = argminz λ2∣∣z∣∣1,因此,我们可以将优化问题构建为:
m
i
n
i
m
i
z
e
α
∣
∣
i
−
Φ
D
(
α
)
∣
∣
2
2
+
λ
1
∣
∣
α
−
E
(
D
(
α
)
)
∣
∣
2
2
+
λ
2
∣
∣
z
∣
∣
1
,
{\rm minimize}_\alpha \ ||i - \Phi D(\alpha) ||^2_2 + \lambda_1 ||\alpha - E(D(\alpha))||_2^2 + \lambda_2|| z ||_1,
minimizeα ∣∣i−ΦD(α)∣∣22+λ1∣∣α−E(D(α))∣∣22+λ2∣∣z∣∣1,
s
u
b
j
e
c
t
t
o
∇
x
y
D
(
α
)
=
z
.
{\rm subject \ to} \ \ \nabla_{xy} D(\alpha) = z.
subject to ∇xyD(α)=z.
在ADMM算法框架下,可以迭代求解。