胥立波，蔣愛民，劉小峰，王 勇

（河海大學物聯(lián)網(wǎng)工程學院，常州213022）

基于Fisher字典學習的運動想象腦電分類算法*

胥立波，蔣愛民，劉小峰，王 勇

（河海大學物聯(lián)網(wǎng)工程學院，常州213022）

目前，在以運動想象為基礎(chǔ)的腦機接口研究中，共空域模式（Common Spatial Pattern，CSP）算法是腦電信號特征提取的主流算法，而如何利用CSP特征進行有效的分類則是該領(lǐng)域的研究熱點之一。近年來，基于稀疏表示的分類方法（Sparse Representation based Classification，SRC）備受國內(nèi)外學者的高度關(guān)注，為此提出了一種基于Fisher字典學習的腦電分類算法。該算法結(jié)合了稀疏重構(gòu)誤差和編碼系數(shù)進行分類。實測數(shù)據(jù)的處理結(jié)果表明，與傳統(tǒng)基于SRC的分類算法相比，所提出的算法能夠取得更為精確的分類結(jié)果。

腦機接口；共空域模式；稀疏表示；Fisher字典學習；重構(gòu)誤差；編碼系數(shù)

1 引 言

近年來，隨著計算機科學、腦科學以及人機交互領(lǐng)域的快速發(fā)展，腦機接口（Brain－Computer Interfaces，BCI）技術(shù)越來越受到國內(nèi)外學者的關(guān)注，它繞開外周神經(jīng)和肌肉組織，實現(xiàn)人腦與計算機或其他外部電子設(shè)備的直接通信，也就是說僅通過人的大腦思維活動即可與外界交流信息，而不需要通過實際動作［1］。這項技術(shù)對于那些有肢體運動障礙的癱瘓病人無疑是一個福音。除了在醫(yī)療康復領(lǐng)域，在軍事和生活娛樂等方面，BCI也有著獨特的應(yīng)用［2］。

特征提取和分類是腦機接口系統(tǒng)中最主要的兩部分。當前，運動想象腦電信號的特征提取主要采用共空域模式（Common Spatial Pattern，CSP）算法［3］；而對于特征分類這一塊，也有許多經(jīng)典算法，如線性判別分析LDA［4］，支持向量機SVM［5］等。

基于稀疏表示的分類技術(shù)（Sparse Representation based Classification，SRC）是近年來新興的信號處理方法，在人臉識別、圖像分類等領(lǐng)域［6］得到了廣泛應(yīng)用。對于運動想象腦電信號分類問題，文獻［7］直接把腦電信號的訓練特征樣本作為字典，然后根據(jù)l1范數(shù)最小化原則［8］求解出表示系數(shù)，最后根據(jù)求得的系數(shù)計算重構(gòu)誤差來確定樣本所屬的類別。這一分類方法只需要利用高效稀疏編碼算法求得表示系數(shù)，而無需對字典進行訓練，因而計算效率較高。但是，這種方法并不能充分利用各類別間的結(jié)構(gòu)信息，丟失了許多隱藏在訓練樣本中的分類信息。此外，由于實測數(shù)據(jù)常常受到噪聲感染，直接把訓練樣本作為字典進行分類會使得求解系數(shù)并非如設(shè)想的那樣稀疏。針對這些問題，提出一種基于Fisher字典學習［9］的方法，通過Fisher鑒別準則［10］從訓練樣本中學習出一個具有很好判別性能的結(jié)構(gòu)化字典，該字典中各類子字典能夠較好地表示該類中的樣本，而對于類外樣本則具有較弱的表示能力。與此同時，在計算每一個樣本特征稀疏表示系數(shù)時增加Fisher判別準則項，保證其稀疏表示系數(shù)具有較大的類間誤差和較小的類內(nèi)誤差，因此在對測試樣本進行分類的時候可以綜合樣本的重構(gòu)誤差值和編碼系數(shù)兩者進行分類。測試結(jié)果表明，所提出的分類算法比傳統(tǒng)的SRC算法具有更佳的效果。

2 基于稀疏表示的分類方法

傳統(tǒng)SRC算法假設(shè)同一類的測試樣本可以由同類中的訓練樣本線性表示，因此它將訓練樣本直接作為字典，然后利用此字典求解稀疏表示系數(shù)，最后根據(jù)重構(gòu)誤差最小原則進行分類。

文獻［7］中，對于給定的兩類EEG信號，首先用CSP算法對每個樣本進行特征提取，其算法思想是利用代數(shù)上矩陣同時對角化的原理，構(gòu)造出一組空域濾波器，使得濾波后的兩類信號一類方差達到極大，另一類方差達到極小，從而達到分類的目的。通過CSP算法提取的特征向量可以表示為：

其中，D1和D2分別表示兩類訓練樣本的特征向量集，k1和k2分別表示兩類訓練樣本的個數(shù)，n為特征向量的維數(shù)。然后把D＝［D1，D2］∈作為字典，D的每一列都是字典的原子。同樣用CSP算法提取出一個測試樣本的特征向量，表示為y。根據(jù)之前的假設(shè)，則有：

