论文标题
通过弱监督分段来解释疾病分类
Explainable Disease Classification via weakly-supervised segmentation
论文作者
论文摘要
基于深度学习的计算机辅助诊断方法(CAD)通常将问题作为图像分类(正常或异常)问题。这些系统在受过训练的特定疾病检测中获得了高度至非常高的精度,但缺乏对提供的决策/分类结果的解释。与决策相对应的激活图与特定疾病的兴趣区域没有很好的相关性。本文研究了这个问题,并提出了一种模仿诊断前寻找证据的临床实践的方法。使用一组混合信息来学习CAD模型:整个培训图像集的类标签,以及可疑区域的粗略本地化,作为较小的培训图像子集以指导学习的额外输入。通过检测到OCT切片的糖尿病黄斑水肿(DME)来说明所提出的方法。在大型公共数据集上进行测试的结果表明,只有三分之一的图像大致分割了流体填充区域,分类准确性与最先进的方法相当,同时以解剖学上准确的热图 /感兴趣区域的形式提供了很好的解释。然后将所提出的溶液适应乳腺X线摄影图像的乳腺癌检测。公共数据集的良好评估结果强调了提出解决方案的普遍性。
Deep learning based approaches to Computer Aided Diagnosis (CAD) typically pose the problem as an image classification (Normal or Abnormal) problem. These systems achieve high to very high accuracy in specific disease detection for which they are trained but lack in terms of an explanation for the provided decision/classification result. The activation maps which correspond to decisions do not correlate well with regions of interest for specific diseases. This paper examines this problem and proposes an approach which mimics the clinical practice of looking for an evidence prior to diagnosis. A CAD model is learnt using a mixed set of information: class labels for the entire training set of images plus a rough localisation of suspect regions as an extra input for a smaller subset of training images for guiding the learning. The proposed approach is illustrated with detection of diabetic macular edema (DME) from OCT slices. Results of testing on on a large public dataset show that with just a third of images with roughly segmented fluid filled regions, the classification accuracy is on par with state of the art methods while providing a good explanation in the form of anatomically accurate heatmap /region of interest. The proposed solution is then adapted to Breast Cancer detection from mammographic images. Good evaluation results on public datasets underscores the generalisability of the proposed solution.