崔石玉，朱志宇

(江蘇科技大學 電子信息學院，江蘇 鎮(zhèn)江 212003)

海水泵是維持船舶艙室設(shè)備運轉(zhuǎn)的關(guān)鍵冷卻裝備，在現(xiàn)代船舶系統(tǒng)中的應(yīng)用越來越廣泛[1]，一旦發(fā)生故障致使冷卻系統(tǒng)癱瘓，會造成巨大經(jīng)濟損失乃至給人身安全帶來隱患。因此海水泵故障診斷的研究具有重要的理論價值和實際意義。

海水泵故障診斷大多是通過監(jiān)測振動信號和電流信號等參數(shù)，選擇和提取原始信號的時頻域特征，通過傅里葉變換、小波變換、經(jīng)驗?zāi)B(tài)分解、主成分分析、流形學習等信號處理方法[2-4]，再利用機器學習方法如人工神經(jīng)網(wǎng)絡(luò)(artificial neural network, ANN)、支持向量機(support vector machine, SVM)等對故障特征進行識別分類[5]。這些傳統(tǒng)方法需要大量的信號處理知識和豐富的專家經(jīng)驗[6]，耗時費力、具有一定的主觀性，會影響故障診斷的精確度。作為人工智能的突破，深度學習具有克服以上缺陷的潛力。目前，深度學習因其強大的學習表征能力和優(yōu)異的識別效果，在很多領(lǐng)域得到應(yīng)用[7]，并在大數(shù)據(jù)處理上取得了突破性的成功。卷積神經(jīng)網(wǎng)絡(luò)(convolutional neural networks, CNN)作為深度學習的一個重要分支，在圖像識別領(lǐng)域具有獨樹一幟的優(yōu)勢[8-9]，在旋轉(zhuǎn)機械故障診斷領(lǐng)域也在逐步替代前一代算法，允許網(wǎng)絡(luò)從數(shù)據(jù)中自適應(yīng)學習非線性故障特征，比傳統(tǒng)機器學習方法具有更強大的表征能力。文獻[10]提出了基于CNN的齒輪箱故障診斷模型，提取齒輪箱各故障狀態(tài)信號的時頻特征作為輸入，構(gòu)建二維CNN網(wǎng)絡(luò)實現(xiàn)故障診斷。Zhang等[11]將一維時間機械振動信號轉(zhuǎn)化為二維圖像，用于CNN訓練和分類。以上算法雖然都用到了CNN，但CNN的輸入仍然是人工提取特征，CNN強大特征學習能力沒有充分利用，限制了故障識別率的提高[12]。同時，一維時間序列的振動信號，各時刻的數(shù)據(jù)點具有關(guān)聯(lián)性，二維形式會破壞這種相關(guān)聯(lián)性，可能使相關(guān)故障信息缺失。目前，以原始振動信號作為一維CNN輸入，基于一維CNN算法實現(xiàn)故障診斷研究尚處于探索階段，公開發(fā)表的成果較少[13-14]。

遷移學習能夠?qū)W習先前任務(wù)的知識和技巧并應(yīng)用到新任務(wù)中[15]，可以解決新領(lǐng)域中訓練數(shù)據(jù)量不足以及訓練數(shù)據(jù)與測試數(shù)據(jù)分布不同等問題，提升故障診斷模型的泛化能力?；趨?shù)(模型)的遷移學習[16-17]可以充分利用模型之間存在的相似性，通過在不同域間共享參數(shù)，來實現(xiàn)遷移學習的效果，代表性方法為多任務(wù)學習。通過神經(jīng)網(wǎng)絡(luò)作為基分類器，多個任務(wù)共享一些神經(jīng)網(wǎng)絡(luò)層及其相關(guān)聯(lián)的模型參數(shù)，輸出層對應(yīng)多個不同的任務(wù)，直觀地實現(xiàn)多任務(wù)學習。Yosinski 等[18]指出 LeNet、AlexNet、VGG、Inception 等都是良好的遷移網(wǎng)絡(luò)?；谔卣鞅硎镜倪w移學習[19-20]是利用源域和目標域的信息，尋找合適的特征表示空間，作為所要遷移知識的載體，來增強目標域的學習效果。與無監(jiān)督遷移成分分析不同，有監(jiān)督遷移學習方法同時利用數(shù)據(jù)中的屬性信息和標簽信息。借鑒深度學習中的微調(diào)方法，即先用源域中的樣本訓練一個基模型，然后利用目標域中的樣本，微調(diào)此基模型。此方法目前主要在圖像識別領(lǐng)域[21-22]流行應(yīng)用。

