arXiv:2601.02819cs.CL2026-01

用标点符号做锚点,让大模型长文本注意力更高效精准

Punctuation-aware Hybrid Trainable Sparse Attention for Large Language Models

论文配图:Punctuation-aware Hybrid Trainable Sparse Attention for Large Language Models
图 1 · 摘自论文原文
  • 以标点作为语义边界,动态选择关键注意力块
  • 32k上下文下稀疏度97.3%时信息损失减少10.8%
  • 适合需要高效处理长文本的模型部署场景

注意力是大语言模型长序列建模的核心机制,但密集注意力因二次复杂度在长序列下不可行。稀疏注意力成为可扩展替代方案,但现有方法依赖粗粒度语义表示进行块选择,模糊了块内语义边界,导致关键信息丢失。为此,我们提出一种原生可训练的稀疏注意力框架——标点感知混合稀疏注意力(PHSA),利用标点符号作为语义边界锚点。具体而言:(1) 设计双分支聚合机制,融合全局语义表示与标点增强的边界特征,在几乎不增加计算开销的前提下保留核心语义结构;(2) 引入极稀疏适应性训练与推理策略,确保极低激活率下的模型稳定性。大量实验表明,PHSA在通用基准和长序列评估中持续优于密集注意力及当前最优稀疏注意力基线(包括InfLLM v2)。以0.6B参数、32k输入序列为例,稀疏度达97.3%时,信息损失降低10.8%。

原文摘要 · Abstract (English)

Attention serves as the fundamental mechanism for long-context modeling in large language models (LLMs), yet dense attention becomes structurally prohibitive for long sequences due to its quadratic complexity. Consequently, sparse attention has received increasing attention as a scalable alternative. However, existing sparse attention methods rely on coarse-grained semantic representations during block selection, which blur intra-block semantic boundaries and lead to the loss of critical information. To address this issue, we propose \textbf{P}unctuation-aware \textbf{H}ybrid \textbf{S}parse \textbf{A}ttention \textbf{(PHSA)}, a natively trainable sparse attention framework that leverages punctuation tokens as semantic boundary anchors. Specifically, (1) we design a dual-branch aggregation mechanism that fuses global semantic representations with punctuation-enhanced boundary features, preserving the core semantic structure while introducing almost no additional computational overhead; (2) we introduce an extreme-sparsity-adaptive training and inference strategy that stabilizes model behavior under very low token activation ratios; Extensive experiments on general benchmarks and long-context evaluations demonstrate that PHSA consistently outperforms dense attention and state-of-the-art sparse attention baselines, including InfLLM v2. Specifically, for the 0.6B-parameter model with 32k-token input sequences, PHSA can reduce the information loss by 10.8\% at a sparsity ratio of 97.3\%.

稀疏注意力长文本标点感知

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。