arXiv:2501.02025cs.LGcs.CV2025-01被引 1

用神经微分方程融合多模态数据,精准预测疾病进展。

RealDiffFusionNet: Neural Controlled Differential Equation Informed Multi-Head Attention Fusion Networks for Disease Progression Modeling Using Real-World Data

  • 引入神经控制微分方程处理不规则时间序列,结合多头注意力对齐多模态信息。
  • 在OSIC和ADNI数据集上,模型测试RMSE低至0.2570,优于基线模型。
  • 适合关注疾病动态建模、医疗时间序列分析的研究者使用。

本文提出一种名为RealDiffFusionNet的新方法,结合神经控制微分方程(Neural CDE)与多头注意力机制,用于处理不规则采样的时间序列数据,并在每个时间点对图像、静态数据等多模态信息进行对齐。采用LSTM作为基线模型,使用两个公开数据集:来自开放影像联盟(OSIC)的肺功能与基线CT数据,以及来自阿尔茨海默病神经影像计划(ADNI)的MRI、人口统计、体格检查和认知评估数据。消融实验表明,多模态数据可提升神经微分方程性能,其测试RMSE更低;多模态神经微分方程优于多模态LSTM。基于注意力的架构中,拼接融合与矩形插值策略表现更优。所提模型在两项数据集上均取得最佳性能,测试RMSE为0.2570。在仅使用结构化数据时,神经微分方程(0.4581)与LSTM(0.471)性能相近;加入MRI特征后,测试RMSE降至0.4372,进一步提升。结果表明,RealDiffFusionNet能有效利用神经微分方程与多模态数据实现疾病进展的高精度预测。

原文摘要 · Abstract (English)

This paper presents a novel deep learning-based approach named RealDiffFusionNet incorporating Neural Controlled Differential Equations (Neural CDE) - time series models that are robust in handling irregularly sampled data - and multi-head attention to align relevant multimodal context (image data, time invariant data, etc.) at each time point. Long short-term memory (LSTM) models were also used as a baseline. Two different datasets were used: a data from the Open-Source Imaging Consortium (OSIC) containing structured time series data of demographics and lung function with a baseline CT scan of the lungs and the second from the Alzheimer's Disease Neuroimaging Initiative (ADNI) containing a series of MRI scans along with demographics, physical examinations, and cognitive assessment data. An ablation study was performed to understand the role of CDEs, multimodal data, attention fusion, and interpolation strategies on model performance. When the baseline models were evaluated, the use of multimodal data resulted in an improvement in Neural CDE performance, with a lower test RMSE. Additionally, the performance of multimodal Neural CDE was also superior to multimodal LSTM. In the attention-based architectures, fusion through concatenation and rectilinear interpolation were found to improve model performance. The performance of the proposed RealDiffFusionNet was found to be superior (0.2570) to all models. For the ADNI dataset, between the Neural-CDE and LSTM models trained only on the structured data, the test RMSE were comparable (0.471 for LSTM vs. 0.4581 Neural-CDE). Furthermore, the addition of image features from patients' MRI series resulted in an improvement in performance, with a lower test RMSE (0.4372 with multimodal vs 0.4581 with structured data). RealDiffFusionNet has shown promise in utilizing CDEs and multimodal data to accurately predict disease progression.

疾病进展建模神经微分方程多模态融合时间序列

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