為解決海水泵工況多變、故障數(shù)據(jù)少、振動特征提取困難等故障診斷問題，本文提出基于參數(shù)遷移與一維卷積神經(jīng)網(wǎng)絡(luò)(one-dimensional convolutional neural network，1DCNN)的海水泵故障診斷方法。該方法利用1DCNN對多工況的海水泵振動信號進行故障分類。首先利用源域樣本訓練一個CNN基分類器模型，再利用基于參數(shù)的遷移學習在源域和目標域間共享模型參數(shù)，然后利用目標域樣本對已訓練好的CNN基模型進行微調(diào)，并給出故障診斷結(jié)果。該方法完全不依賴于人工特征提取和專家知識，可以實現(xiàn)原始信號端到端直接進行故障診斷，最大限度地利用CNN深度網(wǎng)絡(luò)實現(xiàn)故障特征自學習，并利用參數(shù)遷移加速1DCNN模型快速收斂并降低多工況下故障數(shù)據(jù)分布不同對故障診斷的影響，提高了1DCNN模型的識別能力和泛化能力。

1 一維卷積神經(jīng)網(wǎng)絡(luò)

典型的卷積神經(jīng)網(wǎng)絡(luò)由輸入層、卷積層、池化層、全連接層和輸出層構(gòu)成。CNN通過梯度下降法最小化損失函數(shù)對網(wǎng)絡(luò)權(quán)重進行反向調(diào)節(jié)，同時通過頻繁的迭代訓練提高神經(jīng)網(wǎng)絡(luò)的精度。常見的損失函數(shù)有均方根誤差函數(shù)、交叉熵損失函數(shù)、負對數(shù)似然函數(shù)等。在CNN前期特征提取算法中，通過交替使用卷積層和池化層進行輸入數(shù)據(jù)特征的層層提取，使網(wǎng)絡(luò)具有一定程度的位移、尺度、縮放、非線性形變穩(wěn)定性，最后通過全連接層輸出結(jié)果?；A(chǔ)1DCNN結(jié)構(gòu)示意圖，如圖1所示。1DCNN用于在時間序列的特征識別以及提取，同樣具備CNN對特征識別的平移不變性等優(yōu)點，并且大卷積核不會帶來太多參數(shù)以及計算量，為了獲得較大感受野，模型可以采用較大寬度的卷積核，從而更加全面地獲取序列的特征值。

圖1 一維卷積神經(jīng)網(wǎng)絡(luò)結(jié)構(gòu)示意圖

1.1 卷積層

卷積層主要由多個卷積核構(gòu)成。一個卷積核可以認為是一個感受野，CNN通過卷積核運算獲取故障數(shù)據(jù)的特征maps，通過權(quán)值共享方法，同一卷積核以一定的步長與不同局部數(shù)據(jù)進行運算時共享同一組權(quán)重，從而減少計算量。一維卷積核運算原理為

(1)

式中：a為卷積核在x方向上的寬度；F為卷積核參數(shù)向量；G為與卷積核函數(shù)運算的局部向量矩陣；w為卷積核的尺寸。

1.2 池化層

池化層即為降維，能夠減少1DCNN內(nèi)部參數(shù)的計算量，同時能一定程度地防止過擬合，提高特征的魯棒性。通常采用最大池化算子，數(shù)學描述如式(2)所示

(2)

