论文标题
使用自我监督的聚合学习对X射线的异常检测
Anomaly Detection on X-Rays Using Self-Supervised Aggregation Learning
论文作者
论文摘要
使用有监督的学习模式的深度异常检测模型通常在封闭的假设下起作用,并且在训练中遭受过度拟合到以前看到的罕见异常,这阻碍了它们在实际情况下的适用性。此外,获得X射线注释非常耗时,需要对放射科医生进行广泛的培训。因此,在完全无监督或自我监督的方式中训练异常检测将是有利的,从而使放射科医生在报告上花费的时间大幅度减少了。在本文中,我们提出了Salad,这是一种端到端的深度自我监督方法,用于X射线图像的异常检测。所提出的方法基于一种优化策略,在该策略中,鼓励深层神经网络代表嵌入空间中正常数据的原型局部模式。在训练过程中,我们通过记忆库记录了普通训练样本的原型模式。然后,我们的异常得分是通过测量与在记忆库中正常原型模式的加权组合的相似性而得出的,而无需使用任何异常模式。我们对具有挑战性的NIH胸部X射线和MURA数据集进行了广泛的实验,这表明我们的算法将最新方法提高了大幅度。
Deep anomaly detection models using a supervised mode of learning usually work under a closed set assumption and suffer from overfitting to previously seen rare anomalies at training, which hinders their applicability in a real scenario. In addition, obtaining annotations for X-rays is very time consuming and requires extensive training of radiologists. Hence, training anomaly detection in a fully unsupervised or self-supervised fashion would be advantageous, allowing a significant reduction of time spent on the report by radiologists. In this paper, we present SALAD, an end-to-end deep self-supervised methodology for anomaly detection on X-Ray images. The proposed method is based on an optimization strategy in which a deep neural network is encouraged to represent prototypical local patterns of the normal data in the embedding space. During training, we record the prototypical patterns of normal training samples via a memory bank. Our anomaly score is then derived by measuring similarity to a weighted combination of normal prototypical patterns within a memory bank without using any anomalous patterns. We present extensive experiments on the challenging NIH Chest X-rays and MURA dataset, which indicate that our algorithm improves state-of-the-art methods by a wide margin.