最后，根據(jù)式（4）求得的系數(shù)計算兩類樣本信號重構(gòu)后的誤差值，誤差值小的那一類就作為樣本的類別。公式如下：

其中，ac（X）函數(shù)的功能是選出第c類原子對應(yīng)的系數(shù)，并將另一類的系數(shù)置零。從文獻［7］中的實驗結(jié)果可看出，相比于經(jīng)典的線性判別分析（Linear Discriminant Analysis，LDA），SRC算法在分類精度上有了一定提高。

3 提出方法

3．1 基于Fisher字典學習的分類

基于Fisher判別字典學習的分類算法，是從訓練樣本中學習出具有更好判別性能的字典，以便獲得更佳的分類效果。在原始信號經(jīng)過預處理后，利用CSP算法提取信號的特征向量，設(shè)為A＝［A1，A2，．．．，Ac］，其中c表示樣本的類別數(shù)。設(shè)字典為D＝［D1，D2，．．．，Dc］，對應(yīng)系數(shù)矩陣為X＝［X1，X2，．．．，Xc］，整個字典學習過程可以表示為以下優(yōu)化問題：

式（6）中，f（X）表示系數(shù)矩陣的判別約束項。利用Fisher準則構(gòu)建此項

其中，SW（X）表示類內(nèi)誤差，SB（X）表示類間誤差，mi和m分別表示稀疏表示系數(shù)矩陣Xi和X的均值，ni和n分別表示i類樣本和全部樣本的數(shù)目。

3．2 J（D，X）的最優(yōu)化

將式（7）和式（11）代入式（6）中，則目標函數(shù)最終表示為：

雖然，對于（D，X），式（12）表示一非凸優(yōu)化問題，但是當固定其中一個時，則可以將其轉(zhuǎn)化為一系列的凸優(yōu)化問題進行求解。具體步驟如下：

首先，固定D，逐類求解Xi，此時的目標函數(shù)可以表示為：

式（14）中，Mk和M表示k類和所有類別的平均系數(shù)矩陣。研究表明當η＞1－ni／n時，fi（Xi）為嚴格凸函數(shù)［9］。

把式（13）重新寫為：

其中，Q（Xi）＝r（A，D，Xi）＋λ2fi（Xi），τ＝λ1／2，定義，xi，k表示矩陣Xi的第k列。由于Q（Xi）嚴格凸優(yōu)化于Xi，固式（15）可以通過文獻［11］中的迭代投影算法進行求解，具體步驟為：

1．輸入：σ，τ＞0。

3．當未達到收斂條件時，令h＝h＋1，

當固定系數(shù)X時，逐類更新字典Di。此時的目標函數(shù)轉(zhuǎn)化為：

上式是一個二次規(guī)劃問題，可以通過文獻［12］中的算法進行求解，逐步更新字典原子Di。

綜上所述，F(xiàn)isher字典學習法的算法步驟可以總結(jié)如下：

Step1把訓練特征樣本歸一化后作為初始字典中的原子。

Step2固定字典矩陣D，通過迭代投影法逐步求解系數(shù)矩陣Xi。

Step3固定系數(shù)矩陣X，用文獻［12］中算法逐步求解字典原子Di。

Step4當相鄰迭代中的目標函數(shù)值J（D，X）足夠逼近，或者迭代次數(shù)達到預設(shè)值時，輸出X和D，否則返回Step2。

3．3 分類方法

利用訓練特征樣本集，上述Fisher字典學習方法可以獲得具有分類特性的字典。當給定測試樣本特征y時，求解以下問題，從而獲得表示系數(shù)：

其中，α表示待求的系數(shù)向量，γ1和γ2表示加權(quán)常量，表示對應(yīng)字典Di所求得的第子塊系數(shù)均值。最終的分類則需要計算

測試樣本類別選擇使得誤差值ei最小的那一類。

4 實驗結(jié)果分析

實驗采用上海交通大學仿腦計算與機器智能研究實驗室所公開的運動想象腦電數(shù)據(jù)（subject1和subject3）進行實驗分析（下載地址為http：／／bcmi．sjtu．edu．cn／resource．html）。每組數(shù)據(jù)集分別有60個訓練樣本（左右想象運動各30個）和140個測試樣本（左右想象運動各70個）。

首先對兩類訓練樣本和測試樣本分別提取CSP特征，然后利用所提出的Fisher字典學習方法從訓練特征樣本中學習出字典，最后再根據(jù)分類方法對測試特征樣本進行分類。直接根據(jù)指定的訓練和測試樣本進行分類的結(jié)果如表1所示。

表1 直接分類的結(jié)果

為了進一步提升分類效果，又對兩組實驗數(shù)據(jù)進行10次交叉驗證計算，實驗結(jié)果如表2所示。

從表1和表2可以看出，無論是直接按照指定訓練和測試樣本，還是采用交叉驗證的方法，所提出的分類算法性能均優(yōu)于傳統(tǒng)的SRC算法。

