🤖 AI Summary
Traditional knowledge distillation overlooks the dynamic evolution of student models’ cognitive capacity, limiting knowledge transfer efficiency. To address this, we propose Gompertz-CNN—a novel distillation framework that explicitly models the sigmoidal (S-shaped) learning dynamics of students by integrating the Gompertz growth model into the distillation process for the first time. We further design a phase-aware, time-varying loss weighting mechanism that jointly optimizes Wasserstein-based feature alignment and gradient propagation matching, enabling coordinated alignment at both the feature and backward-propagation levels. The resulting framework is end-to-end trainable and supports multi-objective dynamic distillation. Extensive experiments on CIFAR-10 and CIFAR-100 demonstrate significant improvements over state-of-the-art distillation methods, achieving up to 8% and 4% absolute accuracy gains, respectively. Moreover, Gompertz-CNN exhibits strong effectiveness and robustness across diverse teacher–student architecture pairs.
📝 Abstract
This paper introduces a novel dynamic knowledge distillation framework, Gompertz-CNN, which integrates the Gompertz growth model into the training process to address the limitations of traditional knowledge distillation. Conventional methods often fail to capture the evolving cognitive capacity of student models, leading to suboptimal knowledge transfer. To overcome this, we propose a stage-aware distillation strategy that dynamically adjusts the weight of distillation loss based on the Gompertz curve, reflecting the student's learning progression: slow initial growth, rapid mid-phase improvement, and late-stage saturation. Our framework incorporates Wasserstein distance to measure feature-level discrepancies and gradient matching to align backward propagation behaviors between teacher and student models. These components are unified under a multi-loss objective, where the Gompertz curve modulates the influence of distillation losses over time. Extensive experiments on CIFAR-10 and CIFAR-100 using various teacher-student architectures (e.g., ResNet50 and MobileNet_v2) demonstrate that Gompertz-CNN consistently outperforms traditional distillation methods, achieving up to 8% and 4% accuracy gains on CIFAR-10 and CIFAR-100, respectively.