arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1903.03313cs.CV
收藏

一种用于皮肤病变自动分割和分类的互自举模型

A Mutual Bootstrapping Model for Automated Skin Lesion Segmentation and Classification

Northwestern Polytechnical University(西北工业大学);University of Adelaide(阿德莱德大学)

Yutong Xie, Jianpeng Zhang, Yong Xia, Chunhua Shen

摘要:

皮肤病变的自动分割和分类是皮肤癌计算机辅助诊断中两个最基本、最相关的任务。尽管深度学习模型很流行,但它们通常只针对一项任务设计,忽略了联合执行两项任务的潜在好处。在本文中,我们提出了相互引导的深度卷积神经网络(MB-DCNN)模型,同时皮肤病变分割和分类。该模型由粗分割网络(粗SN),掩模引导分类网络(mask-CN)和增强分割网络(增强SN)组成。一方面,粗SN生成粗病变掩模,其为掩模CN提供先验自举以帮助其准确地定位和分类皮肤病变。另一方面,由mask-CN产生的病变定位图然后被馈送到增强的SN,旨在将mask-CN学习的定位信息传递到增强的SN以进行准确的病变分割。以这种方式,分割和分类网络彼此之间相互传递知识,并以自举的方式相互促进。同时,我们还设计了一种新的秩损失,并将其与分割网络中的Dice损失联合使用,以解决类别不平衡和难易像素不平衡引起的问题。我们在ISIC-2017和PH 2数据集上评估了所提出的MB-DCNN模型,并在皮肤病变分割中实现了80.4%和89.4%的Jaccard指数,在皮肤病变分类中实现了93.8%和97.7%的平均AUC,这优于代表性的最先进的皮肤病变分割和分类方法的性能。我们的研究结果表明,通过训练一个统一的模型,以相互引导的方式执行这两项任务,可以同时提高皮肤病变分割和分类的性能。

英文摘要:

Automated skin lesion segmentation and classification are two most essential and related tasks in the computer-aided diagnosis of skin cancer. Despite their prevalence, deep learning models are usually designed for only one task, ignoring the potential benefits in jointly performing both tasks. In this paper, we propose the mutual bootstrapping deep convolutional neural networks (MB-DCNN) model for simultaneous skin lesion segmentation and classification. This model consists of a coarse segmentation network (coarse-SN), a mask-guided classification network (mask-CN), and an enhanced segmentation network (enhanced-SN). On one hand, the coarse-SN generates coarse lesion masks that provide a prior bootstrapping for mask-CN to help it locate and classify skin lesions accurately. On the other hand, the lesion localization maps produced by mask-CN are then fed into enhanced-SN, aiming to transfer the localization information learned by mask-CN to enhanced-SN for accurate lesion segmentation. In this way, both segmentation and classification networks mutually transfer knowledge between each other and facilitate each other in a bootstrapping way. Meanwhile, we also design a novel rank loss and jointly use it with the Dice loss in segmentation networks to address the issues caused by class imbalance and hard-easy pixel imbalance. We evaluate the proposed MB-DCNN model on the ISIC-2017 and PH2 datasets, and achieve a Jaccard index of 80.4% and 89.4% in skin lesion segmentation and an average AUC of 93.8% and 97.7% in skin lesion classification, which are superior to the performance of representative state-of-the-art skin lesion segmentation and classification methods. Our results suggest that it is possible to boost the performance of skin lesion segmentation and classification simultaneously via training a unified model to perform both tasks in a mutual bootstrapping way.

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

↑