arXiv reaDer
胸部CTスキャンによるCOVID-19検出のための3D深層学習モデルの自動モデル設計とベンチマーク
Automated Model Design and Benchmarking of 3D Deep Learning Models for COVID-19 Detection with Chest CT Scans
COVID-19のパンデミックは、数か月にわたって世界的に広がっています。その伝染性と高い病原性は人々の生命を深刻に脅かしているため、COVID-19感染を正確かつ迅速に検出することが重要です。最近の多くの研究では、深層学習(DL)ベースのソリューションが、胸部CTスキャンに基づいてCOVID-19を検出するのに役立つことが示されています。ただし、ほとんどの既存の作業は2Dデータセットに焦点を合わせており、実際のCTスキャンは3D画像であるため、モデルの品質が低下する可能性があります。その上、報告された結果は、比較的不公平な比較で、さまざまなデータセットの広い範囲に及びます。このホワイトペーパーでは、最初に3つの最先端の3Dモデル(ResNet3D101、DenseNet3D121、およびMC3 \ _18)を使用して、公開されている3つの胸部CTスキャンデータセットのベースラインパフォーマンスを確立します。次に、微分可能ニューラルアーキテクチャ検索(DNAS)フレームワークを提案し、ガンベルソフトマックス手法を使用して3D胸部CTスキャン分類の3D DLモデルを自動的に検索し、検索効率を向上させます。さらに、モデルでクラスアクティベーションマッピング(CAM)手法を活用して、結果の解釈可能性を提供します。実験結果は、自動検索されたモデル(CovidNet3D)が、3つのデータセットでベースラインの人間が設計したモデルよりも、モデルサイズが数十倍小さく、精度が高いことを示しています。さらに、この結果は、CAMをCovidNet3D for COVID-19データセットに適切に適用して、医療診断の解釈可能性を提供できることも確認しています。
The COVID-19 pandemic has spread globally for several months. Because its transmissibility and high pathogenicity seriously threaten people's lives, it is crucial to accurately and quickly detect COVID-19 infection. Many recent studies have shown that deep learning (DL) based solutions can help detect COVID-19 based on chest CT scans. However, most existing work focuses on 2D datasets, which may result in low quality models as the real CT scans are 3D images. Besides, the reported results span a broad spectrum on different datasets with a relatively unfair comparison. In this paper, we first use three state-of-the-art 3D models (ResNet3D101, DenseNet3D121, and MC3\_18) to establish the baseline performance on the three publicly available chest CT scan datasets. Then we propose a differentiable neural architecture search (DNAS) framework to automatically search for the 3D DL models for 3D chest CT scans classification with the Gumbel Softmax technique to improve the searching efficiency. We further exploit the Class Activation Mapping (CAM) technique on our models to provide the interpretability of the results. The experimental results show that our automatically searched models (CovidNet3D) outperform the baseline human-designed models on the three datasets with tens of times smaller model size and higher accuracy. Furthermore, the results also verify that CAM can be well applied in CovidNet3D for COVID-19 datasets to provide interpretability for medical diagnosis.
updated: Thu Jan 14 2021 03:45:01 GMT+0000 (UTC)
published: Thu Jan 14 2021 03:45:01 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アソシエイト