UniAttn: Reducing Inference Costs via Softmax Unification for Post-Training LLMs
UniAttn: 通过Softmax统一降低推理成本以适应训练后的LLM
机构 * School of Software, Tsinghua University(清华大学软件学院) ; School of Computer Science, Beijing University of Posts and Telecommunications(北京邮电大学计算机学院) ; Beihang University(北航) ; Department of Automation, Tsinghua University(清华大学自动化系)
专题命中 效率与部署 :post-training(title,abstract);LLM(abstract);large language model(abstract);language model(abstract)
AI总结 UniAttn通过统一Softmax操作降低训练后LLM的推理成本,同时保持性能。
Comments 8 pages, 6 figures. Preprint, under review