论文标题
Skinet:用于皮肤病变诊断的深度学习解决方案,具有不确定性估计和解释性
SkiNet: A Deep Learning Solution for Skin Lesion Diagnosis with Uncertainty Estimation and Explainability
论文作者
论文摘要
皮肤癌被认为是最常见的人类恶性肿瘤。每年在美国记录大约500万例皮肤癌病例。对皮肤病变的早期鉴定和评估具有很大的临床意义,但是不成比例的皮肤科医生患者比率在大多数发展中国家中构成了重大问题。因此,在临床诊断过程中,提出了一种基于深度学习的建筑,即Skinet,其目标是为新训练的医生提供更快的筛查解决方案和帮助。 Skinet的设计和开发背后的主要动机是提供白盒解决方案,解决一个关键的信任和可解释性问题,这对于医学医生更广泛地采用计算机辅助诊断系统至关重要。 Skinet是两阶段的管道,其中病变分割后面是病变分类。在我们的Skinet方法论中,已经采用了蒙特卡洛辍学和测试时间扩展技术来估计认知和态度不确定性,而探索了基于显着的方法来提供有关深度学习模型的事后解释。公开可用的数据集ISIC-2018用于执行实验和消融研究。结果在传统基准上建立了模型的鲁棒性,同时解决了此类模型的黑盒性质,以通过将透明度和信心纳入模型的预测来减轻医学从业者的怀疑。
Skin cancer is considered to be the most common human malignancy. Around 5 million new cases of skin cancer are recorded in the United States annually. Early identification and evaluation of skin lesions is of great clinical significance, but the disproportionate dermatologist-patient ratio poses significant problem in most developing nations. Therefore a deep learning based architecture, known as SkiNet, is proposed with an objective to provide faster screening solution and assistance to newly trained physicians in the clinical diagnosis process. The main motive behind Skinet's design and development is to provide a white box solution, addressing a critical problem of trust and interpretability which is crucial for the wider adoption of Computer-aided diagnosis systems by the medical practitioners. SkiNet is a two-stage pipeline wherein the lesion segmentation is followed by the lesion classification. In our SkiNet methodology, Monte Carlo dropout and test time augmentation techniques have been employed to estimate epistemic and aleatoric uncertainty, while saliency-based methods are explored to provide post-hoc explanations of the deep learning models. The publicly available dataset, ISIC-2018, is used to perform experimentation and ablation studies. The results establish the robustness of the model on the traditional benchmarks while addressing the black-box nature of such models to alleviate the skepticism of medical practitioners by incorporating transparency and confidence to the model's prediction.