EDA365电子论坛网

标题: 利用MATLAB实现白话压缩感知 [打印本页]

作者: pulbieup    时间: 2020-1-13 10:24
标题: 利用MATLAB实现白话压缩感知
: Y) U' _8 K8 K- M5 k
压缩感知介绍  L( [, A9 {+ m5 \  ]; N
压缩感知(Compressive Sensing,CS),有时也叫成Compressive Sampling。相对于传统的奈奎斯特采样定理——要求采样频率必须是信号最高频率的两倍或两倍以上(这就要求信号是带限信号,通常在采样前使用低通滤波器使信号带限),压缩感知则利用数据的冗余特性,只采集少量的样本还原原始数据。& T% |( I; Q3 V
- A9 ~, M9 s9 `  K' e5 e
这所谓的冗余特性,借助MLSS2014马毅老师的课件上的例子来说明,
  a- P% V) M# U7 L* S& {, S
( t9 z) l5 I- R0 C' c ( m# @% t+ E5 s& {& {

7 S/ w! X4 u9 t$ G9 N8 s4 ^因为自然界的数据都存在局部低维结构、周期性、对称性等,因此,传统的固定采样率的采样方法必然存在信息冗余。由于信息冗余的存在,我们就知道有数据压缩那么一门学科。既然这样,为什么要把冗余的数据也采进来,再进行压缩呢,能不能不把冗余的数据采集进来?
2 U- U9 r0 X0 w% G) B9 U$ ~' i' ?, \. D: y  i5 P
压缩感知的思路就是:在采集的过程中就对数据进行了压缩,而且这种压缩能保证不失真(低失真)的恢复原始数据,这与传统的先2倍频率采集信号→存储→再压缩的方式不同,可以降低采集信号的存储空间和计算量。+ n  t, x  P, P
# L8 B! c: h1 U" }& N  K  G
好了,那么就来了解一下压缩感知的具体模型。
' r& q( h6 Q& i- Z  Y" R5 W7 I& R# c0 _9 y
1 {) T2 ]$ p. t: d) u
1. 稀疏表示+ h4 ~. E9 C% s6 L( d. A; T/ p, M
) l  n- V) D% h6 D  f# R
使用压缩感知理论首先要求信号能表示为稀疏信号,如x=[1 0 0 0 1 0],其中只有2个1,可认为是稀疏的。我们将信号通过一个矩阵映射到稀疏空间,( h% k8 O% I; v1 p7 n# L

# |; D0 A( \7 ?/ B+ F+ L设信号x为N维,即,则为NxN维稀疏表达矩阵,s即是将x进行稀疏表示后的Nx1维向量,其中大部分元素值为0。稀疏表示的原理就是通过线性空间映射,将信号在稀疏空间进行表示。- B0 s4 L5 A1 Q: l4 q7 e* s2 w8 M% {; l
  F" b. ?5 `+ K" U% y2 ~, W7 r
比如,信号# A+ z$ j( m  y
: S9 U& R! ~2 j# ^
在时域是非稀疏的,但做傅里叶变换表示成频域后,只有少数几个点为非0(如下图)。则该信号的时域空间为非稀疏,频域空间为稀疏空间,组成的矩阵。一般为正交矩阵。% `. A4 C& E0 u, U) t6 o  j
' L- L- n3 S) S: g$ T" a% `9 q

6 y: C8 S4 F5 v% t; `$ }/ s- k, _, @( w, z4 b* p, z
若稀疏表示后的结果s中只有k个值不为0,则称x的稀疏表示为k-Sparse。上图对x的频域稀疏表示就是2-Sparse。
7 Z, f( Z8 `1 {( T" P7 S
4 J% }( ^' J' k( \" d2 x& M
& x+ E  S# U# }5 k; {& H" L( _2. 感知测量; E0 t- L7 l+ o- z/ \/ \* u% _- E

3 R. U) s" m$ n0 h4 Z# P) {4 E% o0 w0 e8 H2 Z1 |: [
压缩感知的目的是在采集信号时就对数据进行压缩,大牛们的思路集中到了数据采集上——既然要压缩,还不如就从大量的传感器中只使用其中很少的一部分传感器,采集少量的冗余度低的数据。这就是感知测量的通俗的说法,用表达式表示) W; x. d2 j7 ~5 B+ ?, p& U
: N7 [0 T2 J8 J# [, {6 P+ V. b4 }
其中的x就是稀疏表示中的信号,为MxN维的感知矩阵(M表示测量信号的维度),y则表示M(M远小于N才有意义)个传感器的直接测量,因此维度为Mx1。
2 ^8 u6 |* N) q8 H/ j0 g
) g( q$ d8 p! ~% ?9 a将稀疏表示过程和感知测量过程综合起来:
  [5 H5 R) {1 w$ g4 |) ]9 n
+ e" P5 j, M1 K( s* p- }6 H 8 \  W- O4 X9 q5 d8 [# ?/ N
+ R5 a) b6 W! D8 u! \8 k+ N4 \
; F* z  U. {% y9 `# k5 q
数学描述
& l/ Z/ \5 X' {3 _: i- q8 W  f8 w; ^. |9 _

7 N+ E& P1 X7 b' x* u/ c6 l对于压缩感知模型,其中每个量的维度一定要了解(通过维度的变化来理解压缩感知很有效):
8 v2 v4 I  ]2 c8 H* r' U/ }
- u3 a" d3 R$ r2 }) I) U! H 0 l1 b5 u' ?/ C
! c) ]. C  a! w4 l& E
从感知测量中知道:M就是测量的维度(M远小于N)。
7 Q. q  v: X* Z2 K; P6 N( I
! |/ c9 u/ W! Y压缩感知的原信号恢复问题描述为:4 X: j7 m( [% y5 f$ p8 }

: g$ x1 Z5 g1 B: s+ u1 x6 D由已知条件:
8 P; v, l) D& |/ J
' B1 _& I# u) X; V6 v: L  P(1) 测量值y,且,其中e为噪声引入/ b( N% T# L7 h  T

  ]/ g' s$ Z5 y! r(2) s为k-Sparse信号(k未知)) _) v- `* K7 m
) J0 H7 e" A& r% ~" s6 p$ C' O; Z) ?
求解目标:k尽可能小的稀疏表示信号s及对应的。
9 X; ~+ ~- p" y
/ y- N' R( m8 X# K& \; w) X) r用数学形式描述为:/ o+ H; l- Y: d% |- ?9 T

- `% t/ V8 M: r) qe可以看成告诉随机噪声,e~N(0,δ2)。2 a  b( k& S6 a$ p7 f4 \. `
) m8 J) o$ a0 m2 Z
即是要求s使s的0范数(非0值的个数)最小,但0范数优化问题是很难求解的,于是一帮大牛就证明求解1范数也能逼近和上面相同的效果,而求解2范数及其更高的范数则结果相差越来越大(有些人在研究介于0范数与1范数之间的范数求解方法)。因此可转化为1范数求解:
1 [2 U. Y2 f+ L7 j  `8 [) a- Q& a- i7 b# \& ~& v3 g
由拉格朗日乘子法,上面的最优问题可转化成:
6 a5 Q3 r( M" o+ ^  z7 ], ^. \! @8 y5 _) f+ J$ U1 V2 Y
上面的最小值求解问题就可以直接通过凸优化解得结果了。
4 b  p4 R: G+ q' Q/ s6 S" w# I/ G8 m- H7 H. e5 D' k

! R; [$ _4 ~) o+ T$ J程序分析& Z1 M  r! o9 i1 B  f$ g+ ?

" g5 e1 }1 ^" a( L7 a$ n+ o
% l3 d5 C/ h+ u' l# C先下载CVX或spams工具箱,下面的matlab程序分别使用了时域稀疏信号和频域稀疏信号进行测试,两种不同在于时域稀疏信号不用稀疏表达矩阵(因此稀疏表达矩阵使用单位阵即可),而频域稀疏信号则需要先通过稀疏表达矩阵将信号在频域进行稀疏表示,再压缩感知后进行恢复时也要进行反FFT变换到时域。5 u9 O5 q8 L, V# N

( e( W: G* x9 x关于M的选取:测量数M>=K*log(N/K),K是稀疏度,N信号长度,可以近乎完全重构! ?. \* h- p' X9 ~' Z4 J
" M' x0 E4 i. h- {( c
clc
/ ^" Z8 s. @0 J% G% n7 Bclear all# d- E/ a: J7 n; D- H9 D
close all. q4 d: `9 ~# }% Q8 `$ a( v
# z1 M: U, Y- r/ b1 l1 c5 u
%% 产生原始信号及相关参数
2 |2 C. E8 g; t6 cn = 512;                          % 信号长度
# n1 _+ K/ Q& v9 l* [. V# e6 ys = 25;                           % 稀疏度(从下面知道,分时域和频域的情况)+ a6 l2 Y( r# E$ ]+ ?; t9 H) |
m = 5*s;                          % 测量长度 M>=S*log(N/S)& C( d; ~, ^; q" q8 k# M
freq_sparse = 0;                  % 信号频域稀疏则为1
( n5 T' g$ _" w& ?2 ^
) E9 }, [' f6 C& Wif freq_sparse' k8 m( J, ]' _) J7 i: z& o
    t = [0: n-1]';+ e+ o  C+ l8 A+ C/ l+ a. ^
    f = cos(2*pi/256*t) + sin(2*pi/128*t);   % 产生频域稀疏的信号
! A" l/ k6 ^& I+ w, relse2 B# @2 A. d. W* `( H  w
    tmp = randperm(n);    * f1 ?% a% R3 }% }* {
    f = zeros(n,1);0 _' r% Q  Q+ [/ Y
    f(tmp(1:s)) = randn(s,1);     % 产生时域稀疏的信号+ i8 B: }! C- C7 j
end
7 B5 s/ x7 i1 {5 ^
1 V$ Z( K7 Y2 m! M( ?' j%% 产生感知矩阵和稀疏表示矩阵, y. O6 V+ p/ B2 _# D7 \
Phi = sqrt(1/m) * randn(m,n);     % 感知矩阵(测量矩阵): c. T$ R* \) O
% A = get_A_fourier(n, m);
1 p) |+ h% W! e$ P% q3 e- S2 x- @2 b/ P2 c
y = Phi * f;                      % 通过感知矩阵获得测量值9 d( [3 y: M- d7 e! I5 v
                                  % Psi 将信号变换到稀疏域3 e7 m" J( Z( N* H: m# P
if freq_sparse                    % 信号频域可以稀疏表示
, `. ^0 r  C- K/ y: h    Psi = inv(fft(eye(n,n)));     % 傅里叶正变换,频域稀疏正交基(稀疏表示矩阵)
$ b2 K4 V/ P4 ~3 G$ C/ ]else                              % 信号时域可以稀疏表示
& i" G8 A: @  J" u2 U    Psi = eye(n, n);              % 时域稀疏正交基
+ V! X6 x' S- ^* @end  O( d0 D9 p: b7 t9 D1 G
A = Phi * Psi;                    % 恢复矩阵 A = Phi * Psi2 d! ~( z9 z# u- J

6 C& H0 [) `+ ^. {8 K5 T& T%% 优化方法1:使用CVX工具进行凸优化  E( H9 j- O: J( \  R2 s) U+ _5 ]
addpath('../../cvx-w64/cvx/');
9 V3 \; D$ o& ^! k% B- [% Pcvx_startup;                            % 设置cvx环境
  B- D8 a( W- H. M. b4 D$ kcvx_begin2 w# H5 i+ d3 v: F2 @- i! ^
    variable xp(n) complex;               % 如果xp是复数,则添加complex,否则不加
3 n3 \# S& o, t. ]# B8 x5 Y0 w+ O0 K, F    minimize (norm(xp, 1));( a- T3 e' ^) d# l4 A
    subject to
9 ^- A( n7 [* k# V, x1 ?# E        A * xp == y;      7 h5 W" y. \" Z" `4 n* e$ Y
cvx_end
4 B3 m9 U& q5 M, n
! s( B, x% t  j7 E1 B%% 优化方法2:使用spams工具箱进行优化
% y! J; }) X0 P5 u% addpath('spams-matlab/build');; k7 a1 R6 p: Q4 F% ?% s. H
% param.L = 100;7 E7 T5 T" w' ?. |5 f! q; O. j
% param.eps = 0.001;
: ~% V. P: l2 A+ W, H5 W. g% param.numThreads = -1;/ M$ a' s6 N6 o/ z
% ) D# m9 }6 g% H  k* x* f  u) Z2 N5 o
% A=A./repmat(sqrt(sum(A.^2)),[size(A,1) 1]);
$ c! h0 M9 s8 H9 A% xp = mexOMP(y, A, param);       % 正交匹配追踪法Orthogonal Matching Pursuit* c4 h1 `! E5 O1 m
* i! g, N- v1 i) }/ b, x
%% 对比原信号和1 V: I1 i6 M( N1 Z
if freq_sparse
& D. \4 v$ b% ]0 D1 t! I; @    xp = real(ifft(full(xp)));           % 傅里叶逆变换9 e8 B' Z0 o7 U, M; S" q1 W
else
( ~7 G0 ]2 U) R# R, ]$ I
+ m  M7 }$ h' k1 W! t3 {  Lend
) c+ o0 D1 b/ Q% g" M: vplot(f+noise);
8 X+ u" a' t  D" M, v9 Y' F0 zhold on# [6 x4 ^9 }  v
plot(real(xp), 'r.');
/ n2 K" C/ Z; i% O; `legend('Original', 'Recovered');
7 H! y; d: Q0 L/ ^8 A2 X6 V3 u& Z- G, x% C4 ^2 l  U# m% z4 K- e
: ^  R& y! C# F$ W6 R" ~' Y6 ~

( g" P9 _3 Z( Y + [1 N% h$ X) q6 ]: t. X9 ?
" I6 Z7 q+ Y- d, U) c
在恢复时域稀疏信号时,所使用的测量矩阵Phi为初始化的随机阵,因为本身时域就稀疏,而算法直接在时域进行恢复,所以稀疏表达矩阵就是单位阵Psi=E。上面的时域稀疏恢复结果显得没有误差是因为没有给原始信号添加噪声。
& r; d! e7 G; I1 l6 w; n) R7 l% n- J; N. D* }
2. 设程序中的freq_sparse = 1时,观察频域稀疏信号的恢复结果为
9 _  g9 m* A5 Z5 h
7 O* j) O) p& z , V( E% C0 l1 I: U& ?) B" Z

) z. X0 q) D% s* @" i( k6 ?在恢复时域稀疏信号时,所使用的测量矩阵Phi为初始化的随机阵,因为信号只在频域稀疏,所以稀疏变换矩阵为傅里叶变换基,所以稀疏表达矩阵就是Psi = inv(fft(eye(n,n)))。同理,上面的频域稀疏恢复结果显得没有误差是因为没有给原始信号添加噪声。
# D# r  k2 m( r: M( R6 y: x& ^- O/ C
3. 上面都是没有添加噪声的算法结果,我们给信号添加一些噪声,以时域信号为例,
6 ]6 s( j9 w$ Z, J2 i' q$ e  C  n4 f; L
if freq_sparse* b9 ?% d3 f" g6 x$ ]2 T
    t = [0: n-1]';
8 E; X) v. V( v; Y' X4 ], q/ L    f = cos(2*pi/256*t) + sin(2*pi/128*t);   % 产生频域稀疏的信号7 |; v; @  d0 g2 W% A( C9 \7 ^
else
# A" k& b  U. `& E: o    tmp = randperm(n);    * ^* N* p: F# y3 ?' X
    f = zeros(n,1);3 h/ d9 w6 o, G: N$ c
    f(tmp(1:s)) = randn(s,1);     % 产生时域稀疏的信号
% x9 u, u- V7 v# P5 c4 c7 x  bend
9 ~" M" `& M0 N, G# n( B8 T
+ \5 i9 z9 M: c2 g) Nnoise = random('norm', 0, 0.01, [n 1]);
8 Y* O" Q9 p5 Zf = f + noise;                    % 添加噪声* t9 _/ \: i: d* S$ _2 E6 Y( a  S

4 I) m7 `0 `0 p+ C& |: c" Z%% Remaining code not changed. D) W& O; }/ Q3 y( V- X
# p# C1 F7 i- K! \- w2 F
  e6 \8 `7 {4 f; t4 b# ?
从下图的恢复结果看,已经能看到一些恢复误差了,; z" k8 \2 C" e6 j

9 X" ~/ |& c( i' z6 H 9 {& `. l1 E. Z" e

. r$ h) |. K/ v8 {" a0 [
$ W8 \* }" x; X$ a! D
作者: ExxNEN    时间: 2020-1-14 19:54
利用MATLAB实现白话压缩感知




欢迎光临 EDA365电子论坛网 (https://bbs.eda365.com/) Powered by Discuz! X3.2