arXiv:2502.20224eess.IVcs.AI2025-02被引 2

无监督学习+注意力机制,自动诊断糖尿病黄斑水肿

RURANET++: An Unsupervised Learning Method for Diabetic Macular Edema Based on SCSE Attention Mechanisms and Dynamic Multi-Projection Head Clustering

  • 用改进U-Net加SCSE注意力提取病变特征
  • 50维降维后动态多投影头聚类,准确率达84.11%
  • 无需标注数据,适合临床快速筛查

糖尿病黄斑水肿(DME)是糖尿病患者常见并发症,导致视力下降和失明。尽管深度学习在医学图像分析中取得进展,传统DME诊断仍依赖大量标注数据和主观眼科评估,限制了实际应用。为此,本文提出RURANET++,一种基于无监督学习的自动化DME诊断系统。该框架采用优化的U-Net结构,嵌入空间与通道压缩-激励(SCSE)注意力机制以增强病灶特征提取;通过预训练的GoogLeNet模型从视网膜图像中提取深层特征,并进行主成分分析(PCA)降维至50维以提升计算效率。关键创新在于引入一种新型聚类算法,采用多投影头机制显式控制聚类多样性,同时动态调整相似性阈值,从而优化类内一致性与类间区分度。实验结果表明,系统在多项指标上表现优异:最高准确率0.8411、精确率0.8593、召回率0.8411、F1分数0.8390,聚类质量显著。本研究为DME诊断提供了高效的无监督解决方案,具有重要临床意义。

原文摘要 · Abstract (English)

Diabetic Macular Edema (DME), a prevalent complication among diabetic patients, constitutes a major cause of visual impairment and blindness. Although deep learning has achieved remarkable progress in medical image analysis, traditional DME diagnosis still relies on extensive annotated data and subjective ophthalmologist assessments, limiting practical applications. To address this, we present RURANET++, an unsupervised learning-based automated DME diagnostic system. This framework incorporates an optimized U-Net architecture with embedded Spatial and Channel Squeeze & Excitation (SCSE) attention mechanisms to enhance lesion feature extraction. During feature processing, a pre-trained GoogLeNet model extracts deep features from retinal images, followed by PCA-based dimensionality reduction to 50 dimensions for computational efficiency. Notably, we introduce a novel clustering algorithm employing multi-projection heads to explicitly control cluster diversity while dynamically adjusting similarity thresholds, thereby optimizing intra-class consistency and inter-class discrimination. Experimental results demonstrate superior performance across multiple metrics, achieving maximum accuracy (0.8411), precision (0.8593), recall (0.8411), and F1-score (0.8390), with exceptional clustering quality. This work provides an efficient unsupervised solution for DME diagnosis with significant clinical implications.

无监督学习医学图像糖尿病聚类

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