Disease grading and lesion identification are two important tasks for diabetic retinopathy detection. Disease grading uses image-level annotation but lesion identification often needs the fine-grained annotations, which requires a lot of time and effort of professional doctors. Therefore, it is a great challenge to complete disease grading and lesion identification simultaneously with the limited labeled data. We propose a method based on weakly supervised object localization and knowledge driven attribute mining to conduct disease grading and lesion identification using only image-level annotation. We first propose an Attention-Drop-Highlight Layer (ADHL), which enables the CNN to accurately and comprehensively focus on the various lesion features. Then, we design a search space and employ neural architecture search (NAS) to select the best settings of the ADHL, to maximize the performance of the model. Finally, we regard the lesion attributes corresponding to different disease grades as weakly supervised classification labels representing prior knowledge, and propose an Attribute Mining (AM) method to further improve the effect of disease grading and complete lesion identification. Extensive experiments and a user study have proved that our method can capture more lesion features, improve the performance of disease grading, and obtain state-of-the-art results compared to the methods only using imagelevel annotation.
基金:
National Key R&D Program of China [2018YFC0115102]; National Natural Science Foundation of China [61872020, U20A20195]; Beijing Natural Science Foundation Haidian Primitive Innovation Joint Fund [L182016]; Beijing Advanced Innovation Center for Biomedical Engineering [ZF138G1714]; Research Unit of Virtual Human and Virtual Surgery, Chinese Academy of Medical Sciences [2019RU004]; Shenzhen Research Institute of Big Data, Shenzhen [518000]; Global Visiting Fellowship of Bournemouth University
语种:
外文
被引次数:
WOS:
第一作者:
第一作者机构:[1]Beihang Univ, State Key Lab Virtual Real Technol & Syst, Beijing 100191, Peoples R China
通讯作者:
推荐引用方式(GB/T 7714):
Wang Xinliang,Gu Yunchao,Pan Junjun,et al.Diabetic Retinopathy Detection Based on Weakly Supervised Object Localization and Knowledge Driven Attribute Mining[J].OPHTHALMIC MEDICAL IMAGE ANALYSIS, OMIA 2021.2021,12970:32-41.doi:10.1007/978-3-030-87000-3_4.
APA:
Wang, Xinliang,Gu, Yunchao,Pan, Junjun&Jia, Liyun.(2021).Diabetic Retinopathy Detection Based on Weakly Supervised Object Localization and Knowledge Driven Attribute Mining.OPHTHALMIC MEDICAL IMAGE ANALYSIS, OMIA 2021,12970,
MLA:
Wang, Xinliang,et al."Diabetic Retinopathy Detection Based on Weakly Supervised Object Localization and Knowledge Driven Attribute Mining".OPHTHALMIC MEDICAL IMAGE ANALYSIS, OMIA 2021 12970.(2021):32-41