Impact of Positional Encoding: Clean and Adversarial Rademacher Complexity for Transformers under In-Context Regression
By: Weiyi He, Yue Xing
Positional encoding (PE) is a core architectural component of Transformers, yet its impact on the Transformer's generalization and robustness remains unclear. In this work, we provide the first generalization analysis for a single-layer Transformer under in-context regression that explicitly accounts for a completely trainable PE module. Our result shows that PE systematically enlarges the generalization gap. Extending to the adversarial setting, we derive the adversarial Rademacher generalization bound. We find that the gap between models with and without PE is magnified under attack, demonstrating that PE amplifies the vulnerability of models. Our bounds are empirically validated by a simulation study. Together, this work establishes a new framework for understanding the clean and adversarial generalization in ICL with PE.
Similar Papers
SeqPE: Transformer with Sequential Position Encoding
Machine Learning (CS)
Helps AI understand longer texts and images.
A Comparative Study on Positional Encoding for Time-frequency Domain Dual-path Transformer-based Source Separation Models
Audio and Speech Processing
Separates sounds better, but only for short ones.
Tab-PET: Graph-Based Positional Encodings for Tabular Transformers
Machine Learning (CS)
Helps computers learn better from messy data.