用视网膜层结构引导分割,提升弱监督水肿区域识别准确率
A Closer Look at Edema Area Segmentation in SD-OCT Images Using Adversarial Framework
- 基于视网膜层结构与对抗框架结合,改进水肿区域分割
- 在两个公开数据集上显著提升分割精度与鲁棒性
- 适合医学图像分析、弱监督学习研究者参考
针对黄斑水肿(ME)分析中依赖昂贵像素级标注数据的问题,本文提出一种基于弱监督的水肿区域(EA)分割方法。利用光谱域光学相干断层扫描(SD-OCT)图像中水肿与视网膜层之间的强相关性,引入新型层结构引导后处理步骤,并设计测试时自适应(TTA)策略。该方法将密集预测重构为验证水肿边界与视网膜层交点的任务,使结果更符合水肿形状先验。在两个公开数据集上的实验表明,所提方法显著提升了弱监督模型的准确性与泛化能力,缩小了其与全监督模型的性能差距。
原文摘要 · Abstract (English)
The development of artificial intelligence models for macular edema (ME) analy-sis always relies on expert-annotated pixel-level image datasets which are expen-sive to collect prospectively. While anomaly-detection-based weakly-supervised methods have shown promise in edema area (EA) segmentation task, their per-formance still lags behind fully-supervised approaches. In this paper, we leverage the strong correlation between EA and retinal layers in spectral-domain optical coherence tomography (SD-OCT) images, along with the update characteristics of weakly-supervised learning, to enhance an off-the-shelf adversarial framework for EA segmentation with a novel layer-structure-guided post-processing step and a test-time-adaptation (TTA) strategy. By incorporating additional retinal lay-er information, our framework reframes the dense EA prediction task as one of confirming intersection points between the EA contour and retinal layers, result-ing in predictions that better align with the shape prior of EA. Besides, the TTA framework further helps address discrepancies in the manifestations and presen-tations of EA between training and test sets. Extensive experiments on two pub-licly available datasets demonstrate that these two proposed ingredients can im-prove the accuracy and robustness of EA segmentation, bridging the gap between weakly-supervised and fully-supervised models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。