用几何旋转共享参数,让边缘设备跑354倍压缩的视觉专家模型。
ButterflyViT: 354$\times$ Expert Compression for Edge Vision Transformers
- 多个专家共用一个量化原型,通过旋转实现差异,打破线性内存瓶颈。
- 在CIFAR-100上64个专家时内存减少354倍,精度几乎无损失。
- 适合部署在内存受限的边缘设备上的稀疏视觉专家模型。
将稀疏专家混合(MoE)视觉变换器部署到边缘设备仍面临挑战,因其专家内存呈线性增长。传统方法如量化、剪枝和低秩分解仅降低常数因子,无法解决缩放瓶颈。我们提出ButterflyViT,将专家视为统一量化基底的几何旋转变体,而非独立权重矩阵。专家多样性源于对共享容量的不同视角,而非冗余存储。通过学习旋转一个共享三值原型,每个专家内存为$\ ext{O}(d_{\text{model}} \cdot d_{\text{ff}} + N_E \cdot n_\\ell \cdot d)$,在专家数量上呈次线性。针对视觉任务特性,引入空间平滑正则项,惩罚相邻图像块令牌间的路由不规则性,将块相关性转化为训练信号。在CIFAR-100图像分类任务中,ButterflyViT在64个专家下实现354倍内存压缩,精度损失可忽略。该方法证明几何参数化可突破线性缩放限制,使多个专家能在边缘设备上运行。
原文摘要 · Abstract (English)
Deploying sparse Mixture of Experts(MoE) Vision Transformers remains a challenge due to linear expert memory scaling. Linear memory scaling stores $N$ independent expert weight matrices requiring $\mathcal{O}(N_E \cdot d^2)$ memory, which exceeds edge devices memory budget. Current compression methods like quantization, pruning and low-rank factorization reduce constant factors but leave the scaling bottleneck unresolved. We introduce ButterflyViT, a method that treats experts not as independent weight matrices but as geometric reorientations of a unified shared quantized substrate. Diversity among experts arises from viewing different angles of shared capacity, not from redundant storage. By applying learned rotations to a shared ternary prototype, each expert yields $\mathcal{O}(d_{\text{model}} \cdot d_{\text{ff}} + N_E \cdot n_\ell \cdot d)$ memory which is sub-linear in the number of experts. To address the unique challenges of vision, a spatial smoothness regulariser is introduced that penalises routing irregularities between adjacent patch tokens, turning patch correlation into a training signal. Across image classification tasks on CIFAR-100, ButterflyViT achieves 354$\times$ memory reduction at 64 experts with negligible accuracy loss. ButterflyViT allows multiple experts to fit on edge-constrained devices showing that geometric parameterization breaks linear scaling.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。