A Layered Self-Supervised Knowledge Distillation Framework for Efficient Multimodal Learning on the Edge
By: Tarique Dahri , Zulfiqar Ali Memon , Zhenyu Yu and more
Potential Business Impact:
Makes small AI models work better on phones.
We introduce Layered Self-Supervised Knowledge Distillation (LSSKD) framework for training compact deep learning models. Unlike traditional methods that rely on pre-trained teacher networks, our approach appends auxiliary classifiers to intermediate feature maps, generating diverse self-supervised knowledge and enabling one-to-one transfer across different network stages. Our method achieves an average improvement of 4.54\% over the state-of-the-art PS-KD method and a 1.14% gain over SSKD on CIFAR-100, with a 0.32% improvement on ImageNet compared to HASSKD. Experiments on Tiny ImageNet and CIFAR-100 under few-shot learning scenarios also achieve state-of-the-art results. These findings demonstrate the effectiveness of our approach in enhancing model generalization and performance without the need for large over-parameterized teacher networks. Importantly, at the inference stage, all auxiliary classifiers can be removed, yielding no extra computational cost. This makes our model suitable for deploying small language models on affordable low-computing devices. Owing to its lightweight design and adaptability, our framework is particularly suitable for multimodal sensing and cyber-physical environments that require efficient and responsive inference. LSSKD facilitates the development of intelligent agents capable of learning from limited sensory data under weak supervision.
Similar Papers
Learning from Stochastic Teacher Representations Using Student-Guided Knowledge Distillation
Machine Learning (CS)
Makes one computer model smarter than many.
Sample-level Adaptive Knowledge Distillation for Action Recognition
CV and Pattern Recognition
Makes video analysis work on small devices.
Dual-Model Weight Selection and Self-Knowledge Distillation for Medical Image Classification
CV and Pattern Recognition
Makes small computers diagnose diseases from scans.