表2 交叉驗證的分類結(jié)果

5 結(jié)束語

針對傳統(tǒng)SRC方法在字典構(gòu)造過程中不能充分利用隱藏在樣本中的分類信息問題，提出了Fisher字典學習方法。該方法在構(gòu)建結(jié)構(gòu)化字典方面具有兩個優(yōu)勢：第一，每個子字典都有較好的表示對應(yīng)類別樣本的能力；而對于不同類別的樣本，每類子字典則具有較大的重構(gòu)誤差。第二，通過該字典求解出的系數(shù)矩陣具有較小的類內(nèi)誤差和較大的類間誤差。因此在分類時，可綜合利用重構(gòu)誤差和表示系數(shù)對測試樣本進行更為精確的分類。實驗結(jié)果表明，相比傳統(tǒng)的SRC算法，所提出的分類算法具有更高的分類準確率。

［1］ Wolpaw J R，Birbaumer N，Heetderks W J，et al．Braincomputer interface technology：a review of the first international meeting［J］．IEEE transactions on rehabilitation engineering，2000，8（2）：164－173．

［2］ Mak J N，Arbel Y，Minett J W，et al．Optimizing the P300－based brain－computer interface：current status，limitations and future directions［J］．Journal of neural engineering，2011，8（2）：025003．

［3］ Ramoser H，Muller－Gerking J，Pfurtscheller G．Optimal spatial filtering of single trial EEG during imagined hand movement［J］．Rehabilitation Engineering，IEEE Transactions on，2000，8（4）：441－446．

［4］ Muller K，Anderson C W，Birch G E．Linear and nonlinear methods for brain－computer interfaces［J］．Neural Systems and Rehabilitation Engineering，IEEE Transactions on，2003，11（2）：165－169．

［5］ Garrett D，Peterson D A，Anderson C W，et al．Comparison of linear，nonlinear，and feature selection methods for EEG signal classification［J］．Neural Systems and Rehabilitation Engineering，IEEE Transactions on，2003，11（2）：141－144．

［6］ Wright J，Yang A Y，Ganesh A，et al．Robust face recognition via sparse representation［J］．Pattern Analysis and Machine Intelligence，IEEE Transactions on，2009，31（2）：210－227．

［7］ Shin Y，Lee S，Ahn M，et al．Motor imagery based BCI classification via sparse representation of EEG signals［C］．／／Noninvasive Functional Source Imaging of the Brain and Heart＆2011 8th International Conference on Bioelectromagnetism（NFSI＆ICBEM），2011 8th International Symposium on．IEEE，2011．

［8］ Candes E J，Romberg J K，Tao T．Stable signal recovery from incompleteandinaccuratemeasurements［J］．Communications on pure and applied mathematics，2006，59（8）：1207－1223．

［9］ Yang M，Zhang L，F(xiàn)eng X，et al．Sparse representation based Fisher discrimination dictionary learning for image classification［J］．InternationalJournalofComputer Vision，2014，109（3）：209－232．

［10］ Mairal J，Leordeanu M，Bach F，et al．Discriminative sparse image models for class－specific edge detection and image interpretation［M］．Computer Vision－ECCV 2008．Springer Berlin Heidelberg，2008．

［11］ Rosasco L，Verri A，Santoro M，et al．Iterative projection methods for structured sparsity regularization［J］．MIT Technical Report，2009：50－67．

［12］ Yang M，Zhang L，Yang J，et al．Metaface learning for sparse representation based face recognition［C］．／／Image Processing（ICIP），2010 17th IEEE International Conference on．IEEE，2010．

Motor Imagery EEG Classification Algorithm Based on Fisher Dictionary Learning

Xu Libo，Jiang Aimin，Liu Xiaofeng，Wang Yong
（College of Internet of Things Engineering，Hohai University，Changzhou 213022，China）

Now，in the study on motor imagery systems，based on brain－computer interface（BCI），the common spatial pattern（CSP）algorithm is the mainstream algorithm of EEG feature extraction．However，how to use the CSP features to classify effectively is one of the research hotspots in this field．In recent years，the classification method based on sparse representation（SRC）attracts more attention of native scholars and foreign ones．We propose an EEG classification algorithm based on the Fisher dictionary learning which combines the sparse reconstruction error and coding coefficient to classify the signals．The experimental results show that the proposed method can obtain better accuracy rate compared with the traditional classification method SRC．

Brain－computer interface；Common spatial pattern；Sparse representation based classification；Fisher dictionary learning；Reconstruction error；Coding coefficient

10．3969／j．issn．1002－2279．2015．04．014

TP301．6

A

1002－2279（2015）04－0053－04

國家自然科學基金項目（61101158，61471157）；江蘇省自然科學基金項目（BK20141159，BK20141157）

胥立波（1991－），男，江蘇省鹽城市人，碩士研究生，主研方向：生物醫(yī)學信號處理。

2015－01－26