WikiEdge:ArXiv-2409.06585v1/methods
跳转到导航
跳转到搜索
这篇文献的工作部分详细介绍了如何开发和评估用于预测髋关节置换的时序图卷积神经网络模型(TG-CNN)。以下是这部分的主要内容:
- 背景与动机:
- 方法论:
- 数据收集与处理:
- 使用ResearchOne电子健康记录(EHRs)数据,包括临床和行政数据,涵盖151,565名患者的记录。
- 对患者的EHRs进行时间窗口化处理,以预测一年内发生的髋关节置换。
- 选择了最常见的512个Read Codes来构建基于时序图的EHR表示,同时包括了体质指数(BMI)作为预测因子。
- 模型架构:
- 使用TensorFlow构建了一个定制的3D CNN Keras层,利用稀疏线性代数处理512×512×100大小的时序图表示。
- CNN层的输出经过展平、批量归一化,并与LSTM层(捕获长期模式)、dropout层、密集层和人口统计特征(如果包括)进行连接。
- 最终,模型使用分类交叉熵损失和sigmoid函数针对分类目标进行优化。
- 评估与校准:
- 遵循TRIPOD-AI声明报告模型开发和预测模型。
- 在训练集上进行5折交叉验证以选择超参数,并基于验证集的平均准确率优化模型。
- 使用10%的测试数据集进行模型重新校准,以确保预测概率与测试集中结果的真实发生率相匹配。
- 在重新校准后,使用第二个未见过的测试集来验证模型的性能。
- 结果:
- 讨论与结论:
- 讨论了TG-CNN模型在预测髋关节置换风险方面的有效性,以及如何通过识别患者轨迹中的模式来潜在地改善对髋关节相关疾病的理解和管理。
- 强调了模型在临床决策支持工具开发中的潜力,包括促进信任、辅助患者-临床医生讨论、提供个性化医疗和增强患者安全。