Advertisement

ViT Slimming-联合结构搜索与Patch Selection

阅读量:

论文获取地址:https://arxiv.org/abs/2201.00814

项目代码仓库链接:https://github.com/Arnav0400/ViT-Slim

Methods

ViT Slimming作为一种简洁且高效的模型简化策略, 通过融合结构搜索与Patch选择机制,不仅实现了在多个维度和尺度上的模型压缩,同时有效降低了Patch或Token的长度冗余,进而显著减少模型参数数量及计算复杂度。具体实施过程中,在ViT架构中对流动的Tensor引入对应的Soft mask,并在计算时将其进行乘法操作,同时在损失函数中加入对Soft mask的L1正则化约束项:

在此过程中

![](https://ad.itadn.com/c/weblog/blog-img/images/2025-05-31/U6n8rDxTl90ONi4M2

全部评论 (0)

还没有任何评论哟~