提出噪声扩散对称注意力架构,提升大模型性能与效率。
NoiseFormer -- Noise Diffused Symmetric Attention Transformer
- 引入噪声扩散机制优化对称点积注意力,增强表示能力。
- 在GLUE基准上准确率介于原始GPT2与对称注意力之间,模型尺寸显著缩小。
- 适合追求高效推理的大型语言模型部署场景。
Transformer架构因其基于注意力的学习能力和并行化特性,在深度学习和大语言模型领域表现卓越。随着模型规模扩大,内存占用增加,难以在单个GPU或AI加速器上运行,需依赖多设备协同,导致计算成本上升。为此,稀疏注意力技术成为降低模型参数量的重要手段。本文分析了稀疏注意力中的一种方法——对称点积注意力(简称对称注意力),提出一种新型统一架构:噪声扩散对称注意力Transformer(NoiseFormer),在保持对称注意力内存优势的同时,仅引入少量参数与计算开销,显著提升模型精度与推理采样效率。实验基于GPT2 base模型,在多个GLUE基准任务上验证,性能优于纯对称注意力,接近但未超过GPT2 base,且模型规模大幅缩减。
原文摘要 · Abstract (English)
Transformer architecture has been very successful long runner in the field of Deep Learning (DL) and Large Language Models (LLM) because of its powerful attention-based learning and parallel-natured architecture. As the models grow gigantic in terms of memory footprint, difficulties in fitting the model on a device like a GPU or an AI accelerator give rise to the need for multiple computing devices thereby escalating the computing cost. This increased training/inference cost paved the way for efficient model size reduction/parametric reduction deploying Sparse Attention techniques. In this paper, we start analyzing one of the techniques of Sparse Attention called Symmetric Dot-Product Attention (referred to as Symmetric Attention) and propose a novel unified model architecture called Noise Diffused Symmetric Attention Transformer to enhance the model's performance. While maintaining the memory gains of Symmetric Attention, with minute overhead in terms of model parameters and computational overhead, the proposed model brings in enhanced performance in terms of accuracy and inference-time sampling. The proposed model is validated upon GPT2 base model and the results reflect the performance gains falling between plain Symmetric attention and GPT2 base model on a variety of GLUE benchmark tasks in terms of accuracy, with significant model size reduction with respect to the base model.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。