用强化学习自动优化脑电图数据增强,仅用10%标签数据就显著提升自监督学习效果。
RL-BioAug: Label-Efficient Reinforcement Learning for Self-Supervised EEG Representation Learning
- 用强化学习代理自主选择最优数据增强策略,避免传统随机方法丢失信号信息。
- 在睡眠分期和癫痫检测任务上分别提升9.69%和8.80%的宏平均F1分数。
- 适合需要高效利用少量标签数据的脑电图自监督学习研究者。
数据增强质量是对比学习在脑电图(EEG)任务中表现的关键因素。尽管该范式能有效利用无标签数据,但静态或随机增强策略常因脑电信号非平稳性(统计特性随时间变化)而丢失内在信息。为此,我们提出RL-BioAug框架,通过一个标签高效的强化学习(RL)代理自主确定最优增强策略。仅使用10%的标签数据引导代理,即可实现严格自监督的编码器表征学习。实验表明,该方法显著优于随机策略,在Sleep-EDFX和CHB-MIT数据集上分别取得9.69%和8.80%的宏平均F1分数提升。值得注意的是,代理为不同任务选择了最优策略:如睡眠分期任务中以62%概率选择时间掩码,癫痫检测任务中以77%概率选择裁剪与缩放。该框架展现出替代传统启发式增强的潜力,建立数据增强的新自主范式。源代码已开源:https://github.com/dlcjfgmlnasa/RL-BioAug。
原文摘要 · Abstract (English)
The quality of data augmentation serves as a critical determinant for the performance of contrastive learning in EEG tasks. Although this paradigm is promising for utilizing unlabeled data, static or random augmentation strategies often fail to preserve intrinsic information due to the non-stationarity of EEG signals where statistical properties change over time. To address this, we propose RL-BioAug, a framework that leverages a label-efficient reinforcement learning (RL) agent to autonomously determine optimal augmentation policies. While utilizing only a minimal fraction (10%) of labeled data to guide the agent's policy, our method enables the encoder to learn robust representations in a strictly self-supervised manner. Experimental results demonstrate that RL-BioAug significantly outperforms the random selection strategy, achieving substantial improvements of 9.69% and 8.80% in Macro-F1 score on the Sleep-EDFX and CHB-MIT datasets, respectively. Notably, this agent mainly chose optimal strategies for each task--for example, Time Masking with a 62% probability for sleep stage classification and Crop & Resize with a 77% probability for seizure detection. Our framework suggests its potential to replace conventional heuristic-based augmentations and establish a new autonomous paradigm for data augmentation. The source code is available at https://github.com/dlcjfgmlnasa/RL-BioAug.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。