arXiv reaDer
不均衡なデータセットで最適化されたディープワンクラス分類を使用したGI内視鏡画像でのマルチクラス異常検出
Multiclass Anomaly Detection in GI Endoscopic Images using Optimized Deep One-class Classification in an Imbalanced Dataset
ワイヤレスカプセル内視鏡検査は、医師が胃腸(GI)管を非侵襲的に検査するのに役立ちますが、多くの画像を生成するコストがかかります。 KID2やKvasirなどの利用可能なデータセットの多くは、効果的な人工知能(AI)システムのトレーニングを困難にする不均衡の問題に悩まされています。さらに、クラスの数が増えると、問題が悪化します。この研究では、1クラス分類器のアンサンブルを使用して異常を検出します。この方法は、1つのクラスのみのサンプルを使用して単一モデルを学習し、すべてのモデルをアンサンブルしてマルチクラス分類することに重点を置いています。 KID2データセットから、合計1,778の正常、227の炎症、303の血管疾患、および44のポリープ画像が使用されました。最初のステップでは、前処理された画像からオートエンコーダアーキテクチャに基づいて深い特徴が抽出されます。次に、これらの機能は、合成マイノリティオーバーサンプリング手法を使用してオーバーサンプリングされ、クラスタリング構造を識別するために順序付けポイントを使用してクラスタリングされます。 1クラスの分類モデルを作成するために、サポートベクターデータの説明は、F1スコアを改善するためのクラスタリングパラメーターの調整にも使用されるアリコロニー最適化の助けを借りて、各クラスターでトレーニングされます。このプロセスは、マルチクラス分類に使用される最終モデルの各クラスとアンサンブルに適用されます。アルゴリズム全体が5回実行され、異常検出とマルチクラス分類のために、それぞれF1スコア96.3±0.2%とマクロ平均F1スコア85.0±0.4%が得られました。結果は、GoogleNet、AlexNet、Resnet50、VGG16、およびその他の公開されたアルゴリズムと比較され、提案された方法がGI画像のマルチクラスクラス異常検出の競争力のある選択肢であることを示しています。
Wireless Capsule Endoscopy helps physicians examine the gastrointestinal (GI) tract noninvasively, with the cost of generating many images. Many available datasets, such as KID2 and Kvasir, suffer from imbalance issue which make it difficult to train an effective artificial intelligence (AI) system. Moreover, increasing number of classes makes the problem worse. In this study, an ensemble of one-class classifiers is used for detecting anomaly. This method focuses on learning single models using samples from only one class, and ensemble all models for multiclass classification. A total of 1,778 normal, 227 inflammation, 303 vascular diseases, and 44 polyp images have been used from the KID2 dataset. In the first step, deep features are extracted based on an autoencoder architecture from the preprocessed images. Then, these features are oversampled using Synthetic Minority Over-sampling Technique and clustered using Ordering Points to Identify the Clustering Structure. To create one-class classification model, the Support Vector Data Descriptions are trained on each cluster with the help of Ant Colony Optimization, which is also used for tuning clustering parameters for improving F1-score. This process is applied on each classes and ensemble of final models used for multiclass classification. The entire algorithm ran 5 times and obtained F1-score 96.3 +- 0.2% and macro-average F1-score 85.0 +- 0.4%, for anomaly detection and multiclass classification, respectively. The results are compared with GoogleNet, AlexNet, Resnet50, VGG16 and other published algorithms, and demonstrate that the proposed method is a competitive choice for multiclass class anomaly detection in GI images.
updated: Mon Mar 15 2021 16:28:42 GMT+0000 (UTC)
published: Mon Mar 15 2021 16:28:42 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アソシエイト