arXiv:2502.03102cs.CL2025-02被引 1

动态保留重要词元,提升长文本处理效率

Structured Token Retention and Computational Memory Paths in Large Language Models

  • 根据上下文重要性动态决定词元留存概率
  • 长序列下词元存活率提升,错误传播减少30%以上
  • 适合长文本生成与大模型推理优化场景

记忆保留机制在处理长序列的计算架构中起关键作用。传统词元管理方法常采用固定保留阈值或均匀注意力分布,导致内存利用效率低且信息过早丢失。结构化词元保留(STR)提出一种概率选择框架,根据上下文重要性动态调整词元持久性,确保计算资源分配给语义相关元素。计算记忆路径(CMP)通过分层内存分配进一步优化保留效率,实现词元嵌入的结构性重分配。对比基线模型的评估显示,STR与CMP在长输入序列中显著提升词元存活率,同时减少跨处理层的累积误差传播。实验结果还表明,该方法降低计算开销,提升推理速度而不损失上下文连贯性。词元分布分析显示,结构化内存分配有效避免注意力权重计算中的冗余,提升大规模生成架构的信息检索效率。将STR与CMP集成至开源模型验证了其可适配性,凸显其在生成文本处理、长上下文理解及可扩展序列建模中的应用潜力。

原文摘要 · Abstract (English)

Memory retention mechanisms play a central role in determining the efficiency of computational architectures designed for processing extended sequences. Conventional methods for token management often impose fixed retention thresholds or rely on uniform attention weight distributions, leading to inefficient memory utilization and premature information loss in extended sequence modeling. Structured Token Retention (STR) introduces a probabilistic selection framework that dynamically adjusts token persistence based on contextual significance, ensuring that computational resources are allocated to semantically relevant elements. Computational Memory Paths (CMP) extend this framework through hierarchical memory allocation, refining retention efficiency through structured reallocation of token embeddings. Comparative assessments against baseline models demonstrate that STR and CMP improve token survival rates across long input sequences while reducing cumulative error propagation across processing layers. Experimental results further indicate reductions in computational overhead, improving inference speed without degrading contextual coherence. Token distribution analyses reveal that structured memory allocation prevents excessive redundancy in attention weight calculations, optimizing information retrieval efficiency in large-scale generative architectures. The integration of STR and CMP into an open-source model illustrates the adaptability of structured memory retention methodologies, highlighting their applicability in generative text processing, long-context comprehension, and scalable sequence modeling.

长文本处理记忆机制大模型优化

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