式中：P為最大池化得到的特征向量；l為特征maps的寬度；A為卷積層激活后的特征向量矩陣；w為池化區(qū)域的寬度。

1.3 全連接層

全連接層將1DCNN網(wǎng)絡(luò)中多層卷積和池化后的特征maps鋪平為一維向量輸入，然后將每層輸入通過式(3)計算輸出，在輸出層通過式(5)輸出診斷結(jié)果。

(3)

ReLU=max(0,x)

(4)

1.4 輸出層

輸出層采用Softmax分類器解決多分類問題，模型可表述為

(5)

式中：O為輸出結(jié)果矩陣；Wi和bi為第i個分類對應(yīng)的權(quán)重和偏置矩陣。

2 參數(shù)遷移

AlexNet和GoogleNet等深度卷積神經(jīng)網(wǎng)絡(luò)在圖像分類中取得巨大成功，在大型圖像數(shù)據(jù)集ImageNet上獲得充分訓練，學習到了圖像分類識別所需的大量特征。因此，可以運用遷移學習的思想，充分利用預(yù)訓練模型在源域數(shù)據(jù)集上學習到的大量知識，將其用于目標域故障分類識別問題。

模型參數(shù)遷移作為一種常用的遷移方法，去掉預(yù)訓練網(wǎng)絡(luò)的最后一層，將其之前的激活值送入卷積神經(jīng)網(wǎng)絡(luò)等基分類器進行分類訓練；參數(shù)遷移只需重新初始化網(wǎng)絡(luò)的最后的少數(shù)幾層，其余層直接使用預(yù)訓練網(wǎng)絡(luò)的權(quán)值參數(shù)和網(wǎng)絡(luò)參數(shù)，再利用目標域新故障數(shù)據(jù)集對網(wǎng)絡(luò)參數(shù)進行微調(diào)。本文采用參數(shù)精調(diào)的遷移學習方式，將源域基模型修改用于海水泵目標域故障識別的模型，相比于隨機初始化網(wǎng)絡(luò)所有層的權(quán)重參數(shù)和利用訓練數(shù)據(jù)集對網(wǎng)絡(luò)從頭開始訓練的全新學習，精調(diào)有助于卷積神經(jīng)網(wǎng)絡(luò)快速收斂。

參數(shù)遷移模型的結(jié)構(gòu)框架，如圖2所示。給定源域的訓練數(shù)據(jù)DS={(xSi,ySi)|xSi∈X,ySi∈Y,i=1,2,…,nS}和訓練得到的1DCNN基模型fS(·)和模型參數(shù)θS，遷移學習通過利用源域的DS，fS(·)和θS，提高目標域DT的模型fT(·)性能。本文在源域訓練好的1DCNN基模型參數(shù)θS作為目標域1DCNN模型的初始化參數(shù)，用于目標域的測試數(shù)據(jù)的故障診斷，提高目標域1DCNN模型的泛化能力和通用性。

圖2 1DCNN參數(shù)遷移模型結(jié)構(gòu)

3 1DCNN和參數(shù)遷移模型設(shè)計

3.1 模型結(jié)構(gòu)

模型采用AlexNet模型結(jié)構(gòu)，由3個卷積模塊和3個全連接模塊堆疊而成，最后一個全連接層輸出數(shù)量為6，分別對應(yīng)目標域故障類別數(shù)，采用Softmax計算損失，如表1所示。卷積模塊conv_1和conv_2均由卷積層、最大池化層、Dropout層組成，卷積模塊conv_3由卷積層和最大池化層組成。全連接模塊fc_1包含了Dropout層，在訓練時隨機丟棄一些參數(shù)，參照AlexNet，丟棄概率設(shè)為0.5。池化降采樣保持了一定的平移不變性，重疊池化、Dropout緩解了過擬合。神經(jīng)元的激活函數(shù)選擇修正線性單元(ReLU)，其單側(cè)抑制特性提高網(wǎng)絡(luò)的稀疏激活性，可以有效解決梯度彌散，加速網(wǎng)絡(luò)收斂。

表1 一維卷積神經(jīng)網(wǎng)絡(luò)參數(shù)

3.2 模型優(yōu)化

采用交叉熵計算分類損失，衡量預(yù)測的Softmax輸出概率分布和目標類概率分布的相似性。附加L2正則化對權(quán)重參數(shù)進行懲罰，減輕過擬合現(xiàn)象。其數(shù)學描述如式(6)所示

(6)

式中：p(x)為目標類概率分布；q(x)為預(yù)測分布；θ為權(quán)重參數(shù)；λ為正則項系數(shù)；‖θ‖2為防過擬合添加的正則化項。

網(wǎng)絡(luò)模型的優(yōu)化是通過迭代最小化損失函數(shù)實現(xiàn)的。本文采用Adam優(yōu)化算法，通過訓練數(shù)據(jù)迭代更新網(wǎng)絡(luò)權(quán)重。Adam優(yōu)化器同時獲得了 AdaGrad 和 RMSProp 算法的優(yōu)點，即能夠自適應(yīng)地保留學習率提高在稀疏梯度和抗噪的性能，適用于處理非平穩(wěn)的振動信號；同時調(diào)參較少，非常高效。模型Adam優(yōu)化算法步驟如下：

參數(shù)設(shè)定——α=0.001，β1=0.9，β2=0.999，ε=10-8。

要求——α為學習率；β1,β2為超參數(shù)，β1,β2∈[0,1)。

要求——f(θ)為含θ參數(shù)的隨機目標函數(shù)。

要求——θ0；m0←0；v0←0；t←0：初始化時間步。

Whileθt沒有收斂 do

t←t+1；gt←?θft(θt-1)：更新目標函數(shù)的梯度。

end While

3.3 1DCNN參數(shù)遷移模型故障診斷框架

振動時域信號是一維非平穩(wěn)、非線性信號，本文采用1DCNN的參數(shù)遷移模型對海水泵振動故障數(shù)據(jù)進行故障診斷。1DCNN參數(shù)遷移模型故障診斷流程，如圖3所示。

圖3 1DCNN參數(shù)遷移模型故障診斷流程圖

首先將海水泵故障數(shù)據(jù)劃分為源域和目標域數(shù)據(jù)集，之后對源域和目標域數(shù)據(jù)集分別進行讀取并劃分訓練、驗證和測試樣本比例，再對數(shù)據(jù)進行預(yù)處理(包括打亂、標準化Rreshape等)。將源域數(shù)據(jù)集作為1DCNN基模型的輸入，建立1DCNN基模型并利用Adam進行梯度下降迭代調(diào)優(yōu)，使目標函數(shù)最小化，得到1DCNN基模型網(wǎng)絡(luò)參數(shù)θS。調(diào)用1DCNN基模型，并對基模型最后一層(輸出層)微調(diào)，得到1DCNN參數(shù)遷移模型。將MAFAULDA可變工況的目標域數(shù)據(jù)集作為1DCNN參數(shù)遷移模型輸入，同樣地，采用Adam作為優(yōu)化器，使目標函數(shù)即交叉熵損失函數(shù)最小化，不斷迭代優(yōu)化。最終，輸出振動故障診斷結(jié)果。

4 結(jié)果與分析

4.1 試驗數(shù)據(jù)的采集與數(shù)據(jù)集制作

為驗證本文算法實用的可行性，將本文構(gòu)建的算法模型應(yīng)用于海水泵軸承和電動機的故障診斷中。試驗數(shù)據(jù)集來源于中華復(fù)興號客滾輪船的艙室輔機2號海水泵數(shù)據(jù)采集系統(tǒng)，該機最大功率達400 kW，最大流量1 800 m3/h，最大揚程600 m。本文利用三軸振動加速度傳感器采集到的海水泵軸承、電動機徑向加速度數(shù)據(jù)作為振動故障數(shù)據(jù)集，采樣頻率為2 000 Hz。

