基于模糊松弛約束的外觀缺陷多核學(xué)習(xí)技術(shù)

(北京工商大學(xué) 計算機與信息工程學(xué)院，北京 100048)

隨著工業(yè)生產(chǎn)水平的不斷提高，產(chǎn)品外觀質(zhì)量也越來越受到重視。傳統(tǒng)的人工檢測手段已經(jīng)不能滿足現(xiàn)代工業(yè)生產(chǎn)的需求，利用機器視覺技術(shù)對產(chǎn)品外觀缺陷進行檢測是實現(xiàn)工業(yè)現(xiàn)代化的必然趨勢。由于產(chǎn)品外觀缺陷種類較多，采用多種圖像特征融合的方式可以取得更好的檢測效果，并且在檢測中需要對各類缺陷進行學(xué)習(xí)分類[1]。不同的圖像特征，最適合的核函數(shù)不同，同一種特征，選擇不同的核函數(shù)其分類結(jié)果也會有較大差異[2]。

多核學(xué)習(xí)(MKL)針對單核學(xué)習(xí)的劣勢，提出將多個不同的核函數(shù)進行組合獲得一個組合核函數(shù)，采用該組合核函數(shù)作為線性不可分問題的核函數(shù)進行學(xué)習(xí)分類，這樣可以有效地提高分類精度。近幾年來，國內(nèi)外也有不少學(xué)者針對MKL進行了相關(guān)的研究與改進，典型的有：Duan等學(xué)者進一步改進了多核學(xué)習(xí)的交叉驗證方法[3]；Wang等學(xué)者結(jié)合典型相關(guān)分析獲取核映射空間的共有信息，提高了模型的分類精度[4]；Gonen等人針對多核學(xué)習(xí)在整個輸入空間中對某個核都分配相同權(quán)重的問題，利用一種選通模型局部地選擇合適的核函數(shù)，提出一種局部多核學(xué)習(xí)方法[5-6]。張凱軍等人將稀疏MKL算法與Multi-MHKS算法進一步改進，提出一種改進的顯性多核支持向量機(IDMK-SVM)，該方法既考慮到如何充分利用已經(jīng)選用的核函數(shù)中的共有信息，同時還考慮到核函數(shù)的篩選問題，提高了分類精確性[7]；李琦等人針對核函數(shù)選擇對最小二乘支持向量機回歸模型泛化性的影響，提出一種新的基于lp-范數(shù)約束的最小二乘支持向量機多核學(xué)習(xí)方法[8-9]。

針對多核學(xué)習(xí)方法中各個核函數(shù)的權(quán)重求解問題，筆者提出一種基于模糊松弛約束的求解方法。首先提取圖像的幾種不同的特征，對不同的特征選擇合適的核函數(shù)，劃分每個核函數(shù)的模糊權(quán)重，每個權(quán)重包括上下的松弛邊界，求解各個權(quán)重獲得組合核函數(shù)，然后使用SVM對各類外觀缺陷分類。最后將所提出的方法應(yīng)用于實際的生產(chǎn)流水線。結(jié)果表明，所提出的方法可以滿足外觀檢測對于實時性以及檢測精度的要求?？傮w工作流程如圖1所示。

圖1 總體工作流程

1 基于模糊松弛約束的多核學(xué)習(xí)方法

1.1 模型的建立

在建立多核學(xué)習(xí)模型以后，如何確定每個核函數(shù)的權(quán)重直接關(guān)系到最終的缺陷分類結(jié)果。根據(jù)不同的圖像特征對不同的檢測評價指標(biāo)的影響程度，以及外觀的不同表面對檢測精度的不同需求，劃定每個基核權(quán)重的松弛邊界，求解得到組合核函數(shù)的多核學(xué)習(xí)模型，并根據(jù)該模型進行學(xué)習(xí)分類，從而最終將檢測到的模型進行正確分類。具體過程如下。

(1) 建立式(1)所示的初始多核學(xué)習(xí)模型。

(2) 建立模糊約束條件模型：

(1)

式中，A為每幅圖像包含的圖像單元個數(shù)；γi為第i個特征對融合特征性能的影響程度。劃定核函數(shù)松弛權(quán)重范圍如式(2)所示，其中權(quán)重滿足0≤ρ≤1且ρ1+ρ2+ρ3+…+ρm=1。

(2)

根據(jù)不同檢測面的不同需求，選擇權(quán)重松弛邊的上邊界或下邊界：對于正面來說，檢測精度最為重要，確定上下邊界的時候要著重考慮提高精度；對于底面或背面來說，精度要求可以適當(dāng)減低，確定上下邊界的時候需要著重考慮節(jié)約時間；對于頂面或側(cè)面來說，則需要中和兩者考慮。松弛邊界劃定流程如圖2所示。

圖2 松弛邊界劃定流程

