arXiv:2501.16790stat.MLcs.LG2025-01

提出可处理混合数据类型的动态注意力模型,提升复杂序列建模能力。

Exponential Family Attention

  • 用自注意力学习上下文动态相关性,替代静态隐向量
  • 在温度、购物篮和评分数据上重建精度优于现有模型
  • 适合需要建模高维异构序列的科研与工业场景

自注意力机制是支撑大多数大语言模型的Transformer神经网络的核心。本文提出指数族注意力(EFA),一种概率生成模型,可扩展自注意力以处理高维序列、空间或时空数据,支持离散与连续观测的混合类型。EFA的核心思想是将每个观测建模为对所有已有观测(上下文)的条件分布,其相关性通过基于注意力的潜在因子模型以数据驱动方式学习。不同于静态潜在嵌入,EFA利用自注意力捕捉上下文中的动态交互,其中每个上下文观测的相关性依赖于其他观测。我们建立了可识别性结果,并提供了泛化误差的超额损失保证。在真实世界与合成数据集(包括美国城市气温、Instacart购物篮、MovieLens评分)上,EFA在捕捉复杂潜在结构和重构缺失数据方面始终优于现有模型。

原文摘要 · Abstract (English)

The self-attention mechanism is the backbone of the transformer neural network underlying most large language models. It can capture complex word patterns and long-range dependencies in natural language. This paper introduces exponential family attention (EFA), a probabilistic generative model that extends self-attention to handle high-dimensional sequence, spatial, or spatial-temporal data of mixed data types, including both discrete and continuous observations. The key idea of EFA is to model each observation conditional on all other existing observations, called the context, whose relevance is learned in a data-driven way via an attention-based latent factor model. In particular, unlike static latent embeddings, EFA uses the self-attention mechanism to capture dynamic interactions in the context, where the relevance of each context observations depends on other observations. We establish an identifiability result and provide a generalization guarantee on excess loss for EFA. Across real-world and synthetic data sets -- including U.S. city temperatures, Instacart shopping baskets, and MovieLens ratings -- we find that EFA consistently outperforms existing models in capturing complex latent structures and reconstructing held-out data.

注意力机制生成模型序列建模

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