GRFNet开源EEG分类PyTorch模型
AI 总结
一位作者在r/MachineLearning介绍了GRFNet，这是一个面向EEG解码、用于多通道时间序列分类的开源PyTorch模型。该架构结合随机且冻结的卷积核、可学习空间滤波器、膨胀卷积、汇总统计量、学习门控和小型分类头；据称默认的22通道、四分类配置包含3,847个参数。作者表示，基于PhysioNet EEGMMIDB的主体独立评估仍在构建中，与EEGNet和ShallowConvNet的对比计划尚未产生结果。
深度见解
GRFNet提供了一个紧凑且开源的PyTorch EEG解码方案，但现有报告中的性能主张尚未得到验证，因为计划中的主体独立基准测试结果尚未公布。
| Hi all, Through my MSc work on EEG decoding, I built a novel (partially) architecture tailored to the constraints of EEG. The datasets are small, the subject variability is high, and signal acquisition variables differ significantly. I’ve now extracted that model to develop into an open source PyTorch model and would like some input on anything that comes to mind with regard to direction, etc. The design builds on ROCKET, using random, untrained convolutional kernels to yield strong time-series features. Learned spatial filters first mix the input channels, and if you provide sensor positions they initialize from the electrode layout instead of noise. The resulting components pass through a bank of dilated convolutions that are mostly random and frozen, with a few learnable kernels. Several summary statistics are extracted from the bank, combined through a learned gate, and classified by a small head. The default model (22 channels, 4 classes) has 3,847 parameters and It accepts any input length and adapts to the sampling rate. I'm now building a subject-independent evaluation on PhysioNet's EEGMMIDB (109 subjects), where each subject goes entirely into train or test. I'll compare against EEGNet and ShallowConvNet and report scalp topographies, confusion matrices, saliency maps and speed benchmarks. I’m also going to build four variants to compare results and analyse reasons why. I have no benchmark numbers yet (except my MSc work) as i’m early into starting, any input would be greatly appreciated. Would love a couple stars and follows if you think its decent work 😘 submitted by /u/joekrry [link] [comments] |