MiniSeg: An Extremely Minimum Network for Efficient COVID-19 Segmentation

Citations Over TimeTop 1% of 2021 papers

Abstract

The rapid spread of the new pandemic, i.e., COVID-19, has severely threatened global health. Deep-learning-based computer-aided screening, e.g., COVID-19 infected CT area segmentation, has attracted much attention. However, the publicly available COVID-19 training data are limited, easily causing overfitting for traditional deep learning methods that are usually data-hungry with millions of parameters. On the other hand, fast training/testing and low computational cost are also necessary for quick deployment and development of COVID-19 screening systems, but traditional deep learning methods are usually computationally intensive. To address the above problems, we propose MiniSeg, a lightweight deep learning model for efficient COVID-19 segmentation. Compared with traditional segmentation methods, MiniSeg has several significant strengths: i) it only has 83K parameters and is thus not easy to overfit; ii) it has high computational efficiency and is thus convenient for practical deployment; iii) it can be fast retrained by other users using their private COVID-19 data for further improving performance. In addition, we build a comprehensive COVID-19 segmentation benchmark for comparing MiniSeg to traditional methods.

Related Papers

• → An Overview of Overfitting and its Solutions(2019)2,168 cited
• → Lazy Overfitting Control(2013)10 cited
• → Machine Learning Students Overfit to Overfitting(2022)5 cited
• → Overfitting: Causes and Solutions (Seminar Slides)(2020)3 cited
• → Benign Overfitting in Classification: Provably Counter Label Noise with Larger Models(2022)3 cited