摘要:針對變工況條件下因源域和目標域樣本數據分布差異大造成滾動(dòng)軸承故障診斷準確率較低的問(wèn)題,提出一種新的遷移學(xué)習方法——卷積注意力特征遷移學(xué)習(Convolutional Attention-based Feature Transfer Learning, CAFTL),并用于變工況條件下的滾動(dòng)軸承故障診斷。在所提出的CAFTL中,將源域和目標域樣本經(jīng)過(guò)多頭自注意力計算再經(jīng)過(guò)歸一化之后,輸入到卷積神經(jīng)網(wǎng)絡(luò )中得到對應的源域和目標域特征;然后通過(guò)域自適應遷移學(xué)習網(wǎng)絡(luò )將兩域特征投影到同一個(gè)公共特征空間內;接著(zhù),利用由源域有標簽樣本構建的分類(lèi)器進(jìn)行分類(lèi);最后,利用隨機梯度下降(Stochastic Gradient Descent, SGD)方法對CAFTL進(jìn)行訓練和參數更新,得到CAFTL的最優(yōu)參數集后將參數優(yōu)化后的CAFTL用于滾動(dòng)軸承待測樣本的故障診斷。滾動(dòng)軸承故障診斷實(shí)例驗證了所提出的方法的有效性。