基于人工蜂群優(yōu)化高斯過程的運動想象腦電信號分類*

耿雪青,佘青山,韓 笑,孟 明

(杭州電子科技大學智能控制與機器人研究所,杭州 310018)

?

基于人工蜂群優(yōu)化高斯過程的運動想象腦電信號分類*

耿雪青,佘青山*,韓 笑,孟 明

(杭州電子科技大學智能控制與機器人研究所,杭州 310018)

針對傳統(tǒng)的高斯過程采用共軛梯度法確定超參數(shù)時對初值有較強依賴性且易陷入局部最優(yōu)的問題,提出了一種基于人工蜂群優(yōu)化的高斯過程分類方法,用于腦電信號的模式識別。首先,構建高斯過程模型,選擇合適的核函數(shù)且確定待優(yōu)化的參數(shù)。然后,選取識別錯誤率的倒數(shù)為適應度函數(shù),使用人工蜂群算法搜索尋找出限定范圍內(nèi)可以取得最優(yōu)準確率的超參數(shù)。最后,采用參數(shù)優(yōu)化后的高斯過程分類器對樣本分類。分別采用2008年競賽數(shù)據(jù)集BCI Competition Ⅳ Data Set 1和2005年數(shù)據(jù)集BCI Competition Ⅲ Data Set Ⅳa對所提方法進行驗證,并與支持向量機(SVM)、人工蜂群優(yōu)化的支持向量機(ABC-SVM)、高斯過程分類(GPC)方法進行比較,實驗結果表明了所提方法的有效性。

腦電信號,高斯過程分類,人工蜂群,運動想象

腦-機接口技術BCI(Brain Computer-Interface)是不依賴于外周神經(jīng)和肌肉系統(tǒng),直接在大腦和計算機或其他外部設備建立交流和控制通路的人機交互系統(tǒng)[1]。BCI技術已經(jīng)引起了國內(nèi)外研究學者的廣泛關注,在助殘、康復、輔助控制、娛樂等領域有很好的應用前景。模式分類是腦-機接口的關鍵技術之一,由于腦電信號具有微弱性、復雜性和非平穩(wěn)性等特點,有效、快速地區(qū)分不同意識活動的成分是腦電信號處理領域面臨的嚴峻挑戰(zhàn)。因此,探索有效的分類方法以提高腦電模式分類精度仍然具有重要意義。

目前,國內(nèi)外研究者提出很多腦電模式分類的方法,主要包括線性判別分析(LDA)[2]、人工神經(jīng)網(wǎng)絡(ANN)[3]、支持向量機(SVM)[4]等。實踐證明,LDA實現(xiàn)簡單,計算量小,分類速度快,在處理類別較少的EEG數(shù)據(jù)時可取得較好的分類效果。但在處理小樣本問題時,容易陷入維數(shù)災難[2]。ANN的神經(jīng)元連接比較靈活,所以該方法有較高的擬合度和較強的自適應性,它在解決非線性復雜問題時表現(xiàn)出較大的優(yōu)越性,但是該方法容易陷入局部最小,存在過學習現(xiàn)象[3]。SVM以統(tǒng)計學習理論為基礎,巧妙地引入了核函數(shù),避免了維數(shù)災難現(xiàn)象,且能有效地解決小樣本、非線性問題[4]等,因此在腦電信號分類中得到廣泛的應用[4-6]。

