Boosting COVID-19 Severity Detection with Infection-Aware Contrastive Mixup Classification
Abstract
This paper presents our solution for the 2nd COVID-19 Severity Detection Competition. This task aims to distinguish the Mild, Moderate, Severe, and Critical grades in COVID-19 chest CT images. In our approach, we devise a novel infection-aware 3D Contrastive Mixup Classification network for severity grading. Specifically, we train two segmentation networks to first extract the lung region and then the inner lesion region. The lesion segmentation mask serves as complementary information for the original CT slices. To relieve the issue of imbalanced data distribution, we further improve the advanced Contrastive Mixup Classification network by weighted cross-entropy loss. On the COVID-19 severity detection leaderboard, our approach won the first place with a Macro F1 Score of 51.76%. It significantly outperforms the baseline method by over 11.46%.
Cite
Text
Hou et al. "Boosting COVID-19 Severity Detection with Infection-Aware Contrastive Mixup Classification." European Conference on Computer Vision Workshops, 2022. doi:10.1007/978-3-031-25082-8_36Markdown
[Hou et al. "Boosting COVID-19 Severity Detection with Infection-Aware Contrastive Mixup Classification." European Conference on Computer Vision Workshops, 2022.](https://mlanthology.org/eccvw/2022/hou2022eccvw-boosting/) doi:10.1007/978-3-031-25082-8_36BibTeX
@inproceedings{hou2022eccvw-boosting,
title = {{Boosting COVID-19 Severity Detection with Infection-Aware Contrastive Mixup Classification}},
author = {Hou, Junlin and Xu, Jilan and Zhang, Nan and Zhang, Yuejie and Zhang, Xiaobo and Feng, Rui},
booktitle = {European Conference on Computer Vision Workshops},
year = {2022},
pages = {537-551},
doi = {10.1007/978-3-031-25082-8_36},
url = {https://mlanthology.org/eccvw/2022/hou2022eccvw-boosting/}
}