arXiv reaDer
深部空間モデルを使用した MR 画像における脳腫瘍の分類
Classification of Brain Tumours in MR Images using Deep Spatiospatial Models
脳腫瘍は、脳内の異常な細胞の塊またはクラスターであり、隣接する組織に侵入したり、転移を形成したりする可能性があるため、生命を脅かす可能性があります。治療計画を成功させるには正確な診断が不可欠であり、磁気共鳴画像法は、脳腫瘍とその範囲を診断するための主要な画像診断法です。コンピューター ビジョン アプリケーションのディープ ラーニング手法は、近年大幅な改善を示しています。そのほとんどは、モデルをトレーニングするために大量のデータを利用できるという事実によるものであり、モデル アーキテクチャの改善により、教師あり設定。このような深層学習手法を使用した腫瘍の分類は、信頼性の高い注釈付きのオープン データセットが利用できるようになり、大幅な進歩を遂げました。通常、これらの方法は、3D ボリューム MRI を使用する 3D モデル、または各スライスを個別に考慮した 2D モデルです。ただし、スライスの空間次元を個別に扱うことにより、時空間モデルをこのタスクの時空間モデルとして採用することができます。これらのモデルには、計算コストを削減しながら、特定の空間的および時間的関係を学習する機能があります。この論文では、ResNet (2+1)D と ResNet 混合畳み込みという 2 つの時空間モデルを使用して、さまざまなタイプの脳腫瘍を分類しています。これらのモデルは両方とも、純粋な 3D 畳み込みモデルである ResNet18 よりも優れたパフォーマンスを発揮することが観察されました。さらに、腫瘍分類のタスクのためにモデルをトレーニングする前に、別の関連性のないデータセットでモデルを事前トレーニングすると、パフォーマンスが向上することも観察されました。最後に、事前トレーニング済みの ResNet 混合畳み込みは、これらの実験で最良のモデルであることが観察され、0.93 のマクロ F1 スコアと 96.98% のテスト精度を達成すると同時に、計算コストが最小のモデルであることがわかりました。
A brain tumour is a mass or cluster of abnormal cells in the brain, which has the possibility of becoming life-threatening because of its ability to invade neighbouring tissues and also form metastases. An accurate diagnosis is essential for successful treatment planning and magnetic resonance imaging is the principal imaging modality for diagnostic of brain tumours and their extent. Deep Learning methods in computer vision applications have shown significant improvement in recent years, most of which can be credited to the fact that a sizeable amount of data is available to train models on, and the improvements in the model architectures yielding better approximations in a supervised setting. Classifying tumours using such deep learning methods has made significant progress with the availability of open datasets with reliable annotations. Typically those methods are either 3D models, which use 3D volumetric MRIs or even 2D models considering each slice separately. However, by treating the slice spatial dimension separately, spatiotemporal models can be employed as spatiospatial models for this task. These models have the capabilities of learning specific spatial and temporal relationship, while reducing computational costs. This paper uses two spatiotemporal models, ResNet (2+1)D and ResNet Mixed Convolution, to classify different types of brain tumours. It was observed that both these models performed superior to the pure 3D convolutional model, ResNet18. Furthermore, it was also observed that pre-training the models on a different, even unrelated dataset before training them for the task of tumour classification improves the performance. Finally, Pre-trained ResNet Mixed Convolution was observed to be the best model in these experiments, achieving a macro F1-score of 0.93 and a test accuracy of 96.98%, while at the same time being the model with the least computational cost.
updated: Fri May 28 2021 19:27:51 GMT+0000 (UTC)
published: Fri May 28 2021 19:27:51 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アソシエイト