COVID-CT-Dataset:一个关于 COVID-19 的 CT 扫描数据集
机器学习
2020-06-19 v3 计算机视觉与模式识别
图像与视频处理
机器学习
摘要
在 COVID-19 暴发期间,计算机断层扫描(CT)是诊断 COVID-19 患者的有用手段。由于隐私问题,公开可用的 COVID-19 CT 数据集极难获取,这阻碍了基于 CT 的 COVID-19 AI 驱动诊断方法的研发。为解决此问题,我们构建了一个开源数据集——COVID-CT,其包含来自 216 名患者的 349 张 COVID-19 CT 图像以及 463 张非 COVID-19 CT。该数据集的实用性由一位自疫情暴发以来一直诊断和治疗 COVID-19 患者的高级放射科医生确认。我们还进行了实验研究,进一步证明该数据集有助于开发基于 AI 的 COVID-19 诊断模型。利用该数据集,我们开发了基于多任务学习和自监督学习的诊断方法,取得了 0.90 的 F1、0.98 的 AUC 和 0.89 的准确率。据该高级放射科医生所言,具有此类性能的模型已足够用于临床使用。数据和代码可在 https://github.com/UCSD-AI4H/COVID-CT 获取。
引用
@article{arxiv.2003.13865,
title = {COVID-CT-Dataset: A CT Scan Dataset about COVID-19},
author = {Xingyi Yang and Xuehai He and Jinyu Zhao and Yichen Zhang and Shanghang Zhang and Pengtao Xie},
journal= {arXiv preprint arXiv:2003.13865},
year = {2020}
}