郭俊鋒，何 健

(蘭州理工大學(xué)機(jī)電工程學(xué)院，甘肅 蘭州 730050)

隨著現(xiàn)代工業(yè)技術(shù)的快速發(fā)展，機(jī)械設(shè)備振動信號的頻帶越來越寬，對日益大型化、集成化、智能化的機(jī)械設(shè)備采用奈斯特定理進(jìn)行信號采集時(shí)將會產(chǎn)生大量的數(shù)據(jù)，特別是在對遠(yuǎn)程設(shè)備進(jìn)行實(shí)時(shí)監(jiān)測時(shí)，這些數(shù)據(jù)的實(shí)時(shí)傳輸與同步存儲已經(jīng)成為亟待解決的工程技術(shù)問題[1]。

近些年提出的壓縮感知理論[2-6]較好地解決了該問題。壓縮感知理論利用欠采樣技術(shù)，對稀疏信號或可壓縮信號使用遠(yuǎn)低于奈奎斯特采樣定理的采樣頻率進(jìn)行數(shù)據(jù)采集，再通過有效的優(yōu)化算法對原始信號進(jìn)行大概率重構(gòu)。稀疏表示是壓縮感知的先決條件，直接決定壓縮重構(gòu)性能的優(yōu)劣。字典學(xué)習(xí)算法對信號本身進(jìn)行訓(xùn)練，構(gòu)造的過完備學(xué)習(xí)字典對結(jié)構(gòu)復(fù)雜的信號有較好的稀疏表示能力。文獻(xiàn)[7]采用K-SVD(K-singular value decomposition)學(xué)習(xí)字典對振動信號進(jìn)行稀疏表示，有效提升振動信號壓縮重構(gòu)性能；文獻(xiàn)[8]采用SGK(sequential generalization of K-means)學(xué)習(xí)字典代替K-SVD學(xué)習(xí)字典對圖像進(jìn)行稀疏表示，雖然其稀疏表示能力相近，但有效提升了學(xué)習(xí)字典的學(xué)習(xí)速率。SGK算法在稀疏編碼階段采用正交匹配追蹤(OMP)算法對訓(xùn)練樣本原子逐列進(jìn)行稀疏編碼，字典更新階段采用最小二乘法對字典原子進(jìn)行逐列更新。相較K-SVD字典學(xué)習(xí)算法，SGK算法的字典更新方式提升了字典學(xué)習(xí)算法的速率。

本文為進(jìn)一步提升SGK算法的學(xué)習(xí)速率，從稀疏編碼訓(xùn)練樣本原子數(shù)入手，對多列訓(xùn)練樣本原子采用同步正交匹配追蹤法(simultaneous orthogonal matching pursuit,SOMP)[9]進(jìn)行同步稀疏編碼，在保證振動信號壓縮重構(gòu)性能的前提下，有效提升字典訓(xùn)練速率。

1 壓縮感知基本理論

壓縮感知(compressive sensing,CS)理論提出，若一個(gè)信號是稀疏的或在某種稀疏域下具有近似稀疏性，只需用一個(gè)與稀疏域不相關(guān)的測量矩陣將該信號從高維空間投向低維空間，得到一個(gè)遠(yuǎn)小于該信號長度的測量信號，經(jīng)過有效的計(jì)算就可以高概率地精確重構(gòu)原始信號。壓縮感知的前提是Rn空間的信號f是可稀疏的，其數(shù)學(xué)表示形式為：

f=Ψα

(1)

式中：α為稀疏系數(shù)；Ψ為正交基矩陣。此時(shí)α=ΨTf，因此滿足該條件的α中只存在k(k?n)個(gè)非零元。依據(jù)壓縮感知理論，對信號f壓縮采樣可得觀測矢量y：

y=Φf

(2)

式中：Φ為m×n的測量矩陣且與Ψ不相關(guān)，m?n。僅當(dāng)α是k稀疏且k

min‖α‖0s.t.y=Φf=ΦΨα

