用0.5B参数轻量大模型,实时解析脑电情绪并自动生成病历。
EEG Emotion Copilot: Optimizing Lightweight LLMs for Emotional EEG Interpretation with Assisted Medical Record Generation
- 用定制提示结构+剪枝微调,让小模型实现高效脑电情绪识别
- 在0.5B参数下达到比1.5B~7B大模型更优的情绪识别与病历生成效果
- 适合精神健康监测、可本地运行的轻量级医疗辅助系统
在情感计算(AC)与脑机接口(BMI)领域,通过生理与行为信号识别个体情绪状态已成为前沿课题。尽管深度学习在脑电信号情绪识别中取得进展,但在端到端情绪计算方面仍面临实时性、个体适应性与用户交互等挑战。本文提出EEG情绪协作者(EEG Emotion Copilot),基于本地部署的0.5B参数轻量级大语言模型(LLM),直接从脑电信号识别情绪,生成个性化诊疗建议,并支持辅助电子病历自动化。通过创新的提示数据结构、模型剪枝与微调训练策略,显著提升实时性能与计算效率。大量实验表明,该优化后的轻量级模型在情绪识别准确率和病历生成质量上均优于同类规模(如1.5B、1.8B、3B、7B)模型。本系统有望推动情感计算在医疗领域的应用,为心理健康监测提供新方案。代码将开源于https://github.com/NZWANG/EEG_Emotion_Copilot。
原文摘要 · Abstract (English)
In the fields of affective computing (AC) and brain-machine interface (BMI), the analysis of physiological and behavioral signals to discern individual emotional states has emerged as a critical research frontier. While deep learning-based approaches have made notable strides in EEG emotion recognition, particularly in feature extraction and pattern recognition, significant challenges persist in achieving end-to-end emotion computation, including real-time processing, individual adaptation, and seamless user interaction. This paper presents the EEG Emotion Copilot, a system optimizing a lightweight large language model (LLM) with 0.5B parameters operating in a local setting, which first recognizes emotional states directly from EEG signals, subsequently generates personalized diagnostic and treatment suggestions, and finally supports the automation of assisted electronic medical records. Specifically, we demonstrate the critical techniques in the novel data structure of prompt, model pruning and fine-tuning training, and deployment strategies aiming at improving real-time performance and computational efficiency. Extensive experiments show that our optimized lightweight LLM-based copilot achieves an enhanced intuitive interface for participant interaction, superior accuracy of emotion recognition and assisted electronic medical records generation, in comparison to such models with similar scale parameters or large-scale parameters such as 1.5B, 1.8B, 3B and 7B. In summary, through these efforts, the proposed copilot is expected to advance the application of AC in the medical domain, offering innovative solution to mental health monitoring. The codes will be released at https://github.com/NZWANG/EEG_Emotion_Copilot.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。