(3) 在劃分模糊松弛邊界的時候，考慮圖像特征對檢測評價指標(biāo)的“影響程度越大，權(quán)重越高”這一模糊行為，將其量化[10]。調(diào)整第i個特征的影響因子為ωγi，其中ωγi∈(0,γi]，稱其為“虛擬”影響程度。該特征對需求的影響越大，ωγi越大，用ωγi代替模型中的γi，可以滿足影響越大，分配的權(quán)重越大的條件。選取ωγi=γi[1-μk(γi)]，其中k為參數(shù)，反映了對約束條件的重視程度，重視程度越高，k越小。

選取ωγi的理由如下：

①ωγi滿足0<ωγi<γi；

②ωγi是連續(xù)遞減函數(shù)，對于“影響程度大”的隸屬度越大(γi越大)，相對應(yīng)的ωγi則越小，并且為原γi的整數(shù)倍；

③ 該參數(shù)的選擇使得計算更加有利。

(4) 建立“影響較大”的隸屬度函數(shù)。

γ={γi|i=1,2,…,n}，Y={γi|γi大,γi∈γ}，把Y看作是γ上的一個模糊子集，定義隸屬度μ:Y→[0,1]為

(3)

其中：

(4)

該隸屬度函數(shù)確定的理由如下：

① 該隸屬度函數(shù)μ的定義域為[0,+∞)，值域為[0,1]，與實際情況相符合。

② 該隸屬度函數(shù)為連續(xù)遞增函數(shù)，與“影響程度越大”，其隸屬度越大的事實相符合。

③ 該隸屬度函數(shù)μ的導(dǎo)函數(shù)μ′為

(5)

該導(dǎo)函數(shù)為連續(xù)遞減函數(shù)。

求解模型，得到各項參數(shù)，獲得每個基核的權(quán)重?；谀：沙诩s束多核學(xué)習(xí)流程圖如圖3所示。

圖3 模糊松弛約束多核學(xué)習(xí)流程

1.2 模型的求解

設(shè)模型的非基變量指標(biāo)集合為R，基變量指標(biāo)集合為S，令任意有序指標(biāo)m∈R1，其中ρm=0；令任意有序指標(biāo)m∈R2，其中ρm∈(ωm3,ωm4)。R1和R2分別被稱為第一類非基變量指標(biāo)集合以及第二類非基變量指標(biāo)集合，并滿足R=R1∪R2且R1∩R2=?。

把上述ρm代入模型中，得到一個基解{ρm}，其中m∈S，這個基解如果滿足約束條件，則稱其為基可行解。記λm=Um-Lm，其中Um是模型的一個基解，若滿足：

(6)

則基解{ρm}被稱為模型的正則解。若一個基解既是模型的正則解，也是模型的可行解，則該基解被稱為模型的最優(yōu)解。某一特征與權(quán)重對應(yīng)的回路的頂點若增加一個調(diào)整量，則稱該頂點為接納元素；若減少一個調(diào)整量，則稱該頂點為釋放元素。接下來執(zhí)行下述迭代過程。

(1) 令R1=R2=?，{ρm}為一個正則解，若其滿足模型約束條件，且m∈S，則該解即為最優(yōu)解，轉(zhuǎn)入步驟(7)。

(2) 若某一ρpm>ωm4，則轉(zhuǎn)入步驟(3)；若某一ρpm<0，則轉(zhuǎn)入步驟(5)。

(3) 取R1中某一元素使對應(yīng)的ρqm滿足以下條件：

① 存在一條回路，該回路除去ρqm以外全部是基變量；

② (p,m)是釋放元素，(q,m)是接納元素。

轉(zhuǎn)入步驟(4)。

(4) 釋放元素減去Q，其中Q=ρpm-ωm4，余下元素依次交叉+Q和-Q，ρpm變成第二類非基變量，ρqm變成基變量，從而得到一組新的基解并返回步驟(1)。

(5) 取R2中某一元素使對應(yīng)的ρqm滿足以下條件：

① 存在一條回路，該回路除去ρqn以外全部是基變量；

② (p,m)是接納元素，(q,m)是釋放元素。

轉(zhuǎn)入步驟(6)。

(6) 釋放元素減去Q，其中Q=-ρpm，余下元素依次交叉-Q和+Q，ρpm變成第一類非基變量，ρqm變成基變量，從而得到一組新的基解并返回步驟(1)。

(7) 得到最優(yōu)解。

模型求解的算法流程如圖4所示。

圖4 模糊松弛模型求解流程

2 實驗結(jié)果分析

2.1 實驗數(shù)據(jù)來源

在青島海爾集團滾筒洗衣機生產(chǎn)流水線上進行實驗，實驗數(shù)據(jù)為實時采集的洗衣機機身照片，各個角度的圖像共5200張。部分產(chǎn)品外觀圖像如圖5所示。

圖5 部分產(chǎn)品外觀圖像

2.2 產(chǎn)品外觀缺陷種類

