arrow
返回

Efficient knowledge distillation using a shift window target-aware transformer

delete2024-12-24
delete0
PRE
AI
冯
冯敬 (Jing Feng)
W
Wen Eng Ong *
DOI:10.1007/s10489-024-06207-1delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Target-aware Transformer (TaT) knowledge distillation effectively extracts information from intermediate layers but faces high computational costs for large feature maps. While the non-overlapping Patch-group distillation in TaT reduces complexity, it loses boundary information, affecting accuracy. We propose an improved Shifted Windows Target-aware Transformer (Swin TaT) knowledge distillation method, utilizing a hierarchical shift window strategy to preserve boundary information and balance computational efficiency. Our multi-scale approach optimizes Patch-group distillation with dynamic adjustment, ensuring effective local and global feature transfer. This flexible and efficient design enhances distillation performance, addressing previous limitations. The proposed Swin TaT method demonstrates exceptional performance across various architectures, with ResNet18 as the student network. It achieves 73.03% Top-1 accuracy on ImageNet1K, surpassing the SOTA by 1.06% while reducing parameters to approximately 46% less, and improves mIoU by 2.13% on COCOStuff10k.
Keyword:
Knowledge distillation
Target-aware Transformer (TaT)
Shifted window
Patch-group distillation

期刊

Applied Intelligence 封面图
Applied Intelligence
IF:
3.5
论文数:
7.6K
被引数:
1.7W

机构

U
Universiti Sains Malaysia
学者数:
1.5W
论文数: 1.3W
被引数: 131
引用论文

引用论文

err分享
err收藏
Knowledge Distillation: A Survey知识蒸馏: 一项调查
err2021-03-22
err1.5K
PREAI
errGou, Jianping; Yu, Baosheng; Maybank, Stephen J.; Tao, Dacheng
err分享
err收藏
err分享
err收藏
Organic acids can crosslink poly(ionic liquid)s into mesoporous polyelectrolyte complexes
err2013-01-01
err0
errOAAI
errQiang Zhao; Sebastian Soll; Markus Antonietti; Jiayin Yuan
err分享
err收藏
学者 查看更多内容