无需反向传播的绿色学习框架,高效完成心脏超声图像分割与分类
Interpretable and backpropagation-free Green Learning for efficient multi-task echocardiographic segmentation and classification
- 采用无反向传播的Green Learning框架,结合自编码器与XGBoost实现多任务学习
- 在EchoNet-Dynamic数据集上分类准确率达94.3%,分割Dice系数达0.912
- 参数量减少超十倍,兼顾高精度、可解释性与临床部署效率
超声心动图是心力衰竭管理的核心手段,左心室射血分数(LVEF)是指导治疗的关键指标。然而,人工评估LVEF存在显著观察者间差异,现有深度学习模型通常计算开销大、依赖大量数据,且为“黑箱”,难以获得临床信任。本文提出一种无需反向传播的多任务绿色学习(MTGL)框架,实现左心室(LV)分割与LVEF分类的联合建模。该框架融合无监督的VoxelHop编码器进行层次化时空特征提取,配合多级回归解码器与XG-Boost分类器。在EchoNet-Dynamic数据集上,所提MTGL模型达到94.3%的分类准确率和0.912的分割Dice相似系数,显著优于多个先进3D深度学习模型。关键在于,模型参数量减少超过一个数量级,展现出极强的计算效率。本研究证明,绿色学习范式可为复杂医学图像分析提供高精度、高效、可解释的解决方案,推动人工智能在临床实践中的可持续与可信应用。
原文摘要 · Abstract (English)
Echocardiography is a cornerstone for managing heart failure (HF), with Left Ventricular Ejection Fraction (LVEF) being a critical metric for guiding therapy. However, manual LVEF assessment suffers from high inter-observer variability, while existing Deep Learning (DL) models are often computationally intensive and data-hungry "black boxes" that impede clinical trust and adoption. Here, we propose a backpropagation-free multi-task Green Learning (MTGL) framework that performs simultaneous Left Ventricle (LV) segmentation and LVEF classification. Our framework integrates an unsupervised VoxelHop encoder for hierarchical spatio-temporal feature extraction with a multi-level regression decoder and an XG-Boost classifier. On the EchoNet-Dynamic dataset, our MTGL model achieves state-of-the-art classification and segmentation performance, attaining a classification accuracy of 94.3% and a Dice Similarity Coefficient (DSC) of 0.912, significantly outperforming several advanced 3D DL models. Crucially, our model achieves this with over an order of magnitude fewer parameters, demonstrating exceptional computational efficiency. This work demonstrates that the GL paradigm can deliver highly accurate, efficient, and interpretable solutions for complex medical image analysis, paving the way for more sustainable and trustworthy artificial intelligence in clinical practice.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。