研究人员开发了LoKA,一个旨在使低精度算术(特别是FP8)在大型推荐模型(LRMs)中实用的框架。与LLMs不同,LRMs对数值精度敏感,当直接应用FP8时,质量会下降或训练时间延长。LoKA通过系统-模型协同设计方法解决了这个问题,包括通过分析来识别安全的低精度使用,调整模型组件以提高稳定性和效率,以及使用运行时来选择满足精度要求的最快FP8内核。 AI
影响 能够更有效地训练推荐模型,可能带来更快速的个性化AI服务的开发和部署。
排序理由 该集群包含一篇学术论文,详细介绍了用于优化AI模型训练的新技术框架。[lever_c_demoted from research: ic=1 ai=1.0]
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →