Warning: Undefined variable $main_title in /www/wwwroot/msbd/wp-content/themes/onenav/sidebar-sitestop.php on line 26
AI交流(进群备注:)

该项目利用2:4稀疏性模式,通过硬件加速的GPU稀疏性来优化Transformer模型的训练和推理过程。特别针对Squared-ReLU激活函数,该函数天然具有高稀疏性(84-98%),在不损失准确性的情况下,前向和后向传播中的前馈网络(FFN)计算速度提高高达1.3倍。项目通过自定义内核高效处理稀疏操作,并通过张量分割和令牌排列优化后向传播,展示了稀疏性在加速大型语言模型(LLM)中的关键潜力。
2:4 Activation Sparsity Accelerating Transformer Inference and Training的特点:
- 1. 使用2:4稀疏模式实现GPU加速
- 2. 特别针对Squared-ReLU激活,天然具有高稀疏性(84-98%)
- 3. 加速Transformer模型的训练和推理过程
- 4. 前向和后向传播中的FFN层速度提高高达1.3倍
- 5. 无准确性损失,保持与密集模型相当的性能
- 6. 实现自定义内核以高效处理稀疏操作
- 7. 通过张量分割和令牌排列优化后向传播
2:4 Activation Sparsity Accelerating Transformer Inference and Training的功能:
- 1. 在PyTorch中实现,使用torchao库支持稀疏操作
- 2. 需要修改模型以使用Squared-ReLU激活和稀疏FFN层
- 3. 最佳性能需在支持2:4稀疏性的硬件上,如NVIDIA H100 GPU
相关导航

该项目研究揭示了o1类大型语言模型(LLMs)在复杂推理任务中存在的‘欠思考’现象,即模型倾向于频繁切换不同推理思路而未能深入探索有前景的路径。研究通过三个挑战性测试集(MATH500、GPQA Diamond和AIME)和两个代表性开源o1类模型(QwQ-32B-Preview和DeepSeek-R1-671B)的实验,系统分析了这一问题。研究发现错误回答比正确回答使用更多令牌且思路切换频率更高,超过70%的错误回答包含至少一个正确思路。为量化欠思考,研究引入了ξ_UT指标,并提出‘思路切换惩罚(TIP)’解码策略,通过阻止过早的思路转换来提升模型在挑战性数据集上的准确率。实验结果显示TIP策略显著提高了模型性能,且无需微调模型。

Diverse Preference Optimization (DivPO) 是一种在线优化方法,旨在解决大型语言模型(LLM)在后训练阶段普遍存在的多样性坍缩问题。通过在偏好优化过程中引入多样性考量,DivPO 能够生成更多样化的回复,同时保持生成内容的质量。该项目由 Jack Lanchantin, Angelica Chen, Shehzaad Dhuliawala, Ping Yu, Jason Weston, Sainbayar Sukhbaatar, Ilia Kulikov 在 2025 年发表,特别适用于需要多样性的创意生成任务,如故事生成和人物属性生成。