陳玉,楊濤,徐錚
基于RBF神經(jīng)網(wǎng)絡(luò)的睡眠分期方法研究與應(yīng)用
陳玉,楊濤*,徐錚
(福建理工大學(xué) 設(shè)計(jì)學(xué)院, 福州 350118)
提出一種基于徑向基函數(shù)(Radial Basis Function,RBF)神經(jīng)網(wǎng)絡(luò)的睡眠分期方法,設(shè)計(jì)一套能夠根據(jù)用戶(hù)身心恢復(fù)狀態(tài)調(diào)節(jié)喚醒時(shí)間的智能喚醒系統(tǒng),以?xún)?yōu)化用戶(hù)睡眠時(shí)長(zhǎng),減輕醒后不適感。基于心率變異性和睡眠分期等相關(guān)理論知識(shí),通過(guò)低功耗心率帶采集人體心電信號(hào),選取最優(yōu)小波變換對(duì)采集到的心電信號(hào)精準(zhǔn)去噪,對(duì)徑向基函數(shù)神經(jīng)網(wǎng)絡(luò)進(jìn)行反復(fù)訓(xùn)練后,篩選出10個(gè)關(guān)鍵的特征向量,以構(gòu)建睡眠分期模型。睡眠分期信息通過(guò)STM32處理器傳輸?shù)绞謾C(jī)客戶(hù)端,系統(tǒng)根據(jù)預(yù)先設(shè)計(jì)的優(yōu)化喚醒機(jī)制在用戶(hù)身心恢復(fù)到最佳狀態(tài)時(shí)將其喚醒?;谒叻制谀P偷乃惴ㄆ骄R(shí)別準(zhǔn)確率可達(dá)88.9%,卡帕(Kappa)系數(shù)為0.839,相較于其他算法,該算法具有較高的準(zhǔn)確率。該智能喚醒系統(tǒng)的采集成本較低,算法簡(jiǎn)便高效,其喚醒機(jī)制科學(xué)合理,可以使用戶(hù)舒適醒來(lái),對(duì)改善用戶(hù)醒后狀態(tài)具有重要意義。
睡眠分期;心率變異性;小波變換;徑向基函數(shù)神經(jīng)網(wǎng)絡(luò);智能喚醒
隨著經(jīng)濟(jì)的發(fā)展和生活水平的提高,人們?cè)絹?lái)越重視睡眠健康問(wèn)題。睡眠是保持機(jī)體正常運(yùn)轉(zhuǎn)的基礎(chǔ),突然中斷睡眠容易導(dǎo)致人們心情煩躁、精神不佳,不合理的喚醒甚至?xí)绊懶押蟮恼J(rèn)知和決策。研究表明,適宜的睡眠時(shí)長(zhǎng)和高質(zhì)量的睡眠有助于恢復(fù)體力、鞏固記憶。然而,睡眠質(zhì)量的高低與睡眠時(shí)間的長(zhǎng)短并不呈正相關(guān),過(guò)長(zhǎng)的睡眠往往可能適得其反,造成人體軟弱無(wú)力、精神疲憊、代謝率降低等后果。目前,針對(duì)睡眠的研究主要集中在通過(guò)輔助手段促進(jìn)睡眠或監(jiān)測(cè)睡眠信息,并用于臨床診斷,很少關(guān)注到睡眠后的舒適喚醒問(wèn)題。市場(chǎng)上現(xiàn)存的喚醒產(chǎn)品大多不能根據(jù)人體的睡眠狀態(tài)調(diào)節(jié)喚醒時(shí)間,不夠智能化。由此可見(jiàn),針對(duì)睡眠時(shí)長(zhǎng)和睡眠狀態(tài)進(jìn)行評(píng)估,在人體精神和體力恢復(fù)到最佳時(shí)期時(shí)進(jìn)行喚醒具有十分重要的意義。
傳統(tǒng)研究睡眠的工具為多導(dǎo)睡眠監(jiān)護(hù)儀(Polysomnogram,PSG),該方法操作復(fù)雜、成本高,需要在醫(yī)院通過(guò)醫(yī)師操作完成,很難在廣大普通家庭普及。近年來(lái),很多學(xué)者嘗試?yán)脝瓮ǖ赖男碾姡‥lectrocardiogram,ECG)信號(hào)來(lái)分析睡眠,并取得了一定的成果。Mendez等[1]利用隱馬爾可夫模型,Xiao等[2]通過(guò)隨機(jī)森林算法,實(shí)現(xiàn)了三分類(lèi)睡眠分期。還有一些學(xué)者[3-4]基于心率變異性,利用支持向量機(jī)分類(lèi)器,實(shí)現(xiàn)了睡眠階段的自動(dòng)劃分。劉眾等[5]將體動(dòng)信號(hào)與ECG信號(hào)結(jié)合,通過(guò)改進(jìn)的反向傳播(Back Propagation, BP)神經(jīng)網(wǎng)絡(luò)有效實(shí)現(xiàn)了睡眠分期。雖然國(guó)外的一些學(xué)者[6-9]利用多種神經(jīng)網(wǎng)絡(luò)實(shí)現(xiàn)了睡眠階段的自動(dòng)分類(lèi),并獲得了較高的準(zhǔn)確率,但他們?nèi)匀恍枰柚僮鲝?fù)雜的PSG采集腦電信號(hào)。彭程等[10]基于心率變異性,提出了一種根據(jù)用戶(hù)的精神壓力和睡眠狀態(tài)進(jìn)行調(diào)整的音樂(lè)推薦系統(tǒng),有效地增加了用戶(hù)睡眠總時(shí)長(zhǎng),但依然未涉及睡眠后的智能喚醒問(wèn)題。衡瑋等[11]設(shè)計(jì)了基于腦電波監(jiān)測(cè)的智能助眠耳機(jī)系統(tǒng),研究的重點(diǎn)依然放在助眠問(wèn)題上。
以上研究均聚焦于睡眠監(jiān)測(cè)的便捷性和睡眠分期的準(zhǔn)確性,較少關(guān)注睡眠后的智能喚醒問(wèn)題,且系統(tǒng)較復(fù)雜,睡眠分期模型和算法有待進(jìn)一步優(yōu)化。針對(duì)以上不足,文中基于RBF神經(jīng)網(wǎng)絡(luò)算法,提出一種能夠在人體恢復(fù)到最佳時(shí)期時(shí)進(jìn)行叫醒的智能喚醒系統(tǒng),并配以友好的人機(jī)交互界面,以期達(dá)到優(yōu)化睡眠時(shí)長(zhǎng)、改善用戶(hù)醒后狀態(tài)的目的。
智能喚醒系統(tǒng)分為感知層、處理與傳輸層、應(yīng)用層等,系統(tǒng)結(jié)構(gòu)見(jiàn)圖1。感知層對(duì)應(yīng)數(shù)據(jù)采集系統(tǒng),用于采集人體ECG信號(hào),由心率帶(內(nèi)置STM32處理器)與配套的手機(jī)APP(Application software)共同實(shí)現(xiàn)數(shù)據(jù)采集功能。處理與傳輸層對(duì)應(yīng)程序控制系統(tǒng),主要對(duì)采集到的心電信號(hào)進(jìn)行降噪、R波峰定位和特征提取等處理,進(jìn)而采用RBF神經(jīng)網(wǎng)絡(luò)算法實(shí)現(xiàn)睡眠分期。這些功能由Matlab軟件實(shí)現(xiàn),經(jīng)Matlab訓(xùn)練后得到睡眠分期閾值,并傳輸?shù)叫穆蕩?nèi)置的STM32處理器進(jìn)行后續(xù)處理。應(yīng)用層對(duì)應(yīng)音樂(lè)喚醒系統(tǒng),主要將監(jiān)測(cè)到的睡眠階段可視化,并呈現(xiàn)在手機(jī)APP界面上,同時(shí)在判斷用戶(hù)機(jī)體恢復(fù)到最佳狀態(tài)后通過(guò)音樂(lè)實(shí)施喚醒,這些功能由智能喚醒系統(tǒng)應(yīng)用層中的NO SLEEP IN(不賴(lài)床)APP實(shí)現(xiàn)。其中STM32處理器不僅需要接收數(shù)據(jù)采集系統(tǒng)采集到的ECG信號(hào),還需要將處理好的數(shù)據(jù)傳輸?shù)紸PP中,并承擔(dān)在特定時(shí)間驅(qū)動(dòng)程序,實(shí)現(xiàn)音樂(lè)喚醒任務(wù)。
圖1 智能喚醒系統(tǒng)結(jié)構(gòu)
人體的ECG信號(hào)是一種低頻信號(hào),頻率為0.01~ 100 Hz。其中,QRS復(fù)合波的能量主要集中在3~ 40 Hz,且極易受到環(huán)境的干擾。噪聲主要包括工頻干擾、肌電干擾和低頻的基線漂移等[12],在分析ECG信號(hào)前需要對(duì)已獲取的信號(hào)進(jìn)行降噪處理。通過(guò)實(shí)驗(yàn)對(duì)比分析了巴特沃斯低通濾波器、帶通濾波器、中值濾波法、小波變換等4種去噪方法,文中擬選取降噪效果最好的小波變換對(duì)ECG信號(hào)進(jìn)行降噪。小波變換具有多分辨率分析的特點(diǎn),能夠?qū)⑿盘?hào)進(jìn)行多尺度細(xì)分,可改變窗口的大小,也可以同時(shí)局部分析時(shí)域和頻域[13],具有較好的自適應(yīng)性。與傳統(tǒng)的基于傅里葉變換的數(shù)字濾波技術(shù)相比,小波變換在時(shí)頻領(lǐng)域具有不可比擬的優(yōu)勢(shì)。小波變換見(jiàn)式(1)。
式中:為尺度因子,控制小波函數(shù)的伸縮;為時(shí)間平移因子,控制小波函數(shù)的平移;()為基本小波。式(1)對(duì)應(yīng)的逆變換見(jiàn)式(2)。
()不具備唯一性,不同小波基的選取對(duì)濾波效果的影響很大。
Rechtstaffen & Kales(R&K)分期標(biāo)準(zhǔn)將睡眠階段劃分為清醒期(Wake)、快速眼動(dòng)期(Rapid eye movement,REM)和非快速眼動(dòng)期(Non-rapid eye movement,NREM),NREM又包括N1—N4階段,一共分為6期。其中,N1、N2稱(chēng)為淺睡期,N3、N4稱(chēng)為深睡期。2007年,美國(guó)睡眠醫(yī)學(xué)學(xué)會(huì)(AASM)[14]將R&K標(biāo)準(zhǔn)中的N4并入N3,將睡眠階段共分為5期,目前最為常用。由于文中僅將分期結(jié)果用于智能喚醒,因此也將N1與N2合并為1期,將睡眠階段共分為4期進(jìn)行研究,即清醒期、淺睡期、深睡期、快速眼動(dòng)期等4期。按順序依次經(jīng)歷這幾個(gè)睡眠階段為1個(gè)睡眠周期,在一般情況下健康成年人整晚的睡眠會(huì)循環(huán)4~6個(gè)睡眠周期,每個(gè)周期通常為90~ 120 min,不同年齡和性別的人各個(gè)睡眠階段的持續(xù)時(shí)間不同。
RBF神經(jīng)網(wǎng)絡(luò)是一種用于解決模式識(shí)別與分類(lèi)的神經(jīng)網(wǎng)絡(luò),而睡眠階段的劃分實(shí)際上就是一種模式識(shí)別問(wèn)題,近些年已有研究者嘗試?yán)蒙窠?jīng)網(wǎng)絡(luò)來(lái)進(jìn)行睡眠分期[9,15],并獲得了良好的結(jié)果。由于RBF神經(jīng)網(wǎng)絡(luò)不會(huì)局限于局部極值,因此不需要學(xué)習(xí)系數(shù),適于在線實(shí)時(shí)辨識(shí)。與學(xué)習(xí)速度較慢、受到初始權(quán)值的影響較大、隱含層節(jié)點(diǎn)數(shù)較難確定的BP神經(jīng)網(wǎng)絡(luò)相比,RBF神經(jīng)網(wǎng)絡(luò)的收斂速度更快、非線性逼近精度更高,在訓(xùn)練時(shí)權(quán)值不受限制,學(xué)習(xí)算法簡(jiǎn)潔且更易實(shí)現(xiàn)[16],因此這里采用RBF神經(jīng)網(wǎng)絡(luò)來(lái)進(jìn)行睡眠分期。徑向基函數(shù)神經(jīng)網(wǎng)絡(luò)是一種前饋式神經(jīng)網(wǎng)絡(luò),主要包括輸入層、隱含層和輸出層等。輸入層主要用于接收外界信號(hào)。隱含層又叫基函數(shù)層,通過(guò)基函數(shù)實(shí)現(xiàn)非線性變化,一般選用高斯函數(shù)。輸出層對(duì)隱含層的輸出進(jìn)行線性加權(quán),從而使整個(gè)網(wǎng)絡(luò)具有分類(lèi)功能[17]。
RBF神經(jīng)網(wǎng)絡(luò)的結(jié)構(gòu)如圖2所示。由輸入層接收訓(xùn)練樣本x∈R,則RBF網(wǎng)絡(luò)輸出如式(3)所示。
式中:Ri(x)為隱含層的輸出;x為m維輸入向量;Ci為中心向量;σi為歸一化參數(shù),與基函數(shù)的寬度有關(guān);n為隱含層的節(jié)點(diǎn)數(shù)。隱含層實(shí)現(xiàn)x到Ri(x)的非線性映射,輸出層實(shí)現(xiàn)Ri(x)到y(tǒng)j的線性映射,輸出層的輸出如式(4)所示。
式中:w為隱含層與輸出層的連接權(quán)值,即權(quán)重;為輸出節(jié)點(diǎn)數(shù)量。
運(yùn)用RBF神經(jīng)網(wǎng)絡(luò)建立睡眠分期模型,實(shí)際上就是通過(guò)不斷訓(xùn)練網(wǎng)絡(luò),最終確定中心向量C、寬度σ和權(quán)重w。
神經(jīng)網(wǎng)絡(luò)算法的整體性能采用準(zhǔn)確率(Accuracy)和均方誤差(MSE)2個(gè)指標(biāo)進(jìn)行分析。均方誤差指預(yù)測(cè)值與真實(shí)值之差的平方的期望值,用于衡量預(yù)測(cè)誤差,也是一種擬合優(yōu)度指標(biāo),可以用于評(píng)估模型的擬合效果。MSE值越小,則模型的預(yù)測(cè)誤差越小、擬合程度越好。均方誤差的計(jì)算見(jiàn)式(5)。
式中:為總輸入樣本數(shù)量;Y為睡眠階段的標(biāo)定值;Z為睡眠階段的預(yù)測(cè)值。
為了進(jìn)一步分析誤差產(chǎn)生的原因,這里同時(shí)引入混淆矩陣作為衡量依據(jù),具體考察各睡眠階段相關(guān)性的強(qiáng)弱。混淆矩陣是模式識(shí)別領(lǐng)域中常用的圖表形式,是評(píng)價(jià)模型精度的一種標(biāo)準(zhǔn)格式,可有效評(píng)價(jià)分類(lèi)器性能[18]。為了驗(yàn)證RBF神經(jīng)網(wǎng)絡(luò)用于睡眠分期的科學(xué)性、有效性,通過(guò)麥克尼馬爾檢驗(yàn)(Mc Nemar test)方法對(duì)實(shí)驗(yàn)結(jié)果進(jìn)行顯著性差異檢驗(yàn),并計(jì)算卡帕(Kappa)系數(shù),進(jìn)一步確定實(shí)驗(yàn)結(jié)果的一致性程度。麥克尼馬爾檢驗(yàn)是一種非參數(shù)統(tǒng)計(jì)檢驗(yàn)方法,用于比較2個(gè)相關(guān)樣本中差異的統(tǒng)計(jì)顯著性。Kappa系數(shù)是一個(gè)用于一致性檢驗(yàn)的指標(biāo),所檢驗(yàn)的2個(gè)變量分別是實(shí)際標(biāo)記樣本和模型分類(lèi)結(jié)果,因此Kappa系數(shù)可以用來(lái)評(píng)價(jià)模型預(yù)測(cè)結(jié)果與實(shí)際分類(lèi)結(jié)果的一致性程度。在0.75≤Kappa系數(shù)≤1時(shí),表示兩者具有很好的一致性;在0.4≤Kappa系數(shù)<0.75時(shí),表示兩者的一致性一般;在Kappa系數(shù)<0.4時(shí),表示兩者的一致性較差。
接受睡眠監(jiān)測(cè)的研究生志愿者共有4名,年齡為23~28歲,男性2名,女性2名,分別記為志愿者1、志愿者2、志愿者3、志愿者4。每人采集5個(gè)整晚的有效睡眠數(shù)據(jù),共計(jì)20個(gè)晚上,每晚數(shù)據(jù)均大于6 h。在實(shí)驗(yàn)期間保證志愿者的作息規(guī)律,且不服用任何輔助睡眠類(lèi)藥物,志愿者同時(shí)佩戴華為Watch gt 3手表和心率帶同步監(jiān)測(cè)其睡眠信息,將手表與由哈佛醫(yī)學(xué)院CDB中心測(cè)試認(rèn)證的TrueSleep技術(shù)配合使用,TrueSleep技術(shù)采用心肺耦合檢測(cè)睡眠算法,其睡眠檢測(cè)較精準(zhǔn)。專(zhuān)業(yè)醫(yī)師以TrueSleep睡眠分期結(jié)果為參考,判定實(shí)驗(yàn)數(shù)據(jù)的睡眠階段,并做標(biāo)記,將神經(jīng)網(wǎng)絡(luò)判定的睡眠階段與手工判斷結(jié)果進(jìn)行對(duì)比,該標(biāo)記也作為驗(yàn)證睡眠分期算法準(zhǔn)確性的標(biāo)準(zhǔn)。
實(shí)驗(yàn)流程如圖3所示。首先通過(guò)心率帶采集信號(hào),利用小波變換對(duì)采集的ECG信號(hào)進(jìn)行精準(zhǔn)去噪,最大限度地保留清晰有用的心電信號(hào),即數(shù)據(jù)預(yù)處理。其次,利用Matlab中的findpeaks函數(shù)對(duì)R波的峰值進(jìn)行定位,將相鄰R波間隔記為RR間期(R-R Interval,RRI)。接著,以RRI為基礎(chǔ),提取相應(yīng)的時(shí)頻域特征向量。最后,利用RBF神經(jīng)網(wǎng)絡(luò)對(duì)提取的特征值進(jìn)行反復(fù)訓(xùn)練,并篩選出關(guān)鍵的特征向量,以構(gòu)建睡眠分期模型。
采用功耗較低的心率帶采集心電數(shù)據(jù),以降低系統(tǒng)成本,采樣頻率為500 Hz,有效睡眠數(shù)據(jù)共計(jì)20個(gè)夜晚。心率帶由針織材質(zhì)制成,具有可伸縮性,使用時(shí)無(wú)束縛感,基本不會(huì)對(duì)睡眠者的實(shí)驗(yàn)數(shù)據(jù)造成干擾。在信號(hào)采集過(guò)程中心率變化可通過(guò)配套的手機(jī)APP實(shí)時(shí)查看,不同睡眠階段采集到的心率變化如圖4所示。
為了精準(zhǔn)降噪,通過(guò)實(shí)驗(yàn)反復(fù)對(duì)比,這里選取與QRS復(fù)合波最相似的coif4小波對(duì)心電信號(hào)進(jìn)行8個(gè)尺度的小波分解,分解如圖5所示。從圖5可以看出,QRS波的能量主要集中在尺度3~6上,肌電和工頻干擾主要集中在尺度1~2上,而基線漂移則分布在尺度7~8上。這里將噪聲所在的尺度系數(shù)置為0,以消除高低頻噪聲,用尺度3~6上的小波系數(shù)重構(gòu)心電信號(hào),得到清晰干凈的ECG信號(hào)。ECG原始信號(hào)及信號(hào)去噪重構(gòu)示意圖如圖6所示。
心率變異性(Heart Rate Variability,HRV)指2次相鄰的心臟搏動(dòng)周期之間存在的微小差異,換言之就是從本次心跳周期至下一次心跳周期的差異[19]。大量研究表明,HRV與睡眠分期存在密不可分的聯(lián)系,如心率的降低與NREM睡眠中HRV的降低有關(guān),而心率的增加,會(huì)伴隨著清醒期和REM期睡眠中HRV的增加。低頻(LF,0.04~0.15 Hz)與高頻(HF,0.15~ 0.4 Hz)波段的功率比(LF/HF)的降低與NREM睡眠有關(guān),而LF/HF的顯著增加則體現(xiàn)在REM睡眠中[2]。這里對(duì)ECG信號(hào)的特征提取主要基于RR間期的計(jì)算。通過(guò)Matlab中的findpeaks函數(shù)對(duì)ECG信號(hào)中的R波進(jìn)行定位,并計(jì)算出相鄰R波的時(shí)間間隔,記為RR間期(RRI)。由于2個(gè)R波的間隔不會(huì)小于200 ms,因此剔除間隔在0.2~2.0 s范圍之外的異常值,以提高R波檢測(cè)的準(zhǔn)確性。R波峰值定位結(jié)果如圖7所示,最終確定的RR間期序列用于特征向量提取。
這里主要從時(shí)域和頻域2個(gè)方面進(jìn)行特征提取。HRV的時(shí)域特征直接基于RR間期序列計(jì)算得出。為了既精準(zhǔn)劃分睡眠階段又加快系統(tǒng)運(yùn)算速度,在對(duì)RBF神經(jīng)網(wǎng)絡(luò)進(jìn)行多次訓(xùn)練后,篩選了10個(gè)對(duì)睡眠階段的劃分起關(guān)鍵作用的特征向量,然后根據(jù)這10個(gè)特征向量的取值范圍構(gòu)建睡眠分期模型。其中,包括7個(gè)經(jīng)典時(shí)域特征[2,4,11,19](見(jiàn)表1),以及3個(gè)頻域特征。HRV的頻域特征主要反映了ECG信號(hào)中各頻段的能量分布,對(duì)上述提取的RR間期序列進(jìn)行快速傅里葉變換,得到心率的功率譜信息,以時(shí)長(zhǎng)30 s劃分?jǐn)?shù)據(jù),進(jìn)行短時(shí)程頻域分析,進(jìn)而提取0.003~ 0.04 Hz(極低頻)、0.04~0.15 Hz(低頻)、0.15~0.4 Hz(高頻)這3個(gè)經(jīng)典頻域特征。
圖3 心率變異性分析流程
圖4 不同睡眠階段心率變化
圖5 ECG信號(hào)coif4小波8個(gè)尺度分解
圖6 ECG原始信號(hào)及信號(hào)的去噪重構(gòu)
圖7 R波波峰定位結(jié)果
表1 文中所提取的時(shí)域特征
Tab.1 Time domain features extracted for this research
注:表示RR間期序列的均值,mRRI;表示RR間期,RR;表示相鄰RR間期相差大于50 ms的數(shù)量,NN50;表示心率,HR;表示RR間期序列變異系數(shù),CV。
利用RBF神經(jīng)網(wǎng)絡(luò),根據(jù)構(gòu)建好的睡眠分期模型,對(duì)數(shù)據(jù)進(jìn)行睡眠分期實(shí)驗(yàn)。以某晚的睡眠數(shù)據(jù)為例,該數(shù)據(jù)睡眠總時(shí)長(zhǎng)為6.05 h,以30 s為一段樣本,共劃分為726段數(shù)據(jù)樣本,以其中76%的550段數(shù)據(jù)樣本為訓(xùn)練集,以剩下的176段數(shù)據(jù)樣本為測(cè)試集進(jìn)行驗(yàn)證。為了直觀分析每個(gè)睡眠階段具體的預(yù)測(cè)表現(xiàn),將原本按時(shí)間排序的分類(lèi)結(jié)果按睡眠階段的類(lèi)別重新排序,該晚神經(jīng)網(wǎng)絡(luò)睡眠分期預(yù)測(cè)結(jié)果如圖8所示。從圖8可以看出,訓(xùn)練集和測(cè)試集預(yù)測(cè)的準(zhǔn)確率均在93%以上,效果良好,其MSE值分別為0.335 75、0.412 86,預(yù)測(cè)值與真實(shí)值大致吻合、重合度較高,符合分類(lèi)要求。
各睡眠階段分類(lèi)結(jié)果混淆矩陣圖如圖9所示,其中,1表示清醒期,2表示REM期,3表示淺睡期,4表示深睡期。從圖9中可以看出,在訓(xùn)練集中REM期和淺睡期表現(xiàn)較差,在測(cè)試集中深睡期的精準(zhǔn)率和召回率均為97.6%,綜合準(zhǔn)確率最高,預(yù)測(cè)效果最好。
對(duì)4個(gè)志愿者20個(gè)晚上的睡眠分期數(shù)據(jù)進(jìn)行統(tǒng)計(jì),計(jì)算各自睡眠分期的預(yù)測(cè)準(zhǔn)確率和均方誤差。由于訓(xùn)練后的神經(jīng)網(wǎng)絡(luò)模型分類(lèi)效果主要由測(cè)試集體現(xiàn),因此利用麥克尼馬爾檢驗(yàn)方法計(jì)算測(cè)試集的漸進(jìn)顯著性(),并進(jìn)一步計(jì)算其Kappa值(kappa),以驗(yàn)證測(cè)試集實(shí)驗(yàn)結(jié)果的一致性,結(jié)果如表2所示。測(cè)試集的麥克尼馬爾檢驗(yàn)結(jié)果顯示,漸進(jìn)顯著性(平均= 0.1)大于0.05,說(shuō)明RBF神經(jīng)網(wǎng)絡(luò)模型的分類(lèi)結(jié)果與實(shí)際分類(lèi)具有一致性。kappa平均=0.839,大于0.75,說(shuō)明實(shí)驗(yàn)結(jié)果存在顯著一致性,具有統(tǒng)計(jì)學(xué)意義。訓(xùn)練集的準(zhǔn)確率均在90%以上,平均準(zhǔn)確率為92.7%,平均MSE值為0.399 27。測(cè)試集的平均準(zhǔn)確率為88.9%,平均均方誤差為0.511 87??梢?jiàn),整體預(yù)測(cè)準(zhǔn)確率較高,誤差在可接受范圍內(nèi),模型的擬合優(yōu)度良好,具有實(shí)際應(yīng)用價(jià)值。
此外,將文中提出的睡眠分期算法與其他研究者采用的算法進(jìn)行了對(duì)比,如表3所示。從表3可以看出,文中基于RBF神經(jīng)網(wǎng)絡(luò)實(shí)現(xiàn)的睡眠分期算法采集的信息較少,睡眠分期較多,準(zhǔn)確率更高,綜合識(shí)別效果更好,具有顯著的優(yōu)越性。
圖8 RBF神經(jīng)網(wǎng)絡(luò)睡眠分期預(yù)測(cè)結(jié)果
圖9 訓(xùn)練集和測(cè)試集各睡眠階段分類(lèi)結(jié)果混淆矩陣
表2 各志愿者睡眠分期預(yù)測(cè)結(jié)果統(tǒng)計(jì)
Tab.2 Statistics of sleep staging prediction results for each volunteer
表3 不同睡眠分期算法的準(zhǔn)確率對(duì)比
Tab.3 Comparison of accuracy of different algorithms for sleep staging
為了使用戶(hù)在清晨舒適醒來(lái),設(shè)計(jì)了一種智能喚醒機(jī)制,如圖10所示。根據(jù)心理學(xué)方面的研究成果,在人體處于第4個(gè)睡眠周期末時(shí)將其喚醒最理想。由此,在默認(rèn)狀態(tài)下,將智能喚醒系統(tǒng)設(shè)置為在監(jiān)測(cè)到用戶(hù)處于第4個(gè)睡眠周期末時(shí)將其喚醒,以保證用戶(hù)在身體恢復(fù)到最佳狀態(tài)時(shí)醒來(lái)。如果用戶(hù)需要預(yù)設(shè)喚醒時(shí)間段,則系統(tǒng)將用戶(hù)所處的睡眠階段與用戶(hù)預(yù)設(shè)的喚醒時(shí)間綜合起來(lái)作為喚醒的判斷條件。具體來(lái)說(shuō),用戶(hù)先預(yù)設(shè)一個(gè)起床時(shí)間段,由于深睡期一般只發(fā)生在凌晨5點(diǎn)前,因此喚醒系統(tǒng)會(huì)在這個(gè)時(shí)間段內(nèi)先判斷用戶(hù)是否處于淺睡期,如果是則叫醒用戶(hù),如果不是則系統(tǒng)繼續(xù)監(jiān)測(cè);當(dāng)監(jiān)測(cè)到用戶(hù)處于快速眼動(dòng)期(REM)時(shí),將用戶(hù)叫醒,如果不是則系統(tǒng)繼續(xù)監(jiān)測(cè)。這樣,系統(tǒng)既可在用戶(hù)預(yù)設(shè)的時(shí)間段內(nèi)將其喚醒,不耽誤其重要工作,又可使用戶(hù)在恰當(dāng)?shù)乃唠A段醒來(lái),以最佳的狀態(tài)快速進(jìn)入工作。
圖10 智能喚醒機(jī)制
系統(tǒng)將根據(jù)睡眠分期結(jié)果通過(guò)手機(jī)APP在恰當(dāng)?shù)臅r(shí)機(jī)用音樂(lè)將用戶(hù)喚醒。為了最大限度地實(shí)現(xiàn)智能喚醒系統(tǒng)的功能,并創(chuàng)造良好的用戶(hù)體驗(yàn),這里設(shè)計(jì)了與智能喚醒機(jī)制相匹配的NO SLEEP IN(不賴(lài)床)APP界面,如圖11所示。
在視覺(jué)上,整個(gè)界面色調(diào)和諧統(tǒng)一,符合夜間使用需求。在功能設(shè)計(jì)上,對(duì)于智能喚醒需求,該APP設(shè)置了4種模式:默認(rèn)最佳,系統(tǒng)將在用戶(hù)處于第4個(gè)睡眠周期末時(shí)實(shí)施喚醒;半智能,用戶(hù)可預(yù)設(shè)喚醒時(shí)間段及系統(tǒng)監(jiān)測(cè)最佳狀態(tài)的時(shí)長(zhǎng),系統(tǒng)將預(yù)設(shè)時(shí)段與用戶(hù)狀態(tài)結(jié)合起來(lái),綜合判定喚醒時(shí)刻;非智能,僅具有普通鬧鐘功能;自然醒,不具備喚醒功能,系統(tǒng)僅分析用戶(hù)的睡眠狀態(tài)。此外,該APP還可顯示用戶(hù)累計(jì)時(shí)間段的睡眠監(jiān)測(cè)報(bào)告,并具有音樂(lè)助眠、社群交流等功能,以滿(mǎn)足用戶(hù)多樣化的睡眠需求。
基于RBF神經(jīng)網(wǎng)絡(luò)提出了一種準(zhǔn)確率較高、模型擬合優(yōu)度較好的睡眠分期方法,并設(shè)計(jì)了一套低成本高效率的智能喚醒系統(tǒng),優(yōu)化的喚醒機(jī)制可以有效避免用戶(hù)醒后情緒煩躁、狀態(tài)差等問(wèn)題。通過(guò)低功耗心率帶采集人體心電信號(hào),大大節(jié)約了系統(tǒng)成本。在信號(hào)預(yù)處理階段,通過(guò)實(shí)驗(yàn)對(duì)比,優(yōu)選出coif4小波及其分解與重構(gòu)尺度,它對(duì)ECG信號(hào)具有更好的降噪效果,為提升睡眠分期的準(zhǔn)確率奠定了良好的基礎(chǔ)。利用RBF神經(jīng)網(wǎng)絡(luò)算法對(duì)提取的特征值反復(fù)訓(xùn)練,精選出對(duì)睡眠分期起關(guān)鍵作用的特征向量,從而構(gòu)建睡眠分期模型。實(shí)驗(yàn)結(jié)果表明,基于該模型的RBF神經(jīng)網(wǎng)絡(luò)的計(jì)算負(fù)擔(dān)更小、運(yùn)算速度更快,與其他用于睡眠分期的算法相比,它具有更高的準(zhǔn)確率。此外,還基于用戶(hù)體驗(yàn)設(shè)計(jì)了與喚醒機(jī)制相匹配的手機(jī)APP交互界面,提升了整個(gè)系統(tǒng)的智能性和完整性。
文中設(shè)計(jì)的RBF神經(jīng)網(wǎng)絡(luò)結(jié)構(gòu),雖然整體上能較準(zhǔn)確地完成睡眠階段的劃分,但在REM期和清醒期的分類(lèi)表現(xiàn)上相對(duì)較弱,未來(lái)將進(jìn)一步研究神經(jīng)網(wǎng)絡(luò)的算法,以提高REM期和清醒期的分類(lèi)精度。此外,不同性別和年齡的人在睡眠階段劃分標(biāo)準(zhǔn)上也存在差異,因此未來(lái)計(jì)劃擴(kuò)大實(shí)驗(yàn)樣本數(shù)量,將性別和年齡作為特征值進(jìn)行提取,以提高神經(jīng)網(wǎng)絡(luò)睡眠分期的普適性,提升整個(gè)智能喚醒系統(tǒng)的實(shí)用價(jià)值。所設(shè)計(jì)的智能喚醒系統(tǒng)僅從理論上實(shí)現(xiàn)了喚醒機(jī)制和軟件界面設(shè)計(jì)的優(yōu)化,在實(shí)踐上,手機(jī)終端軟件的功能尚未完全實(shí)現(xiàn)。今后將把軟件功能的實(shí)現(xiàn)作為研究重點(diǎn),增強(qiáng)智能喚醒系統(tǒng)的完整性和現(xiàn)實(shí)意義。
[1] MENDEZ M O, MATTEUCCI M, CERUTTI S, et al. Sleep Staging Classification Based on HRV: Time- Variant Analysis[C]// 2009 Annual International Conference of the IEEE Engineering in Medicine and Biology Society. IEEE, 2009: 9-12.
[2] XIAO M, YAN H, SONG J Z, et al. Sleep Stages Classification Based on Heart Rate Variability and Random Forest[J]. Biomedical Signal Processing and Control, 2013, 8(6): 624-633.
[3] 王金海, 孫微, 韋然, 等. 基于心率變異性分析的睡眠分期方法研究[J]. 生物醫(yī)學(xué)工程學(xué)雜志, 2016, 33(3): 420-425.WANG J H, SUN W, WEI R, et al. Study on Sleep Staging Methods Based on Heart Rate Variability Analysis[J]. Journal of Biomedical Engineering, 2016, 33(3): 420-425.
[4] 鄭捷文, 張悅舟, 蘭珂, 等. 基于心率變異性分析的睡眠分期算法研究和驗(yàn)證[J]. 中國(guó)生物醫(yī)學(xué)工程學(xué)報(bào), 2020, 39(4): 432-439. ZHENG J W, ZHANG Y Z, LAN K, et al. Sleep Stage Classification Based on Heart Rate Variability Analysis and Model Performance Validation[J]. Chinese Journal of Biomedical Engineering, 2020, 39(4): 432-439.
[5] 劉眾, 王新安, 李秋平, 等. 基于ECG信號(hào)和體動(dòng)信號(hào)的睡眠分期方法研究[J]. 北京大學(xué)學(xué)報(bào)(自然科學(xué)版), 2021, 57(5): 833-840. LIU Z, WANG X A, LI Q P, et al. Research of Sleep Staging Algorithms Based on ECG and Body Motion Signals[J]. Acta Scientiarum Naturalium Universitatis Pekinensis, 2021, 57(5): 833-840.
[6] ZHU T Q, LUO W, YU F. Convolution-and Attention- Based Neural Network for Automated Sleep Stage Classification[J]. International Journal of Environmental Research and Public Health, 2020, 17(11): 4152.
[7] KWON K, KWON S, YEO W H. Automatic and Accurate Sleep Stage Classification via a Convolutional Deep Neural Network and Nanomembrane Electrodes[J]. Biosensors, 2022, 12(3): 155.
[8] YOU Y Y, ZHONG X Y, LIU G Z, et al. Automatic Sleep Stage Classification: A Light and Efficient Deep Neural Network Model Based on Time, Frequency and Fractional Fourier Transform Domain Features[J]. Artificial Intelligence in Medicine, 2022, 127: 102279.
[9] LEE H, CHOI J, KIM S, et al. A Compressive Sensing- Based Automatic Sleep-Stage Classification System with Radial Basis Function Neural Network[J]. IEEE Access, 2019, 7: 186499-186509.
[10] 彭程, 常相茂, 仇媛. 基于心率變異性分析的睡眠音樂(lè)推薦系統(tǒng)[J]. 計(jì)算機(jī)應(yīng)用, 2020, 40(5): 1539-1544. PENG C, CHANG X M, QIU Y. Heart Rate Variability Analysis Based Sleep Music Recommendation System[J]. Journal of Computer Applications, 2020, 40(5): 1539- 1544.
[11] 衡瑋, 徐自強(qiáng), 陳璐瑤, 等. 基于腦電波監(jiān)測(cè)智能助睡眠耳機(jī)設(shè)計(jì)[J]. 計(jì)算機(jī)應(yīng)用與軟件, 2020, 37(8): 219-226.HENG W, XU Z Q, CHEN L Y, et al. Design of Intelligent sleep-Assisted Earphone Based on Brain Wave Monitoring[J]. Computer Applications and Software, 2020, 37(8): 219-226.
[12] 冀常鵬, 李蓓蕾. 心電信號(hào)識(shí)別方法的研究與實(shí)現(xiàn)[J]. 計(jì)算機(jī)應(yīng)用與軟件, 2018, 35(3): 173-176. JI C P, LI B L. Research and Implement of ECG Signal Identification Method[J]. Computer Applications and Software, 2018, 35(3): 173-176.
[13] 付文亭, 鄧體俊, 陳海生. 小波變換系數(shù)奇異值分解數(shù)字水印防偽算法[J]. 包裝工程, 2015, 36(21): 128-132. FU W T, DENG T J, CHEN H S. A Watermarking Algorithm for Security Based on Discrete Wavelet Transform and Singular Value Decomposition[J]. Packaging Engineering, 2015, 36(21): 128-132.
[14] BERRY R B, BUDHIRAJA R, GOTTLIEB D J, et al. Rules for Scoring Respiratory Events in Sleep: Update of the 2007 AASM Manual for the Scoring of Sleep and Associated Events. Deliberations of the Sleep Apnea Definitions Task Force of the American Academy of Sleep Medicine[J]. Journal of Clinical Sleep Medicine: JCSM: Official Publication of the American Academy of Sleep Medicine, 2012, 8(5): 597-619.
[15] CHEN X Y, HE J, WU X Q, et al. Sleep Staging by Bidirectional Long Short-Term Memory Convolution Neural Network[J]. Future Generation Computer Systems, 2020, 109: 188-196.
[16] 唐艷, 孫劉杰, 王勇. 基于SIFT變換和RBF神經(jīng)網(wǎng)絡(luò)的彩色全景圖拼接算法[J]. 包裝工程, 2018, 39(21): 216-221. TANG Y, SUN L J, WANG Y. Color Panorama Mosaic Algorithm Based on SIFT Transform and RBF Neural Network[J]. Packaging Engineering, 2018, 39(21): 216- 221.
[17] 李丹, 翟震. 改進(jìn)的RBF神經(jīng)網(wǎng)絡(luò)PID算法在電磁振機(jī)中的應(yīng)用[J]. 包裝工程, 2019, 40(7): 192-196. LI D, ZHAI Z. Application of Improved RBF Neural Network PID Algorithm in Electromagnetic Vibrating Machine[J]. Packaging Engineering, 2019, 40(7): 192-196.
[18] 楊韞儀, 裴卉寧, 李書(shū)航, 等. 基于多項(xiàng)Logistic回歸的座面材質(zhì)預(yù)測(cè)模型[J]. 包裝工程, 2021, 42(16): 182-187.YANG Y Y, PEI H N, LI S H, et al. Seat Surface Material Prediction Model Based on Multiple Logistic Regression[J]. Packaging Engineering, 2021, 42(16): 182-187.
[19] 應(yīng)少飛. 基于心率變異性的多導(dǎo)睡眠監(jiān)護(hù)儀[D]. 成都: 電子科技大學(xué), 2020: 27-33. YING S F. PSG Base on Heart Rate Variability[D]. Chengdu: University of Electronic Science and Technology of China, 2020: 27-33.
Research and Application of Sleep Staging Method Based on RBF Neural Network
CHEN Yu, YANG Tao*, XU Zheng
(School of Design, Fujian University of Technology, Fuzhou 350118, China)
The work aims to propose a sleep staging method based on a radial basis function (RBF) neural network and use it to design an intelligent wake-up system that can adjust the wake-up time according to the user's recovery state, in order to optimize the user's sleep duration and reduce their discomfort after waking up. Based on theoretical knowledge of heart rate variability and sleep staging, the electrocardiogram (ECG) signal was collected from the human body through a low-power heart rate band, and the optimal wavelet transform was selected to precisely denoise the collected ECG signal. The radial basis function (RBF) neural network was trained repeatedly to filter out 10 key feature vectors, so as to build a model of sleep staging. The sleep staging information was transmitted to the mobile phone client via a STM32 processor, and the system woke up the user according to designed optimized wake-up mechanism when the user's body and mind recovered to the optimal state. The results showed that the algorithm based on the sleep staging model had an average accuracy of 88.9% with a Kappa coefficient of 0.839, which was higher than that of other algorithms. The intelligent wake-up system has a lower collection cost, a simpler and more efficient algorithm, and a scientific and reasonable wake-up mechanism, which enables the user to wake up comfortably and is of great significance in improving the user's state after awakening.
sleep staging; heart rate variability; wavelet transform; radial basis function neural network; intelligent wake-up
TP391.4;TB472
A
1001-3563(2024)04-0371-09
10.19554/j.cnki.1001-3563.2024.04.041
2023-09-16
福建省自然科學(xué)基金(2023J01947);福建省社會(huì)科學(xué)基金(FJ2021B187);福建理工大學(xué)科研啟動(dòng)基金(GY-S20089)