提出GDKVM模型,提升超声心动图视频分割精度与实时性
GDKVM: Echocardiography Video Segmentation via Spatiotemporal Key-Value Memory with Gated Delta Rule
- 采用时空键值记忆与门控增量规则,高效建模帧间关联
- 在CAMUS和EchoNet-Dynamic数据集上达到最优Dice分数,优于现有方法
- 适合临床心脏功能量化分析,尤其适用于高噪声或形变场景
准确分割超声心动图序列中的心腔对心脏功能的定量分析至关重要,有助于临床诊断与治疗。成像噪声、伪影以及心脏的形变与运动给分割算法带来挑战。尽管基于卷积神经网络、Transformer和时空记忆网络的方法提升了分割精度,但往往难以平衡长时序时空依赖建模与计算效率及细粒度特征表示之间的关系。本文提出GDKVM,一种新型超声心动图视频分割架构。该模型采用线性键值关联(LKVA)有效建模帧间相关性,并引入门控增量规则(GDR)高效存储中间记忆状态。关键像素特征融合(KPFF)模块在多尺度下整合局部与全局特征,增强对边界模糊和噪声干扰的鲁棒性。我们在两个主流超声心动图视频数据集(CAMUS和EchoNet-Dynamic)上验证了GDKVM,并与多种先进方法进行对比。实验结果表明,GDKVM在分割精度与鲁棒性方面均优于现有方法,同时保证实时性能。代码已开源:https://github.com/wangrui2025/GDKVM。
原文摘要 · Abstract (English)
Accurate segmentation of cardiac chambers in echocardiography sequences is crucial for the quantitative analysis of cardiac function, aiding in clinical diagnosis and treatment. The imaging noise, artifacts, and the deformation and motion of the heart pose challenges to segmentation algorithms. While existing methods based on convolutional neural networks, Transformers, and space-time memory networks have improved segmentation accuracy, they often struggle with the trade-off between capturing long-range spatiotemporal dependencies and maintaining computational efficiency with fine-grained feature representation. In this paper, we introduce GDKVM, a novel architecture for echocardiography video segmentation. The model employs Linear Key-Value Association (LKVA) to effectively model inter-frame correlations, and introduces Gated Delta Rule (GDR) to efficiently store intermediate memory states. Key-Pixel Feature Fusion (KPFF) module is designed to integrate local and global features at multiple scales, enhancing robustness against boundary blurring and noise interference. We validated GDKVM on two mainstream echocardiography video datasets (CAMUS and EchoNet-Dynamic) and compared it with various state-of-the-art methods. Experimental results show that GDKVM outperforms existing approaches in terms of segmentation accuracy and robustness, while ensuring real-time performance. Code is available at https://github.com/wangrui2025/GDKVM.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。