画像前処理とハイパーパラメータ調整を伴う画像分類のための畳み込みニューラルネットワーク(CNN)の方法を提案した。この方法は、より複雑なモデルアーキテクチャでありながら、COVID-19診断の予測パフォーマンスを向上させることを目的としています。まず、CNNモデルには、4つの類似した畳み込み層が含まれ、その後に平坦化と2つの密な層が続きます。この作業は、コンピュータ断層撮影スキャンの2Dスライスを単純に分類することに基づくより複雑でないソリューションを提案します。アーキテクチャが単純であるにもかかわらず、提案されたCNNモデルは、スライスのケースを予測する際に、最先端を超える改善された定量的結果を示しました。結果は、COV-19-CT-DBデータセットの注釈付きCTスライスで達成されました。次に、元のデータセットは、解剖学的構造に関連するスライスのマスキング、CTボリュームから代表的でないスライスの削除、およびハイパーパラメータの調整によって処理されました。スライス処理では、固定サイズの長方形の領域を使用して、画像内の解剖学的関連の関心領域をトリミングし、2値化されたスライスの白いピクセル数に基づくしきい値を使用して、代表的でないスライスを画像から削除しました。 3D-CTスキャン。学習率スケジュールと指数関数的減衰およびスライスフリッピング技術を備えたCNNモデルが、処理されたスライスに展開されました。提案された方法は、2Dスライスの予測を行うために使用され、患者レベルでの最終診断のために、診断を行うために各CTスキャンのスライスに多数決が適用されました。提案された方法のマクロF1スコアは、検証セットおよびCOV-19CT-DBデータセットからの以前に表示されなかった画像のテストパーティションで、ベースラインアプローチおよびその他の代替案をはるかに上回りました。
A method of a Convolutional Neural Networks (CNN) for image classification with image preprocessing and hyperparameters tuning was proposed. The method aims at increasing the predictive performance for COVID-19 diagnosis while more complex model architecture. Firstly, the CNN model includes four similar convolutional layers followed by a flattening and two dense layers. This work proposes a less complex solution based on simply classifying 2D-slices of Computed Tomography scans. Despite the simplicity in architecture, the proposed CNN model showed improved quantitative results exceeding state-of-the-art when predicting slice cases. The results were achieved on the annotated CT slices of the COV-19-CT-DB dataset. Secondly, the original dataset was processed via anatomy-relevant masking of slice, removing none-representative slices from the CT volume, and hyperparameters tuning. For slice processing, a fixed-sized rectangular area was used for cropping an anatomy-relevant region-of-interest in the images, and a threshold based on the number of white pixels in binarized slices was employed to remove none-representative slices from the 3D-CT scans. The CNN model with a learning rate schedule and an exponential decay and slice flipping techniques was deployed on the processed slices. The proposed method was used to make predictions on the 2D slices and for final diagnosis at patient level, majority voting was applied on the slices of each CT scan to take the diagnosis. The macro F1 score of the proposed method well-exceeded the baseline approach and other alternatives on the validation set as well as on a test partition of previously unseen images from COV-19CT-DB dataset.