arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2407.04068cs.CV
收藏

CLIP-DR:基于排序感知提示的文本知识引导糖尿病视网膜病变分级

CLIP-DR: Textual Knowledge-Guided Diabetic Retinopathy Grading with Ranking-aware Prompting

University of Exeter(埃克塞特大学);University of Liverpool(利物浦大学);Ningbo Institute of Materials Technology and Engineering, Chinese Academy of Sciences(中国科学院宁波材料技术与工程研究所);Institute of High Performance Computing, Agency for Science, Technology and Research (A*STAR)(新加坡科技研究局高性能计算研究所)

Qinkai Yu, Jianyang Xie, Anh Nguyen, He Zhao, Jiong Zhang, Huazhu Fu, Yitian Zhao, Yalin Zheng, Yanda Meng

AI总结:

本文提出CLIP-DR框架,通过排序感知提示和相似度矩阵平滑模块,利用CLIP的文本知识及序数信息,解决糖尿病视网膜病变分级中的数据变异性和类别长尾问题,并在GDRBench上取得优越性能。

摘要:

糖尿病视网膜病变(DR)是糖尿病的一种并发症,通常需要数十年才会发展到威胁视力的程度。准确且稳健地检测DR严重程度对于糖尿病的及时管理和治疗至关重要。然而,当前大多数DR分级方法对数据变异性(例如彩色眼底图像)的鲁棒性不足,给准确且稳健的分级带来了显著困难。在本工作中,我们基于三个观察提出了一种新颖的DR分级框架CLIP-DR:1)近期预训练的视觉语言模型(如CLIP)在各种下游任务中展现出显著的泛化能力,可作为有效的基线模型。2)DR的图像-文本对分级通常遵循可辨识的自然顺序,但大多数现有DR分级方法主要忽略了这一方面。3)DR严重程度等级之间的长尾分布使分级过程复杂化。本工作提出了一种新颖的排序感知提示策略,以帮助CLIP模型利用序数信息。具体而言,我们在两个不同的排序方向上,依次为相邻的文本-图像对设计可学习提示。此外,我们在CLIP结构中引入了相似度矩阵平滑模块,以平衡类别分布。最后,我们在GDRBench基准上与多种最先进方法进行了广泛比较,证明了我们的CLIP-DR具有鲁棒性和优越性能。实现代码可在\url{https://github.com/Qinkaiyu/CLIP-DR}获取。

英文摘要:

Diabetic retinopathy (DR) is a complication of diabetes and usually takes decades to reach sight-threatening levels. Accurate and robust detection of DR severity is critical for the timely management and treatment of diabetes. However, most current DR grading methods suffer from insufficient robustness to data variability (\textit{e.g.} colour fundus images), posing a significant difficulty for accurate and robust grading. In this work, we propose a novel DR grading framework CLIP-DR based on three observations: 1) Recent pre-trained visual language models, such as CLIP, showcase a notable capacity for generalisation across various downstream tasks, serving as effective baseline models. 2) The grading of image-text pairs for DR often adheres to a discernible natural sequence, yet most existing DR grading methods have primarily overlooked this aspect. 3) A long-tailed distribution among DR severity levels complicates the grading process. This work proposes a novel ranking-aware prompting strategy to help the CLIP model exploit the ordinal information. Specifically, we sequentially design learnable prompts between neighbouring text-image pairs in two different ranking directions. Additionally, we introduce a Similarity Matrix Smooth module into the structure of CLIP to balance the class distribution. Finally, we perform extensive comparisons with several state-of-the-art methods on the GDRBench benchmark, demonstrating our CLIP-DR's robustness and superior performance. The implementation code is available \footnote{\url{https://github.com/Qinkaiyu/CLIP-DR}

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