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.

Rethinking Positional Encoding in Language Pre-training

A Simple and Effective Positional Encoding for Transformers

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

The Locality and Symmetry of Positional Encodings

Rethinking Position Embedding Methods in the Transformer Architecture

Learning positional encodings in transformers depends on initialization

PoPE: Legendre Orthogonal Polynomials Based Position Encoding for Large Language Models

Rethinking Positional Encoding in Tree Transformer for Code Representation.

Latent Positional Information is in the Self-Attention Variance of Transformer Language Models Without Positional Embeddings

Algebraic Positional Encodings

Eliminating Position Bias of Language Models: A Mechanistic Approach

Technical Report: Impact of Position Bias on Language Models in Token Classification

A Frustratingly Easy Improvement for Position Embeddings Via Random Padding

An Empirical Study on the Impact of Positional Encoding in Transformer-based Monaural Speech Enhancement

Improve Transformer Models with Better Relative Position Embeddings

Rethinking and Improving Relative Position Encoding for Vision Transformer

2D-TPE: Two-Dimensional Positional Encoding Enhances Table Understanding for Large Language Models

Efficient transformer with reinforced position embedding for language models

A Morphology-Based Investigation of Positional Encodings

Attention Alignment and Flexible Positional Embeddings Improve Transformer Length Extrapolation

The Impact of Positional Encoding on Length Generalization in Transformers