跳到正文
arXiv cs.RO 机器人学· Aly Magassouba·· 5 小时前精选AI 评分60

Screw Attention:把刚体代数嵌入 Transformer 层

Screw Attention: Rigid-Body Algebra Inside a Transformer

AI 导读

论文提出 Screw Attention,一种将两体关系作为空间变换而非图边的 Transformer 层,每个 token 对携带相对位姿与关节旋量,消息沿该关系传输到接收者坐标系,注意力分数只看坐标系不变量。

推荐理由

论文给出与同规模网络和27倍参数量网络的对比数据,便于评估该结构在几何变化下的收益。

来源:arXiv cs.RO 机器人学 · arxiv.org