EfficientNet has shown exceptional performance in computer vision, yet its application in cytopathological image classification remains underexplored. This study introduces a novel self-distillation framework integrated into EfficientNet variants to enhance thyroid cytopathology classification, particularly for diagnosing Papillary Thyroid Carcinoma. Our approach incorporates Gate Classifiers into the last three blocks of EfficientNet during training, refining feature representations through a multi-loss strategy combining Mean Squared Error (MSE), Jessen Shannon divergence (JSD) and cross-entropy (CE) loss. By retaining only the final Gate Classifier at inference, our method improves classification accuracy while maintaining computational efficiency. Experimental results demonstrate nearly 7% higher Precision than compared to the baseline approach [1], with a throughput of 458 samples in just 55 s, underscoring its potential for real-world deployment.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Self-distilled EfficientNet: A Novel Training Framework for Thyroid Image Classification

  • Quoc Hieu Nguyen,
  • De Van Nguyen,
  • Tien Dung Vu,
  • Trong Hieu Nguyen,
  • Dinh Tung Pham,
  • Dinh Hoan Trinh,
  • Aladine Chetouani,
  • Marie Luong

摘要

EfficientNet has shown exceptional performance in computer vision, yet its application in cytopathological image classification remains underexplored. This study introduces a novel self-distillation framework integrated into EfficientNet variants to enhance thyroid cytopathology classification, particularly for diagnosing Papillary Thyroid Carcinoma. Our approach incorporates Gate Classifiers into the last three blocks of EfficientNet during training, refining feature representations through a multi-loss strategy combining Mean Squared Error (MSE), Jessen Shannon divergence (JSD) and cross-entropy (CE) loss. By retaining only the final Gate Classifier at inference, our method improves classification accuracy while maintaining computational efficiency. Experimental results demonstrate nearly 7% higher Precision than compared to the baseline approach [1], with a throughput of 458 samples in just 55 s, underscoring its potential for real-world deployment.