AI消息速览

Lévy Attention:单次前向传播即可获得连续时间注意力机制的不确定性估计

事件日期 2026-08-19 · 学术前沿 · 已接受

事件日期2026-08-19
信息日期2026-08-19
入库日期2026-08-20
通道学术前沿
状态已接受
来源arXiv 论文

知识卡片:Lévy Attention:单次前向传播即可获得连续时间注意力机制的不确定性估计

  • 一句话结论:Lévy Attention 将注意力层重新表述为关于非齐次泊松随机测度的随机积分,使模型在单次前向传播中产生预测的同时,以闭式解形式输出无需额外训练头的不确定性估计。

  • 事件概述或研究问题:面向不规则采样时间序列的深度模型可以在任意连续时间戳回答查询,但并未报告每个答案应被信任多远。论文提出在注意力层内部直接补齐这一缺口,而不是依赖外部不确定性模块。

  • 方法/产品要点:

    • 提出 Lévy Attention 交叉注意力算子,其输出是对一个非齐次泊松随机测度的随机积分。
    • 查询-键兼容性在连续的(时间×通道)索引空间上构成强度,测度在该强度下散布原子,输出对这些原子处的插值值场取平均。
    • 在期望上,该算子等价于一种 mollified cosine-kernel attention(平滑余弦核注意力),因此可替换 softmax 层,并用精确梯度训练。
    • 泊松构造以闭式保留了 softmax 丢弃的证据量 $\Lambda_q$(总兼容性质量)与分歧 $\mathrm{tr},\Sigma_V(q)$(值散布)。
    • 通过精确方差恒等式,二者组合为 $\hat{\sigma}(q)=\sqrt{\mathrm{tr},\Sigma_V(q),\varphi(\Lambda_q)}$,即采样算子的均方根偏差,由确定性前向传播直接发出,无需训练头。
  • 主要结果或产业意义:

    • 经验结果显示,分歧项携带主要信号;证据因子在密集数据上几乎无信息,在稀疏数据上则具有很强的信息性。
    • 在 t-PatchGNN 上替换该算子,与匹配对照相比最多损失 5.6% 准确率,在最稀疏数据集上没有损失。
    • 免费获得的分歧信号在匹配的五种子套件上优于 20 次 MC dropout。
    • $\hat{\sigma}$ 可缩放出一个校准的高斯分布,其零样本 CRPS 优于五十次抽取的采样器;split-conformal 包装达到名义覆盖率。
    • 单次前向传播在 1.4 秒内对 3,383 名未见过患者按可信度排序。
  • 为什么重要:现有深度模型通常需要多次采样(如 MC dropout)或额外的不确定性头才能获得不确定性估计;Lévy Attention 将不确定性量化嵌入注意力算子本身,以单次前向传播的代价提供闭式不确定性。这对连续时间序列预测以及医疗等高风险排序场景尤为有价值。与已有相关卡片中图像生成、风格迁移、真实-虚假分类的主题相比,本条聚焦于连续时间注意力模型的内在不确定性估计,属于不同的增量方向。

  • 局限与不确定性:目前仅有论文摘要与元数据;具体实验设置、基线细节、代码复现性以及注意力算子替换在其他架构上的通用性仍需核实。论文中“成本至多 5.6% 准确率”“零样本 CRPS 优于 50 次采样”等表述均为摘要中的原始结果,尚未经过独立验证。

  • 可用于图书/PPT/简报的角度:

    • 用“注意力机制如何自己知道‘不知道’”来介绍本工作。
    • 对比“单次前向传播”与“20 次 MC dropout”的不确定性估计成本。
    • 强调不确定性从“附加模块”变为“结构内禀属性”的范式变化。
    • 以“1.4 秒排序 3,383 名患者”展示高效可信度排序的落地潜力。
  • 原始材料:

    • 英文标题:Lévy Attention: Single-Pass Predictive Uncertainty for Continuous-Time Attention
    • 英文关键词(据摘要提取):continuous-time attention; predictive uncertainty; Poisson random measure; irregularly-sampled time series; single-pass uncertainty
    • 来源:arXiv:2608.19171v1 [cs.LG]
    • URL: https://arxiv.org/abs/2608.19171v1
    • PDF: https://arxiv.org/pdf/2608.19171v1
    • 作者:Sotirios P. Chatzis, Loukas Papadoulas
    • 提交/更新日期:2026-08-19