用分解与自编码技术,从少量超声心动图中提前预测心衰时间。
Heart Failure Prediction using Modal Decomposition and Masked Autoencoders for Scarce Echocardiography Databases
- 结合高阶动态模态分解与掩码自编码,增强数据并提取特征
- 在稀缺数据下训练视觉变换器,预测准确率优于主流模型
- 适合医疗影像少样本场景,尤其适用于心衰早期预警
心血管疾病仍是全球死亡主因,每年约导致1800万例死亡。心力衰竭(HF)对医疗系统提出早期、快速、有效预测的迫切需求。本文提出一种自动系统,融合高阶动态模态分解(HODMD)与掩码自编码器(MAE),将任务从心脏病分类拓展至尚未被充分研究的心衰发生时间预测。该系统分两阶段:第一阶段利用HODMD对超声心动图视频序列进行数据增强与特征提取,生成大量可用于深度学习训练的标注图像;第二阶段构建并训练视觉变换器(ViT),采用自监督与有监督学习相结合的MAE策略,实现从零开始的有效训练,即使在数据稀缺情况下亦表现优异。所设计模型可实时分析超声心动图图像,预测心衰发生时间。实验表明,该方法在稀缺数据库上显著提升预测精度,优于多个成熟的ViT与卷积神经网络(CNN)架构。源代码将集成至ModelFLOWs-app软件下一版本(https://github.com/modelflows/ModelFLOWs-app)。
原文摘要 · Abstract (English)
Heart diseases remain the leading cause of mortality worldwide, implying approximately 18 million deaths according to the WHO. In particular, heart failures (HF) press the healthcare industry to develop systems for their early, rapid, and effective prediction. This work presents an automatic system based on a novel framework which combines Modal Decomposition and Masked Autoencoders (MAE) to extend the application from heart disease classification to the more challenging and specific task of heart failure time prediction, not previously addressed to the best of authors' knowledge. This system comprises two stages. The first one transforms the data from a database of echocardiography video sequences into a large collection of annotated images compatible with the training phase of machine learning-based frameworks and deep learning-based ones. This stage includes the use of the Higher Order Dynamic Mode Decomposition (HODMD) algorithm for both data augmentation and feature extraction. The second stage builds and trains a Vision Transformer (ViT). MAEs based on a combined scheme of self-supervised (SSL) and supervised learning, so far barely explored in the literature about heart failure prediction, are adopted to effectively train the ViT from scratch, even with scarce databases. The designed neural network analyses in real-time images from echocardiography sequences to estimate the time of happening a heart failure. This approach demonstrates to improve prediction accuracy from scarce databases and to be superior to several established ViT and Convolutional Neural Network (CNN) architectures. The source code will be incorporated into the next version release of the ModelFLOWs-app software (https://github.com/modelflows/ModelFLOWs-app).
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。