2006年,Rasmussen等人[7]通過理論研究和數(shù)值實驗分析,正式提出了高斯過程機器學習方法。高斯過程是基于貝葉斯框架的新型核學習機,除了對于高維數(shù)、小樣本、非線性等復雜問題具有較好適應性的優(yōu)點外,它還具有容易實現(xiàn)、預測輸出具有概率意義的優(yōu)點[8]。所以該方法一經(jīng)提出,就在機器學習領域引起廣泛關注,并被用于回歸和分類[8-9],但在腦電信號分類研究中應用較少。另外,高斯過程的預測和分類精度都依賴于超參數(shù)的選擇,傳統(tǒng)的高斯過程采用共軛梯度法確定超參數(shù),其效果依賴于初值,易陷入局部最優(yōu)。為此,文獻[10]將遺傳算法與高斯過程回歸算法相結合,用于隧道圍巖變形預測,提高了預測精度。文獻[11]采用粒子群算法優(yōu)化高斯過程回歸模型的超參數(shù),用于滑坡位移時序的預測,取得了較好的效果。文獻[12]采用人工蜂群算法搜索高斯過程回歸模型的最優(yōu)超參數(shù),用于海上遠程精確打擊體系作戰(zhàn)效能評估中。

針對上述情況,本文基于智能優(yōu)化與機器學習相集成的思想,提出了一種基于人工蜂群優(yōu)化高斯過程的運動想象腦電信號分類算法。首先通過實驗對比了遺傳算法、粒子群算法、人工蜂群算法的性能特點,然后將綜合性能較優(yōu)的人工蜂群算法與高斯過程分類算法相結合,用于腦電信號的特征分類中,最后分別在兩屆腦電公開競賽數(shù)據(jù)集上進行實驗測試,驗證了所提算法的有效性。

1 實驗數(shù)據(jù)

本文分別采用2008年第四屆腦-機接口競賽數(shù)據(jù)BCI Competition Ⅳ Data Set 1和2005年第3屆競賽數(shù)據(jù)BCI Competition Ⅲ Data Set Ⅳa進行實驗分析。

①BCI Competition Ⅳ Data Set 1。該數(shù)據(jù)集由Berlin BCI研究組提供[13]。4名受試者(a,b,f和g)按照要求執(zhí)行運動想象任務,共采集59個通道的腦電信號數(shù)據(jù),采樣頻率為100 Hz。每個受試者按照電腦屏幕的提示執(zhí)行左手、右手、單或雙腳3種運動想象任務中的兩類。具體而言,受試者a和f想象左手、腳運動,受試者b和g想象左手、右手運動。每個受試者執(zhí)行200次實驗,每類任務執(zhí)行100次。

②BCI Competition Ⅲ Data Set Ⅳa。該數(shù)據(jù)集也由Berlin BCI研究組提供[14]。5名受試者(aa,al,av,aw和ay)根據(jù)顯示器提示完成右手和腳的運動想象任務。采用118個通道記錄腦電信號數(shù)據(jù),采樣頻率為100 Hz。每個受試者執(zhí)行280次實驗,其中訓練集的試驗次數(shù)分別為168,224,84,56 和28,而剩下的實驗數(shù)據(jù)作為測試集。

2 方法

2.1 高斯過程分類

高斯過程(Gaussian Process)是指任意時刻點集的聯(lián)合概率密度都服從高斯分布的隨機過程[15]。高斯過程可以看作是對高斯分布的特殊推廣,只是前者是基于向量,而后者是針對函數(shù)。所以,高斯過程也像高斯分布一樣,完全由它的均值和協(xié)方差確定。

簡言之,高斯過程分類就是通過建立高斯過程模型得到后驗概率以實現(xiàn)對樣本特征的類別判斷。其中,高斯過程模型的建立主要由潛變量函數(shù)(Latent Function)的定義、似然函數(shù)的定義以及潛變量函數(shù)后驗概率計算三部分組成[16]。

對于腦電樣本集D={(xi,yi),i=1,2,…,m},xi,為腦電特征樣本,yi為類別標簽,取值范圍為{1,-1},m為特征值個數(shù),可定義潛函數(shù)f(x)來描述xi和yi的映射關系:

p(y=1|x)=φ[f(x)]

(1)

