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

引导稀疏解码器潜在特征以控制动态头部剪枝在视觉变换器中(学生摘要)

Steering Sparse Autoencoder Latents to Control Dynamic Head Pruning in Vision Transformers (Student Abstract)

Yousung Lee, Dongsoo Har

AI总结:

本文提出结合稀疏解码器与动态剪枝的新框架,通过稀疏潜在特征实现头部剪枝的可解释和可控,提升ViTs的效率和可解释性。

摘要:

动态头部剪枝在视觉变换器(ViTs)中通过移除冗余注意力头提高效率,但现有剪枝策略往往难以解释和控制。本文提出一种新框架,将稀疏解码器(SAEs)与动态剪枝结合,利用其能力将密集嵌入解耦为可解释且可控的稀疏潜在特征。具体而言,我们训练SAE在ViT的最终层残差嵌入上,并通过不同策略放大稀疏潜在特征以改变剪枝决策。其中,按类别引导揭示了紧凑、类特定的头部子集,例如碗在减少头部使用(0.72到0.33)的同时提高精度(76%到82%)通过头部h2和h5。这些结果表明,稀疏潜在特征使动态剪枝能够实现类特定的控制,有效弥合了ViTs中的剪枝效率和机制可解释性。

英文摘要:

Dynamic head pruning in Vision Transformers (ViTs) improves efficiency by removing redundant attention heads, but existing pruning policies are often difficult to interpret and control. In this work, we propose a novel framework by integrating Sparse Autoencoders (SAEs) with dynamic pruning, leveraging their ability to disentangle dense embeddings into interpretable and controllable sparse latents. Specifically, we train an SAE on the final-layer residual embedding of the ViT and amplify the sparse latents with different strategies to alter pruning decisions. Among them, per-class steering reveals compact, class-specific head subsets that preserve accuracy. For example, bowl improves accuracy (76% to 82%) while reducing head usage (0.72 to 0.33) via heads h2 and h5. These results show that sparse latent features enable class-specific control of dynamic pruning, effectively bridging pruning efficiency and mechanistic interpretability in ViTs.

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

↑