arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1908.08242cs.CV
收藏

Uncertainty-Guided Domain Alignment for Layer Segmentation in OCT Images

School of Information Science and Engineering, Xiamen University(厦门大学信息科学与技术学院);Tencent YouTu Lab(腾讯优图实验室);State Key Laboratory of Ophthalmology, Zhongshan Ophthalmic Center, Sun Yat-sen University(中山大学中山眼科中心眼科学国家重点实验室);Dept. of Computer Science and Engineering, The Chinese University of Hong Kong(香港中文大学计算机科学与工程学系)

Jiexiang Wang, Cheng Bian, Meng Li, Xin Yang, Kai Ma, Wenao Ma, Jin Yuan, Xinghao Ding, Yefeng Zheng

英文摘要:

Automatic and accurate segmentation for retinal and choroidal layers of Optical Coherence Tomography (OCT) is crucial for detection of various ocular diseases. However, because of the variations in different equipments, OCT data obtained from different manufacturers might encounter appearance discrepancy, which could lead to performance fluctuation to a deep neural network. In this paper, we propose an uncertainty-guided domain alignment method to aim at alleviating this problem to transfer discriminative knowledge across distinct domains. We disign a novel uncertainty-guided cross-entropy loss for boosting the performance over areas with high uncertainty. An uncertainty-guided curriculum transfer strategy is developed for the self-training (ST), which regards uncertainty as efficient and effective guidance to optimize the learning process in target domain. Adversarial learning with feature recalibration module (FRM) is applied to transfer informative knowledge from the domain feature spaces adaptively. The experiments on two OCT datasets show that the proposed methods can obtain significant segmentation improvements compared with the baseline models.

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

↑