arXiv reaDer
不均衡なデータセットでの皮膚がん検出のためのディープニューラルネットワークのアンサンブルを使用した転移学習
Transfer Learning with Ensembles of Deep Neural Networks for Skin Cancer Detection in Imbalanced Data Sets
医用画像から皮膚がんを正確に検出するためのいくつかの機械学習技術が報告されています。これらの手法の多くは、事前にトレーニングされた畳み込みニューラルネットワーク(CNN)に基づいており、限られた量のトレーニングデータに基づいてモデルをトレーニングできます。ただし、これらのモデルの分類精度は、悪性腫瘍からの代表的な画像の不足によって依然として厳しく制限される傾向があります。複数のCNNモデル(一部は事前トレーニング済み、一部は手元のデータのみでトレーニング済み)と、入力画像に関連付けられたメタデータの形式の補助データを組み合わせた、新しいアンサンブルベースのCNNアーキテクチャを提案します。メタ学習者。提案されたアプローチは、限られた不均衡なデータを処理するモデルの能力を向上させます。 2056人の患者からの33126のダーモスコピー画像を含むデータセットを使用して、提案された手法の利点を示します。提案された手法のパフォーマンスを、F1メジャー、ROC曲線下面積(AUC-ROC)、およびPR曲線下面積(AUC-PR)の観点から評価し、7つの異なるベンチマーク手法のパフォーマンスと比較します。 、最近の2つのCNNベースの手法を含みます。提案された手法は、すべての評価指標の点で有利に比較されます。
Several machine learning techniques for accurate detection of skin cancer from medical images have been reported. Many of these techniques are based on pre-trained convolutional neural networks (CNNs), which enable training the models based on limited amounts of training data. However, the classification accuracy of these models still tends to be severely limited by the scarcity of representative images from malignant tumours. We propose a novel ensemble-based CNN architecture where multiple CNN models, some of which are pre-trained and some are trained only on the data at hand, along with auxiliary data in the form of metadata associated with the input images, are combined using a meta-learner. The proposed approach improves the model's ability to handle limited and imbalanced data. We demonstrate the benefits of the proposed technique using a dataset with 33126 dermoscopic images from 2056 patients. We evaluate the performance of the proposed technique in terms of the F1-measure, area under the ROC curve (AUC-ROC), and area under the PR-curve (AUC-PR), and compare it with that of seven different benchmark methods, including two recent CNN-based techniques. The proposed technique compares favourably in terms of all the evaluation metrics.
updated: Mon May 17 2021 09:33:03 GMT+0000 (UTC)
published: Mon Mar 22 2021 06:04:45 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト