arXiv:2508.13476cs.LG2025-08

用可解释嵌入分析癫痫患者发作特征,预测手术效果。

Classifying Clinical Outcome of Epilepsy Patients with Ictal Chirp Embeddings

  • 通过t-SNE降维并保留局部结构,生成可视觉化的时间-频域特征嵌入。
  • 随机森林与kNN在最优病例识别中达88.8%准确率,区分难易程度和治疗成败。
  • 结合SHAP生成特征重要性图,揭示关键波形特征如何决定聚类分布。

本研究提出一个流程,利用t-SNE对多种预后情景下的啁啾特征进行可解释可视化。数据集包含基于啁啾的时间、频谱及频率度量。通过基于学生t分布的相似性优化,t-SNE在保持局部邻域关系的同时缓解了拥挤问题。基于二维t-SNE嵌入,构建了三个分类任务:(1) 区分临床成功与失败/未切除;(2) 分离高难度与低难度病例;(3) 识别最优病例(即成功且临床难度最低)。采用随机森林、支持向量机、逻辑回归和k近邻四类分类器,使用分层5折交叉验证进行训练与评估。在各任务中,随机森林与kNN表现更优,在最优病例检测中最高达到88.8%准确率。此外,通过将SHAP解释应用于预测t-SNE坐标的模型,生成特征影响敏感度图,揭示了嵌入空间中特定啁啾属性驱动区域聚类与类别分离的空间局部特征重要性,为数据潜在结构提供了洞察。该集成框架展示了可解释嵌入与局部特征归因在临床分层与决策支持中的潜力。

原文摘要 · Abstract (English)

This study presents a pipeline leveraging t-Distributed Stochastic Neighbor Embedding (t-SNE) for interpretable visualizations of chirp features across diverse outcome scenarios. The dataset, comprising chirp-based temporal, spectral, and frequency metrics. Using t-SNE, local neighborhood relationships were preserved while addressing the crowding problem through Student t-distribution-based similarity optimization. Three classification tasks were formulated on the 2D t-SNE embeddings: (1) distinguishing clinical success from failure/no-resection, (2) separating high-difficulty from low-difficulty cases, and (3) identifying optimal cases, defined as successful outcomes with minimal clinical difficulty. Four classifiers, namely, Random Forests, Support Vector Machines, Logistic Regression, and k-Nearest Neighbors, were trained and evaluated using stratified 5-fold cross-validation. Across tasks, the Random Forest and k-NN classifiers demonstrated superior performance, achieving up to 88.8% accuracy in optimal case detection (successful outcomes with minimal clinical difficulty). Additionally, feature influence sensitivity maps were generated using SHAP explanations applied to model predicting t-SNE coordinates, revealing spatially localized feature importance within the embedding space. These maps highlighted how specific chirp attributes drive regional clustering and class separation, offering insights into the latent structure of the data. The integrated framework showcases the potential of interpretable embeddings and local feature attribution for clinical stratification and decision support.

癫痫预测可解释性特征分析t-SNE

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。