arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2107.00296eess.IVcs.CVcs.LG

Explainable Diabetic Retinopathy Detection and Retinal Image Generation

Beihang University(北京航空航天大学);Peng Cheng Laboratory(鹏城实验室);RIKEN AIP(理化学研究所人工智能研究中心);University of Tokyo(东京大学);Ningbo Institute of Industrial Technology, CAS(中国科学院宁波工业技术研究院)

Yuhao Niu, Lin Gu, Yitian Zhao, Feng Lu

英文摘要:

Though deep learning has shown successful performance in classifying the label and severity stage of certain diseases, most of them give few explanations on how to make predictions. Inspired by Koch's Postulates, the foundation in evidence-based medicine (EBM) to identify the pathogen, we propose to exploit the interpretability of deep learning application in medical diagnosis. By determining and isolating the neuron activation patterns on which diabetic retinopathy (DR) detector relies to make decisions, we demonstrate the direct relation between the isolated neuron activation and lesions for a pathological explanation. To be specific, we first define novel pathological descriptors using activated neurons of the DR detector to encode both spatial and appearance information of lesions. Then, to visualize the symptom encoded in the descriptor, we propose Patho-GAN, a new network to synthesize medically plausible retinal images. By manipulating these descriptors, we could even arbitrarily control the position, quantity, and categories of generated lesions. We also show that our synthesized images carry the symptoms directly related to diabetic retinopathy diagnosis. Our generated images are both qualitatively and quantitatively superior to the ones by previous methods. Besides, compared to existing methods that take hours to generate an image, our second level speed endows the potential to be an effective solution for data augmentation.

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

↑