吳春志， 馮輔周， 吳守軍， 陳湯， 江鵬程

(陸軍裝甲兵學(xué)院 車輛工程系， 北京 100072)

0 引言

行星傳動機(jī)構(gòu)具有體積小、重量輕、傳動比大、承載能力強(qiáng)、傳動效率高等諸多優(yōu)點(diǎn)，廣泛應(yīng)用于直升機(jī)主減速器、艦船、自行火炮、飛機(jī)、裝甲車輛等軍用裝備中[1-2]。在實(shí)際運(yùn)行過程中，行星傳動機(jī)構(gòu)承受動態(tài)重載，運(yùn)行環(huán)境復(fù)雜多變，導(dǎo)致太陽齒輪，行星齒輪，齒圈等關(guān)鍵部件容易發(fā)生故障。這些故障如果不能被及時排除，將會導(dǎo)致嚴(yán)重的安全事故[3]。

設(shè)計(jì)故障診斷分類模型時，各個故障狀態(tài)的樣本數(shù)量通常保持一致，而在真正的應(yīng)用環(huán)境中，類別分布并不均衡，故障數(shù)據(jù)往往相對較少。當(dāng)數(shù)據(jù)集不均衡時，傳統(tǒng)的分類器通常偏向大多數(shù)類，因此無法正確地對少數(shù)類進(jìn)行分類，當(dāng)訓(xùn)練數(shù)據(jù)高度不平衡時，甚至可能無法檢測到少數(shù)類。處理樣本不均衡問題有3種主要方法：第1種是修改或者創(chuàng)建一種新的算法，加強(qiáng)對少數(shù)群體的學(xué)習(xí)；第2種方法是在數(shù)據(jù)或算法級別使用代價敏感算法，以便最大限度地降低誤差；第3種也是更通用的方法是通過欠采樣、過采樣或混合方法重新平衡類分布，從而在數(shù)據(jù)級別進(jìn)行修改[4]。在少量故障樣本下的行星齒輪箱故障診斷中，研究人員通常采用的是第1種方法。楊宇等[5]提出了一種基于增量半監(jiān)督(VPMCD)的故障診斷方法，該方法可以在小診斷樣本的情況下對齒輪故障模式進(jìn)行分類。Yang等[6]應(yīng)用經(jīng)驗(yàn)?zāi)B(tài)分解算法提取故障特征，并采用支持向量機(jī)(SVM)對小樣本情況下的齒輪故障進(jìn)行分類。在故障樣本少的條件下，SVM和灰色模型常用作故障診斷算法[7-8]。然而這些方法在對特定對象進(jìn)行故障特征提取的同時還要在模式識別算法上做相適應(yīng)的改變，無法滿足更多更復(fù)雜情況下的診斷需求。

深度學(xué)習(xí)以其無需人工經(jīng)驗(yàn)選取可自動學(xué)習(xí)特征的特點(diǎn)逐漸成為解決此類問題的最佳選擇。2014年，Goodfellow等[9]提出了生成式對抗網(wǎng)絡(luò)(GAN)，它可以生成高分辨率的真實(shí)感圖像，在此基礎(chǔ)上又逐漸演變出深度卷積生成對抗網(wǎng)絡(luò)(DCGAN)、Wassertein 生成對抗網(wǎng)絡(luò)(WGAN)、條件生成對抗網(wǎng)絡(luò)(CGAN)等百余種變體算法。通過這些算法可以擴(kuò)展故障集，實(shí)現(xiàn)樣本均衡。

本文重點(diǎn)研究過采樣技術(shù)用以解決樣本不均衡下行星傳動機(jī)構(gòu)的故障診斷問題。使用改進(jìn)的Wasserstein生成式對抗網(wǎng)絡(luò)構(gòu)造樣本生成模型，與GAN相比確保了訓(xùn)練的穩(wěn)定性和生成樣本的多樣性，再通過卷積神經(jīng)網(wǎng)絡(luò)(CNN)進(jìn)行故障分類識別。為了評估WGAN算法，在加州大學(xué)歐文分校(UCI)人工數(shù)據(jù)集上對所提方法和其他過采樣方法進(jìn)行對比分析，然后在某型坦克行星變速箱故障試驗(yàn)臺上進(jìn)行驗(yàn)證。

1 深度學(xué)習(xí)算法

1.1 卷積神經(jīng)網(wǎng)絡(luò)

CNN是由哺乳動物視覺皮層細(xì)胞感受野啟發(fā)而建立的一種前饋神經(jīng)網(wǎng)絡(luò)，現(xiàn)在已經(jīng)是圖片和視頻識別領(lǐng)域應(yīng)用最為廣泛的方法。一個傳統(tǒng)的CNN模型如圖1所示，由一個輸入層，兩組交替出現(xiàn)的卷積層和池化層，以及全連接層組成。卷積層中每一個特征圖都對應(yīng)一個卷積核，這些卷積核通過一組權(quán)重來卷積前一層的輸入并組成一組特征輸出，成為下一層的輸入。與傳統(tǒng)濾波器需要人工設(shè)定參數(shù)不同，CNN的權(quán)值和偏置通過反向傳播(BP)算法進(jìn)行訓(xùn)練。池化層主要是為了實(shí)現(xiàn)兩個目的：一是大幅減少輸入卷積層的空間維度，使權(quán)重參數(shù)減少75%，從而降低計(jì)算成本；二是控制過擬合，使測試集準(zhǔn)確度更接近訓(xùn)練集準(zhǔn)確度。主流的池化方式有最大值池化、平均池化、L2-norm池化。經(jīng)過兩組卷積層和池化層后，會接一個全連接層。與傳統(tǒng)的神經(jīng)網(wǎng)絡(luò)類似，全連接層可以應(yīng)用到不同的分類模型。全連接層后接一個隱藏層，最后由Softmax回歸層完成分類。CNN以其出色的特征自動提取并分類的能力受到廣大研究人員的青睞，在機(jī)械故障診斷領(lǐng)域得到了廣泛的應(yīng)用[10-11]，本文所用模式識別方法統(tǒng)一采用CNN，與傳統(tǒng)方法對比可見文獻(xiàn)[10-11]。

1.2 GAN和WGAN網(wǎng)絡(luò)

圖1 典型卷積神經(jīng)網(wǎng)絡(luò)結(jié)構(gòu)圖Fig.1 Typical convolutional neural network structure

在GAN誕生之前，非監(jiān)督學(xué)習(xí)的基本思想是根據(jù)數(shù)據(jù)集估計(jì)樣本的分布p(x)，然后對p(x)進(jìn)行采樣，生成與樣本分布類似的新樣本，這種方法難以解決高維樣本的生成問題。GAN作為非監(jiān)督深度學(xué)習(xí)的代表，創(chuàng)造性地提出了生成器G和判別器D. 生成器G學(xué)習(xí)樣本x的分布pd，并從噪聲數(shù)據(jù)z中生成新樣本G(z)得到生成樣本分布pg，G(z)通過訓(xùn)練自己，希望以假亂真，讓判別器D判別不出到底哪個是真、哪個是假。而判別器D獨(dú)自訓(xùn)練自己，希望能分辨出真實(shí)的數(shù)據(jù)分布和生成器G(z)給的數(shù)據(jù)分布。二者進(jìn)行博弈從而優(yōu)化模型，GAN的目標(biāo)函數(shù)loss可以描述為

(1)

式中：Ex～pd和Ex～pg是期望值；Ex～pd[lgD(x)]表示樣本x分布pd的對數(shù)損失函數(shù)，最大化這一項(xiàng)意味著D(x)→1；Ex～pg[lg(1-D(G(z)))]則是根據(jù)從噪聲數(shù)據(jù)z生成樣本分布pg構(gòu)建的對數(shù)損失函數(shù)，最小化這一項(xiàng)意味著D(G(z))→0.

圖2 WGAN-CNN結(jié)構(gòu)圖Fig.2 Structure of WGAN-CNN

在訓(xùn)練GAN模型時，先固定G再訓(xùn)練D，使得D的準(zhǔn)確率達(dá)到最大；然后再固定D，訓(xùn)練G，使得D的準(zhǔn)確率達(dá)到最小。當(dāng)且僅當(dāng)pd=pg時達(dá)到全局最優(yōu)解，也就是從噪聲中生成的數(shù)據(jù)分布和真實(shí)樣本分布基本達(dá)到一致。

為了解決梯度消失的問題，Arjovsky等提出了WGAN，引入了Wasserstein距離，由于其相對Kullback-Leibler散度與Jensen-Shannon散度具有優(yōu)越的平滑特性，理論上可以解決梯度消失的問題。在WGAN中最后的目標(biāo)函數(shù)loss可以描述為

(2)

1.3 WGAN-CNN