式中:φ為累積高斯(CumulativeGaussian)函數(shù),一般取Sigmoid函數(shù)[16],以保證概率值落在[0,1]區(qū)間。f(x)服從高斯過程,可表示為f(x,θ)～GP(0,K)。該高斯過程均值為0,K為正定協(xié)方差函數(shù),也叫核函數(shù),θ為核函數(shù)K的參數(shù)。由于潛函數(shù)的觀測數(shù)據(jù)是相互獨立的伯努利分布變量,其似然函數(shù)為:

(2)

潛函數(shù)f(x)的先驗分布可描述為:

p(f|x,θ)=N(0,K)

(3)

式中:K為核函數(shù),θ為超參數(shù),在高斯建模的訓練學習過程中選擇合適的核函數(shù)和最優(yōu)超參數(shù)尤為重要,因為它們影響高斯過程分類器的性能。最常用的核函數(shù)為高斯核函數(shù),可表示為:

(4)

式中:σf和lk共同組成超參數(shù)θ,d表示數(shù)據(jù)的維數(shù)。σf為潛函數(shù)的幅值,用于衡量局部相關的程度。lk為長度尺度(LengthScale),用于衡量第k維數(shù)據(jù)兩兩之間的相關性。確定了潛函數(shù)的先驗分布和似然函數(shù),則可以根據(jù)貝葉斯公式,得到潛函數(shù)f(x)的后驗分布:

(5)

對于給定的測試樣本x*,可以計算與之對應的潛函數(shù)f*的條件概率:p(f*|D,θ,x*)=∫p(y*|f,x,θ,x*)p(f|D,θ)df

(6)

進一步得到測試樣本x*的類別判斷概率y*:

p(y*|D,θ,x*)=∫p(y*|f*)p(f*|D,θ,x*)df

(7)

值得注意的是,由于定義的似然函數(shù)p(y|f)為Sigmoid函數(shù),所以不能通過求解得到潛函數(shù)的后驗分布和類別判斷概率。針對此問題,可以采用近似逼近的方法取代原來的后驗分布值進行計算,得到對應樣本的類別概率。

2.2 人工蜂群算法

人工蜂群算法(ArtificialBeeColony,ABC)[17-18]是受蜂群協(xié)作采蜜行為啟發(fā)而提出的一種群智能算法。該算法的生物學依據(jù)是:雇傭蜂,觀察蜂和偵察蜂通過信息交流和角色轉換的方式完成在不同自然環(huán)境下的采蜜。由于該算法原理簡單,易于實現(xiàn),適應性強,所以它引起了廣大學者的關注,并被成功用于優(yōu)化問題中[18]。

人工蜂群算法是一個迭代尋優(yōu)的過程。對于具體的優(yōu)化問題,生物學中蜂群采蜜行為涉及的因素被抽象為相應的變量,信息交流行為被抽象為相應的算法控制。其中,蜜源被抽象為潛在解,蜜源的數(shù)量即為解空間的大小,蜜源的質(zhì)量好壞對應解的適應度的高低。雇傭蜂發(fā)現(xiàn)蜜源并分享信息的行為被抽象為潛在解的搜尋和適應度計算。觀察蜂根據(jù)雇傭蜂提供的信息選擇蜜源的行為被抽象為:根據(jù)適應度的概率公式進行輪盤賭選擇,以提高算法收斂速度。雇傭蜂在蜜源質(zhì)量沒有改善時,則放棄現(xiàn)有蜜源,本身轉變?yōu)閭刹旆涞男袨?被抽象為:潛在解經(jīng)過多次迭代到達閾值卻沒得到改善,則放棄當前解,以避免算法陷入局部最優(yōu)解。另外,在算法實現(xiàn)時,在不影響算法的思想的前提下,自然采蜜的行為細節(jié)被理想化。即假設雇傭蜂和觀察蜂數(shù)目相等均為蜂群種族大小的一半,等于蜜源的數(shù)量,且同一時間內(nèi)一個蜜源只被一只引領蜂開采[17]。

針對給定的尋優(yōu)問題,人工蜂群算法的具體步驟如下:

①初始化:初始化蜂群大小為CS,目標函數(shù)最大評估次數(shù)為MCN,最大不更新次數(shù)為Limit,待優(yōu)化參數(shù)個數(shù)為Dim,待優(yōu)化參數(shù)的上下限分別為ub,lb。則解空間大小為CS,雇傭蜂和觀察蜂的個數(shù)為CS/2。初始化解為:

xij=lb+rand(ub-lb)

(8)

式中:rand表示范圍為[0,1]的均勻隨機分布,i∈{1,2,…,CS},j∈{1,2,…,Dim}。

②雇傭蜂搜尋新解:雇傭蜂在初始解xij鄰域產(chǎn)生新的解vij:

vij=xij+φ(xij-xkj)

(9)

式中:i≠k,φ=(rand-0.5)×2,即為范圍為[-1,1]的均勻隨機分布。通過目標函數(shù)和適應度函數(shù)計算出初始解和新解的適應度,如果新解的適應度高于初始解,則用新解代替初始解,否則不取代。

③觀察蜂選擇解:觀察蜂根據(jù)雇傭蜂傳遞的解和適應度信息計算選擇解的概率:

(10)

式中:fiti表示第i解的適應度。觀察蜂根據(jù)輪盤賭的方式選擇解。接著觀察蜂在所選解周圍的鄰域周圍產(chǎn)生新解,計算選擇解和新解的適應度后,依據(jù)貪心算法在新解和原選擇解之間選擇解。

④偵察蜂出現(xiàn):如果某些解經(jīng)過Limit次循環(huán)后,解的質(zhì)量沒有改善,則雇傭蜂轉換角色為偵察蜂,放棄該解并隨機產(chǎn)生新解,且將迭代次數(shù)置0。

⑤結束算法:判斷迭代次數(shù)是否達到MCN,若未達到,則算法正常進行,否則結束算法,輸出最優(yōu)解。

2.3 人工蜂群優(yōu)化高斯過程算法

如前所述,人工蜂群算法是一種基于全局搜索的群智能算法,具有操作簡單,搜索精度較高和魯棒性較強的優(yōu)點。利用ABC優(yōu)化GPC,確定最優(yōu)的核函數(shù)構建模型,能使GPC表現(xiàn)出更優(yōu)越的性能。

使用ABC-GPC算法時,高斯過程的核函數(shù)選擇如式(4),人工蜂群主要用來優(yōu)化高斯過程核函數(shù)的超參數(shù)σf和lk。但是,由于σf主要影響信號的幅值,而高斯分類的輸出結果被Sigmoid函數(shù)限定在[0,1]范圍內(nèi),所以σf對分類結果并沒有大的影響[19],可以把σf值設為1。所以,本文使用人工蜂群只對參數(shù)lk進行優(yōu)化以提升分類精度。

由于優(yōu)化高斯過程分類核函數(shù)中超參數(shù)的目的是為了獲得更高的分類準確率,所以依據(jù)人工蜂群中選擇適應度函數(shù)的基本形式,在腦電信號模式分類時選取的適應度函數(shù)為:

(11)

式中:acc為腦電訓練特征樣本送入高斯過程分類器得到的分類正確率。

圖1 人工蜂群優(yōu)化高斯過程算法流程

對于給定的運動想象腦電數(shù)據(jù),基于人工蜂群優(yōu)化的高斯過程分類算法的流程如圖1所示。

具體算法步驟如下:①根據(jù)高斯過程參數(shù)優(yōu)化需求,初始化人工蜂群算法的控制參數(shù);②將人工蜂群的初始解和腦電訓練特征樣本集送入高斯過程分類器,得到分類準確率,進一步計算適應度;③雇傭蜂在初始解周圍產(chǎn)生新解,計算適應度,然后根據(jù)貪心算法,在新解和初始解中選擇確定的解;④觀察蜂從雇傭蜂處得到解并通過輪盤賭的方式選擇解,并計算適應度和生成新解,根據(jù)貪心算法確定優(yōu)化解;⑤偵察蜂出現(xiàn)以避免算法陷入局部最優(yōu);⑥人工蜂群算法迭代結束,將求得的最優(yōu)解保存;⑦將腦電測試特征樣本集送入高斯過程分類器,將上一步保存的最優(yōu)解送入核函數(shù),即可得到最后的分類結果。

