paper-with-me

홈 › Papers

A Length-Extrapolatable Transformer

2022-12-20 · Yutao Sun, Li Dong, Barun Patra, Shuming Ma, Shaohan Huang, Alon Benhaim, Vishrav Chaudhary, Xia Song, Furu Wei

Position modeling plays a critical role in Transformers. In this paper, we focus on length extrapolation, i.e., training on short texts while evaluating longer sequences. We define attention resolution as an indicator of extrapolation. Then we propose two designs to improve the above metric of Transformers. Specifically, we introduce a relative position embedding to explicitly maximize attention resolution. Moreover, we use blockwise causal attention during inference for better resolution. We evaluate different Transformer variants with language modeling. Experimental results show that our model achieves strong performance in both interpolation and extrapolation settings. The code will be available at https://aka.ms/LeX-Transformer.

📄 PDF Abstract BibTeX arXiv:2212.10554

Code (5)

microsoft/torchscale 공식 구현 pytorch
abacusai/long-context pytorch
conceptofmind/palm pytorch
fkodom/yet-another-retnet pytorch
lucidrains/rotary-embedding-torch pytorch

Tasks

Language ModelingLanguage ModellingPosition

Methods 이 논문이 사용한 방법론

Multi-Head Attention 설명 없음
Attention 설명 없음
Linear Layer A Linear Layer is a projection $\mathbf{XW + b}$.
BPE Byte Pair Encoding, or BPE, is a subword segmentation algorithm that encodes rare and unknown words as sequences of subword units. The intuition is that various word…
Dense Connections Dense Connections, or Fully Connected Connections, are a type of layer in a deep neural network that use a linear operation where every input is connected to every output…
Residual Connection 설명 없음
Label Smoothing Label Smoothing is a regularization technique that introduces noise for the labels. This accounts for the fact that datasets may have mistakes in them, so maximizing the…
Position-Wise Feed-Forward Layer 설명 없음

Similar Papers 제목 키워드 기반

LEDiT: Your Length-Extrapolatable Diffusion Transformer without Positional Encoding

2025-03-06 · Shen Zhang, Yaning Tan, Siyuan Liang, Linze Li 외

Diffusion transformers(DiTs) struggle to generate images at resolutions higher than their training resolutions. The primary obstacle is that the explicit positional encodings(PE), such as RoPE, need extrapolation which d…

Dissecting Transformer Length Extrapolation via the Lens of Receptive Field Analysis

2022-12-20 · Ta-Chung Chi, Ting-Han Fan, Alexander I. Rudnicky, Peter J. Ramadge

Length extrapolation permits training a transformer language model on short sequences that preserves perplexities when tested on substantially longer sequences. A relative positional embedding design, ALiBi, has had the …

Language ModelingLanguage Modelling

Attention Alignment and Flexible Positional Embeddings Improve Transformer Length Extrapolation

2023-11-01 · Ta-Chung Chi, Ting-Han Fan, Alexander I. Rudnicky

An ideal length-extrapolatable Transformer language model can handle sequences longer than the training length without any fine-tuning. Such long-context utilization capability relies heavily on a flexible positional emb…

Code CompletionLanguage ModelingLanguage ModellingQuestion Answering+1

Length Extrapolation of Transformers: A Survey from the Perspective of Positional Encoding

2023-12-28 · Liang Zhao, Xiachong Feng, Xiaocheng Feng, Weihong Zhong 외

Built upon the Transformer, large language models (LLMs) have captured worldwide attention due to their remarkable abilities. Nevertheless, all Transformer-based models including LLMs suffer from a preset length limit an…

PositionSurvey

TimelyGPT: Extrapolatable Transformer Pre-training for Long-term Time-Series Forecasting in Healthcare

2023-11-29 · Ziyang Song, Qincheng Lu, Hao Xu, He Zhu 외

Large-scale pre-trained models (PTMs) such as BERT and GPT have recently achieved great success in Natural Language Processing and Computer Vision domains. However, the development of PTMs on healthcare time-series data …

DiagnosticTime SeriesTime Series ForecastingTrajectory Prediction