arXiv:2410.21160eess.IVcs.CV2024-10

用卡尔曼滤波改进注意力机制,精准分割视网膜微小血管。

KaLDeX: Kalman Filter based Linear Deformable Cross Attention for Retina Vessel Segmentation

  • 引入卡尔曼滤波的可变形卷积与交叉注意力融合,增强对细小血管的捕捉能力。
  • 在5个数据集上平均准确率达97.25%至98.89%,显著优于现有方法。
  • 适合眼科医学图像分析、血管精细化分割任务的研究者与开发者。

在眼科学影像领域,精准的血管分割对多种眼病的诊断与管理至关重要。当前基于深度学习的血管分割模型已接近人工水平,但在识别微小血管方面仍存在挑战。由于CNN模型需多次下采样,高分辨率图像中的精细细节不可避免丢失。本研究旨在设计一种新结构以精确捕捉细微血管。提出KaLDeX网络,结合基于卡尔曼滤波的线性可变形交叉注意力(LDCA)模块,嵌入UNet++框架。核心由两部分构成:基于卡尔曼滤波的线性可变形卷积(LD)与交叉注意力(CA)。LD模块自适应聚焦于易被忽略的细血管;CA模块通过融合LD提取的细节特征与UNet++的高层语义特征,提升对血管整体结构的理解。最后采用基于持久同调的拓扑损失函数,约束分割结果的拓扑连续性。在DRIVE、CHASE_BD1、STARE及OCTA-500的3mm和6mm数据集上评估,平均准确率分别为97.25%、97.77%、97.85%、98.89%和98.21%。实验证明该方法在多个血管分割数据集上超越当前最优模型。源代码见:https://github.com/AIEyeSystem/KalDeX。

原文摘要 · Abstract (English)

Background and Objective: In the realm of ophthalmic imaging, accurate vascular segmentation is paramount for diagnosing and managing various eye diseases. Contemporary deep learning-based vascular segmentation models rival human accuracy but still face substantial challenges in accurately segmenting minuscule blood vessels in neural network applications. Due to the necessity of multiple downsampling operations in the CNN models, fine details from high-resolution images are inevitably lost. The objective of this study is to design a structure to capture the delicate and small blood vessels. Methods: To address these issues, we propose a novel network (KaLDeX) for vascular segmentation leveraging a Kalman filter based linear deformable cross attention (LDCA) module, integrated within a UNet++ framework. Our approach is based on two key components: Kalman filter (KF) based linear deformable convolution (LD) and cross-attention (CA) modules. The LD module is designed to adaptively adjust the focus on thin vessels that might be overlooked in standard convolution. The CA module improves the global understanding of vascular structures by aggregating the detailed features from the LD module with the high level features from the UNet++ architecture. Finally, we adopt a topological loss function based on persistent homology to constrain the topological continuity of the segmentation. Results: The proposed method is evaluated on retinal fundus image datasets (DRIVE, CHASE_BD1, and STARE) as well as the 3mm and 6mm of the OCTA-500 dataset, achieving an average accuracy (ACC) of 97.25%, 97.77%, 97.85%, 98.89%, and 98.21%, respectively. Conclusions: Empirical evidence shows that our method outperforms the current best models on different vessel segmentation datasets. Our source code is available at: https://github.com/AIEyeSystem/KalDeX.

视网膜分割卡尔曼滤波注意力机制医学图像

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