arXiv:2410.02808eess.IVcs.AI2024-10中稿 · BIBM 2024被引 3

用卡尔曼滤波优化可变形卷积,提升视网膜血管分割精度

KLDD: Kalman Filter based Linear Deformable Diffusion Model in Retinal Image Segmentation

  • 基于卡尔曼滤波改进线性可变形卷积,动态调整感受野适应血管结构
  • 在DRIVE、CHASE_DB1等数据集上,Dice系数最高达0.923,优于传统U-Net
  • 适合需要高精度微血管分割的医学影像分析场景

基于AI的血管分割在眼科疾病筛查与治疗中日益重要。以U-Net为基础的深度学习模型虽已取得良好效果,但在传统下采样模块中常丢失小血管和毛细血管。为此,本文提出一种基于卡尔曼滤波的线性可变形扩散(KLDD)模型用于视网膜血管分割。模型通过扩散过程迭代优化分割结果,利用可变形卷积在特征提取模块中灵活调整感受野,以适应细长血管结构。具体地,先使用线性可变形卷积提取血管信息;为更优优化卷积坐标位置,引入卡尔曼滤波增强对血管结构的感知能力。随后,将提取的血管特征作为条件输入扩散模型,通过交叉注意力聚合模块(CAAM)与通道软注意力模块(CSAM)强化生成能力。在DRIVE、CHASE_DB1以及OCTA-500数据集(3mm、6mm子集)上的实验表明,该模型性能优于现有方法,最大Dice系数达0.923。

原文摘要 · Abstract (English)

AI-based vascular segmentation is becoming increasingly common in enhancing the screening and treatment of ophthalmic diseases. Deep learning structures based on U-Net have achieved relatively good performance in vascular segmentation. However, small blood vessels and capillaries tend to be lost during segmentation when passed through the traditional U-Net downsampling module. To address this gap, this paper proposes a novel Kalman filter based Linear Deformable Diffusion (KLDD) model for retinal vessel segmentation. Our model employs a diffusion process that iteratively refines the segmentation, leveraging the flexible receptive fields of deformable convolutions in feature extraction modules to adapt to the detailed tubular vascular structures. More specifically, we first employ a feature extractor with linear deformable convolution to capture vascular structure information form the input images. To better optimize the coordinate positions of deformable convolution, we employ the Kalman filter to enhance the perception of vascular structures in linear deformable convolution. Subsequently, the features of the vascular structures extracted are utilized as a conditioning element within a diffusion model by the Cross-Attention Aggregation module (CAAM) and the Channel-wise Soft Attention module (CSAM). These aggregations are designed to enhance the diffusion model's capability to generate vascular structures. Experiments are evaluated on retinal fundus image datasets (DRIVE, CHASE_DB1) as well as the 3mm and 6mm of the OCTA-500 dataset, and the results show that the diffusion model proposed in this paper outperforms other methods.

视网膜分割扩散模型可变形卷积卡尔曼滤波

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