“压缩传感”引论
香港大学电机电子工程学系
高效计算方法研究小组
沙威 2008 年 11 月 20 日
Email: wsha@eee.hku.hk
Personal Website: http://www.eee.hku.hk/~wsha
到了香港做博士后,我的研究领域继续锁定计算电磁学。我远离小波和计算时谐分析有
很长的时间了。尽管如此,我仍然陆续收到一些年轻学者和工程人员的来信,和我探讨有关
的问题。对这个领域,我在理论上几乎有很少的贡献,唯一可以拿出手的就是几篇网上发布
的帖子和一些简单的入门级的代码。但是,从小波和其相关的理论学习中,我真正懂得了一
些有趣的知识,并获益良多。我深切地感到越来越多的学者和工程师开始使用这个工具解决
实际的问题,我也发现互联网上关于这方面的话题多了起来。但是,我们永远不能只停留在
某个阶段,因为当今学术界的知识更新实在太快。就像我们学习了一代小波,就要学习二代
小波;学习了二代小波,就要继续学习方向性小波(X-let)。我也是在某个特殊的巧合下不
断地学习某方面的知识。就像最近,我的一个友人让我帮她看看“压缩传感”(Compressive
Sensing)这个话题的时候,我的兴趣又一次来了。我花了一个星期,阅读文献、思考问题、
编程序、直到写出今天的帖子。我希望这篇帖子,能对那些没进入且迫切想进入这个领域的
学者和工程师有所帮助。并且,我也希望和我一个星期前一样,对这个信号处理学界的“一
个大想法”(A Big Idea)丝毫不了解的人,可以尝试去了解它。我更希望,大家可以和我
探讨这个问题,因为我到现在甚至不完全确定我对压缩传感的某些观点是否正确,尽管我的
简单的不到 50 行的代码工作良好。在这个领域中,华裔数学家陶哲轩和斯坦福大学的统计
学家 David Donoho 教授做出了重要的贡献。在这个引言中,我用简单的关键字,给出我们
为什么需要了解甚至是研究这个领域的原因。那是因为,我们从中可以学习到,下面的这些:
矩阵分析、统计概率论、拓扑几何、优化与运筹学、泛函分析、时谐分析(傅里叶变换、小
波变换、方向性小波变换、框架小波)、信号处理、图像处理等等。所以,我们有什么理由,
拒绝这个有意思的东西呢?让我们开始吧。
传统思路——正交变换
对于一维的信号
1×∈ NRx
,大多数情况下,信息是冗余的。我们可以通过正交变换的
方法来压缩它。正变换:
y Ψ= ,反变换
x
x
HΨ=
y
。这里,
ΨΨ
H
H
=ΨΨ=
I
NNC ×∈Ψ
,
,
I 是单位矩阵。对于
1×∈ NCy
,能量较 x 集中,本质上去除了 x 中的相关性。因此,我们只
保留 K 个较大的分量,而把其它
KN − 个置为零。通过反变换,我们能够近乎完美的重建
原始信号。因为,那
KN − 个变换域系数的贡献,实在微乎其微。具有这样性质的信号被
称为 K “稀疏”(Sparsity)的。于是,我们有了如下编码解码的策略:
编码:构造 Ψ ,做正变换
y Ψ= ,保留 y 中最重要的 K 个分量,和其对应的位置。
x
解码:把 K 个分量放回到对应的位置,其它位置填 0,构造 HΨ ,反变换
ˆ Ψ=
x
H ˆ
y
。
而解码能否近乎得到原始信号呢?显然,我们希望
||
x
−
||ˆ
x
2
||
=
y
−
||ˆ
y
2
δ≤
,δ是一个小
的常数。但更有效的是用相对误差
||
y
−
||ˆ
y
2
||/
y
||
2
δ≤
。
但这种编码解码方法有些缺点:1、考虑到香农(Shannon)采样定理,为了获得很好的
信号分辨率,采样间隔会很小,造成了原始信号长度会很长,因此变换过程会消耗很长的时
间。2、 K 个需要保留的重要分量的位置,是随着信号的不同而不同的。因此,这种策略是
“自适应”(Adaptive)的,且需要分配多余的空间存储这些位置。3、一旦在传输过程中 K
个分量中的某几个丢失了,后果可想而知。如果我们制作一个音频设备,1 将带来电力的消
耗和用户的不满,2 将带来存储空间的增加,3 将带来较差的抗干扰能力。
新的思路——压缩传感
压缩传感(Compressive Sensing)是一个很有意思的新的方向。它也正成为信号处理
领域的“A Big Idea”。对于信号
1×∈ NRx
,我们可以找到它的 M 个线性测量(Liner
Measurement),
s Φ= 。
x
∈Φ
NMR ×
。这里,Φ 的每一行可以看作是一个传感器(Sensor),
它与信号相乘,拾取(Acquisition)了信号的一部分信息。拥有了这 M 个测量和 Φ ,我
们就可以近乎完美的重构原始信号了。听起来“相当”传奇,事实上,它基于如下严格的数
学最优化(Optimization)问题:
目标函数
0||ˆ||min y ,且满足等式约束
ΦΨ ˆ
yH =
s
或者,可以写成
||min
s
ΦΨ−
H
||ˆ
y
2
λ+
||ˆ||
y
0
求解该最优化问题,得到变换域的 yˆ ,然后反变换,便可以得到时域的 xˆ 。公式中的 2 是我
们熟悉的 2-范数,而 0 是什么呢?是 0-范数,也就是向量 yˆ 中非零元素的个数。看起来很
有道理,因为 yˆ 是待求的变换域向量,它是 K 稀疏的。使 yˆ 非零元素的个数尽量小,也就
是保留了尽量少的重要的 K 个分量,显然这几个分量可以近乎完美重构 x 。我们回到传统
的思路,这 K 个分量是我们在变换域“自适应”找的,而该优化算法也可以使我们找到这 K
个分量。
这就足够了吗?显然不行,我们仍然没有探讨测量矩阵 Φ 需要满足的性质。我们用极
限分析法。如果我们把 Φ 构造成和 HΨ 极端相似(Coherence)的矩阵,也就是拿出 Ψ 的
前 M 行。用这个算法求 yˆ ,我们将得到
ˆ
y
=
⎛
⎜⎜
⎝
0
1
×
Ms
MN
(
−
1)
×
,这显然是错误的。也就是说,你
⎞
⎟⎟
⎠
强迫的认为前 M 个变换域分量是重要的。而事实是,重要的 K 个分量的位置我们事先是不
知道的,是随着信号的不同而不同的。当然,你可以将 Φ 恰好构造成对应最重要分量的 K 行,
1
得到正确的结果。而这种的做法要付出的概率代价 K
NC
。也就是说,你必须穷举 K
NC 次,才
能得到你想要的结果。但是,即使你有幸碰到了它,也并不能肯定这个结果就是对的。因此,
我们选择 Φ 和 HΨ 极端不相似(Extremely Incoherence)。于是, Φ 很大程度上和随机
(Randomness)这个词相联系,它可以是满足高斯分布的白噪声矩阵,或贝努里分布的 1± 矩
阵 ( 也 称 作 Noiselet ) 等 等 。 除 此 之 外 , 我 们 希 望 线 性 测 量 有 稳 定 的 能 量 性 质 :
1
≤−
δ
||
||ˆ
yH
ΦΨ
||ˆ||
y
2
2
1
+≤
δ
,也就是它要保持 K 个重要分量的长度。综合上面的,我们有
了如下编码解码的策略:
编码:构造 Φ ,生成测量
s Φ= ,保留 s 。
x
解码:构造同样的 Φ ,构造任一种正交变换 HΨ ,根据 s 重构 x 。
压缩传感的优势:1、非自适应(Non-Adaptive)的,一开始就可以传输长度较短的信
号,甚至突破采样定理的极限。2、抗干扰, s 中任何一项都是重要的,或者说不重要的。
丢失了某几项,仍然可以完美重构。它的缺点:1、实际中, s 的长度一般是重要分量长度
的 4 倍,才能近乎完美重构。数学上更严格的,
M 4≈
K
或者
KM ≥
N
(
log 2 K
)
。2、重构
(恢复)算法是 NP 问题。即使将 0-范数转化为 1-范数,由于其不可微性(Indifferentiable),
算法的计算复杂度仍然很高。它的应用前景广泛:低成本数码相机和音频采集设备;节电型
音频和图像采集设备;天文(图像本身就稀疏,例如天空的星星);网络;军事(用很简易
的摄像机随机记录场景,可以完全重构军事地图);超宽带(雷达信号处理)。这里,值得指
出的是,美国的工程学家已经设计出了实际的产品。
快速算法——正交匹配追踪
对于 0-范数的优化问题,实际上是 NP 问题,就是在多项式时间内难以求解,甚至无法
验证解的可靠性。于是,我们必须将 0-范数换一下,变成 1-范数。为什么不是 2-范数呢,
那样就会简单多了。毕竟 2-范数的优化问题可以转化成 2 次型问题,而 1-范数,
||ˆ||
y
1
∑=
i
|ˆ|
iy
,在 0 点处不可导,因此无论是梯度算法、矩阵求导等等手段都变得相形见
绌。因此,基于 1-范数的优化算法需要特殊处理,且复杂度很高。下面我们来解释下为什
么要用 1-范数,而不是 2-范数。我们令恢复(Recovery)矩阵
T ΦΨ=
H
,则等式约束可
重写为:
yT =ˆ
s
。 yˆ 中未知数有 N 个,方程只有 M 个,且
M << 。因此,方程有无穷
N
多解。从几何上说,
ˆ
yT
=− s
0
是一个超平面,为了简化,在 2-D 问题中(
1=K , yˆ 只有
两个元素待求)可认为它就是一条直线。而范数约束呢?0-范数是一个十字架,因此它的最
外侧(范数的最小值)是 4 个点。所以其和直线的交点,必然在坐标轴上。也就是说,能使
yˆ 产生更多的 0,这正是我们想要的“稀疏”的结果。2 范数是一个圆,因此它的最外侧边
界和直线的交点(就是切线的概念),以压倒性的概率不在坐标轴上,除了直线的斜率恰好
为 0 或者无穷大。其实直线的斜率恰好为 0 或者无穷大,是不可能的,因为 Φ 和 HΨ 极端
不相似。只有 Φ 取 Ψ 的某一行时,两者相似,才会发生斜率恰好为 0 或者无穷大的情况(因
此,你的胜算只有
/1 C ,但你不知道哪个是对的。)。依上所述,用 2-范数优化的结果,使
1
2
yˆ 几乎没有 0,这是我们不期望的。而 1-范数是一个菱形,四个角都在坐标轴上,因此它和
直线的交点以压倒性的概率落在坐标轴上。这就是我们要用 1-范数的原因。事实上, p 范
0
数满足,
≤ p ,它的外边界都向中心凹(Concave)。而
<
1
1>p ,外边界向外凸(Convex)。
所以前者的外边界和直线的交点无疑落在坐标轴上。根据这个几何解释,我们可以将问题转
化成:
||min
||ˆ
yTs
−
λ+
||ˆ||
y
1
2
这显然是一个非线性(Non-Linear)的凸(Convex)优化问题。
众所周知,对于优化问题,我们一般用梯度的方法来求解。而对
1||ˆ|| y ,在 0 点导数不
存在,因为这个点正好位于两条直线的交点上,左右导数不相等。这也正是很多数学方法的
考虑。像子梯度(Subgradient)法、平滑近似法(Smooth Approximation)等等。我们暂
不谈这些方法,因为它需要特殊的数学背景。我们谈一谈工程领域最常用的正交匹配追踪法
(Orthogonal Matching Pursuit)。它的思想本质上还是来自于这个 K “稀疏”。我们绕了
一圈,还是为了找这 K 个关键的分量。既然是关键,显然它的系数的绝对值应该比其它
KN − 个分量大得多。为了简单起见,我们先假设
1=K ,唯一非零元素 qyˆ 在 yˆ 中对应的
位置在 q 。于是 yTˆ 就是恢复矩阵T 的第 q 列 qT 与 yˆ 中的非零元素 qyˆ 的乘积,即
yT
q
=ˆ
q
s
q
。
且
||
s
−
s
q
||
2
||/
s
||
2
δ<
。 换 句 话 说 , T 的 第 q 列 与 s 的 相 似 程 度 最 高 , 即
|
<
sT
,
q
|
|
=>
sT
H
q
|
|
>>
sT
H
r
|
|
<=
sT
,
r
|,
>
r
≠
q
。所以,我们只要计算恢复矩阵T 的所有列与
s 的内积,找到内积绝对值最大的那列就行了,该列对应的位置就是 q 。根据最小二乘法,
ˆ
y
q
=
TT
(
H
q
q
1)
−
sT
H
q
,就是使
||
ˆ
q yTs −
q
2||
最小的那个 qyˆ 。呵呵,感觉到了吗,实际上这
有点或者非常像施密特(Schimidt)正交化方法。余量
r
n
−=
s
sT
,
<
q
TT
,
<
q
q
>
>
T
q
,始终同 qT 正
交。这也是为什么这个方法叫“正交”匹配追踪的意思了。而匹配,显然是你找到了最大的
|
<
sTq
,
|
>
。对于
1>K 呢,其实是差不多的。我们找到余量 nr 同T 中所有列向量最大的那
个 即 可 ( 但 第 一 次 找 到 的 那 列 要 排 除 , 因 为 它 已 经 保 留 了 下 来 。)。 于 是 , 找 到 使
||
s
−
(
TT
,
q
q
1
2
⎛
)
⎜⎜
⎝
ˆ
y
ˆ
y
q
2
q
1
⎞
⎟⎟
⎠
||
2
最小的那个
⎛
⎜⎜
⎝
ˆ
y
q
ˆ
y
2
q
1
。这里, 1qT 是我们第一次找到的那一列, 2qT 是
⎞
⎟⎟
⎠
我们新找到的那一列(也要记住它的列号 2q )。可以看出,
ˆ
y
q
=
ˆ
y
ˆ
y
q
2
q
1
⎛
⎜⎜
⎝
⎞
⎟⎟
⎠
被更新了,由原来
的一个变成两个了,也就是我们找到两个在变换域最关键的元素和其在 yˆ 中对应的位置了。
令
T =
q
(
TT
2,
q
q
)1
,余量 nr 又一次被写为:
r
n
−=
s
sT
,
<
q
TT
,
<
q
q
>
>
T
q
。我们再一次看到了施密
特正交化的影子。继续上面的步骤,直至找到变换域所有 K 个最重要的分量。也就是说,
正交匹配追踪的迭代次数
Km ≥ 。实际操作上只要满足
||
rn
||
2
||/
s
||
δ<2
,迭代就可以中
止了。最后,假设正交变换 Ψ 是傅里叶变换,我们来分析下计算复杂度。生成恢复矩阵T ,
采用 FFT 快速算法,需要计算复杂度
MNO
(
log
2 N
(
))
;找到 T 中匹配的列向量需要
O
(mNM
)
次操作,注意 m 是算法需要的迭代次数;求解 qyˆ 和 nr 需要
2MmO
(
)
次操作。考
虑到
Km ~ ,
KM
~
log
(
KN
/
)
2
,总计算复杂度可以控制在
NO
(
[log
2 N
(
2
)]
)
以下。到
此为止,我们完成了压缩传感所有基础的理论和操作。
小结——压缩传感的特点
最 后 , 让 我 们 总 结 一 下 , 压 缩 传 感 理 论 的 关 键 字 : 稀 疏 ( Sparsity )、 不 相 关
( Incoherence )、 随 机 性 ( Randomness )、 非 自 适 应 ( Non-Adaptivity )、 非 线 性
(Non-Linearity)、不可微(Indifferentiability)。从这些词语可以看出,压缩传感理论
是对传统理论的颠覆。这种颠覆最令人振奋的表现,就是它突破了香农采样定理的极限,能
以随机采样的方式用更少的数据采样点(平均采样间隔低于采样定理的极限),来完美地恢
复原始信号。科学也就是在对传统理论不断地颠覆和修正中才得以进步和发展。
为了大家更好的理解压缩传感,在我的个人网站上,给出了一个没有经过优化的简单的
操作代码,希望对大家有所帮助。此外,由于此帖含有大量公式,我用了类似 Latex 的语法
敲出了它,一个更好阅读的 pdf 版本也放在了我的个人网页上。最后,如果阅读这篇帖子的
人,试图转贴的话,请注明这篇帖子的出处和作者,以此作为对作者工作小小的肯定和支持。
(程序和帖子的 PDF 版下载地址:http://www.eee.hku.hk/~wsha/Freecode/freecode.htm)