(3)

正交基矩陣Ψ對結(jié)構(gòu)復(fù)雜信號稀疏表示性能較差，而用字典學(xué)習(xí)算法對信號本身特征進(jìn)行訓(xùn)練的過完備字典D對信號有較好的稀疏表示性能。壓縮感知理論提出[10]，若稀疏變化域?yàn)镈∈Rn×K，可將公式(2)變?yōu)閒=Dα,α∈θ且θ={α:‖α‖0≤k}，于是可將式(3)變?yōu)?/p>

min‖α‖0s.t.y=Φf=ΦDα=Tα

(4)

式中：T為感知矩陣。

2 機(jī)械振動信號稀疏表示字典學(xué)習(xí)快速算法的實(shí)現(xiàn)

2.1 傳統(tǒng)字典學(xué)習(xí)算法模型

K-SVD、SGK字典學(xué)習(xí)算法的核心思想都源于自聚類算法，字典學(xué)習(xí)算法由初始化字典的構(gòu)造、稀疏編碼以及字典更新3個(gè)部分組成，求解目標(biāo)函數(shù)通常被表示為：

(5)

K-SVD與SGK算法選擇固定過完備字典(例如過完備余弦字典)或初始化部分訓(xùn)練樣本原子用以構(gòu)造初始化字典D1，在稀疏編碼階段采用OMP算法在已知初始字典D1上對訓(xùn)練樣本原子逐列稀疏編碼求解稀疏系數(shù)矩陣A，在字典更新階段固定稀疏系數(shù)矩陣A對字典原子逐列更新，對過完備字典以及稀疏系數(shù)矩陣進(jìn)行反復(fù)交替優(yōu)化，直至達(dá)到停止條件后得到最優(yōu)過完備學(xué)習(xí)字典D。

2.2 字典學(xué)習(xí)快速算法

1)隨機(jī)選擇部分訓(xùn)練樣本原子構(gòu)造初始字典D1并且確定字典學(xué)習(xí)算法參數(shù)，其中D1∈Rn×K。

2)確定訓(xùn)練樣本X中同步稀疏編碼樣本原子個(gè)數(shù)m1，并對X進(jìn)行分塊處理得到若干個(gè)分塊矩陣Sg，具體形式如下：

X=(x1,x2,…,xN)=(S1,S2,…,SN1)

(6)

式中：X∈Rn×N；xj∈Rn；Sg∈Rn×m1,N1=N/m1，其中N為信號長度即信號離散向量采集的個(gè)數(shù)。

3)采用SOMP算法在D1上對Sg進(jìn)行同步稀疏編碼，求取稀疏系數(shù)矩陣A。

SOMP算法是OMP算法的拓展，每次對多列樣本原子Sg在D1上進(jìn)行同步稀疏編碼，其具體步驟如下。

步驟1，設(shè)置初始?xì)埐顁1=Sg，矩陣Λ=?,?為空集。

步驟2，固定已知字典D1，對多列樣本原子Sg進(jìn)行同步稀疏編碼，每次迭代根據(jù)式(7)選擇最佳字典原子dt。

(7)

步驟3，每次迭代選擇的最佳字典原子dt存入Λ中，再根據(jù)式(8)求解Sg的稀疏系數(shù)Ag：

Ag=(ΛTΛ)-1ΛTSg

(8)

步驟4，根據(jù)公式(9)對殘差r1進(jìn)行更新：

r1=Sg-ΛAg

(9)

重復(fù)以上步驟直至達(dá)到停止條件，求得Sg的稀疏系數(shù)Ag。

4)采用以上方式對訓(xùn)練樣本X進(jìn)行同步稀疏編碼,求得稀疏系數(shù)矩陣A。

5)固定稀疏系數(shù)矩陣A，對初始字典D1的字典原子進(jìn)行更新，其具體步驟如下。

