arXiv reaDer
Parallel Capsule Networks for Classification of White Blood Cells
Capsule Networks(CapsNets)は、畳み込みニューラルネットワーク(CNN)のいくつかの欠点を克服するために提案された機械学習アーキテクチャです。ただし、CapsNetは、画像が小さいデータセットや識別対象のオブジェクトのバックグラウンドノイズが最小限であるデータセットでは、主にCNNを上回っています。この作業では、ネットワークを分岐して特定のカプセルを分離し、各分岐が異なるエンティティを識別できるようにするという概念を活用する、新しいアーキテクチャである並列CapsNetsを紹介します。私たちは、カプセルの異なる層を持つネットワークのパフォーマンスを研究して、2つの現在のタイプのCapsNetアーキテクチャに私たちの概念を適用しました。急性骨髄性白血病の画像(15クラス)の公開された非常に不均衡なデータセットで設計をテストしました。私たちの実験では、従来のCapsNetはベースラインCNN(ResNeXt-50)と同様のパフォーマンスを示していますが、不安定性の問題を示しています。対照的に、並列CapsNetはResNeXt-50をしのぐことができ、より安定しており、従来のCapsNetとResNeXt-50の両方よりも優れた回転不変性を示します。
Capsule Networks (CapsNets) is a machine learning architecture proposed to overcome some of the shortcomings of convolutional neural networks (CNNs). However, CapsNets have mainly outperformed CNNs in datasets where images are small and/or the objects to identify have minimal background noise. In this work, we present a new architecture, parallel CapsNets, which exploits the concept of branching the network to isolate certain capsules, allowing each branch to identify different entities. We applied our concept to the two current types of CapsNet architectures, studying the performance for networks with different layers of capsules. We tested our design in a public, highly unbalanced dataset of acute myeloid leukaemia images (15 classes). Our experiments showed that conventional CapsNets show similar performance than our baseline CNN (ResNeXt-50) but depict instability problems. In contrast, parallel CapsNets can outperform ResNeXt-50, is more stable, and shows better rotational invariance than both, conventional CapsNets and ResNeXt-50.
updated: Thu Aug 05 2021 14:30:44 GMT+0000 (UTC)
published: Thu Aug 05 2021 14:30:44 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト