Nov, 2022

HeatViT:面向视觉Transformer的硬件高效自适应Token剪枝

TL;DR本文提出了一种硬件高效的图像自适应标记修剪框架HeatViT,以在嵌入式FPGA上实现高效而准确的ViT加速,通过显著重复利用现有硬件组件来实现标记选择器,使用固定点量化和多阶段训练策略来优化插入标记选择器的变压器块,使模型在硬件上不仅提高了准确性和推理延迟,而且还可在与现有计算成本相似的情况下获得更高的准确性或在与相似模型准确性相似的情况下实现更高的计算量减少。