arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2506.07055cs.CV

用于边缘高效多模态学习的分层自监督知识蒸馏框架

A Layered Self-Supervised Knowledge Distillation Framework for Efficient Multimodal Learning on the Edge

Fast School of Computing, National University of Computer and Emerging Sciences(国家计算机与新兴科学大学快速计算学院);Universiti Malaya(马来亚大学);School of Professional Education and Executive Development, The Hong Kong Polytechnic University(香港理工大学专业教育与执行发展学院);COMSATS University Islamabad, Wah Campus(伊斯兰堡COMSATS大学瓦校区);Intelligent Transportation Systems University of Tennessee-Oak Ridge Innovation Institute’s Energy Storage and Transportation Convergent Research Initiative(智能交通系统田纳西大学奥克岭创新研究所能源存储与交通融合研究计划);Center for research in Computer Vision, University of Central Florida(计算机视觉研究中心,佛罗里达中央大学)

Tarique Dahri, Zulfiqar Ali Memon, Zhenyu Yu, Mohd. Yamani Idna Idris, Sheheryar Khan, Sadiq Ahmad, Maged Shoman, Saddam Aziz, Rizwan Qureshi

AI总结:

提出 LSSKD 框架,通过中间特征图辅助分类器生成自监督知识并跨阶段一对一迁移,无需预训练教师网络,在多个图像基准和少样本场景中取得先进性能,且推理无额外开销,适合边缘多模态感知部署。

摘要:

我们提出了用于训练紧凑型深度学习模型的分层自监督知识蒸馏(Layered Self-Supervised Knowledge Distillation,LSSKD)框架。与依赖预训练教师网络的传统方法不同,我们的方法将辅助分类器附加到中间特征图上,生成多样化的自监督知识,并实现跨不同网络阶段的一对一迁移。在 CIFAR-100 上,我们的方法相比最先进的 PS-KD 方法平均提升 4.54%,相比 SSKD 提升 1.14%;在 ImageNet 上相比 HASSKD 提升 0.32%。在少样本学习场景下于 Tiny ImageNet 和 CIFAR-100 上进行的实验也取得了最先进结果。这些发现证明了我们的方法无需大型过度参数化教师网络即可增强模型泛化能力和性能的有效性。重要的是,在推理阶段,所有辅助分类器均可移除,不会产生额外计算成本。这使我们的模型适合在经济实惠的低算力设备上部署小型语言模型。由于其轻量级设计和适应性,我们的框架特别适合需要高效、响应迅速推理的多模态感知和信息物理环境。LSSKD 促进了能够在弱监督下从有限感官数据中学习的智能体的开发。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