arXiv:2411.05597cs.CVcs.LG2024-11中稿 · ML-CDS workshop, M…被引 5

用视网膜血管图+多模态自监督学习,提升中风预测准确率

Predicting Stroke through Retinal Graphs and Multimodal Self-supervised Learning

  • 构建视网膜血管图与临床数据融合的对比学习框架
  • 自监督学习使AUROC提升3.78%,且训练时间显著减少
  • 适合医学影像分析与心血管疾病早期筛查研究者

早期识别中风对干预至关重要,需依赖可靠模型。本文提出一种高效视网膜图像表示方法,结合临床信息以全面评估心血管健康,利用大规模多模态数据获得新医疗洞见。该方法是首个整合图结构与表格数据的对比学习框架,采用从视网膜图像生成的血管图进行高效表征。结合多模态对比学习,通过融合多源数据并利用对比学习实现迁移学习,显著提升中风预测准确率。所用自监督学习技术可有效利用无标签数据,降低对大规模标注数据的依赖。框架在自监督到监督学习中实现3.78%的AUROC提升。基于图级别的表示方法优于图像编码器,同时大幅减少预训练和微调时间。结果表明,视网膜图像是一种低成本提升心血管疾病预测能力的有效手段,为未来视网膜与脑血管关联研究及基于图的视网膜血管表征应用开辟路径。

原文摘要 · Abstract (English)

Early identification of stroke is crucial for intervention, requiring reliable models. We proposed an efficient retinal image representation together with clinical information to capture a comprehensive overview of cardiovascular health, leveraging large multimodal datasets for new medical insights. Our approach is one of the first contrastive frameworks that integrates graph and tabular data, using vessel graphs derived from retinal images for efficient representation. This method, combined with multimodal contrastive learning, significantly enhances stroke prediction accuracy by integrating data from multiple sources and using contrastive learning for transfer learning. The self-supervised learning techniques employed allow the model to learn effectively from unlabeled data, reducing the dependency on large annotated datasets. Our framework showed an AUROC improvement of 3.78% from supervised to self-supervised approaches. Additionally, the graph-level representation approach achieved superior performance to image encoders while significantly reducing pre-training and fine-tuning runtimes. These findings indicate that retinal images are a cost-effective method for improving cardiovascular disease predictions and pave the way for future research into retinal and cerebral vessel connections and the use of graph-based retinal vessel representations.

中风预测视网膜图像自监督学习图神经网络

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