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

利用解耦表示实现糖尿病视网膜病变的未知域泛化

Generalizing to Unseen Domains in Diabetic Retinopathy with Disentangled Representations

Monash University(莫纳什大学);UNC-Chapel Hill(北卡罗来纳大学教堂山分校)

Peng Xia, Ming Hu, Feilong Tang, Wenxue Li, Wenhao Zheng, Lie Ju, Peibo Duan, Huaxiu Yao, Zongyuan Ge

AI总结:

本文提出一种解耦表示框架,将不同域的视网膜图像特征分解为语义特征与域噪声,并通过原型插值、数据感知权重和像素级语义对齐损失提升模型在未知域上的糖尿病视网膜病变分级泛化能力。

摘要:

糖尿病引发的糖尿病视网膜病变(DR)具有导致视力受损的重大风险。准确且有效的DR分级有助于该疾病的治疗。然而,现有模型在未知域上因域偏移而出现显著的性能下降。以往方法通过简单的视觉变换模拟域风格,并通过学习鲁棒表示来缓解域噪声,从而解决这一问题。然而,域偏移不仅包含图像风格,还忽略了由种族、年龄和诊断标准等隐性因素引起的偏差。在我们的工作中,我们提出了一种新颖的框架,将来自不同域的成对数据的表示解耦为语义特征和域噪声。生成的增强表示由原始视网膜语义和来自其他域的域噪声组成,旨在生成符合真实临床需求的增强表示,并融合来自不同域的丰富信息。随后,为了提高解耦表示的鲁棒性,我们采用类别原型和域原型对解耦表示进行插值,同时设计了数据感知权重以关注稀有类别和稀有域。最后,我们设计了一种鲁棒的像素级语义对齐损失,用于对齐从特征中解耦出的视网膜语义,在类内多样性和密集类特征之间保持平衡。在多个基准上的实验结果表明,我们的方法在未知域上具有有效性。代码实现可在 https://github.com/richard-peng-xia/DECO 获取。

英文摘要:

Diabetic Retinopathy (DR), induced by diabetes, poses a significant risk of visual impairment. Accurate and effective grading of DR aids in the treatment of this condition. Yet existing models experience notable performance degradation on unseen domains due to domain shifts. Previous methods address this issue by simulating domain style through simple visual transformation and mitigating domain noise via learning robust representations. However, domain shifts encompass more than image styles. They overlook biases caused by implicit factors such as ethnicity, age, and diagnostic criteria. In our work, we propose a novel framework where representations of paired data from different domains are decoupled into semantic features and domain noise. The resulting augmented representation comprises original retinal semantics and domain noise from other domains, aiming to generate enhanced representations aligned with real-world clinical needs, incorporating rich information from diverse domains. Subsequently, to improve the robustness of the decoupled representations, class and domain prototypes are employed to interpolate the disentangled representations while data-aware weights are designed to focus on rare classes and domains. Finally, we devise a robust pixel-level semantic alignment loss to align retinal semantics decoupled from features, maintaining a balance between intra-class diversity and dense class features. Experimental results on multiple benchmarks demonstrate the effectiveness of our method on unseen domains. The code implementations are accessible on https://github.com/richard-peng-xia/DECO.

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

↑