用解剖先验和注意力机制提升眼底OCT图像水肿分割精度
Prior-AttUNet: Retinal OCT Fluid Segmentation Based on Normal Anatomical Priors and Attention Gating
- 结合生成先验与注意力门控,动态优化特征提取
- 在三类设备上平均Dice达93.47%~95.18%,边界识别更准
- 适合眼科影像自动化分析,兼顾精度与推理速度
黄斑水肿是年龄相关性黄斑变性和糖尿病性黄斑水肿等致盲性疾病的典型病理特征,准确分割对临床诊断至关重要。针对光学相干断层扫描(OCT)图像中液体区域边界模糊、跨设备异质性强的问题,本文提出Prior-AttUNet模型,融合生成式解剖先验与分割网络。该框架采用双路径结构:变分自编码器提供多尺度正常解剖先验,主干网络结合密集连接块与空间金字塔池化模块以捕获丰富上下文信息;同时引入一种由解剖先验引导的三重注意力机制,在解码阶段动态调节特征重要性,显著提升边界分割能力。在公开的RETOUCH基准数据集上评估,模型在Cirrus、Spectralis和Topcon三类设备上的平均Dice相似系数分别为93.93%、95.18%和93.47%。模型计算量仅为0.37 TFLOPs,兼顾高精度与高效推理,展现出作为自动化临床分析工具的潜力。
原文摘要 · Abstract (English)
Accurate segmentation of macular edema, a hallmark pathological feature in vision-threatening conditions such as age-related macular degeneration and diabetic macular edema, is essential for clinical diagnosis and management. To overcome the challenges of segmenting fluid regions in optical coherence tomography (OCT) images-notably ambiguous boundaries and cross-device heterogeneity-this study introduces Prior-AttUNet, a segmentation model augmented with generative anatomical priors. The framework adopts a hybrid dual-path architecture that integrates a generative prior pathway with a segmentation network. A variational autoencoder supplies multi-scale normative anatomical priors, while the segmentation backbone incorporates densely connected blocks and spatial pyramid pooling modules to capture richer contextual information. Additionally, a novel triple-attention mechanism, guided by anatomical priors, dynamically modulates feature importance across decoding stages, substantially enhancing boundary delineation. Evaluated on the public RETOUCH benchmark, Prior-AttUNet achieves excellent performance across three OCT imaging devices (Cirrus, Spectralis, and Topcon), with mean Dice similarity coefficients of 93.93%, 95.18%, and 93.47%, respectively. The model maintains a low computational cost of 0.37 TFLOPs, striking an effective balance between segmentation precision and inference efficiency. These results demonstrate its potential as a reliable tool for automated clinical analysis.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。