Abstract:The attention mechanism is central to the transformer's ability to capture complex dependencies between tokens of an input sequence. Key to the successful application of the attention mechanism in transformers is its choice of positional encoding (PE). The PE provides essential information that distinguishes the position and order amongst tokens in a sequence. Most prior investigations of PE effects on generalization were tailored to 1D input sequences, such as those presented in natural language, where adjacent tokens (e.g., words) are highly related. In contrast, many real world tasks involve datasets with highly non-trivial positional arrangements, such as datasets organized in multiple spatial dimensions, or datasets for which ground truth positions are not known, such as in biological data. Here we study the importance of learning accurate PE for problems which rely on a non-trivial arrangement of input tokens. Critically, we find that the choice of initialization of a learnable PE greatly influences its ability to discover accurate PEs that lead to enhanced generalization. We empirically demonstrate our findings in a 2D relational reasoning task and a real world 3D neuroscience dataset, applying interpretability analyses to verify the learning of accurate PEs. Overall, we find that a learned PE initialized from a small-norm distribution can 1) uncover interpretable PEs that mirror ground truth positions, 2) learn non-trivial and modular PEs in a real-world neuroscience dataset, and 3) lead to improved downstream generalization in both datasets. Importantly, choosing an ill-suited PE can be detrimental to both model interpretability and generalization. Together, our results illustrate the feasibility of discovering accurate PEs for enhanced generalization.

SHAPE: Shifted Absolute Position Embedding for Transformers

A Simple and Effective Positional Encoding for Transformers

Improve Transformer Models with Better Relative Position Embeddings

Improving Transformers using Faithful Positional Encoding

TransPoser: Transformer as an Optimizer for Joint Object Shape and Pose Estimation

The Impact of Positional Encoding on Length Generalization in Transformers

Rethinking Position Embedding Methods in the Transformer Architecture

Bidirectional Transformer with Absolute-Position Aware Relative Position Encoding for Encoding Sentences

Improving Position Encoding of Transformers for Multivariate Time Series Classification

Geometry-aware 3D pose transfer using transformer autoencoder

Algebraic Positional Encodings

Disentangling Patterns and Transformations from One Sequence of Images with Shape-invariant Lie Group Transformer

Shifted Chunk Transformer for Spatio-Temporal Representational Learning

GridPE: Unifying Positional Encoding in Transformers with a Grid Cell-Inspired Framework

A bio-inspired positional embedding network for transformer-based models

Rethinking and Improving Relative Position Encoding for Vision Transformer

What Do Position Embeddings Learn? An Empirical Study of Pre-Trained Language Model Positional Encoding

PT43D: A Probabilistic Transformer for Generating 3D Shapes from Single Highly-Ambiguous RGB Images

Position Coupling: Improving Length Generalization of Arithmetic Transformers Using Task Structure

Rethinking Positional Encoding in Language Pre-training

Learning positional encodings in transformers depends on initialization