DINT Transformer

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Cang, Yueyang, Liu, Yuhang, Zhang, Xiaoteng, Zhao, Erlu, Shi, Li
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866916588707381248
author Cang, Yueyang
Liu, Yuhang
Zhang, Xiaoteng
Zhao, Erlu
Shi, Li
author_facet Cang, Yueyang
Liu, Yuhang
Zhang, Xiaoteng
Zhao, Erlu
Shi, Li
contents DIFF Transformer addresses the issue of irrelevant context interference by introducing a differential attention mechanism that enhances the robustness of local attention. However, it has two critical limitations: the lack of global context modeling, which is essential for identifying globally significant tokens, and numerical instability due to the absence of strict row normalization in the attention matrix. To overcome these challenges, we propose DINT Transformer, which extends DIFF Transformer by incorporating a differential-integral mechanism. By computing global importance scores and integrating them into the attention matrix, DINT Transformer improves its ability to capture global dependencies. Moreover, the unified parameter design enforces row-normalized attention matrices, improving numerical stability. Experimental results demonstrate that DINT Transformer excels in accuracy and robustness across various practical applications, such as long-context language modeling and key information retrieval. These results position DINT Transformer as a highly effective and promising architecture.
format Preprint
id arxiv_https___arxiv_org_abs_2501_17486
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle DINT Transformer
Cang, Yueyang
Liu, Yuhang
Zhang, Xiaoteng
Zhao, Erlu
Shi, Li
Computation and Language
Artificial Intelligence
Machine Learning
DIFF Transformer addresses the issue of irrelevant context interference by introducing a differential attention mechanism that enhances the robustness of local attention. However, it has two critical limitations: the lack of global context modeling, which is essential for identifying globally significant tokens, and numerical instability due to the absence of strict row normalization in the attention matrix. To overcome these challenges, we propose DINT Transformer, which extends DIFF Transformer by incorporating a differential-integral mechanism. By computing global importance scores and integrating them into the attention matrix, DINT Transformer improves its ability to capture global dependencies. Moreover, the unified parameter design enforces row-normalized attention matrices, improving numerical stability. Experimental results demonstrate that DINT Transformer excels in accuracy and robustness across various practical applications, such as long-context language modeling and key information retrieval. These results position DINT Transformer as a highly effective and promising architecture.
title DINT Transformer
topic Computation and Language
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2501.17486