aifollow.news
返回 Rohan Paul
Rohan Paul· @rohanpaul_ai · X· · 原发布时间 AI 评分54

腾讯论文提出按神经元贡献压缩混合专家模型

AI 导读

腾讯一篇新论文发现,数学、编程等领域能力分布在不同专家的神经元中,而非集中于单个专家。研究提出按神经元的路由贡献进行剪枝;在6个模型的实验中,保留75%的神经元后,模型保留了原有性能的91%至98%。

原帖引用 · 原文

@rohanpaul_ai

New Tencent paper shows domain skills (like math or coding ) in Mixture of Experts models sit in scattered neurons across expert.

so cutting weak neurons instead of whole experts shrinks the models while keeping most skills.

In their experiment, Mixture of Experts AI models kept 91% to 98% of their performance after their least useful 25% of expert neurons were cut.

MoE models send each token to a few experts, which suggests each expert handles a domain like math or code. Other work finds routing mostly tracks token type instead.

They scored each neuron's routed contribution with a measure called RAMP. Neurons picked for one domain overlapped 58% to 74% across samples of that domain, versus 15% to 24% across domains, and sat in mixed experts.

RAMPr prunes by these scores across all experts and kept 91% to 98% of original performance with 75% of neurons left, across 6 models. With 50% left, GPU memory fell 36% to 43% and batched prefill throughput rose 17% to 39%.

If you compress MoE models, prune neurons inside experts by their routed contribution instead of treating each expert as a single unit.

来源:Rohan Paul · x.com

论文
发现内容有误?提交纠错