步驟1，對D1的第j列字典原子dj進(jìn)行更新時(shí)，提取稀疏系數(shù)矩陣A中第j行的非零元素列下標(biāo)數(shù)組Wk，并將其第j行的所有非零元素賦值為零。

步驟2，依據(jù)列下標(biāo)數(shù)組Wk提取A以及訓(xùn)練樣本X中對應(yīng)列標(biāo)號的所有列向量分別構(gòu)成矩陣Aj，Xj，并通過式(10)求得殘差矩陣Ej。

Ej=Xj-D1Aj

(10)

(11)

重復(fù)以上步驟直至D1中所有字典原子更新，得到新的字典D1。

6)重復(fù)步驟3～5直至達(dá)到預(yù)定條件，得到最優(yōu)過完備學(xué)習(xí)字典D。

3 試驗(yàn)結(jié)果分析

3.1 重構(gòu)效果評價(jià)

本文在相同壓縮率(compressionrate，CR)CR下，采用相對重構(gòu)誤差σ以及皮爾遜相關(guān)系數(shù)r對振動信號重構(gòu)性能進(jìn)行評價(jià)，其定義如下。

1)壓縮率CR用于衡量振動信號壓縮度，CR越大，信號壓縮率越高。定義為：

(12)

式中：n為原始振動信號f的長度；m為壓縮測量信號y的長度。

2)相對重構(gòu)誤差σ是指振動信號的絕對誤差與原信號的比值，其定義為：

(13)

3)皮爾遜相關(guān)系數(shù)r衡量原始信號與重構(gòu)信號的波形相似度，其定義為：

(14)

本文采用美國西儲大學(xué)軸承數(shù)據(jù)庫的數(shù)據(jù)驗(yàn)證本文算法的有效性。在相同條件下，采用K-SVD、SGK以及本文算法對訓(xùn)練學(xué)習(xí)字典的時(shí)間進(jìn)行對比，并基于3種算法構(gòu)造的學(xué)習(xí)字典對振動信號進(jìn)行壓縮重構(gòu)。

3.2 同步稀疏編碼多列樣本原子數(shù)的確定

選取電機(jī)驅(qū)動端12點(diǎn)鐘方向加速傳感器所采集的OR007@6信號數(shù)據(jù)進(jìn)行試驗(yàn)測試，該信號為周期信號。將信號0～121991采樣點(diǎn)擴(kuò)展至0～731964采樣點(diǎn)，其中0～620000采樣點(diǎn)為字典訓(xùn)練樣本信號，620001～731964采樣點(diǎn)為測試信號。字典原子長度n=512，字典原子個(gè)數(shù)K=800，訓(xùn)練樣本X原子個(gè)數(shù)N0=1 200，待分解信號稀疏表示時(shí)線性組合原子數(shù)L=10、字典訓(xùn)練迭代次數(shù)J=10、壓縮重構(gòu)迭代次數(shù)L1=10。為保證壓縮感知測試信號中至少包含一個(gè)完整振動信號周期，選取以640001起始的1 024個(gè)采樣點(diǎn)為測試信號段。

不同m1下訓(xùn)練過完備學(xué)習(xí)字典所用時(shí)間見表1，從表中可以看出當(dāng)m1為2～5時(shí)，字典訓(xùn)練時(shí)間趨于穩(wěn)定減少，當(dāng)m1=6時(shí)，訓(xùn)練時(shí)間急劇增加，因此確定m1為2～5列訓(xùn)練樣本原子。

表1 不同稀疏編碼原子數(shù)m1下字典訓(xùn)練時(shí)間

采用不同待分解信號稀疏表示時(shí)線性組合原子數(shù)L進(jìn)行學(xué)習(xí)字典訓(xùn)練，學(xué)習(xí)字典訓(xùn)練所需時(shí)間如圖1所示。從圖中可以看出，L值的選取直接影響本文算法訓(xùn)練過完備學(xué)習(xí)字典的時(shí)間。當(dāng)m1=2時(shí)，每次同步稀疏編碼2列訓(xùn)練樣本原子，必包含一個(gè)振動周期，并且學(xué)習(xí)字典訓(xùn)練時(shí)間隨著L值的增加穩(wěn)定增長，由此確定本文算法中同步稀疏編碼多列樣本原子數(shù)m1為2。