目標域數(shù)據(jù)集振動一維信號時域波形圖，如圖4所示。振動故障數(shù)據(jù)集，如表2所示。模型的訓練與測試均在TensorFlow深度學習框架下完成的。利用Google 云端硬盤提供的免費GPU運行Python代碼。模型參數(shù)中，批量大小batch_size設(shè)為512，step設(shè)置為50。為試驗方便，將采集的數(shù)據(jù)以1 024個點為一個樣本，將每類信號劃分400個樣本，并按照7∶1∶2的比例劃分訓練集、驗證集和測試集。

表2 振動故障數(shù)據(jù)集

圖4 故障振動信號時域波形圖

4.2 診斷結(jié)果與分析

對比兩種模型學習方式(全新學習、參數(shù)遷移+微調(diào))，3組初始學習率(0.001，0.050，0.100)，海水泵工況在負載為75 kW、3種轉(zhuǎn)速條件下采集到的振動故障試驗數(shù)據(jù)，共進行18組試驗。全新學習時，1DCNN網(wǎng)絡(luò)模型的所有層的權(quán)重參數(shù)需要隨機初始化，再利用目標域數(shù)據(jù)集對其從頭開始訓練。參數(shù)遷移訓練時，源域數(shù)據(jù)集是在負載75 kW，轉(zhuǎn)速1 792 r/min下獲取的振動數(shù)據(jù)。最后一層的權(quán)值參數(shù)與偏置參數(shù)隨機初始化，其他各層的W和b則從在源域訓練得到的1DCNN基模型中加載。1DCNN參數(shù)遷移模型經(jīng)5次的50輪訓練后，各組試驗的診斷結(jié)果如表3所示。

表3 1DCNN參數(shù)遷移模型診斷結(jié)果

如表3所示，在全新學習方式下，所有層的參數(shù)都是隨機初始化，適合的學習率可以使訓練快速接近最優(yōu)解，使其在相同的訓練輪數(shù)下獲得更高的準確率。當學習率設(shè)置分別為0.050和0.100時，全新學習模型的訓練精度在72%左右，測試精度只有17%左右，存在嚴重過擬合現(xiàn)象。而學習率為0.001時的訓練和測試準確率是最高的，全新學習模型平均準確率達到了93%以上。

如表3所示，在參數(shù)遷移模型中，由于網(wǎng)絡(luò)前端各層參數(shù)都獲得優(yōu)質(zhì)訓練，訓練初始時刻已接近最優(yōu)解，故模型的精度得到一定的保障。學習率為0.001時的訓練和測試準確率是最高的，比其他同條件的不同學習率分別提高了12%和38%左右，模型平均精度更是達到了97%左右，平均識別準確率為96.13%。將學習率為0.001的全新學習和參數(shù)遷移學習方式下的各組試驗對應(yīng)比較，容易發(fā)現(xiàn)參數(shù)遷移學習的各組試驗的模型測試準確率高出全新學習的各組試驗約10～50個百分點。如圖5(a)和圖5(b)所示，設(shè)置學習率為0.001，將A、B、C三種不同工況數(shù)據(jù)集輸入到1DCNN遷移模型，經(jīng)過5個epoch后，模型完全達到穩(wěn)定，訓練精度逼近98%，損失函數(shù)趨于0.03且平穩(wěn)。因此，1DCNN參數(shù)遷移模型對不同工況下的海水泵的故障都有較高的診斷識別率，同時高診斷精度不存在過擬合問題。

圖5 學習率為0.001，1DCNN參數(shù)遷移模型結(jié)果

4.3 模型性能的影響因素分析

4.3.1 學習率對模型性能影響

