HDKD: Hybrid Data-Efficient Knowledge Distillation Network for Medical Image Classification

TL;DR

This paper introduces HDKD, a hybrid knowledge distillation method that combines CNN and transformer models for medical image classification, improving accuracy and efficiency with a novel lightweight convolutional block.

Contribution

The paper proposes a novel hybrid distillation framework using a CNN teacher and a hybrid student, enabling effective feature transfer without additional computational costs.

Findings

HDKD outperforms state-of-the-art models on medical datasets.

The hybrid student leverages both CNN and transformer strengths.

The lightweight MBCSA block enhances efficiency and performance.

Abstract

Vision Transformers (ViTs) have achieved significant advancement in computer vision tasks due to their powerful modeling capacity. However, their performance notably degrades when trained with insufficient data due to lack of inherent inductive biases. Distilling knowledge and inductive biases from a Convolutional Neural Network (CNN) teacher has emerged as an effective strategy for enhancing the generalization of ViTs on limited datasets. Previous approaches to Knowledge Distillation (KD) have pursued two primary paths: some focused solely on distilling the logit distribution from CNN teacher to ViT student, neglecting the rich semantic information present in intermediate features due to the structural differences between them. Others integrated feature distillation along with logit distillation, yet this introduced alignment operations that limits the amount of knowledge transferred due to mismatched architectures and increased the computational overhead. To this end, this paper presents Hybrid Data-efficient Knowledge Distillation (HDKD) paradigm which employs a CNN teacher and a hybrid student. The choice of hybrid student serves two main aspects. First, it leverages the strengths of both convolutions and transformers while sharing the convolutional structure with the teacher model. Second, this shared structure enables the direct application of feature distillation without any information loss or additional computational overhead. Additionally, we propose an efficient light-weight convolutional block named Mobile Channel-Spatial Attention (MBCSA), which serves as the primary convolutional block in both teacher and student models. Extensive experiments on two medical public datasets showcase the superiority of HDKD over other state-of-the-art models and its computational efficiency. Source code at: https://github.com/omarsherif200/HDKD