Journal of Optoelectronics · Laser, Volume. 35, Issue 9, 993(2024)

Lightweight thoracic disease classification algorithm based on mixed knowledge distillation

LAI Yu1, LI Qiang1, NIE Weizhi2, BAI Yunpeng3, and ZHAO Feng3
Author Affiliations
  • 1School of Microelectronics, Tianjin University, Tianjin 300072, China
  • 2School of Electrical and Information Engineering, Tianjin University, Tianjin 300072, China
  • 3Department of Cardiovascular Surgery, Tianjin Chest Hospital, Tianjin 300222, China
  • show less

    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.

    Tools

    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

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category:

    Received: Feb. 20, 2023

    Accepted: Dec. 20, 2024

    Published Online: Dec. 20, 2024

    The Author Email:

    DOI:10.16136/j.joel.2024.09.0056

    Topics