计算机科学 > 机器学习
[提交于 2025年1月2日
]
标题: RealDiffFusionNet:使用真实世界数据进行疾病进展建模的神经受控微分方程多头注意力融合网络
标题: RealDiffFusionNet: Neural Controlled Differential Equation Informed Multi-Head Attention Fusion Networks for Disease Progression Modeling Using Real-World Data
摘要: 本文提出了一种名为RealDiffFusionNet的新型基于深度学习的方法,结合了神经控制微分方程(Neural CDE)- 时间序列模型,这些模型在处理不规则采样数据时具有鲁棒性,并结合多头注意力机制,在每个时间点对相关多模态上下文(图像数据、时间不变数据等)进行对齐。长短期记忆(LSTM)模型也被用作基线模型。使用了两个不同的数据集:一个来自开源成像联盟(OSIC)的数据集,包含人口统计数据和肺功能的结构化时间序列数据以及肺部基线CT扫描;另一个来自阿尔茨海默病神经影像学倡议(ADNI)的数据集,包含一系列MRI扫描以及人口统计数据、体格检查和认知评估数据。进行了消融研究以了解CDEs、多模态数据、注意力融合和插值策略对模型性能的作用。当评估基线模型时,使用多模态数据提高了神经CDE的性能,测试RMSE更低。此外,多模态神经CDE的性能也优于多模态LSTM。在基于注意力的架构中,通过拼接和矩形插值进行融合被发现可以提高模型性能。所提出的RealDiffFusionNet的性能被发现优于所有模型(0.2570)。对于ADNI数据集,在仅在结构化数据上训练的神经CDE和LSTM模型之间,测试RMSE是相当的(LSTM为0.471,神经CDE为0.4581)。此外,添加患者MRI序列中的图像特征提高了性能,测试RMSE更低(多模态为0.4372,结构化数据为0.4581)。RealDiffFusionNet在利用CDEs和多模态数据准确预测疾病进展方面显示出前景。
文献和引用工具
与本文相关的代码,数据和媒体
alphaXiv (什么是 alphaXiv?)
CatalyzeX 代码查找器 (什么是 CatalyzeX?)
DagsHub (什么是 DagsHub?)
Gotit.pub (什么是 GotitPub?)
Hugging Face (什么是 Huggingface?)
带有代码的论文 (什么是带有代码的论文?)
ScienceCast (什么是 ScienceCast?)
演示
推荐器和搜索工具
arXivLabs:与社区合作伙伴的实验项目
arXivLabs 是一个框架,允许合作伙伴直接在我们的网站上开发和分享新的 arXiv 特性。
与 arXivLabs 合作的个人和组织都接受了我们的价值观,即开放、社区、卓越和用户数据隐私。arXiv 承诺这些价值观,并且只与遵守这些价值观的合作伙伴合作。
有一个为 arXiv 社区增加价值的项目想法吗? 了解更多关于 arXivLabs 的信息.