通過對生產(chǎn)車間現(xiàn)場的深入調(diào)研和跟蹤，再結(jié)合工業(yè)產(chǎn)品工藝特征的不同，總結(jié)出常見的外觀缺陷主要包括五大類[11-12]，分別是機身劃傷、表面凹凸坑、拼裝縫隙過大、商標(biāo)錯誤、色差這幾大類型[13]。每一類還包含不同的表現(xiàn)形式，如圖6所示。

圖6 不同種類缺陷實例

2.3 圖像特征提取與核函數(shù)的選擇

本文的研究共提取3種圖像特征，分別為SIFT特征、多尺度等價模式LBP特征和HSV顏色特征。對應(yīng)的核函數(shù)分別為高斯核函數(shù)、Sigmoid核函數(shù)、線性核函數(shù)。采用3種核函數(shù)建立缺陷分類模型。通過反復(fù)交叉驗證，確定3種核函數(shù)的最優(yōu)參數(shù)如表1所示。

表1 3種核函數(shù)最優(yōu)參數(shù)選擇

2.4 缺陷分類實驗

采用基于模糊松弛約束的多核學(xué)習(xí)方法進行缺陷分類試驗，選取5類缺陷共計200個樣本進行實驗，從樣本集中抽取100個作為訓(xùn)練集，其他的作為測試集，步驟如下：

① 對樣本圖像進行預(yù)處理；

② 提取樣本圖像的圖像特征；

③ 選取核函數(shù)，采用模糊松弛約束的求解方法，得到組合核函數(shù)；

④ 進行分類實驗，得到結(jié)果。

2.5 單核學(xué)習(xí)與多核學(xué)習(xí)對比

采用多核學(xué)習(xí)方法，針對多特征融合以后，單一核函數(shù)不能滿足分類需求，為了對比單核系統(tǒng)與多核系統(tǒng)性能上的差異，采用對照實驗法，分別單獨采用3種核函數(shù)建立分類模型對缺陷分類，再采用組合核函數(shù)對缺陷分類。分類結(jié)果如表2所示，其中百分?jǐn)?shù)表示分類精確度。

表2 單核學(xué)習(xí)方式與多核學(xué)習(xí)方式實驗對比

其中，4種分類方式仿真圖像如圖7～圖10所示,其中圖中的“+”、“*”、“?”和“□”均表示測試數(shù)據(jù)，“?”表示被分類到1中的測試數(shù)據(jù)，“□”表示被分類到2中的測試數(shù)據(jù)，支持向量被“○”圈出。

圖7 線性核函數(shù)

圖8 Sigmoid核函數(shù)

圖9 高斯核函數(shù)

圖10 組合核函數(shù)

根據(jù)表1中的確定參數(shù)，確定組合核函數(shù)。經(jīng)過反復(fù)交叉驗證，該組合函數(shù)得到的分類曲線沒有過擬合現(xiàn)象，分類效果較好。

2.6 實測結(jié)果

在系統(tǒng)運行時間段內(nèi)，誤檢率可以控制在10%左右，為了能夠得到更加真實客觀的測試數(shù)據(jù)，采用隨機抽樣法。統(tǒng)計某一時段內(nèi)，連續(xù)檢測500臺洗衣機的各類缺陷的漏檢率與誤檢率統(tǒng)計，如表3所示。表中檢測的缺陷種類包括五大類，每一大類中還有若干小類，同時，還給出了出現(xiàn)次品以后的差異原因、處理方式以及解決方案。

2.7 有無松弛約束實驗對比

引入松弛邊界是為了能夠解決大尺寸工業(yè)產(chǎn)品不同表面需要采用不同的檢測標(biāo)準(zhǔn)的問題，即根據(jù)所需要的標(biāo)準(zhǔn)，選擇松弛邊界的上邊界或是下邊界。為了驗證加入松弛邊界后系統(tǒng)性能的優(yōu)越性，實驗對比了模糊松弛約束求取核函數(shù)權(quán)重與模糊約束求取權(quán)重后的實測結(jié)果，結(jié)果表明，模糊松弛約束的性能要優(yōu)于模糊約束。

表3 檢測結(jié)果統(tǒng)計

3 結(jié)束語

本文提出了一種基于模糊松弛約束的多核學(xué)習(xí)缺陷分類方法。首先提取圖像特征，針對每種特征選擇適合的核函數(shù)，然后給每個特征劃定一個模糊邊界，根據(jù)一件產(chǎn)品不同表面的不同檢測需求的特點，為模糊邊界劃定上下松弛邊界以提高系統(tǒng)性能，最后求解模型得到分類結(jié)果。實驗對比了單核系統(tǒng)與多核系統(tǒng)的分類精度，以及有無松弛約束的檢測性能，實驗結(jié)果表明，多核系統(tǒng)的分類精度要高于單核系統(tǒng)，當(dāng)系統(tǒng)中加入松弛約束時可以提高對產(chǎn)品外觀缺陷的檢測精度并且滿足外觀缺陷的實時檢測。