arXiv reaDer
脳腫瘍分類のためのトリプレット対照学習
Triplet Contrastive Learning for Brain Tumor Classification
脳腫瘍は、成人と子供の両方に影響を与える一般的で致命的な癌の形態です。したがって、脳腫瘍をさまざまなタイプに分類することは、医師が処方する治療に大きく影響するため、重要なタスクです。これに照らして、医療画像技術、特に深い畳み込みネットワークとそれに続く分類層を適用する技術が開発され、脳腫瘍タイプのコンピューター支援分類が可能になりました。この論文では、分類などの下流のタスクに使用できる、脳腫瘍タイプの深い埋め込みを直接学習する新しいアプローチを紹介します。トリプレット損失バリアントを使用することに加えて、私たちのアプローチは、脳腫瘍画像解析ドメインのデータ不足の問題を効果的に改善するために、まれなケースのデータ拡張モジュールと組み合わせて、監視されていない事前トレーニングの実行に対照学習を適用します。私たちは、27の異なる腫瘍クラスからなる広範な脳腫瘍データセットで私たちの方法を評価します。そのうち13はまれであると定義されています。すべての実験で共通のエンコーダーを使用して、アプローチをベースライン分類レイヤーベースのモデルと比較し、その結果は、測定されたすべてのメトリックにわたるアプローチの有効性を十分に証明しています。
Brain tumor is a common and fatal form of cancer which affects both adults and children. The classification of brain tumors into different types is hence a crucial task, as it greatly influences the treatment that physicians will prescribe. In light of this, medical imaging techniques, especially those applying deep convolutional networks followed by a classification layer, have been developed to make possible computer-aided classification of brain tumor types. In this paper, we present a novel approach of directly learning deep embeddings for brain tumor types, which can be used for downstream tasks such as classification. Along with using triplet loss variants, our approach applies contrastive learning to performing unsupervised pre-training, combined with a rare-case data augmentation module to effectively ameliorate the lack of data problem in the brain tumor imaging analysis domain. We evaluate our method on an extensive brain tumor dataset which consists of 27 different tumor classes, out of which 13 are defined as rare. With a common encoder during all the experiments, we compare our approach with a baseline classification-layer based model, and the results well prove the effectiveness of our approach across all measured metrics.
updated: Sun Aug 08 2021 11:26:34 GMT+0000 (UTC)
published: Sun Aug 08 2021 11:26:34 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アソシエイト