圖6(a)和圖6(b)是不同學習率對1DCNN(全新學習)模型的性能影響曲線圖。從圖中可以看出，當初始學習率為0.001 0和0.010 0時，模型的準確率和損失率經(jīng)過20個epoch后達到穩(wěn)定，損失率趨近于0，訓練精度趨近于95%，比相同條件下其他學習率試驗高出20%。但是，學習率為0.010 0時會出現(xiàn)訓練周期性小幅度上下振蕩的現(xiàn)象，沒有學習為0.001 0時結(jié)果穩(wěn)定，學習率為0.000 1時模型訓練精度最高達到90%，損失率也沒有達到最優(yōu)，在經(jīng)過50個epoch后仍然沒有趨于平穩(wěn)，在圖中準確率和損失率曲線中都有體現(xiàn)。而學習率為0.050 0時，訓練精度大概只有70%，損失率曲線可以看出訓練已經(jīng)嚴重過擬合。因此，適當?shù)膶W習率可以使模型訓練快速接近最優(yōu)解且模型性能更好。

圖6 不同學習率對模型的性能比較圖

圖6(c)和圖6(d)是不同學習率對1DCNN參數(shù)遷移模型性能影響曲線圖。從圖中可以看出，初始學習率為0.001 0和0.010 0時，模型的精度趨近98%，比相同條件下其他學習率試驗高出10%～20%。但是，學習率為0.010 0時會出現(xiàn)訓練周期性小范圍波動的現(xiàn)象；學習率為0.000 1時模型訓練在經(jīng)過20個epoch后精度達到92%。而學習率為0.050 0和0.100 0時，出現(xiàn)了過擬合現(xiàn)象。綜上，無論是全新學習還是參數(shù)遷移模型，學習率可在[0.001 0,0.010 0]內(nèi)選擇最佳，訓練的準確率最高。

4.3.2 優(yōu)化器的選擇

在優(yōu)化器選擇試驗中繼續(xù)采用表1中的基準模型參數(shù)設(shè)置，本試驗對目前最常用的SGD, AdaGrad, RMSProp和Adam共4種優(yōu)化器進行測試，試驗結(jié)果如圖7和表4所示。

圖7可明顯看出：Adam自適應(yīng)優(yōu)化器的性能優(yōu)于其他3種優(yōu)化器。圖7(a)中，當批大小batch_size=8時，經(jīng)過50 輪epoch 后只有SGD對應(yīng)的準確率曲線還未收斂，大概在經(jīng)過100輪epoch后才會趨于平穩(wěn)，其準確率達到96.3%。而在圖7(b)中，當批大小batch_size=512時，SGD和Adagrad優(yōu)化器的準確率還不足30%，陷入局部最優(yōu)，效果都很差。Adagrad處理稀疏梯度的優(yōu)勢在本文中也沒有體現(xiàn)。RMSprop的效果也不錯，略低于Adam，其發(fā)揮了自身處理非線性非平穩(wěn)信號的優(yōu)勢。表4主要比較了batch_size大小對SGD和Adam精度和時間的影響。SGD適合小批次的模型，但需要經(jīng)過很長時間的訓練，由圖7(a)也反映出僅50輪訓練后未收斂，大概在經(jīng)過100輪訓練后SGD精度逼近96%。而Adam優(yōu)化器對batch_size大小不敏感。之所以選擇batch_size=512主要考慮實時性因素。大批次的模型訓練時，Adam在保證高精度的同時能夠更快收斂穩(wěn)定。

圖7 優(yōu)化器性能曲線對比圖

綜合來看，在算法時間和準確率方面，由于結(jié)合RMSprop處理非平穩(wěn)目標特點和自適應(yīng)調(diào)節(jié)學習率特點，Adam優(yōu)化器性能最好，尤其在前期準確率上升很快，在15個epoch左右趨于穩(wěn)定。綜上，本文采用Adam優(yōu)化算法最佳。

4.3.3 參數(shù)遷移和微調(diào)對模型性能的影響

在參數(shù)遷移和微調(diào)對模型性能影響試驗中，采用表1中的1DCNN基模型參數(shù)，其中epoch設(shè)置為100。試驗對無參數(shù)遷移、無微調(diào)和本文提出的算法模型進行測試，試驗結(jié)果如圖8所示。

圖8 參數(shù)遷移和微調(diào)對模型性能的影響

