An Improved COVID-19 Lung X-Ray Image Classification Algorithm Based on ConvNeXt Network
Fuxiang Liu, Chen Zang, Junqi Shi, Weiyu He, Yupeng Liang, Lei Li
Abstract
Aiming at the new coronavirus that appeared in 2019, which has caused a large number of infected patients worldwide due to its high contagiousness, in order to detect the source of infection in time and cut off the chain of transmission, we developed a new Chest X-ray (CXR) image classification algorithm with high accuracy, simple operation and fast processing for COVID-19. The algorithm is based on ConvNeXt pure convolutional neural network, we adjusted the network structure and loss function, added some new Data Augmentation methods and introduced attention mechanism. Compared with other classical convolutional neural network classification algorithms such as AlexNet, ResNet-34, ResNet-50, ResNet-101, ConvNeXt-tiny, ConvNeXt-small and ConvNeXt-base, the improved algorithm has better performance on COVID dataset.