arXiv reaDer
A Mutual Bootstrapping Model for Automated Skin Lesion Segmentation and Classification
 自動化された皮膚病変のセグメンテーションと分類は、皮膚癌のコンピューター支援診断で最も重要な2つの関連するタスクです。普及しているにもかかわらず、ディープラーニングモデルは通常1つのタスクのみを対象に設計されており、両方のタスクを共同で実行することの潜在的な利点を無視しています。本論文では、同時皮膚病変のセグメンテーションと分類のための相互ブートストラップディープ畳み込みニューラルネットワーク(MB-DCNN)モデルを提案します。このモデルは、粗いセグメンテーションネットワーク(coarse-SN)、マスク誘導分類ネットワーク(mask-CN)、および強化されたセグメンテーションネットワーク(enhanced-SN)で構成されています。一方、粗いSNはマスクCNの事前のブートストラップを提供する粗い病変マスクを生成し、皮膚病変を正確に特定して分類するのに役立ちます。一方、マスクCNによって生成された病変局在化マップは、正確な病変セグメンテーションのためにマスクCNによって学習された局在化情報を拡張SNに転送することを目的として、拡張SNに供給されます。このようにして、セグメンテーションネットワークと分類ネットワークの両方が相互に知識を相互に転送し、ブートストラップ方式で相互に促進します。また、新しいランク損失を設計し、セグメンテーションネットワークのダイス損失と共同で使用して、クラスの不均衡と困難なピクセルの不均衡に起因する問題に対処します。 ISIC-2017およびPH2データセットで提案された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.
updated: Wed Feb 12 2020 01:18:30 GMT+0000 (UTC)
published: Fri Mar 08 2019 07:57:40 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト