基于多種卷積核特征提取自適應(yīng)融合的滾動(dòng)軸承故障診斷方法

尚東方,申浩,王正

(1.中國(guó)礦業(yè)大學(xué) 安全工程學(xué)院,江蘇 徐州 221116;2.黃河水利職業(yè)技術(shù)學(xué)院,河南 開封 475001;3.上海電機(jī)學(xué)院 電氣學(xué)院,上海 201306)

隨著科技發(fā)展的日新月異,電動(dòng)機(jī)設(shè)備的功能越來越強(qiáng)大,設(shè)施越來越復(fù)雜,作為電動(dòng)機(jī)關(guān)鍵部件的電機(jī)軸承極易損壞,對(duì)其故障診斷的要求也越來越高[1]。目前,對(duì)于電機(jī)軸承故障診斷的主要方法有基于模型的方法和基于數(shù)據(jù)的方法。傳統(tǒng)的機(jī)器學(xué)習(xí)方法在故障診斷領(lǐng)域取得了不錯(cuò)成果,如文獻(xiàn)[2]提出一種基于信息熵的軸承故障特征提取方法,將提取的故障特征向量作為優(yōu)化支持向量機(jī)(SVM)的輸入,使用集成經(jīng)驗(yàn)?zāi)B(tài)分解提取各階分量的信息熵,通過主成分分析(PCA)提取主要信息,利用Grid Search優(yōu)化SVM參數(shù)并對(duì)故障信息進(jìn)行分類,從而快速有效地識(shí)別了不同的軸承故障信息。

隨著AlexNet的問世,深度學(xué)習(xí)方法逐漸進(jìn)入故障診斷領(lǐng)域并成為眾多學(xué)者的研究對(duì)象[3-5],如文獻(xiàn)[6-8]將深度信念網(wǎng)絡(luò)(Deep Belief Network,DBN)以及堆疊自編碼器(Stacked Auto Encoder,SAE)應(yīng)用于故障診斷領(lǐng)域,著重于提取輸入的故障特征,均取得了較好的診斷效果。目前,將卷積神經(jīng)網(wǎng)絡(luò)(Convolutional Neural Networks,CNN)應(yīng)用于軸承故障診斷領(lǐng)域的研究更為廣泛：文獻(xiàn)[9]將多通道原始軸承信號(hào)經(jīng)過短時(shí)傅里葉變換轉(zhuǎn)換為二維時(shí)頻圖像,送入深度殘差網(wǎng)絡(luò)進(jìn)行故障分類,準(zhǔn)確率得到了很大提升;文獻(xiàn)[10]提出了一種具有殘差結(jié)構(gòu)的一維擴(kuò)張卷積網(wǎng)絡(luò)方法,將鋸齒形擴(kuò)張卷積引入神經(jīng)網(wǎng)絡(luò)模型,并在網(wǎng)絡(luò)中添加注意力模塊進(jìn)行特征增強(qiáng),很大程度上提高了故障識(shí)別率。上述方法均是將軸承信號(hào)輸入卷積神經(jīng)網(wǎng)絡(luò)模型提取故障特征從而達(dá)到診斷效果,但實(shí)際工況復(fù)雜,干擾因素較多[11],使用單一卷積核構(gòu)成的卷積神經(jīng)網(wǎng)絡(luò)模型的特征提取能力有限,很大程度上會(huì)堆疊重復(fù)特征,存在較多的冗余信息。

針對(duì)以上問題，本文提出一種基于多卷積核(Multiple Convolution Kernels，MCK)卷積神經(jīng)網(wǎng)絡(luò)的滾動(dòng)軸承故障診斷方法(MCK-CNN)。經(jīng)過預(yù)網(wǎng)絡(luò)特征提取后產(chǎn)生輸出特征圖，再將其分別輸入基于Involution卷積核搭建的NET1和常規(guī)Convolution卷積搭建的NET2進(jìn)行故障特征提取，然后通過CBAM模塊自適應(yīng)融合二者提取的特征，使網(wǎng)絡(luò)更加關(guān)注有判別能力的特征以提高網(wǎng)絡(luò)診斷能力，最后通過Softmax函數(shù)輸出分類結(jié)果。

1 理論背景

1.1 二維卷積

二維卷積的目的是提取圖片中的特征,卷積核相當(dāng)于過濾器,通過在特征圖上的不停移動(dòng)提取輸入特征圖的信息。其卷積過程如圖1所示。

圖1 二維卷積

卷積的運(yùn)算過程可表示為

(1)

1.2 Involution卷積

Involution卷積沒有像Convolution卷積一樣采用固定尺寸的權(quán)重作為可學(xué)習(xí)參數(shù),而是基于特征圖生成對(duì)應(yīng)的Involution核,從而確定卷積核的尺寸與輸入特征矩陣的大小在空間維度上是可以對(duì)齊的[12]。Involution算子可以表示為

(2)

式中：X為輸入圖片的通道數(shù);H為Involution卷積核,H∈RH×W×K×K×G,通用形式為Hi,j=φ(XΨi,j);φ為Involution卷積核生成函數(shù);Ψi,j為坐標(biāo)(i,j)鄰域的集合,表示特征圖上包含Xi,j的某個(gè)鄰域,生成圖如圖2所示。

圖2 Involution卷積

1.3 卷積注意力模塊

卷積注意力模塊(Convolutional Block Attention Module,CBAM) 包含2個(gè)獨(dú)立的子模塊[13],通道注意力模塊(Channel Attention Module,CAM)和空間注意力模塊(Spartial Attention Module,SAM)如圖3所示。

圖3 CBAM結(jié)構(gòu)圖

給定一個(gè)特征圖作為輸入,CBAM的運(yùn)算過程總體分為2個(gè)部分：1)對(duì)輸入按通道進(jìn)行全局最大池化和均值池化,將池化后的2個(gè)一維向量送入全連接層運(yùn)算后相加,生成一維通道注意力并與輸入元素相乘,獲得通道注意力調(diào)整后的特征圖F′;2)將F′按空間進(jìn)行全局最大池化和均值池化,將池化生成的2個(gè)二維向量拼接后進(jìn)行卷積操作,最終生成二維空間注意力并與F′按元素相乘得到輸出F″。

2 基于MCK-CNN的滾動(dòng)軸承故障診斷方法

2.1 MCK-CNN模型

本文采用的神經(jīng)網(wǎng)絡(luò)模型結(jié)合了Involution卷積和Convolution卷積,而且可通過CBAM注意力機(jī)制進(jìn)行自適應(yīng)特征融合,如圖4所示：將通過小波變換產(chǎn)生的時(shí)頻圖裁剪為64*64*3的尺寸輸入神經(jīng)網(wǎng)絡(luò)模型;經(jīng)過NET0中9個(gè)5*5的卷積核提取特征以及擴(kuò)充維度之后產(chǎn)生60*60*9尺寸的輸出,通過最大池化層下采樣壓縮特征并去除冗余信息后得到30*30*9尺寸的輸出特征圖并分別作為NET1和NET2的輸入;在NET1中,Involution卷積根據(jù)輸入特征圖尺寸產(chǎn)生對(duì)應(yīng)的Involution卷積核并進(jìn)行逐個(gè)像素點(diǎn)的特征提取,得到30*30*9尺寸的Involution輸出特征;在NET2中,輸入分別經(jīng)過6個(gè)1*1的卷積核和9個(gè)1*1的卷積核升降維度后產(chǎn)生30*30*9尺寸的Convolution輸出特征;通過CBAM注意力模塊自適應(yīng)分配Involution和Convolution特征所占權(quán)重并進(jìn)行融合,通過Softmax函數(shù)輸出分類結(jié)果。

圖4 神經(jīng)網(wǎng)絡(luò)模型結(jié)構(gòu)圖

2.2 算法流程

基于MCK-CNN的滾動(dòng)軸承故障診斷方法包含3個(gè)模塊,其流程如圖5所示：

圖5 故障診斷流程圖

1)數(shù)據(jù)處理。通過電動(dòng)機(jī)驅(qū)動(dòng)端的傳感器獲取滾動(dòng)軸承一維振動(dòng)信號(hào)并通過小波變換轉(zhuǎn)換為二維時(shí)頻圖,經(jīng)過裁剪、翻轉(zhuǎn)等操作后將數(shù)據(jù)集劃分為訓(xùn)練集、驗(yàn)證集和測(cè)試集,進(jìn)行網(wǎng)絡(luò)初始化后將訓(xùn)練集、驗(yàn)證集輸入神經(jīng)網(wǎng)絡(luò)模型。

2)網(wǎng)絡(luò)訓(xùn)練。將劃分后的數(shù)據(jù)集經(jīng)過卷積層NET0進(jìn)行簡(jiǎn)單的特征提取以及維度擴(kuò)充,得到輸出特征圖后分別送入基于Convolution卷積的NET2和基于Involution卷積的NET1進(jìn)一步提取特征,由CBAM模塊自適應(yīng)融合NET2和NET1分別產(chǎn)生的特征,最后展平成一維送入全連接層并通過Softmax函數(shù)輸出分類結(jié)果。訓(xùn)練完成后輸出準(zhǔn)確率曲線和損失函數(shù)曲線,保存網(wǎng)絡(luò)權(quán)重。

3)網(wǎng)絡(luò)性能測(cè)試。將測(cè)試集輸入訓(xùn)練好的網(wǎng)絡(luò)模型中,計(jì)算得到混淆矩陣。

3 試驗(yàn)分析

3.1 軸承數(shù)據(jù)集