圖1 不同m1下基于不同L值的學(xué)習(xí)字典時(shí)間

3.3 本文算法有效性分析

在相同條件下，采用K-SVD、SGK以及本文算法訓(xùn)練過完備學(xué)習(xí)字典，并在壓縮率為60%時(shí)對測試振動信號進(jìn)行壓縮重構(gòu)。從圖2與表2可以看出，K-SVD、SGK、本文算法訓(xùn)練的學(xué)習(xí)字典對振動信號的壓縮重構(gòu)性能幾乎相同，但本文算法訓(xùn)練字典用時(shí)最少，是K-SVD算法的43%，是SGK算法的55%。

圖2 不同學(xué)習(xí)字典算法下振動信號壓縮重構(gòu)效果

表2 不同算法下學(xué)習(xí)字典訓(xùn)練時(shí)間以及壓縮重構(gòu)性能

為驗(yàn)證本文算法在不同學(xué)習(xí)字典迭代次數(shù)下的有效性，采用K-SVD、SGK以及本文算法對以上實(shí)驗(yàn)訓(xùn)練樣本基于不同J值進(jìn)行學(xué)習(xí)字典訓(xùn)練并采用相同的振動信號測量值進(jìn)行壓縮重構(gòu)。從圖3與表3可以看出，隨著J值的增加，本文算法訓(xùn)練字典時(shí)間相較于K-SVD以及SGK算法明顯減少，并且采用3種算法訓(xùn)練的過完備字典對振動信號的壓縮重構(gòu)性能幾乎相同，進(jìn)一步驗(yàn)證了本文算法在不同的J值下仍具備有效性。

圖3 不同迭代次數(shù)J下的學(xué)習(xí)字典訓(xùn)練時(shí)間

表3 振動信號基于不同迭代次數(shù)下的壓縮重構(gòu)性能

3.4 本文算法可靠性分析

為驗(yàn)證本文算法的可靠性，對10種不同故障類別數(shù)據(jù)分別采用K-SVD算法、SGK算法以及本文算法進(jìn)行過完備學(xué)習(xí)字典訓(xùn)練，并在壓縮率為60%時(shí)對測試部分振動信號進(jìn)行壓縮重構(gòu)，試驗(yàn)結(jié)果見表4。從表4中看出，在不同類別故障數(shù)據(jù)下基于本文算法、K-SVD算法以及SGK算法構(gòu)造的過完備字典對不同故障振動信號的壓縮重構(gòu)性能相近，但本文算法訓(xùn)練學(xué)習(xí)字典用時(shí)較短。該實(shí)驗(yàn)有效證明本文算法構(gòu)造的過完備學(xué)習(xí)字典可以有效應(yīng)用于不同振動信號的壓縮重構(gòu)。

表4 不同故障類別下不同字典學(xué)習(xí)算法的字典訓(xùn)練時(shí)間以及壓縮重構(gòu)性能

4 結(jié)束語

本文針對機(jī)械振動信號在壓縮采集過程中K-SVD、SGK字典學(xué)習(xí)算法訓(xùn)練過完備學(xué)習(xí)字典時(shí)間過長的問題，提出一種字典學(xué)習(xí)快速算法，對SGK字典學(xué)習(xí)算法的稀疏編碼部分采用同步正交匹配追蹤法同步稀疏編碼多列樣本原子。利用美國西儲大學(xué)軸承數(shù)據(jù)庫數(shù)據(jù)證明了在相同條件下本文算法與SGK算法、K-SVD算法所構(gòu)造的過完備字典對振動信號的壓縮重構(gòu)壓性能幾乎相同，但本文算法訓(xùn)練過完備學(xué)習(xí)字典的時(shí)間明顯更短。