2.4 基于優(yōu)化高斯過程的腦電分類

根據(jù)腦電信號易受噪聲干擾以及非線性等特點,對運動想象腦電信號的處理主要包括預處理、特征提取和分類階段。

①預處理:巴特沃斯濾波器是數(shù)字帶通濾波器的一種,由于其實現(xiàn)簡單,濾波效果顯著,所以在運動想象腦電信號的預處理中得到廣泛應用[20],受試者進行運動想象時,大腦中心區(qū)域會產(chǎn)生8 Hz～12 Hz的mu節(jié)律和18 Hz～25 Hz的beta節(jié)律,本文使用8 Hz～30 Hz[21]的5階巴特沃茲濾波器對腦電數(shù)據(jù)進行濾波。

②特征提取:共同空間模式CSP(Common Spatial Pattern)算法是一種廣泛用于腦電信號處理的特征提取方法[21]。CSP算法對協(xié)方差矩陣同時聯(lián)合對角化,尋找一組空間濾波器,使經(jīng)過它濾波投影后的信號中一類信號的方差最大,另一類信號的方差最小,從而最大化類間樣本的距離。本文使用CSP提取腦電信號的特征向量。在實驗中,CSP的濾波器個數(shù)m取3。

③分類:對于提取出的腦電特征數(shù)據(jù),分別使用支持向量機(SVM)、基于人工蜂群優(yōu)化的支持向量機(ABC-SVM)、高斯過程分類器(GPC)、基于人工蜂群優(yōu)化的高斯過程分類器(ABC-GPC)對不同任務下的腦電特征向量進行識別和比對分析。其中,采用ABC-GPC時,人工蜂群算法的適應度函數(shù)如式(11),參數(shù)設置如表1所示。

表1 人工蜂群的參數(shù)設置

3 實驗結果與分析

為了證明方法的有效性,本文采用兩個實驗:實驗一是用來表明人工蜂群算法及其他優(yōu)化算法的性能優(yōu)劣;實驗二是對比人工蜂群優(yōu)化的高斯過程分類方法用于腦電信號的分類時,與其他常用或者優(yōu)化過的分類方法的優(yōu)劣。

3.1 優(yōu)化算法實驗

為了評估人工蜂群算法的性能,選擇函數(shù)優(yōu)化領域中兩個具有代表性的標準測試函數(shù)進行測試。表2給出了所選擇函數(shù)的表達式及相應特性。

表2 測試函數(shù)

圖2 測試函數(shù)三維波形圖

在表2中f1(x)為連續(xù)性單模態(tài)函數(shù),只有一個全局最小值沒有局部最小值,用來測定算法的尋優(yōu)精度,f2(x)為復雜的多模態(tài)模型,有多個局部最小值,全局最優(yōu)值難以確定,所以可用來驗證算法是否易陷入局部最優(yōu)。為了更直觀的說明兩個函數(shù)的特點,做出函數(shù)的三維波形圖如圖2所示。

分別用遺傳算法、粒子群算法和人工蜂群算法對上面所提測試函數(shù)求最優(yōu)值。其中,粒子群和人工蜂群算法的參數(shù)設置依據(jù)參考文獻[22],遺傳算法的種群規(guī)模取100,最大迭代次數(shù)為100 000,交叉概率為0.5,變異概率為0.045,競賽規(guī)模為10。每種算法重復實驗30次,將30次最優(yōu)值的平均值記錄如下表。為了簡化表示,當數(shù)值小于10-12時,則將該值記作0值。上面3種算法的最優(yōu)值結果如表3所示。

表3 優(yōu)化算法的最優(yōu)值結果

觀察表3的實驗結果,可以發(fā)現(xiàn):對于函數(shù)f1(x)3種算法的尋優(yōu)性能,人工蜂群優(yōu)于粒子群,粒子群優(yōu)于遺傳算法,說明人工蜂群的優(yōu)化精度最高。對于函數(shù)f2(x),粒子群算法最優(yōu)值求取結果不僅差于人工蜂群算法和遺傳算法,且與實際最優(yōu)值有較大差距,說明該算法可能已陷入局部最優(yōu)值。遺傳算法較粒子群算法表現(xiàn)出較好的全局尋優(yōu)能力,而人工蜂群算法很好的避免了此問題,所以本文提出使用人工蜂群算法優(yōu)化高斯過程分類器的方法。

3.2 腦電分類實驗

為了說明所提算法的有效性,本文采用BCI Competition Ⅳ Data Set 1和BCI Competition Ⅲ Data Set Ⅳa兩組數(shù)據(jù)集對所提分類算法進行驗證。

3.2.1 數(shù)據(jù)集Data Set 1

選用受試者a,b,f和g提供的4組標定數(shù)據(jù)進行實驗。參照文獻[20]的推薦,本文選取11個通道的腦電信號進行分析,分別對應10～20系統(tǒng)的“FC3”,“FC4”,“Cz”,“C3”,“C4”,“C5”,“C6”,“T7”,“T8”,“CCP3”,“CCP4”,時間段選取4 s,每個受試者執(zhí)行200次實驗,每類任務執(zhí)行100次,從200次實驗中隨機取120次用于訓練學習,80次用作測試集。

圖3 ABC-GPC方法100次分類實驗準確率統(tǒng)計

為了使實驗結果更直觀,將本文所提的分類方法得出的分類準確率統(tǒng)計結果以箱線圖的方式畫出,如圖3所示。其中,箱線圖的兩端分別表示分類過程重復進行100次迭代后準確率的上下四分數(shù),中線表示準確率的中位數(shù),“+”表示溫和異常點。

觀察圖3可知,除了受試者b的準確率略低外,其他3名受試者的中位線都在0.85以上,說明平均準確率結果都高于85%,其中最高準確率接近92%。

為了進一步說明所提方法的效果,將4種分類方法的分類過程重復試驗100次,并統(tǒng)計平均準確率和標準差如圖4所示。圖4中,柱狀圖表示平均準確率,柱狀圖上的短線表示標準差。

圖4 數(shù)據(jù)集Data Set 1中不同受試者的分類準確率

從圖4中可以看出,不同的分類方法對于不同受試者的分類性能優(yōu)劣不同。圖4中對于受試者b和f,ABC-GPC算法的平均準確率相對傳統(tǒng)GPC方法分別提高了0.2%、0.3%;對于受試者a,這兩種方法分類準確率接近,且都高于SVM和ABC-SVM;對于受試者g,ABC-GPC略低于其他3種方法,總體而言ABC-GPC算法比其他3種分類算法具有相對更好的分類性能。

3.2.2 數(shù)據(jù)集Data Set Ⅳa

本文選用aa,av,ay 3個受試者,他們的實驗次數(shù)如表4所示。

表4 3個受試者實驗次數(shù)說明

圖5 數(shù)據(jù)集Data Set IVa中不同受試者的分類準確率

圖5給出的分類結果,不管是對于受試者個體還是平均值而言,GPC類方法的分類準確率都優(yōu)于SVM和ABC-SVM,這可能是因為該數(shù)據(jù)集主要是小樣本數(shù)據(jù),說明GPC類方法在處理小樣本問題時,表現(xiàn)出比SVM更好的性能。另外,ABC-GPC法的平均準確率比傳統(tǒng)GPC方法高2%,尤其是在受試者av上表現(xiàn)出了明顯優(yōu)越性。

4 結論

模式分類是運動想象EEG信號識別不同想象任務的關鍵環(huán)節(jié)之一。高斯過程算法雖然分類優(yōu)勢明顯并且被廣泛的研究,但是該算法具有優(yōu)化迭代次數(shù)不確定,容易陷入局部最優(yōu)的局限性。本文提出了人工蜂群算法優(yōu)化高斯過程的運動想象腦電信號分類方法。使用人工蜂群優(yōu)化高斯過程的超參數(shù),以克服傳統(tǒng)高斯過程超參數(shù)的優(yōu)化迭代次數(shù)不確定,且易陷入局部最優(yōu)的特點。使用兩組腦電競賽數(shù)據(jù)集對所提方法進行實驗,并與SVM、ABC-SVM以及傳統(tǒng)高斯過程方法進行對比,結果表明本文方法能較好地對腦電信號進行分類,因此該方法可以作為一種有效的腦電信號分類方法。然而,本文所提的分類方法在使用人工蜂群優(yōu)化參數(shù)時,耗時較大,這是后期算法研究需要改進的地方。

[1] Marchetti M,Priftis K. Brain-Computer Interfaces in Amyotrophic Lateral Sclerosis:A Metanalysis[J]. Clinical Neurophysiology,2015,126(6):1255-1263.

[2] Chen W,Shen C P,Chiu M J,et al. Epileptic EEG Visualization and Sonification Based on Linear Discriminate Analysis[C]//Proceedings of IEEE Conference on Engineering in Medicine and Biology Society,Milan,2015:4466-4469.

[3] Zuhair M,Thomas S,Keshri A K,et al. Automatic Identification of an Epileptic Spike Pattern in an EEG Signals Using ANN[C]//Proceedings of the 3rd International Conference on Soft Computing for Problem Solving,India,2014:915-923.

[4] Brinkmann B H,Patterson E E,Vite C,et al. Forecasting Seizures Using Intracranial EEG Measures and SVM in Naturally Occurring Canine Epilepsy[J]. PloS One,2015,10(8):e0133900.

[5] 王金甲,賈敏. 輔助訓練的半監(jiān)督線性支持向量機用于EEG分類[J]. 儀器儀表學報,2013,34(4):768-773.

[6] 張毅,祝翔,羅元,等. 基于SVM多分類技術的肌電輔助腦電智能輪椅控制系統(tǒng)[J]. 中南大學學報(自然科學版),2013,S2:73-76.

[7] Rasmussen C E,Williams C K I. Gaussian processes for machine learning[M]. The MIT press,Cambridge,2006.

[8] 夏戰(zhàn)國,夏士雄,蔡世玉,等. 類不均衡的半監(jiān)督高斯過程分類算法[J]. 通信學報,2013,34(5):42-51.

[9] 何志昆,劉光斌,趙曦晶,等. 高斯過程回歸方法綜述[J]. 控制與決策,2013,28(8):1121-1129.

[10] 劉開云,方昱,劉保國,等. 隧道圍巖變形預測的進化高斯過程回歸模型[J]. 鐵道學報,2011,33(12):101-106.

[11] 徐沖,劉保國,劉開云,等. 基于粒子群-高斯過程回歸耦合算法的滑坡位移時序分析預測智能模型[J]. 巖土力學,2011,32(6):1669-1675.

[12] 張樂,劉忠,張建強,等. 基于人工蜂群算法優(yōu)化的改進高斯過程模型[J]. 國防科技大學學報,2014,36(1):154-160.

[13] Blankertz B,Dornhege G,Krauledat M,et al. The Non-Invasive Berlin Brain-Computer Interface:Fast Acquisition of Effective Performance in Untrained Subjects[J]. NeuroImage,2007,37(2):539-550.

