Journal of Optoelectronics · Laser, Volume. 35, Issue 9, 993(2024)
Lightweight thoracic disease classification algorithm based on mixed knowledge distillation
Existing lightweight networks for classifying thoracic diseases have a large number of parameters and require significant hardware resources. This paper proposes a lightweight algorithm for classifying thoracic diseases based on mixed knowledge distillation (KD) training strategy. Firstly, the algorithm incorporates an optimized residual shrinkage module into the MobileViT base network and employs soft thresholding to filter background noise in X-ray images. Then a mixed knowledge distillation training strategy is proposed, utilizing multi-level attention maps and similarity activation matrices as supervisory signals to enhance the ability of lightweight networks to recognize thoracic diseases. Finally, the focal loss function is employed to address the imbalance between positive and negative samples in the dataset. Experimental results on the ChestX-Ray14 dataset demonstrate that the average AUC value for the RMSNet student model trained with distilled knowledge to recognize 14 types of thoracic diseases is 0.836. The number of parameters and FLOPs are only 0.96 M and 0.27 G, respectively. These results indicate that the proposed algorithm improves classification accuracy while maintaining lightweight, enabling the network to run with less hardware.
Get Citation
Copy Citation Text
LAI Yu, LI Qiang, NIE Weizhi, BAI Yunpeng, ZHAO Feng. Lightweight thoracic disease classification algorithm based on mixed knowledge distillation[J]. Journal of Optoelectronics · Laser, 2024, 35(9): 993
Category:
Received: Feb. 20, 2023
Accepted: Dec. 20, 2024
Published Online: Dec. 20, 2024
The Author Email: