通过共享二值原型旋转实现专家压缩,内存仅随专家数线性增长。
ButterflyMoE: Compression-Scalable Ternary Experts via Structured Butterfly Orbits
- 用共享二值原型加旋转参数化专家,避免独立存储矩阵。
- 8专家时内存压缩80倍,256专家时达150倍,精度仍超稠密基线。
- 适合边缘设备部署,尤其适用于低比特训练与大规模扩展场景。
当前混合专家(MoE)架构存在线性内存膨胀问题:N个独立专家权重矩阵需$/mathcal{O}(N ullet d^2)$内存,超出边缘设备预算。现有压缩方法如量化、剪枝和低秩分解仅降低常数因子,无法解决缩放瓶颈。本文提出ButterflyMoE,将专家视为共享量化基底的几何旋转,专家多样性来自视角差异而非冗余存储。通过学习旋转作用于共享二值原型,每个专家内存从$/mathcal{O}(d^2)$降至$/mathcal{O}(d ullet /log d)$,总内存为$/mathcal{O}(d^2 + N ullet d ullet /log d)$。关键发现:量化训练旋转可抑制激活异常,稳定极低比特训练。在语言建模基准上,8专家时实现80倍内存压缩,且性能优于同等内存的稠密模型;扩展至256专家时内存压缩达150倍。该方法使线性缩放的常数因子随专家数增加而减小。
原文摘要 · Abstract (English)
In current Mixture of Experts (MoE) architectures, linear memory scaling is present, the memory grows as the number of experts increases. $N$ independent expert weight matrices require $\mathcal{O}(N \cdot d^2)$ memory which exceeds the memory budget of edge devices. Current compression methods like quantization, pruning, and low-rank factorization reduce constant factors, but the scaling bottleneck is still unresolved. We introduce ButterflyMoE, a method which treats experts not as independent matrices but as geometric reorientations of a shared quantized substrate. Diversity amongst the experts arises from viewing different angles of the shared capacity and not from redundant storage. Learned rotations are applied to a shared ternary prototype. With this, each expert yields $\mathcal{O}(d^2 + N \cdot d \log d)$ memory-reducing per-expert cost from $\mathcal{O}(d^2)$ to $\mathcal{O}(d \log d)$. The key insight is that training these rotations with quantization reduces activation outliers and stabilizes extreme low-bit training where other static methods collapse. Across language modeling benchmarks, ButterflyMoE achieves 80$\times$ memory reduction at 8 experts with a highly favorable memory-accuracy tradeoff.At this 80x compression ButterflyMoE outperforms an equal memory dense baseline, showing that orbital parameterization extracts fundamentally more utility per byte. When scaled up to 256 experts, ButterflyMoE asymptotically compresses the memory by 150 $\times$. ButterflyMoE reduces the constant factor of linear scaling with compression ratio growing with the expert count.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。