為驗(yàn)證MCK-CNN模型的性能和泛化能力,采用CWRU軸承數(shù)據(jù)集[14]以及實(shí)驗(yàn)室采集的軸承數(shù)據(jù)進(jìn)行試驗(yàn)分析,數(shù)據(jù)集的劃分見表1。CWRU軸承數(shù)據(jù)集軸承型號(hào)為6205,采樣頻率為48 kHz,轉(zhuǎn)速為1 797 r/min,N代表正常軸承,OR,IR,B分別代表外圈、內(nèi)圈、鋼球故障,007,014,021分別表示直徑0.178,0.356,0.534 mm的故障。實(shí)驗(yàn)室軸承測(cè)試平臺(tái)如圖6所示,在電動(dòng)機(jī)驅(qū)動(dòng)端的橫向、徑向、軸向分別安裝傳感器采集得到三通道的振動(dòng)信號(hào),采樣頻率為10 kHz,轉(zhuǎn)速為1 420 r/min,試驗(yàn)軸承型號(hào)為6206-2RS,故障類型如圖7所示。

表1 試驗(yàn)軸承數(shù)據(jù)集

(a) 正常軸承 (b) 保持架故障

3.2 模型性能測(cè)試

試驗(yàn)配置如下：CPU為Intel(R)Core(TM)i5-10400 CPU@2.90GHz,顯卡為RTX2060,運(yùn)行程序軟件為PyCharm Community Edition 2020.3 x64。神經(jīng)網(wǎng)絡(luò)模型設(shè)置參數(shù)為：學(xué)習(xí)率設(shè)為0.002,迭代次數(shù)為50,batchsize為32。由于輸入圖片64*64*3的尺寸較小,在Involution卷積操作中的壓縮率r和組數(shù)G均取為3。

為更好地驗(yàn)證MCK-CNN模型的故障診斷性能,在保證NET0的基礎(chǔ)上設(shè)置了只有NET1(Involution卷積)或NET2(Convolution卷積)的網(wǎng)絡(luò)進(jìn)行對(duì)比分析,結(jié)果如圖8所示：

1)對(duì)于實(shí)驗(yàn)室軸承數(shù)據(jù)集,由于故障類型僅有4類,各模型的故障檢測(cè)準(zhǔn)確率均較穩(wěn)定,在只有Convolution卷積進(jìn)行特征提取的狀態(tài)下準(zhǔn)確率即可達(dá)到94%,在Involution卷積進(jìn)行二次特征提取的狀態(tài)下準(zhǔn)確率達(dá)到了97%;當(dāng)把二者提取的特征通過CBAM模塊進(jìn)行自適應(yīng)融合之后,模型準(zhǔn)確率達(dá)到了99%,故障識(shí)別率得到了較大程度的提高。

2)對(duì)于CWRU軸承數(shù)據(jù)集,在10種故障類型情況下,MCK-CNN模型的故障識(shí)別率達(dá)到了99.25%,仍高于僅通過單一卷積核提取特征的網(wǎng)絡(luò)模型,說明其能夠有效進(jìn)行滾動(dòng)軸承故障識(shí)別。

3.3 網(wǎng)絡(luò)模型對(duì)比

為進(jìn)一步驗(yàn)證本文算法的優(yōu)越性,選取2個(gè)較為前沿的神經(jīng)網(wǎng)絡(luò)模型算法進(jìn)行對(duì)比：1)通過短時(shí)傅里葉變換將一維軸承振動(dòng)信號(hào)轉(zhuǎn)為時(shí)頻圖再輸入改進(jìn)的深度殘差網(wǎng)絡(luò)(STFT-DRNN)模型;2)將一維軸承振動(dòng)信號(hào)輸入由空洞卷積搭建的神經(jīng)網(wǎng)絡(luò)(FAC-CNN)模型。

由于實(shí)驗(yàn)室軸承數(shù)據(jù)集的故障類型較少,3種網(wǎng)絡(luò)均取得了較好的故障識(shí)別率,因此在相同試驗(yàn)條件下采用CWRU軸承數(shù)據(jù)集進(jìn)行試驗(yàn)。學(xué)習(xí)率為0.02,batch_size為32,迭代50次,試驗(yàn)結(jié)果如圖9所示：MCK-CNN達(dá)到了99.25%的故障識(shí)別率,比FAC-CNN,STFT-DRNN分別高出1.50%和2.75%;另外,相對(duì)于同樣將軸承一維振動(dòng)信號(hào)轉(zhuǎn)換為二維時(shí)頻圖的STFT-DRNN,MCK-CNN模型的訓(xùn)練時(shí)間僅為其50.4%,極大縮短了模型訓(xùn)練耗時(shí);因此,本文所提MCK-CNN模型可以高效、準(zhǔn)確地實(shí)現(xiàn)滾動(dòng)軸承故障診斷。

4 結(jié)束語(yǔ)

建立了一種用于滾動(dòng)軸承故障診斷的多卷積核特征提取自適應(yīng)融合的神經(jīng)網(wǎng)絡(luò)模型(MCK-CNN),并采用公開軸承數(shù)據(jù)集驗(yàn)證了該方法在理論上和實(shí)際上的可行性。采用多種卷積核共同提取輸入信號(hào)特征時(shí),由于每類卷積核的故障特征提取方式不同,側(cè)重點(diǎn)不同,可以更全面、充分的提取故障特征,自適應(yīng)融合則可以提高模型的訓(xùn)練效率,從而高效、準(zhǔn)確地實(shí)現(xiàn)滾動(dòng)軸承的故障診斷,具備較好的實(shí)用性。