如圖8所示，全新學習模型中，訓練準確率經(jīng)過100個epoch后達到90%左右，模型基本收斂，測試準確率只有80%，泛化能力不夠好。無微調(diào)的模型中，約經(jīng)過40個epoch后訓練準確率增加了6.3%，損失率也降低了15%，說明參數(shù)遷移發(fā)揮了作用，不僅提高了準確率，并且節(jié)省了近60個epoch網(wǎng)絡(luò)收斂時間。本文提出的加入?yún)?shù)遷移和微調(diào)的算法模型的訓練精度逼近98%，測試精度在95%左右，經(jīng)過5個epoch模型就收斂，相比于無精調(diào)模型，明顯加快收斂速度。說明參數(shù)遷移和微調(diào)對于加速網(wǎng)絡(luò)收斂和提高精度有明顯促進的作用。

結(jié)合表3可以看出，在適當?shù)膶W習率設(shè)定前提下，在全新學習模式下，訓練初始階段的測試準確率只有80%，經(jīng)過50個epoch后，模型的測試準確率才逐漸趨于穩(wěn)定。在參數(shù)遷移和微調(diào)學習模式下，訓練初始階段的測試準確率達到90%以上，之后網(wǎng)絡(luò)迅速收斂，經(jīng)過約5個epoch準確率急速爬升，經(jīng)過約10個epoch便達到準確率峰值，并趨于穩(wěn)定。節(jié)省了約4/5的訓練時間，平均準確率提高了10%～20%左右。這歸功于1DCNN模型的參數(shù)遷移學習和微調(diào)，加速了模型的快速收斂并提高了模型訓練的精度。

4.4 故障預(yù)測與分析

對故障診斷性能較優(yōu)的第18個試驗(1DCNN參數(shù)遷移，初始學習率為0.001，數(shù)據(jù)集C)中經(jīng)過50個epoch訓練獲得的模型進行預(yù)測，精準率P、召回率R、F1的計算公式如式(7)，計算得到的混淆矩陣、P、R、F1得分如表5所示。

表5 1DCNN參數(shù)遷移模型的混淆矩陣

P=TP/TP+FP，

R=TP/TP+FN，

(7)

由表5的混淆矩陣可知，模型的平均分類準確率為95.59%，F(xiàn)1得分平均分類率為95.87%，分類性能(F1得分)由高到低排名依次是無故障、不對中故障、外圈磨損、滾動體磨損、不平衡故障和內(nèi)圈磨損。由混淆矩陣可以看出，分類錯誤主要為軸承內(nèi)圈磨損和轉(zhuǎn)子不平衡故障，其中不平衡故障識別率為92.6%，有7.4%誤認為是軸承內(nèi)圈故障，軸承內(nèi)圈故障中有17.3%被誤認為是不平衡故障。主要是因為兩類故障的一維振動信號波形類似，可能是在海水泵機組在75 kW時由于功率太大，需要的能量越多，振動的越劇烈，長此以往使軸在運轉(zhuǎn)和高溫下變形，產(chǎn)生動不平衡力和離心力，離心慣性力通過軸承作用到海水泵機組機械器件上，引起振動產(chǎn)生噪聲，加速軸承磨損，長此以往可能已產(chǎn)生復(fù)合故障，故而在分類識別中會降低診斷精度，較難區(qū)分歸屬故障類別。

圖9是不同負載下的試驗測試結(jié)果。圖9(a)是在海水泵機組在30 kW負載時的故障預(yù)測結(jié)果。圖9(b)是海水泵機組在75 kW時的故障預(yù)測結(jié)果。從表5和圖9(b)中看到，除復(fù)合故障之外，其余故障位置均能100%識別出來，且綜合精度達到95%以上，說明產(chǎn)生的復(fù)合故障屬于輕微故障，器件磨損還不太嚴重。因此，需要采取預(yù)防保護措施，延長海水泵機械器件的壽命和使用。與75 kW負載情況不同，負載在30 kW下的故障位置幾乎都能判別出來，經(jīng)式(7)計算，得到precision值為99.38%，F(xiàn)1得分整體分類性能為99.37%。故障分類精度很高，說明本文算法1DCNN參數(shù)遷移模型的確學習到了表征故障的非線性顯著特征，即使是在復(fù)雜多變工況條件下，故障識別精度仍然很高，在實際工程應(yīng)用中實現(xiàn)快速定位具有重大的工程意義。

