[關(guān)鍵詞]遷移學(xué)習(xí);域適應(yīng);聯(lián)合分布;故障診斷
[中圖分類號]TH17 ;TP18 [文獻標志碼]A [文章編號]2095–6487(2024)11–0131–04
軸承故障會直接影響旋轉(zhuǎn)機械的性能,處理不當(dāng)甚至?xí)<安僮魅藛T的生命。因此,診斷軸承健康狀況非常重要。近年來,隨著機器學(xué)習(xí)的蓬勃發(fā)展,基于深度學(xué)習(xí)的旋轉(zhuǎn)機械故障診斷逐漸用于工業(yè)領(lǐng)域并取得了良好的效果[1]。在實際工業(yè)生產(chǎn)環(huán)境中,由于旋轉(zhuǎn)機械設(shè)備運行工況復(fù)雜多變,使得原始的振動信號具有不同的特征分布并且部分工況數(shù)據(jù)采集困難,這無法滿足深度學(xué)習(xí)中要求特征獨立同分布且標簽數(shù)據(jù)要充足的要求[2]。遷移學(xué)習(xí)可以從其他相關(guān)數(shù)據(jù)集(即有足夠的標記樣本但不同的分布)中學(xué)習(xí)知識來構(gòu)建當(dāng)前故障分類任務(wù)的診斷模型,并且它可以有效地解決數(shù)據(jù)缺失狀況下的故障診斷問題[3]。無監(jiān)督域自適應(yīng)作為遷移學(xué)習(xí)的一個子域,使用度量函數(shù)測量和減少域之間的分布差異,通過學(xué)習(xí)有標記的訓(xùn)練樣本和無標記的測試樣本的共享特征,緩解樣本分布不一致的問題。但大多數(shù)模型關(guān)注的是域間整體邊緣分布的對齊,而沒有考慮到故障類別之間的不匹配。
因此,本文結(jié)合邊緣分布與條件分布對齊方法,構(gòu)建聯(lián)合分布域適應(yīng)診斷模型,考慮到單一度量準則對指導(dǎo)提取域不變特征不足的問題,將最大均值差異度量(MMD)與相關(guān)對齊準則(CORAL)結(jié)合成新的度量準則,充分提取域不變特征,該模型通過模型迭代收集源域和目標域數(shù)據(jù)的可遷移特征,達到對無標簽?zāi)繕擞驍?shù)據(jù)的健康狀況識別目的。
3實驗研究
3.1數(shù)據(jù)集介紹
JNU軸承故障數(shù)據(jù)集共有4 種健康狀態(tài)類型:正常狀態(tài)(N)、內(nèi)圈故障(IF)外圈故障(OF)、滾動體故障(BF),每種故障類型有3 種不同的轉(zhuǎn)速,即3種工況,分別為600 r/min,800 r/min 和1 000 r/min 下采集的滾動軸承故障數(shù)據(jù),采樣頻率為50kHz,采樣時間為20 s。本文選取其中兩種轉(zhuǎn)速做遷移域適應(yīng)實驗數(shù)據(jù),具體劃分如表1所示。
源域和目標域中每個類別的樣本數(shù)為1000,則源域和目標域各有4000個樣本??紤]到實際中故障樣本較少,采用滑動采樣技術(shù)對原始數(shù)據(jù)進行分割,對故障樣本進行擴充,以獲得足夠的故障信息,樣本長度為3072,滑動步數(shù)為256。本文直接使用原始振動樣本作為故障診斷模型的輸入。
3.2實驗結(jié)果及分析
所有實驗均采用均方根Prop(RMSProp)優(yōu)化器對網(wǎng)絡(luò)結(jié)構(gòu)進行訓(xùn)練,交叉熵損失函數(shù)作為目標函數(shù),網(wǎng)絡(luò)模型設(shè)置學(xué)習(xí)率設(shè)為0.001。訓(xùn)練epoch 設(shè)置為300,訓(xùn)練批次batchsize 設(shè)置為256。權(quán)衡參數(shù)設(shè)置為μ=0.1,λ=0.01。當(dāng)網(wǎng)絡(luò)的損失值不再下降或訓(xùn)練次數(shù)達到300 時,保存網(wǎng)絡(luò)參數(shù),使用無標簽的目標域數(shù)據(jù)進行測試。
為驗證本文聯(lián)合分布遷移域適應(yīng)故障診斷方法對診斷精度的影響,設(shè)計3組對比實驗,分別為①無JDA、無CORAL 度量;② 僅無JDA ;③ 僅無CORAL,每次結(jié)果為5 次實驗結(jié)果的平均值,結(jié)果如表2 所示。
分析表2 可知,本文方法相較于對照組,在診斷準確率上與分類損失方面有較大的提升,本文模型準確率曲線與損失值迭代曲線如圖2(a)、(b)所示。
4結(jié)論與展望
針對基于統(tǒng)計度量的的域適應(yīng)法中的的度量準則進行改進,將最大均值差異與相關(guān)對齊準則相結(jié)合組成新的度量準則,提高網(wǎng)絡(luò)模型的提取域不變特征的能力。在遷移域適應(yīng)中,使用聯(lián)合分布自適應(yīng)方法,在使得域間邊緣分布對齊的同時,考慮到類間的條件分布對齊,充分減小源域與目標域數(shù)據(jù)的分布差異,提高診斷的精度。通過實驗證明了本文方法的有效性。
在實際生產(chǎn)環(huán)境中,對滾動軸承振動信號的采集越來越多,數(shù)據(jù)分布差異也越來越大,這對域適應(yīng)方法提出了更高的挑戰(zhàn)。本文并未對聯(lián)合分布中邊緣分布與條件分布做動態(tài)適應(yīng)工作,以適應(yīng)更對復(fù)雜多變的數(shù)據(jù)分布,后續(xù)會繼續(xù)對動態(tài)域適應(yīng)的領(lǐng)域做進一步研究工作。