Screw Attention: Rigid-Body Algebra Inside a Transformer

Screw Attention introduces a transformer layer that treats the relation between two bodies as a spatial transform rather than a graph edge, enabling each token to represent a body with a pose and its relative pose or joint screw. This design ensures equivariance to independent frame changes and allows a single layer to capture rigid‑body velocity recursion. Experiments on simulated manipulation tasks show that Screw Attention matches or outperforms other network architectures, achieving high success rates on LIBERO‑Spatial with far fewer parameters and maintaining performance under frame convention changes and pose noise.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv AI.

• llms
• robotics