LitCovNet: Attention-Based Lightweight Convolutional Network for Covid and Lung Disease Classification from Chest X-ray Images
摘要
Developing computer-aided diagnosis systems that balance accuracy and efficiency remains a challenge in COVID-19 and respiratory disease classification. To develop effective lightweight models, we investigated various optimization strategies, including the convolutional block attention module, depthwise separable convolutions, global average pooling, residual connections, and knowledge distillation. Leveraging these optimization strategies, we introduce two models, LitCovNet 1 and LitCovNet 2, and evaluate them on the COVID-19 Radiography Database and COVID-QU-Ex datasets. LitCovNet 2 achieves 98.78% accuracy on the Radiography Database and 95.64% on COVID-QU-Ex with only 67,349 parameters, while LitCovNet 1 attains 95.70% on COVID-QU-Ex with 131,861 parameters. Despite being 350