perseveranceLX / HKBU_HPML_COVID-19

Source code of paper "Benchmarking Deep Learning Models and Automated Model Design for COVID-19 Detection with Chest CT Scans".

Home Page:https://www.medrxiv.org/content/10.1101/2020.06.08.20125963v2

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

HKBU_HPML_COVID-19

Source code of this paper.

Table of Contents

1. Link to Clean-CC-CCII (based on CC-CCII ver. 1.0)

FTP Server: http://47.92.107.188/Datasets/COVID-DATA-CT/

Google Drive: https://drive.google.com/drive/folders/1qOWNdi5eRpuJClPimwIHvCV8z2RN7HQB?usp=sharing

OneDrive: https://hkbuedu-my.sharepoint.com/:f:/g/personal/shwang_link_hkbu_edu_hk/EoieIbNgTjNGh0-sbsx3kj8BxzurNH994kwX2R6mjf3q1g?e=miYpfz

Raw Data (CC-CCII) http://ncov-ai.big.ac.cn/download

2. Experimental Results

2.1 Benchmark Deep Learning Models

If you want to run the benchmark experiments, you can refer to the directory of covid19_pipeline.

  • The pipeline of benchmarking deep learning-based models.

pipeline

  • Performance comparison between different models

model_perf

  • Performance comparison between ResNet3d models with different depth

model_depth

  • Performance comparison between models trained by scan data comprising a different number of slices.

mdoel_slice

  • The model accuracy before and after using MixUp data augmentation method.

model_mixup

2.2 Automated Model Design

The code of NAS will be released very soon ...

  • NAS pipeline

NAS pipeline

  • Search space

Search space

  • The performance comparison between baseline models and models designed by NAS

nas_vs_manual

3. Citation

Benchmarking Deep Learning Models and Automated Model Design for COVID-19 Detection with Chest CT Scans

@article {He et al.benchmark,
	author = {He, Xin and Wang, Shihao and Shi, Shaohuai and Chu, Xiaowen and Tang, Jiangping and Liu, Xin and Yan, Chenggang and Zhang, Jiyong and Ding, Guiguang},
	title = {Benchmarking Deep Learning Models and Automated Model Design for COVID-19 Detection with Chest CT Scans},
	elocation-id = {2020.06.08.20125963},
	year = {2020},
	doi = {10.1101/2020.06.08.20125963},
	publisher = {Cold Spring Harbor Laboratory Press},
	URL = {https://www.medrxiv.org/content/early/2020/06/09/2020.06.08.20125963},
	eprint = {https://www.medrxiv.org/content/early/2020/06/09/2020.06.08.20125963.full.pdf},
	journal = {medRxiv}
}

About

Source code of paper "Benchmarking Deep Learning Models and Automated Model Design for COVID-19 Detection with Chest CT Scans".

https://www.medrxiv.org/content/10.1101/2020.06.08.20125963v2


Languages

Language:Jupyter Notebook 91.4%Language:Python 8.4%Language:Shell 0.2%