arXiv reaDer
BERT機能を使用した2段階のCOVID19分類
Two-Stage COVID19 Classification Using BERT Features
ダブルBERT特徴抽出を使用した肺CTスキャンスライス画像からの自動COVID1-19診断フレームワークを提案します。最初のBERT特徴抽出では、3D-CNNを最初に使用してCNN内部特徴マップを抽出します。グローバル平均プーリングを使用する代わりに、後期BERT時間プーイングを使用して、これらの特徴マップの時間情報を集約し、その後に分類レイヤーを使用します。この3D-CNN-BERT分類ネットワークは、最初に、すべての元のCTスキャンボリュームからサンプリングされた固定数のスライス画像でトレーニングされます。第2段階では、3D-CNN-BERT埋め込み特徴が、すべてのCTスキャンボリュームのすべてのスライス画像で抽出され、これらの特徴が固定数のセグメントに平均化されます。次に、別のBERTネットワークを使用して、これらの複数の機能を1つの機能に集約し、その後に別の分類レイヤーを配置します。両方のステージの分類結果を組み合わせて、最終出力を生成します。検証データセットでは、0.9164のマクロF1スコアを達成しています。
We propose an automatic COVID1-19 diagnosis framework from lung CT-scan slice images using double BERT feature extraction. In the first BERT feature extraction, A 3D-CNN is first used to extract CNN internal feature maps. Instead of using the global average pooling, a late BERT temporal pooing is used to aggregate the temporal information in these feature maps, followed by a classification layer. This 3D-CNN-BERT classification network is first trained on sampled fixed number of slice images from every original CT scan volume. In the second stage, the 3D-CNN-BERT embedding features are extracted on all slice images of every CT scan volume, and these features are averaged into a fixed number of segments. Then another BERT network is used to aggregate these multiple features into a single feature followed by another classification layer. The classification results of both stages are combined to generate final outputs. On the validation dataset, we achieve macro F1 score of 0.9164.
updated: Wed Jun 29 2022 19:01:37 GMT+0000 (UTC)
published: Wed Jun 29 2022 19:01:37 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アソシエイト