因果格拉斯曼序列建模
Gr(2,r)
⊗
挑战自注意力机制:通过几何流形构建更具可解释性且高效的深度学习框架。
因果格拉斯曼序列建模
Causal Grassmann Sequence Modeling
Source: Attention Is Not What You Need (arXiv:2512.19428)
psychology_alt
核心变革:从张量到流形
传统Transformer的"不可解释性"源于其复杂的张量提升过程(Tensor Lifting)。新架构将词元状态视为低维流形上的几何对象。
传统 Attention
grid_on
• 二次方复杂度 O(L²)
• 成对交互过于密集
• 难以追踪数学规律
• 成对交互过于密集
• 难以追踪数学规律
Grassmann Flow
waves
• 线性复杂度 O(L)
• 局部几何流形映射
• 显式几何不变性
• 局部几何流形映射
• 显式几何不变性
architecture
架构原理:格拉斯曼混合层
通过普吕克坐标(Plücker coordinates)捕捉局部几何特征,信息在低秩子空间中流动。
Input H ∈ ℝL×d → Low-dim Z ∈ ℝL×r → Gr(2, r) Manifold
线性降维
Linear Reduction
arrow_forward
多尺度配对
Multi-scale Pair
arrow_forward
普吕克编码
Plücker Embed
arrow_forward
门控融合
Gated Fusion
verified
核心优势
-
speed
线性计算复杂度
避开注意力机制的 O(L²) 成本,与序列长度呈线性比例,适合长序列建模。 -
insights
显式几何不变性
模型在有限维流形(Grassmannian)上操作,便于数学分析和解释,不再"不可追踪"。 -
memory
高效的信息流
通过低秩子空间的受控变形传播信息,而非简单的权重加权。
bar_chart
实验表现:媲美甚至超越 Transformer
在语言建模和自然语言推理任务上,该架构表现出极强的竞争力。
10-15%
Wikitext-2 困惑度差距
(更接近基线)
(更接近基线)
85.5%
SNLI 准确率
(略优于基线)
(略优于基线)
- 特定分类任务中表现略胜一筹
💬 讨论回复 (1)
推荐
🌟 智谱 GLM-5 已上线
我正在智谱大模型开放平台 BigModel.cn 上打造 AI 应用,智谱新一代旗舰模型 GLM-5 已上线,在推理、代码、智能体综合能力达到开源模型 SOTA 水平。
🎁 领取 2000万 Tokens