將圖1中所示模型優(yōu)化為圖2右側(cè)部分所示的一維CNN模型。優(yōu)化后的模型由輸入層、CNN層組(由卷積層和池化層組成)、全連接層、輸出層組成。其中CNN層組可以根據(jù)所需分類數(shù)據(jù)的復(fù)雜度增加或減少，本模型使用2個CNN層組。第1個CNN層組中卷積核數(shù)量為N1，第2個CNN層組中卷積核數(shù)量為N2，相應(yīng)卷積核的權(quán)值向量和偏置值分別為W和b，激活函數(shù)采用修正線性單元ReLU.

(3)

(4)

(5)

第2個CNN層的輸出為

(6)

CNN訓(xùn)練的目的是為了使最終的損失達(dá)到最小，模型采用的損失函數(shù)為交叉熵，優(yōu)化器選用AdamOptimizer. 對于不均衡樣本的分類問題，首先采用WGAN對少數(shù)類樣本進(jìn)行生成(見圖2左側(cè)部分)，再結(jié)合CNN進(jìn)行分類識別。

2 UCI數(shù)據(jù)集對比分析

2.1 過采樣算法

為了測試WGAN在處理樣本不均衡問題中的效果，本節(jié)將使用8個經(jīng)典的UCI人工數(shù)據(jù)集，人為減少某一類的樣本數(shù)量構(gòu)成不同不均衡比例(IR)的數(shù)據(jù)集，并與其他傳統(tǒng)的過采樣方法進(jìn)行對比，使用CNN方法來分類識別。通過計(jì)算調(diào)和平均值Fe、幾何平均值Gm以及AUC作為過采樣算法的評價指標(biāo)。

過采樣為的是增加少數(shù)類樣本，最直接的方法是對樣本進(jìn)行隨機(jī)復(fù)制，但重復(fù)的樣本會造成過擬合，為了解決這個問題，一種隨機(jī)生成無重復(fù)少數(shù)類樣本的樣本生成技術(shù)(SMOTE)方法[12]應(yīng)運(yùn)而生，沿著連接少數(shù)類樣本的線段生成合成樣本，數(shù)據(jù)生成過程的直接方法是使用捕獲實(shí)際數(shù)據(jù)分布的生成模型。接著一些SMOTE的變體方法也被提出來用以改進(jìn)SMOTE的過生成問題。Bunkhumpornpat等[13]提出Safe-level SMOTE方法生成的少數(shù)類樣本在“安全層”以內(nèi)，避免了樣本重復(fù)。Han等[14]提出Borderline-SMOTE方法識別分類邊界的少數(shù)類樣本并進(jìn)行過采樣，提高了邊界樣本的識別率。對于類內(nèi)有多個簇的樣本，Cieslak等[15]提出了Cluster-SMOTE，首先使用k-means算法將少數(shù)類聚類成m個子聚類，然后將SMOTE方法應(yīng)用于每個子聚類。

2.2 多分類評價指標(biāo)

如表1混淆矩陣所示，在二分類問題中，測試精確度Precision、召回率Recall、Fe、Gm分別為

(7)

(8)

(9)

(10)

式中：FN表示被判定為負(fù)樣本，但事實(shí)上是正樣本；FP表示被判定為正樣本，但事實(shí)上是負(fù)樣本；TN表示被判定為負(fù)樣本，事實(shí)上也是負(fù)樣本；TP表示被判定為正樣本，事實(shí)上也是正樣本；Precision表示的是預(yù)測為正的樣本中有多少是真正的樣本；Recall表示樣本中的正例有多少被預(yù)測正確了；β為調(diào)和參數(shù)，一般取1.Gm考慮了分類的正確率，如果分類器忽略少數(shù)類并且有利于多數(shù)類，則分類器將獲得較低的Gm；AUC是受試者工作特征(ROC)圖下的區(qū)域，對二類的分布不敏感，這使得它適合作為不平衡問題的分類性能指標(biāo)。

表1 混淆矩陣表

通過繪制真陽性率(TPR)與假陽性率(FPR)之間的關(guān)系來獲得ROC圖，定義為

(11)

式中：Np是少數(shù)類數(shù)量；NN是多數(shù)類數(shù)量。

對于多分類情況，可以將多分類問題拆分成多個二分類問題，將所得到的多個二分類指標(biāo)Fe、Gm以及AUC求幾何平均值后得到最終的分類評價指標(biāo)Fa、Ga、AUCa.

2.3 算法對比

