簡介這套基于 MATLAB 與機器學習的心電疲勞檢測項目面向畢業(yè)設計、課程設計和項目開發(fā)場景以心電信號為輸入完成疲勞狀態(tài)的識別與評估。壓縮包共二十四個文件總計約四點一七 MB其中 .m 與 .mlx 腳本覆蓋信號預處理、特征提取、模型訓練與結(jié)果可視化.xlsx 提供心電采集數(shù)據(jù)和時域特征數(shù)據(jù).mat 存放訓練好的模型.md 與 .txt 則給出說明和注釋整體結(jié)構清晰便于按模塊研讀。項目源碼經(jīng)過嚴格測試可放心參考并在其基礎上擴展。內(nèi)容涉及小波閾值去噪、Pan-Tompkins QRS 檢測、ARMA 系數(shù)、LSTM 建模及混淆矩陣評估等關鍵環(huán)節(jié)能幫助學習者快速理解從原始心電數(shù)據(jù)到疲勞識別模型的完整鏈路也適合作為相關課程設計或答辯展示的落地模板。目前已有四十四人瀏覽學習便于在真實項目場景中對照使用與二次開發(fā)。1. 心電疲勞檢測為什么說這是一條“能畢業(yè)又不勸退”的技術路線如果你正為畢業(yè)設計或者課程設計選題發(fā)愁又不想一上來就碰腦電、眼動這些昂貴設備那基于 Matlab 機器學習實現(xiàn)的心電疲勞檢測是一個性價比很高的方向。核心思路并不復雜把心電ECG信號切成若干時間窗從每個窗口里提取心率變異性HRV等特征再用分類器輸出“疲勞 / 清醒”的判別結(jié)果。你需要的材料也很常規(guī)一個帶疲勞標簽的心電數(shù)據(jù)集、Matlab 2019b 及以上版本、以及基本的信號處理常識。這個方向適合兩類人一類是需要快速跑通“采集—特征—模型—評估”全流程的在校學生另一類是打算在可穿戴場景里做疲勞狀態(tài)監(jiān)測的開發(fā)者。它真正的難點其實不在分類器而在兩個容易被忽略的問題疲勞標簽怎么定義才算合理以及特征提取和驗證方式怎么繞開數(shù)據(jù)泄漏。這篇文章把這兩件事的前因后果講透。2. 疲勞為什么能在心電信號里露餡HRV特征與生理學解釋2.1 疲勞檢測的兩個前提標簽怎么定信號從哪來做疲勞檢測的第一個問題是什么叫疲勞這是一個主觀狀態(tài)沒有金標準。常見的打標方式有三種主觀量表如 KSS 嗜睡量表、疲勞自評問卷、任務時長連續(xù)工作 2 小時以上視為疲勞、客觀行為指標反應時變長、眼動變慢。畢業(yè)設計階段最穩(wěn)妥的做法是讓受試者在清醒和疲勞兩種狀態(tài)下各采集一段心電然后按“段”打標簽而不是按“秒”打標簽。因為疲勞是一個緩慢變化的過程逐秒標注既不可靠也沒必要。第二個問題是為什么選心電而不是腦電腦電對設備要求高眼動需要攝像頭和圖像算法而心電只需要貼片電極或胸帶Matlab 里讀入數(shù)據(jù)直接就能處理。更重要的是心電信號雖然看起來只是一串 QRS 波形但 R 峰之間的時間間隔RR 間期隱含著自主神經(jīng)系統(tǒng)的調(diào)控信息而疲勞恰恰會改變這種調(diào)控模式。這就是心電能做疲勞檢測的生理學基礎。2.2 自主神經(jīng)系統(tǒng)的痕跡HRV 為什么跟著疲勞變化心率變異性HRV衡量的是相鄰心跳間隔的波動程度。你可能直覺上認為心跳越規(guī)律越好實際上恰恰相反健康的、清醒狀態(tài)下的心臟RR 間期有豐富的波動而疲勞或壓力狀態(tài)下交感神經(jīng)占優(yōu)勢副交感神經(jīng)被抑制RR 間期的波動幅度會明顯下降。換句話說疲勞的人心跳可能并不快但心跳間隔變得“機械而單調(diào)”。HRV 特征可以分成三組時域特征直接對 RR 間期做統(tǒng)計頻域特征把 RR 間期序列變換到頻譜上觀察不同頻段的能量非線性特征則用熵或散點圖描述序列的復雜程度。其中時域特征最容易理解也是入門時優(yōu)先復現(xiàn)的。疲勞發(fā)生時SDNNRR 間期標準差和 RMSSD相鄰 RR 差異的均方根通常會下降pNN50 也會變小這些趨勢在很多公開研究里都被反復驗證過。2.3 用 Matlab 提取 HRV 特征的最小代碼下面的代碼演示了從一段 ECG 信號到 HRV 特征的全過程。這里假設你已經(jīng)完成了基礎的帶通濾波ecg是濾波后的心電列向量fs是采樣率。%% 1. 檢測 R 峰位置 % MinPeakDistance 設置為 0.6s對應最大心率 100 bpm防止把 T 波誤檢為 R 峰 [~, locs] findpeaks(ecg, ... MinPeakHeight, 0.3 * max(ecg), ... MinPeakDistance, round(0.6 * fs), ... MinPeakProminence, 0.15 * max(ecg)); %% 2. 計算 RR 間期單位秒并轉(zhuǎn)成毫秒序列 rr diff(locs) / fs; rr_ms rr * 1000; %% 3. 時域 HRV 特征 sdnn std(rr_ms); % RR 間期標準差 rmssd sqrt(mean(diff(rr_ms).^2)); % 相鄰 RR 差異的均方根 nn50 sum(abs(diff(rr_ms)) 50); % 相鄰差超過 50ms 的個數(shù) pnn50 nn50 / length(rr_ms) * 100; % pNN50 百分比這里MinPeakProminence的作用比預期更重要。ECG 中 T 波的幅度有時會接近 R 峰的 50%僅靠高度閾值區(qū)分不了 R 峰和 T 波。MinPeakDistance是從時間上限制兩個 R 峰的最小間距MinPeakProminence則要求候選峰相對于周圍波形有足夠的突出程度。采樣率 250 Hz 時round(0.6 * 250)就是 150 個采樣點。如果你的數(shù)據(jù)包含運動員或高心率受試者需要把 0.6s 調(diào)小到 0.45s 左右否則會把真實心跳漏檢。2.4 建議優(yōu)先提取的特征清單下面這張表列出的特征是疲勞檢測里最常被論文引用、也最容易復現(xiàn)的。建議第一個版本先做這些。特征名計算方式疲勞時的一般趨勢備注SDNNRR 間期標準差下降反映整體變異性RMSSD相鄰 RR 差的均方根下降反映副交感活性最穩(wěn)定pNN50相鄰差 50ms 的占比下降對短時窗敏感窗口太短會失真LF / HF頻段能量0.04–0.15 Hz / 0.15–0.4 HzLF/HF 常升高需要 RR 間期重采樣SampEn樣本熵下降非線性特征量級小但區(qū)分度高頻域特征的坑在 RR 間期是非等間隔采樣不能直接做 FFT必須先重采樣成等間隔序列再算功率譜這一步在下一章詳細展開。3. 從原始心電到機器學習輸入R 峰檢測、滑窗切片的 Matlab 路線圖3.1 數(shù)據(jù)集怎么組織先想清楚再動手很多人在跑通模型之后才發(fā)現(xiàn)數(shù)據(jù)文件夾結(jié)構不合理導致訓練腳本里全是if strcmp(subject, subj07)這樣的補丁。我的建議是一開始就按“受試者—條件—時間段”三級目錄組織。比如data/subj01/fatigue/seg1.mat表示 1 號受試者在疲勞狀態(tài)下采集的某一段心電。Matlab 里用dir批量遍歷這三個層級腳本邏輯會干凈很多。數(shù)據(jù)集的原始信號一般要存成單獨的.mat文件里面至少包含三個變量ecg原始信號、fs采樣率、label0 表示清醒1 表示疲勞。標簽放在每一段信號的 meta 里而不是放在全局變量里這樣后續(xù)做滑窗切分的時候每個窗口能繼承它所屬時間段的標簽。3.2 預處理三步去基線、帶通濾波、重采樣ECG 原始信號里通?；熘龢訓|西50 Hz 工頻干擾、肌電噪聲、以及因呼吸和電極移動造成的基線漂移。疲勞檢測關心的是 RR 間期所以預處理的核心目標是讓 R 峰檢測穩(wěn)定而不是把波形還原得多漂亮。%% 預處理基線漂移 帶通濾波 % 假設 ecg_raw 是原始信號fs 是采樣率 % 1) 用局部中位數(shù)估計基線減法去漂移 base movmedian(ecg_raw, round(0.2 * fs)); ecg_detrend ecg_raw - base; % 2) 0.5~40 Hz 帶通保留 QRS 主能量去掉肌電和高頻噪聲 ecg_filt bandpass(ecg_detrend, [0.5 40], fs);movmedian的窗口長度按 0.2 秒設置這個長度遠小于一個心搏周期因此估計出的“基線”不會把 QRS 波吞掉只跟蹤緩慢變化的漂移。bandpass的低頻設 0.5 Hz是因為呼吸造成的基線變化頻率通常在 0.2~0.5 Hz 之間低于 0.5 Hz 的成分丟掉不影響 QRS高頻設 40 Hz 是為了保留 QRS 的陡峭邊沿同時濾掉大部分肌電噪聲。如果你的數(shù)據(jù)采樣率只有 125 Hz上限就要降到 30 Hz 左右否則濾波結(jié)果會出現(xiàn)邊緣抖動。3.3 R 峰檢測閉著眼睛用 findpeaks 一定會翻車用findpeaks直接找 R 峰第一個版本一定會遇到兩類誤檢把 T 波當成 R 峰或者把運動偽跡當成 QRS。處理思路是在檢測之后加一個合理性校驗而不是盲目調(diào)閾值。%% R 峰檢測 合理性校驗 [~, locs] findpeaks(ecg_filt, ... MinPeakDistance, round(0.5 * fs), ... MinPeakProminence, 0.2 * max(ecg_filt)); rr diff(locs) / fs; % 剔除 RR 間期小于 0.4s心率150bpm或大于 2s心率30bpm的異常點 valid rr 0.4 rr 2.0; locs_valid locs([false; valid]); % 第一個 R 峰沒有對應的 RR 間期直接保留參數(shù)說明MinPeakDistance取 0.5 秒是下限對應 120 bpm這個值可以保證不會在同一心搏里檢出兩個峰同時不會漏掉心率稍快的受試者。MinPeakProminence取 0.2 倍幅度理由是 T 波的突出程度通常小于這個比例。合理性校驗這一步絕對不能省運動偽跡常常會讓某一段的 RR 間期直接掉到 0.2 秒如果不剔除后面算出來的 SDNN 會虛高。3.4 滑窗切片與標簽繼承特征矩陣的構造R 峰檢測完成后下一步是構造機器學習需要的特征矩陣。疲勞檢測一般不用整段信號做一條樣本而是把信號切成 30~60 秒的窗口每個窗口算一組特征特征矩陣的一行就是一個樣本。%% 滑動窗口切分 RR 間期序列構造特征矩陣 win_len 30; % 窗口長度秒 step 10; % 步長秒窗口重疊 2/3 % 先把 R 峰時間轉(zhuǎn)成連續(xù)時間軸 t_r locs_valid / fs; features zeros(0, 5); % 5 個特征SDNN, RMSSD, pNN50, LF/HF, SampEn labels []; for t0 0 : step : (max(t_r) - win_len) idx t_r t0 t_r t0 win_len; rr_win rr(valid) * 1000; % 對應窗口內(nèi)的 RR 間期ms if length(rr_win) 20 continue; % 窗口內(nèi)心搏太少特征不可靠 end f extract_hrv_features(rr_win); % 自定義函數(shù)見下方 features(end1, :) f; labels(end1, :) seg_label; % 整段時間段共用一個標簽 end窗口長度選 30 秒是經(jīng)驗值太短20 秒時 RMSSD 和 pNN50 的統(tǒng)計穩(wěn)定性很差太長5 分鐘則喪失了疲勞隨時間變化的檢測能力。步長 10 秒讓相鄰窗口有重疊一定程度上能平滑預測結(jié)果的抖動。extract_hrv_features里做的事情就是把上一章提到的 5 個特征全部算出來RD 間期標準差、RMSSD、pNN50、LF/HF 以及樣本熵。頻域特征這一步經(jīng)常有同學直接對 RR 間期序列調(diào)用fft得到的結(jié)果完全沒有意義因為 RR 間期是事件序列時間間隔不均勻。正確做法是先按 4 Hz 重采樣成均勻序列再算功率譜。Matlab 里用interp1就能實現(xiàn)重采樣頻率選 4 Hz 即可覆蓋 0.04~0.4 Hz 的 HRV 頻帶。4. 用 Matlab 訓練疲勞分類模型特征篩選、分類器對比與交叉驗證參數(shù)4.1 特征矩陣準備好了先做兩件事歸一化與正確劃分特征矩陣的行是窗口列是特征。第一個容易犯的錯是把所有樣本混在一起做歸一化然后再劃分訓練集和測試集。這看起來只是順序問題實際上會造成數(shù)據(jù)泄漏測試集的均值和標準差已經(jīng)通過歸一化參與到了訓練過程中導致交叉驗證分數(shù)虛高。正確做法是先劃分、再對訓練集算歸一化參數(shù)、最后應用到測試集上。第二個容易犯的錯是隨機劃分樣本。同一個受試者的相鄰時間窗高度相關隨機劃分會把同一個人的窗口同時分進訓練集和測試集模型很容易“認出”這個人。正確的劃分粒度是受試者保證同一個人的所有窗口要么全在訓練集要么全在測試集。這是拿去做畢設答辯時評委最常追問的一個點。4.2 按受試者劃分數(shù)據(jù)的 Matlab 寫法%% 按受試者分組劃分而不是按樣本行隨機劃分 subjects unique(subject_id); % subject_id 是和特征矩陣等長的列向量 cv cvpartition(numel(subjects), KFold, 5); for k 1 : cv.NumTestSets test_subjects subjects(cv.test(k)); test_idx ismember(subject_id, test_subjects); train_idx ~test_idx; X_train features(train_idx, :); y_train labels(train_idx, :); X_test features(test_idx, :); y_test labels(test_idx, :); % 只對訓練集估計歸一化參數(shù) mu mean(X_train); sigma std(X_train); X_train_norm (X_train - mu) ./ sigma; X_test_norm (X_test - mu) ./ sigma; % 在這里訓練模型并記錄每一折的指標 endcvpartition(numel(subjects), KFold, 5)的對象是受試者編號的列表而不是特征矩陣的行數(shù)。ismember負責把受試者編號映射回樣本行的邏輯索引。歸一化參數(shù)mu和sigma必須在每一折里重新計算這個細節(jié)決定了交叉驗證結(jié)果是否可信。4.3 一個能當默認基線的分類器集成樹對于特征數(shù)不多幾十個、樣本量不大幾百到幾千行的疲勞檢測任務集成樹是一個很好的默認基線。它不需要精細調(diào)參對特征量綱不敏感還能天然處理特征之間的非線性關系。%% 集成樹分類5 折交叉驗證 rng(42); % 固定隨機種子結(jié)果可復現(xiàn) mdl fitcensemble(X_train_norm, y_train, ... Method, Bag, ... NumLearningCycles, 50, ... Learners, templateTree(MinLeafSize, 3)); % 查看訓練集上的交叉驗證準確率 cvmdl crossval(mdl); loss kfoldLoss(cvmdl, LossFun, ClassifError); acc 1 - loss;Method選Bag即隨機森林的 bagging 形式NumLearningCycles是樹的數(shù)量。50 棵樹對疲勞檢測這個規(guī)模的數(shù)據(jù)足夠再多收益很小且訓練時間變長。MinLeafSize設 3 是防止單棵樹的葉子太大導致欠擬合這個值可以視樣本量調(diào)整樣本量越小葉子應該越小否則模型學不到細節(jié)。4.4 特征太多時先用 relieff 做一輪篩選如果做完特征工程后特征數(shù)量超過幾十個我一般會先做一次有監(jiān)督的特征篩選而不是直接全部丟進分類器。疲勞檢測里有些特征之間存在強相關性比如 SDNN 和總功率譜能量經(jīng)常高度相關同時放進去不會提升精度卻會讓模型更難解釋。%% 用 relieff 做特征重要性排序 [idx, weights] relieff(features, labels, 10); % 輸出按重要性排序的特征索引取前 8~12 個 top_features idx(1:min(12, numel(idx)));relieff的第三個參數(shù)是最近鄰數(shù)量默認 10 即可。它的原理是隨機選樣本在同類和異類中各找最近鄰比較特征在這些鄰居上的差異差異越大的特征權重越高。篩選之后重新跑 4.3 的集成樹你會發(fā)現(xiàn)交叉驗證分數(shù)要么持平要么略微上升但模型解釋起來輕松得多。4.5 分類器對比給畢業(yè)設計準備的基線表疲勞檢測的公開數(shù)據(jù)集一般沒有大到能支撐深度學習經(jīng)典的機器學習分類器已經(jīng)夠用。下面是四種常見分類器在這個任務上的定位方便你在論文里做對比實驗時參考。分類器Matlab 函數(shù)適合場景需要關注的參數(shù)集成樹fitcensemble默認基線魯棒NumLearningCycles, MinLeafSize支持向量機fitcsvm特征維度高、樣本量小KernelScale, BoxConstraint線性判別fitcdiscr特征近似高斯分布、追求可解釋無特別參數(shù)K 近鄰fitcknn數(shù)據(jù)量小、不做實時部署NumNeighbors, DistanceSVM 在小樣本場景經(jīng)常表現(xiàn)不錯但要用KernelScale, auto讓它自動估計核寬度否則手動給定的尺度往往不合適。疲勞檢測的輸出如果只需要“疲勞 / 清醒”二分類SVM 和集成樹的差距通常不大沒必要過度糾結(jié)分類器選型。真正決定上限的是特征質(zhì)量和驗證方式。5. 心電疲勞檢測的常見問題排查5 條翻車實錄與修復方法5.1 數(shù)據(jù)泄漏交叉驗證準確率 98%答辯現(xiàn)場被打臉現(xiàn)象模型在交叉驗證里準確率接近滿分但換成新受試者數(shù)據(jù)后準確率跌到 60% 左右。原因這是疲勞檢測里最隱蔽的坑。大多數(shù)情況下是因為歸一化或特征篩選在整個數(shù)據(jù)集上完成測試集的信息混入了訓練流程另一個常見原因是同受試者的窗口被隨機分到了訓練集和測試集模型實際上在記憶受試者身份而不是在識別疲勞狀態(tài)。解決嚴格按照 4.2 節(jié)的“先按受試者分組、再在訓練集內(nèi)部做歸一化”的流程重寫評估腳本。檢查方法很簡單打印出每一折訓練集和測試集的受試者 ID確認沒有重疊。這個動作花十分鐘但能避免答辯時被評委一句話問住。5.2 R 峰誤檢T 波被當成 R 峰HRV 特征全線失真現(xiàn)象提取出的 RR 間期序列里頻繁出現(xiàn) 0.3 秒左右的“心跳間隔”SDNN 和 RMSSD 異常偏大疲勞趨勢完全反了。原因findpeaks只認峰值不認波形形態(tài)。在疲勞后期心電信號形態(tài)會因肌肉緊張和呼吸變化而變形T 波幅度升高很容易滿足峰值條件。如果同時沒有設置MinPeakDistance就會把 T 波誤檢成 R 峰。解決除了加MinPeakDistance我在預處理里額外加了一步 RR 間期合理性校驗剔除小于 0.4 秒和大于 2.0 秒的間隔并把被剔除間隔兩側(cè)的 R 峰一并標記為可疑。同時在特征提取前打印一個簡單的直方圖肉眼確認 RR 間期的分布峰值在 0.6~1.0 秒之間這屬于心電分析的直覺檢查能發(fā)現(xiàn)大多數(shù)誤檢問題。5.3 樣本不平衡模型永遠輸出“清醒”準確率還挺高現(xiàn)象疲勞樣本只占總樣本的 20%訓練出的模型在測試集上準確率 80% 以上但混淆矩陣顯示所有的疲勞樣本都被判成了清醒。原因這是典型的類別不平衡問題。分類器默認優(yōu)化整體準確率對占比大的類別更“友好”少數(shù)類樣本被犧牲了。疲勞數(shù)據(jù)采集成本高、受試者配合度有限疲勞段天然少于清醒段。解決先在訓練時給少數(shù)類加權重。集成樹可以用fitcensemble后處理或者直接換用fitcsvm并設置Weights, w把疲勞類的權重設為清醒類的 3 倍左右。評價指標也要改不要只看準確率要看疲勞類的召回率和 F1-score。Matlab 里用confusionchart畫混淆矩陣一眼就能看出問題在哪。5.4 頻域特征頻譜混亂直接用 FFT 處理 RR 間期現(xiàn)象用fft(rr_ms)算出的 LF/HF 忽大忽小完全沒有規(guī)律。原因RR 間期是事件序列時間間隔不均勻。休息時心跳間隔約 1 秒但每一秒都不完全一致直接對原始 RR 序列做 FFT采樣時間點不固定頻譜已經(jīng)失真了。解決先用interp1把 RR 間期重采樣到 4 Hz 等間隔序列再做功率譜估計。Matlab 里pburg或pwelch都行pburg對短數(shù)據(jù)更友好。注意重采樣前先去除異常 RR 間期否則一個 2 秒的偽跡會被插值成大片低頻能量。5.5 特征全部堆上去維度高、樣本少、過擬合現(xiàn)象特征工程做了 30 個特征交叉驗證分數(shù)反而比只用 5 個特征時低測試集波動很大。原因樣本量窗口數(shù)可能只有幾百而特征維度到了幾十模型進入過擬合區(qū)間。尤其是非線性特征之間相關性高重復信息被模型當成相互獨立的證據(jù)放大了噪聲。解決先做相關性矩陣把相關性大于 0.85 的特征去掉一組再用 4.4 節(jié)的relieff排序只保留前 10~12 個。在疲勞檢測這個任務上精簡特征后的模型在跨受試者測試里通常比全特征模型更穩(wěn)。5.6 避坑自查清單做完一輪實驗后按下面這個清單逐項檢查能攔截大多數(shù)翻車事故檢查項通過標準受試者分組每一折的訓練集與測試集受試者 ID 零重疊歸一化時機每一折內(nèi)先 fit 訓練集再 transform 測試集RR 間期分布直方圖主峰位于 0.5~1.2 秒無大量 0.4 秒以下短間隔類別不平衡混淆矩陣中疲勞類召回率不是 0頻域特征重采樣為等間隔序列后再算功率譜6. 把疲勞檢測模型真正用起來滑動平滑、置換檢驗與最小可行驗證畢設交稿或者課程設計驗收之前我建議花半天時間做三件小事它們能顯著提升項目的說服力。第一給預測結(jié)果加滑動平均。模型的輸出是每個窗口的疲勞概率相鄰窗口概率波動往往很大直接按 0.5 閾值硬切會產(chǎn)生頻繁的狀態(tài)跳變。我在部署時一般對概率序列做長度為 3~5 的滑動平均再套閾值狀態(tài)切換就平滑了。這個技巧不改變模型本身卻讓演示效果完全不同。第二做一次置換檢驗。把標簽隨機打亂 100 次每次重新訓練模型并記錄準確率如果真實準確率落在隨機打亂后的分布邊緣說明模型并沒有學到有效信息。Matlab 里用一個for循環(huán)加上randperm就能實現(xiàn)這個實驗放在論文里是很有說服力的基線驗證。第三用最小可行模型驗證整個流程。我會先用 SDNN、RMSSD、pNN50 這 3 個最基礎的時域特征跑通一遍確認疲勞組的特征值確實低于清醒組再做頻域和非線性特征。如果基礎特征連趨勢都反了問題大概率不在模型而在 R 峰檢測或標簽定義上這時候繼續(xù)加特征只會讓錯誤更隱蔽。我自己的習慣是每次拿到一份新的心電數(shù)據(jù)先打印一頁報告RR 間期直方圖、兩個窗口的時域特征對比、以及分類器的混淆矩陣三樣東西花不了兩分鐘卻能避免在錯誤的方向上調(diào)試一整天。做這類項目慢不是問題方向錯了才是。希望這些經(jīng)驗能在你的心電疲勞檢測項目里派上用場祝順利。本文還有配套的精品資源點擊獲取