简介这份本科毕业设计资源聚焦基于Transformer的运动想象脑电信号分类面向人工智能与生物医学工程交叉方向的本科生及脑机接口入门研究者帮助解决EEG信号深层模式挖掘与多类别运动想象识别问题。压缩包共31个文件约18.45MB以23个Python脚本为核心涵盖CNNTransformer模型构建、EEGNet与Conformer对比、空间注意力、CSP特征提取及t-SNE可视化等模块另含2个MATLAB预处理脚本、2个xlsx实验数据表、1个npy训练数据、1个pth模型权重及xml、md说明文件结构完整便于复现。目前已有275人学习下载。读者可据此获得从数据预处理、时频特征提取到模型训练与k折交叉验证的完整方案并借助AUC、箱线图、脑热力图与CAM分析理解分类依据适合作为毕业设计参考或脑电分类项目的起步模板。1. 运动想象脑电分类从Transformer入场说起运动想象脑电信号分类说白了就是让人在脑子里“过一遍”左手或右手的动作算法通过头皮上采集的脑电波判断他到底在想哪只手。这件事在脑机接口领域做了二十多年传统方案从共空间模式加SVM到后来CNN、EEGNet精度在公开数据集上基本卡在七成上下。Transformer进来之后局面有了变化——它靠自注意力机制直接建模通道间与时间片间的长程依赖不再依赖手工设计卷积核尺寸。如果你正在做本科毕业设计选题落在“基于Transformer的运动想象脑电信号分类”那这篇笔记就是把你从数据集下载、预处理、模型搭建、训练调参到避坑的整条链路走一遍。适合已经学过深度学习基础、跑过至少一个PyTorch或TensorFlow小项目、但还没碰过脑电信号这个模态的本科生和初级工程师。读完你能得到一个可复现的最小系统也能判断这个方向值不值得继续投入。2. 数据到手先别急着喂模型运动想象脑电的预处理链路2.1 公开数据集选型与通道取舍做运动想象分类绕不开两个公开数据集BCI Competition IV 2a 和 2b。2a 是9名被试、22通道、左右手/双脚/舌头四分类2b 是9名被试、3通道C3、Cz、C4、左右手二分类。本科毕设如果时间紧建议从2b入手通道少、二分类、基线清晰跑通全流程后再迁移到2a。常见做法是只取C3、Cz、C4三个通道因为运动想象的事件相关去同步/同步现象在感觉运动皮层最显著这三个电极正好覆盖该区域。如果你用2a22通道全上不是不行但Transformer的输入token数会变成22乘以时间片数显存和过拟合风险都陡增。我一般会先做通道筛选计算每个通道在任务态和静息态之间的方差比保留排名前8到12的通道再送入模型。2.2 带通滤波与滑动窗口切分原始脑电采样率2a是250Hz2b是250Hz。运动想象的有效频段集中在8到30Hz具体分mu节律8-13Hz和beta节律13-30Hz。预处理第一步是带通滤波用Butterworth四阶零相位滤波避免相位失真。代码示例如下import numpy as np from scipy.signal import butter, filtfilt def bandpass_filter(data, lowcut8.0, highcut30.0, fs250, order4): data: shape (n_trials, n_channels, n_times) 返回同样shape的滤波后数据 nyq 0.5 * fs low lowcut / nyq high highcut / nyq b, a butter(order, [low, high], btypeband) # filtfilt做零相位滤波前后向各滤一次 filtered filtfilt(b, a, data, axis-1) return filtered逻辑说明butter设计滤波器系数filtfilt避免滤波带来的时间延迟。参数上lowcut和highcut根据你的任务调整左右手运动想象用8-30Hz是安全选择如果只关注mu节律可以设8-13Hz但会丢失beta频段的信息。注意filtfilt要求数据长度至少是滤波器阶数的3倍否则报错。滤波之后做滑动窗口切分。运动想象范式通常是提示后0.5秒到2.5秒为任务期我一般取提示后0.5秒到3.5秒用长度2秒、步长0.1秒的滑窗切分这样每个trial能生成多个样本数据增强效果比单用trial级样本好。标签继承trial标签。2.3 标准化与伪迹剔除脑电信号幅度在微伏级不同被试、不同session之间漂移很大。逐通道做z-score标准化是标配减去训练集均值、除以训练集标准差验证集和测试集用同样的均值和标准差。千万别在全部数据上算均值和标准差再划分那是数据泄露测试精度会虚高。伪迹剔除方面眼电和肌电是主要污染源。简单做法是计算每个trial的峰峰值超过100微伏的trial直接丢弃。更精细的用ICA分解把 frontal 区域权重高的成分去掉但ICA计算量大本科毕设如果时间紧峰峰值法够用。注意剔除比例不要超过总trial的20%否则类别不平衡会加剧。3. Transformer怎么搭从输入嵌入到分类头的逐层拆解3.1 脑电信号的token化把时间片映射成向量Transformer原本是为序列设计的脑电信号本身就是时间序列但有个问题每个时间点只是一个标量直接一个点一个token序列太长且信息密度低。常见做法是把时间维切成若干段每段做线性投影得到一个embedding。比如2秒数据、250Hz采样率共500个时间点切成50段每段10个点然后用一个线性层把10维映射到64维得到50个token。通道维怎么处理两种方案一是把通道也拼进每段的特征里比如3通道乘以10个点等于30维再投影到64维二是通道独立做token再在Transformer里靠注意力融合。我一般用第一种因为参数量少且通道间关系在投影前已经混合。import torch import torch.nn as nn class EEGPatchEmbedding(nn.Module): def __init__(self, n_channels3, patch_len10, embed_dim64, n_patches50): super().__init__() self.patch_len patch_len self.n_patches n_patches # 输入维度 通道数 * 每段点数 self.projection nn.Linear(n_channels * patch_len, embed_dim) # 可学习的位置编码 self.pos_embed nn.Parameter(torch.randn(1, n_patches, embed_dim) * 0.02) def forward(self, x): # x: (batch, n_channels, n_times) batch, ch, times x.shape # 切成n_patches段每段patch_len个点 x x.reshape(batch, ch, self.n_patches, self.patch_len) # 重排成 (batch, n_patches, ch*patch_len) x x.permute(0, 2, 1, 3).reshape(batch, self.n_patches, -1) x self.projection(x) # (batch, n_patches, embed_dim) x x self.pos_embed return x逻辑说明reshape把时间维切成段permute调整维度顺序让通道和段内点拼在一起线性层做投影。位置编码用可学习参数比正弦编码在脑电这种非自然语言序列上更灵活。参数上patch_len和n_patches要满足patch_len乘以n_patches等于总时间点数embed_dim通常取64或128太大容易过拟合太小表达力不够。3.2 编码器层数与注意力头数的取舍Transformer编码器由多头自注意力和前馈网络组成。本科毕设场景下编码器层数建议2到4层注意力头数4到8。层数太多参数量上去小数据集上过拟合严重层数太少长程依赖建模能力不足。我一般从2层4头起步看验证集精度再决定是否加深。前馈网络的隐藏维度通常是embed_dim的2到4倍比如embed_dim64隐藏层取128或256。class TransformerEncoder(nn.Module): def __init__(self, embed_dim64, n_heads4, n_layers2, ff_dim128, dropout0.3): super().__init__() encoder_layer nn.TransformerEncoderLayer( d_modelembed_dim, nheadn_heads, dim_feedforwardff_dim, dropoutdropout, batch_firstTrue ) self.encoder nn.TransformerEncoder(encoder_layer, num_layersn_layers) def forward(self, x): # x: (batch, n_patches, embed_dim) return self.encoder(x)逻辑说明直接调用PyTorch的TransformerEncoderLayerbatch_firstTrue让输入维度是(batch, seq, feature)。dropout设0.3到0.5脑电小数据集上正则化很重要。注意PyTorch默认的注意力实现是O(n^2)复杂度n_patches50时计算量可控如果切到200段以上考虑用Flash Attention或减少段数。3.3 分类头与损失函数的选择编码器输出是(batch, n_patches, embed_dim)分类需要聚合成一个向量。常见做法是取平均池化或取第一个token类似BERT的[CLS]。我一般用平均池化因为脑电没有明确的“分类token”语义。池化后接一个线性层映射到类别数。class EEGTransformer(nn.Module): def __init__(self, n_channels3, patch_len10, n_patches50, embed_dim64, n_heads4, n_layers2, n_classes2, dropout0.3): super().__init__() self.embedding EEGPatchEmbedding(n_channels, patch_len, embed_dim, n_patches) self.encoder TransformerEncoder(embed_dim, n_heads, n_layers, ff_dimembed_dim*2, dropoutdropout) self.classifier nn.Sequential( nn.LayerNorm(embed_dim), nn.Linear(embed_dim, n_classes) ) def forward(self, x): x self.embedding(x) # (batch, n_patches, embed_dim) x self.encoder(x) # (batch, n_patches, embed_dim) x x.mean(dim1) # 平均池化 return self.classifier(x)损失函数用交叉熵类别不平衡时加权重。优化器用AdamW学习率1e-3到1e-4权重衰减1e-2到1e-4。学习率调度用余弦退火训练50到100个epoch早停耐心设10到15。4. 训练与评估让模型真正学到运动想象特征而不是噪声4.1 被试独立与被试依赖的评估协议运动想象脑电有个核心问题不同被试的脑电模式差异极大被试依赖同一被试的数据混合划分训练测试精度可以到85%以上被试独立留一被试交叉验证可能掉到65%。本科毕设如果只报被试依赖精度答辩时容易被质疑泛化能力。我建议两个协议都跑被试依赖看模型拟合能力被试独立看泛化。被试独立用留一法9个被试轮流做测试集报告平均精度和标准差。4.2 训练循环与关键超参数训练循环用标准PyTorch流程但有几个脑电特有的注意点。第一batch size不要太大16到32即可因为样本数本身不多。第二每个epoch后打乱数据但同一trial切出的滑窗样本尽量分到同一折避免信息泄露。第三梯度裁剪设1.0防止梯度爆炸。from torch.utils.data import DataLoader, TensorDataset import torch.optim as optim def train_model(model, train_X, train_y, val_X, val_y, epochs80, lr1e-3, batch_size32): device torch.device(cuda if torch.cuda.is_available() else cpu) model model.to(device) optimizer optim.AdamW(model.parameters(), lrlr, weight_decay1e-2) scheduler optim.lr_scheduler.CosineAnnealingLR(optimizer, T_maxepochs) criterion nn.CrossEntropyLoss() train_loader DataLoader(TensorDataset(torch.FloatTensor(train_X), torch.LongTensor(train_y)), batch_sizebatch_size, shuffleTrue) best_val_acc 0.0 patience_counter 0 for epoch in range(epochs): model.train() for xb, yb in train_loader: xb, yb xb.to(device), yb.to(device) optimizer.zero_grad() logits model(xb) loss criterion(logits, yb) loss.backward() torch.nn.utils.clip_grad_norm_(model.parameters(), 1.0) optimizer.step() scheduler.step() # 验证 model.eval() with torch.no_grad(): val_logits model(torch.FloatTensor(val_X).to(device)) val_pred val_logits.argmax(dim1).cpu().numpy() val_acc (val_pred val_y).mean() if val_acc best_val_acc: best_val_acc val_acc patience_counter 0 torch.save(model.state_dict(), best_model.pth) else: patience_counter 1 if patience_counter 15: print(fEarly stop at epoch {epoch}) break return best_val_acc逻辑说明AdamW带权重衰减余弦退火让学习率平滑下降梯度裁剪防爆炸早停防过拟合。参数上学习率1e-3适合从零训练如果加载预训练权重可以降到1e-4权重衰减1e-2在脑电小数据上正则效果明显。4.3 精度上不去时先查什么如果被试依赖精度低于70%按顺序查一、标签是否对齐滑窗切分后标签有没有错位二、标准化是否用了全局统计量导致泄露三、学习率是否太大loss震荡四、数据增强是否过度把判别性特征抹掉了。如果被试独立精度低但被试依赖高那是泛化问题考虑加域适应层或对抗训练但本科毕设做到这一步已经够毕业了。5. 避坑与排查运动想象Transformer训练里最容易翻车的五件事5.1 现象训练loss下降但验证精度始终50%左右原因数据泄露导致训练集和验证集分布不一致或者标签在滑窗切分时错位。常见于先切窗再划分数据集同一个trial的窗口同时出现在训练和验证集。解决按trial划分数据集再对训练集切窗做增强验证集用trial级样本或独立窗口。5.2 现象模型输出全部预测为同一类原因类别极度不平衡或者损失函数没有加权。运动想象数据里左右手通常均衡但剔除伪迹后可能失衡。解决计算类别权重传入CrossEntropyLoss的weight参数或者用Focal Loss。另外检查最后一层是否初始化正常全零初始化会导致输出恒定。5.3 现象验证精度波动极大不同随机种子差10个点原因小数据集上模型初始化敏感或者batch size太小导致梯度噪声大。解决固定随机种子用5折交叉验证报告平均精度batch size提到32或64加Dropout和权重衰减。如果还是波动考虑用集成方法训练5个模型取投票。5.4 现象GPU显存溢出尤其是用2a的22通道时原因token数等于通道数乘以时间片数22通道乘以50段等于1100个token自注意力矩阵是1100乘1100显存爆炸。解决先做通道筛选降到8到12通道或者用通道独立的token方案每个通道单独做注意力再融合或者减小patch数量增大patch_len。5.5 现象被试独立评估时某些被试精度接近随机原因脑电信号被试间差异大某些被试的运动想象特征本身就不明显。解决这不是模型问题是数据问题。报告结果时剔除或单独说明这些被试或者用域适应方法对齐不同被试的特征分布。本科毕设里如实报告并分析原因比强行刷高精度更可信。6. 把精度再推一截通道注意力与数据增强的实战技巧如果你已经跑通基线被试依赖精度在75%到80%之间想再往上走有两个方向投入产出比最高。第一个是通道注意力在patch embedding之后加一个通道注意力模块让模型自动学习哪些通道对当前任务更重要。具体做法是对每个通道的特征做全局平均池化过一个小MLP得到通道权重再乘回原特征。这个模块参数量极小但在2a数据集上通常能涨2到3个点。第二个是数据增强脑电信号不能像图像那样随便旋转裁剪但可以加高斯噪声、做时间维的随机缩放、或者用mixup在样本对之间做线性插值。我一般用mixup加轻微高斯噪声噪声标准差取信号标准差的0.1倍太大反而掉点。还有一个容易被忽略的技巧学习率预热。前5个epoch从1e-5线性升到1e-3再余弦退火。Transformer对初始学习率敏感预热能明显稳定训练。另外如果你用2b数据集3通道输入时patch_len可以设小一点比如5个点一段增加token数让注意力有更多交互机会但注意显存。最后说一个我踩过的坑不要盲目堆层数。我试过8层编码器参数量是2层的4倍但被试独立精度反而掉了3个点因为过拟合了。后来固定用2层4头配合强正则泛化最好。这个方向值得做但别指望Transformer一上就碾压CNN预处理和评估协议的设计往往比模型结构更决定最终结果。希望帮到你。本文还有配套的精品资源点击获取