[14] Dornhege G,Blankertz B,Curio G,et al. Boosting Bit Rates in Non-Invasive EEG Single-Trial Classifications by Feature Combination and Multi-Class Paradigms[J]. IEEE Transactions on Biomedical Engineering,2004,51(6):993-1002.

[15] Wang B,Wan F,Mak P U,et al. EEG Signals Classification for Brain Computer Interfaces Based on Gaussian Process Classifier[C]//Proceedings of IEEE Conference on Information,Communications and Signal Processing,Macau,2009:1-5.

[16] 賀建軍,張俊星,賈思齊,等. 一種新高斯過程分類算法[J]. 控制與決策,2014,29(9):1587-1592.

[17] Karaboga D. A Idea Based on Bee Swarm for Numerical Optimization[R]. Türkiye:zErciyes University,Engineering Faculty,Computer Engineering Department,2005.

[18] Sharma M,Chandra S. Application of Artificial Bee Colony Algorithm for Numerical Optimization Technique[C]//Proceedings of IEEE Conference on Advance Computing,Banglore,2015:1267-1272.

[19] Wang L,Li C. Spectrum-Based Kernel Length Estimation for Gaussian Process Classification[J]. IEEE Transactions on Cybernetics,2014,44(6):805-816.

[20] Samek W,Kawanabe M,Muller K R. Divergence-Based Framework for Common Spatial Patterns Algorithms[J]. IEEE Transactions on Biomedical Engineering,2014(7):50-72.

[21] Park C,Looney D,Ahrabian A,et al. Classification of Motor Imagery BCI Using Multivariate Empirical Mode Decomposition[J]. IEEE Transactions on Neural Systems and Rehabilitation Engineering,2013,21(1):10-22.

[22] Karaboga D,Basturk B. On the Performance of Artificial Bee Colony(ABC)Algorithm[J]. Applied Soft Computing,2008,8(1):687-697.

耿雪青(1990-),女,山西陽泉人,碩士,從事腦電信號處理與分析方面的研究,1565782139@qq.com;

佘青山(1980-),男,湖北荊州人,副教授,博士,從事生物信息處理與分析、機器學習、模式識別等領域的研究,qsshe@hdu.edu.cn;

韓 笑(1991-),女,河南南陽人,碩士,從事腦電信號處理與分析方面的研究,754163862@qq.com。

Classification of Motor Imagery EEG Based on Gaussian Process Optimized with Artificial Bee Colony*

GENGXueqing,SHEQingshan*,HANXiao,MENGMing

(Institute of Intelligent Control and Robotics,Hangzhou Dianzi University,Hangzhou 310018,China)

The conjugate gradient method is used to determine the parameters in the traditional Gaussian process. However,the conjugate gradient method has a strong dependence on the initial value and is easy to fall into local optimum. In order to solve the problem,a Gaussian process classification(GPC)method is proposed based on artificial bee colony(ABC)optimization and applied for pattern recognition of EEG signals. Firstly,Gaussian process model is constructed,and suitable kernel function is chosen and the parameters to be optimized are specified. Then the reciprocal of the recognition error rate is selected as fitness function,and the parameters which are used to obtain optimal accuracy in a limited range are found out by employing the ABC algorithm. Finally,the Gaussian process classifier with optimized parameters is used to classify the samples. The efficiency of the propose method has been demonstrated by comparison with support vector machine(SVM),support vector machine optimized with Artificial bee colony(ABC-SVM)and GPC algorithms on both BCI Competition Ⅳ Data Set 1 in 2008 and BCI Competition Ⅲ Data Set Ⅳa in 2005.

EEG signal,Gaussian process classification,artificial bee colony,motor imagery

項目來源:浙江省自然科學基金資助項目(LY15F010009,LY14F030023);國家自然科學基金資助項目(61201302)

2016-07-01 修改日期:2016-11-18

TP391

A

1004-1699(2017)03-0378-07

C:7230

10.3969/j.issn.1004-1699.2017.03.008