圖9 不同負載下的試驗測試結(jié)果

4.5 與其他算法對比

為驗證本文算法與目前主流故障診斷算法的識別性能，將SVM、多層感知器(multi-layer perceptron, MLP)、Zheng等研究中的遷移局部保持投影(transfer locality preserving projections, TLPP)和1DCNN作為對比算法進行試驗。在傳統(tǒng)故障診斷方法中通常先對原始數(shù)據(jù)集進行手工體征提取，再將提取的特征輸入到算法完成分類預(yù)測。選取5種方法進行特征值選取，如表6所示。按照表6的特征提取算子對故障數(shù)據(jù)進行提取，再分別用于SVM，MLP，TLPP算法進行故障診斷。

表6 特征提取方法

每種算法反復(fù)進行試驗10次，試驗結(jié)果如表7所示。從表7可以看出，由于SVM和MLP較淺層結(jié)構(gòu)，無法學習到深層特征，限制了識別率的提升。Zheng等研究中的核局部保持投影強大的非線性提取能力和遷移學習強大的跨任務(wù)、跨領(lǐng)域?qū)W習能力使TLPP的故障識別率打破傳統(tǒng)機器學習算法的壁壘。但是其數(shù)據(jù)特征值的選取和預(yù)處理需要耗費大量的計算和時間，在實際工程領(lǐng)域難以應(yīng)用。1DCNN模型參數(shù)隨機初始化和從頭開始訓練新任務(wù)的特點影響了故障診斷的精度，因此其識別率較本文算法仍有一定差距。同時，本文算法的識別率結(jié)果高和實時性高，進一步驗證了本文算法良好的的魯棒性和泛化能力。

表7 不同算法試驗結(jié)果

5 結(jié) 論

針對傳統(tǒng)識別方法存在特征提取耗時費力的問題，本文利用參數(shù)遷移和1DCNN對海水泵故障一維振動信號進行診斷分類，并就模型性能的影響因素進行分析，得到以下結(jié)論：

(1)卷積神經(jīng)網(wǎng)絡(luò)能夠完全擺脫人工提取特征和專家經(jīng)驗的主觀性，直接將原始故障數(shù)據(jù)作為模型輸入，自動逐層非線性提取故障特征，自動輸出故障分類結(jié)果，這種“端到端”的模型結(jié)構(gòu)具有更好的通用性和應(yīng)用性。并且算法具有較高的故障識別率和故障定位能力，平均識別準確率達到95.93% 以上。

(2)適合的學習率可以使訓練快速接近最優(yōu)解，使其在相同的訓練輪數(shù)下獲得更高的準確率。若初始學習率設(shè)置不當，會造成模型訓練振蕩甚至發(fā)散。

(3)相對于全新學習，參數(shù)遷移+微調(diào)模型充分利用在源域上學到的知識，顯著加速網(wǎng)絡(luò)的收斂和提高分類性能。本文算法節(jié)省1/2～4/5的訓練和測試時間，有利于實現(xiàn)在線實時監(jiān)測和快速智能診斷應(yīng)用。通過混淆矩陣可直接量化故障誤判的位置和數(shù)量。

(4)該算法具有良好的魯棒性和泛化性能，即使轉(zhuǎn)速和負載發(fā)生變化，仍然保持較高的故障識別率。

目前的1DCNN遷移模型局限于故障單一的海水泵故障一維信號。實際應(yīng)用中海水泵收集的同一振動信號可能會出現(xiàn)多種故障復(fù)合的故障特征。今后，將進一步地豐富海水泵故障信號數(shù)據(jù)集，采用多源信號共同參與海水泵的故障診斷，以進一步地提高模型的準確率與實用性。