從經(jīng)典的UCI人工數(shù)據(jù)集中選出8個常用數(shù)據(jù)集，如表2所示。其中數(shù)據(jù)集1、6、7是多分類問題，其余是二分類問題，數(shù)據(jù)集1～3、6、7中少數(shù)類是由原樣本中隨機(jī)選出產(chǎn)生的，造成從1∶1.38～1∶6不同的IR. 然后采用隨機(jī)復(fù)制(Random)、SMOTE、Borderline-SMOTE、Cluster SMOTE、Safe_level SMOTE、WGAN 6種過采樣方法對少數(shù)類進(jìn)行過采樣，最后經(jīng)過CNN分類并計(jì)算Fa、Ga、AUCa. 對每種數(shù)據(jù)集所得到的每類評價指標(biāo)進(jìn)行1～6的排名，求得排名的平均值表如表3所示。從表3中可以看出，WGAN相對于其他5種過采樣方法，在UCI人工數(shù)據(jù)集上的分類表現(xiàn)最好。雖然WGAN模型在訓(xùn)練階段學(xué)習(xí)樣本構(gòu)造生成模型時用時相對較長，一旦模型訓(xùn)練完畢，則可以源源不斷地帶來所需樣本，此時的樣本生成時間可忽略不計(jì)。

表2 所選UCI數(shù)據(jù)集的詳細(xì)信息表

表3 過采樣方法評價指標(biāo)排名平均值表

3 行星齒輪箱故障診斷

3.1 實(shí)驗(yàn)介紹

某型坦克行星變速箱故障模擬試驗(yàn)臺實(shí)物圖如圖3所示。試驗(yàn)臺機(jī)械傳動裝置由變頻電動機(jī)、變頻器柜、錐齒輪傳動箱、行星變速箱、轉(zhuǎn)速轉(zhuǎn)矩儀、加載發(fā)電機(jī)等組成。電機(jī)輸出經(jīng)變頻柜調(diào)節(jié)后的轉(zhuǎn)速及扭矩，動力經(jīng)錐齒輪傳動箱輸入到行星變速箱內(nèi)。錐齒輪傳動箱的作用是改變力的傳遞方向，使試驗(yàn)臺整體合理布置。行星變速箱兩側(cè)的輸出端與轉(zhuǎn)速轉(zhuǎn)矩儀相連，使用發(fā)電機(jī)對其進(jìn)行加載，液壓站負(fù)責(zé)對行星變速箱提供潤滑油壓及換擋壓力。實(shí)驗(yàn)設(shè)置兩種故障狀態(tài)，分別是K3排的太陽輪斷齒故障(齒數(shù)為30的太陽輪故障，簡稱Z30)以及與太陽輪嚙合的行星輪斷齒故障(齒數(shù)為15的行星輪故障，簡稱Z15)，斷齒故障設(shè)置如圖4所示。

圖3 行星變速箱故障模擬試驗(yàn)臺Fig.3 Fault simulation test rig for planetary gearbox

圖4 齒輪故障設(shè)置Fig.4 Gear failure setting

測試系統(tǒng)裝置包括主控平臺、32通道信號采集儀、振動傳感器和相應(yīng)的數(shù)據(jù)線。通過主控平臺控制行星變速箱故障模擬試驗(yàn)臺的啟停、轉(zhuǎn)速調(diào)節(jié)、加載扭矩等工程，采用相應(yīng)的數(shù)據(jù)采集軟件實(shí)現(xiàn)對信號采集儀的通道設(shè)置。采樣頻率為20 kHz，采樣時間30 s，選用的四檔、工況為轉(zhuǎn)速1 500 r/min、負(fù)載900 N·m的數(shù)據(jù)為測試數(shù)據(jù)，每種狀態(tài)的時域圖如圖5所示。由于時域信號采樣頻率高，同一周期內(nèi)點(diǎn)數(shù)較多且規(guī)律性不明顯，因此對截取的多段時域信號樣本做傅里葉變換，用頻譜信號作為訓(xùn)練集樣本。文獻(xiàn)[10]指出頻譜信號相較于振動信號直接作訓(xùn)練樣本，在應(yīng)用CNN時能得到更高的準(zhǔn)確率。

圖5 振動信號時域圖Fig.5 Vibration signals

3.2 數(shù)據(jù)分析

