跳到正文
arXiv · cs.AI· Walter Simoncini·· 12 天前精选AI 评分53

arXiv 论文提出免重训练方法让预训练 Transformer 可按算力预算弹性伸缩

Your Transformer Is Secretly Elastic

来源与原文

arXiv · cs.AI

作者:Walter Simoncini

发布时间:(北京时间)

阅读原始来源 · arxiv.org ↗
完整链接与内容说明

https://arxiv.org/abs/2610.20588v1

本站目前仅有 AI 导读与摘要,未展示原文正文;技术细节、代码和完整结论请查看原始来源。

AI 导读

Walter Simoncini 在 arXiv 发布论文,提出免重训练方法把预训练 Transformer 变成可在连续算力预算间自适应的单一模型,核心是对功能单元排序问题做可微松弛,在统一尺度上联合排序每个 MLP 神经元和注意力头。在 22M 到 8B 参数的 27 个视觉与文本 Transformer 编码器上,方法在最高 60% 稀疏度下平滑退化并优于 one-shot 基线;DINOv3 ViT-H+/16 剪枝到 50% 稀疏度移除 427M 参数,ImageNet-1k linear-probing 精度仅降 0.8%;AugReg ViT-B/16 在 60% 稀疏度下于 A100 GPU 上吞吐提升最高 1.80 倍、单查询延迟降低 1.26 倍。

推荐理由

论文给出免重训练的弹性剪枝方法,在27个模型上验证60%稀疏度下的精度与吞吐数据,可复现。

来源:arXiv · cs.AI · arxiv.org