论文标题
基于生物标志物激活图的可解释的糖尿病性视网膜病变诊断
Interpretable Diabetic Retinopathy Diagnosis based on Biomarker Activation Map
论文作者
论文摘要
深度学习分类器提供了基于光学相干断层扫描(OCT)及其血管造影(OCTA)自动诊断糖尿病性视网膜病(DR)的最准确手段。这些模型的功能部分归因于包含隐藏的图层,这些隐藏层提供了实现所需任务所需的复杂性。但是,隐藏层也使算法输出难以解释。在这里,我们介绍了基于生成对抗性学习的新型生物标志物激活图(BAM)框架,该框架使临床医生可以验证和理解分类器的决策。基于当前的临床标准,将包括456次黄斑扫描的数据集分为不可提及的DR。用于评估我们的BAM的DR分类器首先是根据此数据集对培训的。 BAM生成框架是通过梳理两个U形发电机来为该分类器提供有意义的解释性设计设计的。对主发电机进行了训练,以将引用的扫描作为输入,并产生一个将分类器分类为不可介绍的输出。然后将BAM构造为主发电机的输出和输入之间的差异图像。为了确保BAM仅突出显示分类器利用生物标志物的助理发电机进行了相反的训练,从而产生了扫描,该扫描将被分类器分类为分类器从不可回顾的扫描中归类为引用。生成的BAMS突出了已知的病理特征,包括非灌注区域和视网膜液。基于这些亮点的完全可解释的分类器可以帮助临床医生更好地利用和验证自动化的DR诊断。
Deep learning classifiers provide the most accurate means of automatically diagnosing diabetic retinopathy (DR) based on optical coherence tomography (OCT) and its angiography (OCTA). The power of these models is attributable in part to the inclusion of hidden layers that provide the complexity required to achieve a desired task. However, hidden layers also render algorithm outputs difficult to interpret. Here we introduce a novel biomarker activation map (BAM) framework based on generative adversarial learning that allows clinicians to verify and understand classifiers decision-making. A data set including 456 macular scans were graded as non-referable or referable DR based on current clinical standards. A DR classifier that was used to evaluate our BAM was first trained based on this data set. The BAM generation framework was designed by combing two U-shaped generators to provide meaningful interpretability to this classifier. The main generator was trained to take referable scans as input and produce an output that would be classified by the classifier as non-referable. The BAM is then constructed as the difference image between the output and input of the main generator. To ensure that the BAM only highlights classifier-utilized biomarkers an assistant generator was trained to do the opposite, producing scans that would be classified as referable by the classifier from non-referable scans. The generated BAMs highlighted known pathologic features including nonperfusion area and retinal fluid. A fully interpretable classifier based on these highlights could help clinicians better utilize and verify automated DR diagnosis.