為了直觀地了解在不同IR下CNN對樣本的分類情況，分別取Z15訓(xùn)練集數(shù)量為30、120、600組進(jìn)行CNN訓(xùn)練，模型結(jié)構(gòu)參數(shù)如表4所示。將所有測試樣本經(jīng)過CNN模型的中間層時得到的特征向量作t分布領(lǐng)域嵌入算法(t-SNE)計(jì)算，降維后得到3個維度的特征，分別為特征值1、特征值2、特征值3，展示如圖6所示。從圖6中可以直觀地看到，當(dāng)IR為1∶20時，正常狀況和Z15斷齒基本上沒有分開，當(dāng)IR逐漸降低時，3種狀況被很好地分開。

表4 訓(xùn)練集數(shù)量及CNN結(jié)構(gòu)參數(shù)

圖6 不同IR分類散點(diǎn)圖Fig.6 Different IR classification scatter plots

由于在臺架試驗(yàn)中可以得到足夠多的樣本，為了模擬實(shí)際操作中某個故障樣本稀少的情況，將Z15斷齒故障樣本取30組，其余樣本取600組，而測試樣本都為400組。以30組樣本作為WGAN的訓(xùn)練集(結(jié)構(gòu)參數(shù)見表4)，訓(xùn)練生成模型，生成的樣本如圖7所示。從圖7中可以看出，生成的頻譜圖對于頻率幅值較低的部分生成得不夠細(xì)致，但是對于頻譜圖幅值較高的部分還原得較好，在整體上基本上反應(yīng)了原始樣本的特征，可以作為原始樣本的擴(kuò)展樣本。

圖7 WGAN生成振動信號頻譜和原信號頻譜對比圖Fig.7 Spectra of WGAN vibration signal and original signal

圖8 不同IR下評價指標(biāo)變化曲線 Fig.8 Evaluation indexes of different IRs

為了證明WGAN生成頻譜樣本的有效性，全部以生成樣本構(gòu)造訓(xùn)練集，以原始樣本作為測試集進(jìn)行實(shí)驗(yàn)。不同過采樣方法對Z15的單個類別統(tǒng)計(jì)如表5所示。由于同一狀態(tài)不同樣本的幅值譜較為接近，因此采用Random可以得到相對較好的結(jié)果，而WGAN表現(xiàn)最好。選取Z15斷齒故障作為少數(shù)類，在不同IR下，分別計(jì)算Z15單個類別的Fe、Gm、AUC、Precision、Recall以及整體的Fa、Ga、AUCa、Precision、Recall，如圖8所示。隨著IR降低，除Precision外，Z15的評價指標(biāo)逐漸接近平均值，并在在IR達(dá)到1∶2以上時，評價指標(biāo)接近1，這說明故障樣本的不均衡嚴(yán)重影響了分類器的判斷，也說明了WGAN生成的樣本的有效性，在IR等于1時，分類各個指標(biāo)達(dá)到了100%. 同時通過對比Z15的Precision和Recall隨IR的變化情況可知，對于不均衡樣本的分類評價，單純求Precision容易陷入誤區(qū)，應(yīng)考慮其他評價指標(biāo)做綜合評判。

表5 過采樣方法對Z15測試結(jié)果統(tǒng)計(jì)表

4 結(jié)論

本文針對在實(shí)際運(yùn)行過程中行星變速箱故障樣本不均衡的問題，提出了使用WGAN-CNN算法作為不平衡數(shù)據(jù)的故障診斷分類方法，同時將二分類評價指標(biāo)擴(kuò)展到多分類問題中。經(jīng)過8個經(jīng)典的UCI人工數(shù)據(jù)集以及樣本不均衡下行星變速箱齒輪故障實(shí)驗(yàn)，得到如下結(jié)論：

1) 在8個人工數(shù)據(jù)集上采用Random、SMOTE、Borderline-SMOTE、Cluster SMOTE、Safe_level SMOTE 5種常用的過采樣算法和WGAN算法進(jìn)行比較，WGAN算法在Fa、Ga、AUCa上的平均排名最高、效果最好。

2) 設(shè)置Z15故障頻譜信號為少數(shù)類樣本，采用WGAN-CNN模型對行星變速箱進(jìn)行故障診斷。通過對比不同IR下行星變速箱的分類情況以及評價指標(biāo)變化情況，驗(yàn)證了樣本不均衡問題會嚴(yán)重影響模型的分類結(jié)果，同時表明模型生成的頻譜信號基本上反應(yīng)了原始樣本的特征，可以作為原始樣本的擴(kuò)展樣本，解決了故障樣本不均衡問題。