研究人员开发了HiFA4,一种在Ascend HIF4 NPU上以4位执行FlashAttention操作的新型训练后设计,旨在提高LLM推理效率。该方法结合了两种关键机制:用于重新缩放注意力权重的Smooth-QK和用于累积softmax归一化器的P-Reordering。在包括Qwen3-8B和Gemma2-9B在内的五个LLM上的评估表明,HiFA4显著降低了量化引起的准确性回归和决策漂移,并在MMLU得分方面取得了显著改进。 AI
影响 这项研究通过在不显著损失准确性的情况下实现低比特量化,有望在专用硬件上实现更高效的LLM推理。
排序理由 该集群描述了在arXiv上的一篇学术论文中提出的一种新颖技术方法。
- Ascend HIF4 NPUs
- Flashattention
- gemma2:9b
- HiFA4
- llama3.1:8b
- Massive Multitask Language Understanding
- mistral:7b
- Phi-4B
- P-Reordering
- Qwen3_8B
- Smooth-QK
AI 生成摘要 · Google Gemini · 来自 2 个来源。 我们如何撰写摘要 →