Intelligent Method for COVID-19 Diagnosis: Construction and Comparative Analysis of ResKAN18
TL;DR
This paper proposes ResKAN18, a hybrid structure that embeds the learnable spline function of the Kolmogorov–Arnold network (KAN) into the ResNet18 classification head for intelligent diagnosis of COVID-19 in chest X-ray images.
Abstract
In response to the challenge of balancing accuracy and generalization in medical image classification using convolutional neural networks (CNNs), this paper proposes ResKAN18, a hybrid structure that embeds the learnable spline function of the Kolmogorov–Arnold network (KAN) into the ResNet18 classification head for intelligent diagnosis of COVID-19 in chest X-ray images. ResKAN18 includes three variants: ResKAN18—Large (four layers of KAN, hidden-layer dimensions [256, 128, 64]), ResKAN18—Standard (four layers of KAN, hidden-layer dimensions [128, 64, 32]), and ResKAN18—Simple (three layers of KAN, hidden-layer dimensions [64, 32]), which can achieve a flexible balance between accuracy and efficiency with different depths of KAN. A systematic comparison has been conducted between four classic CNN baselines including ResNet18, VGG16, DenseNet121, ShuffleNetV2, and three ResKAN18 variants on a benchmark dataset containing 3880 chest X-rays (COVID-19, normal, viral pneumonia). The results have shown that ResKAN18—Large can achieve an accuracy of 98.80% on the independent test set, which is 1.21% higher than ResNet18 and 0.69% higher than DenseNet121—its parameter count is 13.97M, inference delay is 8.25 ms/image, and training–validation accuracy difference is only 1.50%. The accuracy and performance stability of the dataset under random partitioning conditions are superior to the other two variants and all classic CNN baselines. All ResKAN18 variants have achieved zero missed diagnoses for COVID-19, while ResNet18 has shown missed diagnoses (0.9944). Taking into account the trade-off between accuracy, generalization, and inference efficiency, ResKAN1—Large is recommended as the default configuration, while for edge deployment scenarios with severely limited resources, ResKAN18—Simple can provide a cost-effective alternative with an extremely low latency of 2.50 ms/image and only 3.1% parameter increment compared to ResNet18.