跳到正文
原文
arXiv cs.LG· Zhuolin Fu·· 22 天前AI 评分33

神经期望最大化与元模型:从概率推断视角统一 Transformer 架构

Neural expectation-maximization and meta-models

AI 导读

研究者将 Transformer 架构解释为结合最大乘积环路信念传播(LBP)与期望最大化(EM)的混合概率推断过程。自注意力机制实现了在动态马尔可夫随机场(MRF)上的可微 LBP 近似,而跨网络层级的递进则充当隐式 M 步。基于此视角提出的元模型(meta-models)设计范式,在统一框架下涵盖了自注意力、混合专家(MoE)及门控单元等机制。

来源:arXiv cs.LG · arxiv.org