One Transformer Sublayer Wrapped By mHC
→
Hpre · Read
从 n 条残差流里加权读取,聚合成主干层能吃下的 1 条 C 维输入。
xmain = Hpre · xl
→
F · Attention / FFN
主干网络照常工作,不需要知道外面有几条残差流。
y = F(xmain)
→
Hpost · Write
把主干层的新结果按比例广播回 n 条残差流。
Hpost ⊙ y
Hres · 残差内循环混合
行和=1 · 列和=1
Sinkhorn 把原始路由矩阵拉回 Birkhoff polytope,约束车道间交换不整体放大信号。
xl+1 =
Hres · xl
+ Hpost ⊙ F(Hpre · xl)
→