压力影响脑电图模型预测精神分裂症的准确性,通过校正压力伪影可提升性能。
The Effect of Acute Stress on the Interpretability and Generalization of Schizophrenia Predictive Machine Learning Models
- 用XGBoost建模并结合XAI分析,识别记录时的压力状态
- 压力波动降低模型泛化能力,校正后预测准确率显著提升
- 适合关注临床数据质量与模型鲁棒性的研究者阅读
精神分裂症是一种严重精神障碍,早期诊断对改善预后至关重要。其复杂性使发病和进展预测困难。脑电图(EEG)已成为研究该病的重要工具,机器学习在诊断中应用日益广泛。本文评估了机器学习模型预测精神分裂症的准确性,并探讨了脑电记录期间急性应激对模型性能的影响。构建了四个XGBoost模型:一个用于压力预测,两个用于静息态和任务态下的精神分裂症分类,一个用于两种条件下的联合预测。采用XAI技术分析结果。实验通过健康对照组和独立健康筛查对照组测试模型泛化能力。压力模型识别出高压力受试者,并将其排除于后续分析。提出一种新方法,调整脑电频段功率以去除压力伪影,从而提升预测模型性能。结果显示,急性应激在不同脑电记录中存在差异,影响模型表现与准确性;在训练中考虑并补偿这种压力变化后,模型泛化能力显著提升。研究强调了全面健康筛查及记录过程中患者状态管理的重要性。记录期间或由记录引发的应激可能损害模型泛化,建议将压力视为额外生理伪影进行数据预处理。所提出的校正压力伪影的方法显著提升了基于脑电图训练模型的预测性能。
原文摘要 · Abstract (English)
Introduction Schizophrenia is a severe mental disorder, and early diagnosis is key to improving outcomes. Its complexity makes predicting onset and progression challenging. EEG has emerged as a valuable tool for studying schizophrenia, with machine learning increasingly applied for diagnosis. This paper assesses the accuracy of ML models for predicting schizophrenia and examines the impact of stress during EEG recording on model performance. We integrate acute stress prediction into the analysis, showing that overlapping conditions like stress during recording can negatively affect model accuracy. Methods Four XGBoost models were built: one for stress prediction, two to classify schizophrenia (at rest and task), and a model to predict schizophrenia for both conditions. XAI techniques were applied to analyze results. Experiments tested the generalization of schizophrenia models using their datasets' healthy controls and independent health-screened controls. The stress model identified high-stress subjects, who were excluded from further analysis. A novel method was used to adjust EEG frequency band power to remove stress artifacts, improving predictive model performance. Results Our results show that acute stress vary across EEG sessions, affecting model performance and accuracy. Generalization improved once these varying stress levels were considered and compensated for during model training. Our findings highlight the importance of thorough health screening and management of the patient's condition during the process. Stress induced during or by the EEG recording can adversely affect model generalization. This may require further preprocessing of data by treating stress as an additional physiological artifact. Our proposed approach to compensate for stress artifacts in EEG data used for training models showed a significant improvement in predictive performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。