AMD发布 AI小语言模型:6900亿token、推测解码提速3.88倍
来源:快科技 发表于2024-10-01 17:20:08 编辑:时寒峰

  原标题:AMD发布 AI小语言模型:6900亿token、推测解码提速3.88倍

  10月1日消息,AMD发布了自己的 小语言模型(SLM),名为“AMD-135M”。

  相比于越来越庞大的大语言模型(LLM),它体积小巧,更加灵活,更有针对性,非常适合私密性、专业性很强的企业部署。

  AMD-135小模型隶属于Llama家族,有两个版本:

AMD发布      AI小语言模型:6900亿token、推测解码提速3.88倍

  一是基础型“AMD-Llama-135M”,拥有多达6700亿个token,在八块Instinct MIM250 64GB加速器上训练了六天。

  二是延伸型“AMD-Llama-135M-code”,额外增加了专门针对编程的200亿个token,同样硬件训练了四天。

  创建与部署流程

  它使用了一种名为“推测解码”(speculative decoding)的方法,通过较小的草稿模型,在单次前向传播中生成多个候选token,然后发送给更大的、更 的目标模型,进行验证或纠正。

  这种方法可以同时生成多个token,不会影响性能,还可以降低内存占用,但因为数据交易更多,功耗也会增加。

  AMD还使用AMD-Llama-135M-code作为CodeLlama-7b的草案模型,测试了推测解码使用与否的性能。

  比如在MI250加速器上,性能可提升最多约2.8倍,锐龙AI CPU上可提升最多约3.88倍,锐龙AI NPU上可提升最多约2.98倍。

  推测解码

  AMD-135M小模型的训练代码、数据集等资源都已经开源,遵循Apache 2.0。

  按照AMD的说法,它的性能与其他开源小模型基本相当或略有领先,比如Hellaswag、SciQ、ARC-Easy等任务超过Llama-68M、LLama-160M,Hellaswag、WinoGrande、SciQ、MMLU、ARC-Easy等任务则基本类似GTP2-124MN、OPT-125M。

投稿邮箱:lukejiwang@163.com
相关推荐
消息称字节跳动采购10万颗华为GPU!性能、能效超NVID
消息称字节跳动采购10万颗华为GPU!性能、能效超NVID

原标题:消息称字节跳动采购10万颗华为GPU!性能、能效超NVIDIA A100 据报道,字

3C数码56秒前

一场大火 iPhone印度代工厂无限期停产
一场大火 iPhone印度代工厂无限期停产

原标题:一场大火 iPhone印度代工厂无限期停产 10月1日消息,负责iPhone零件生产

3C数码1小时前

AMD发布      AI小语言模型:6900亿token、推测解码提速3
AMD发布 AI小语言模型:6900亿token、推测解码提速3

原标题:AMD发布 AI小语言模型:6900亿token、推测解码提速3.88倍 10月1日消息

3C数码1小时前

曝PS5 Pro恐因价格高致预购量不佳:甚至没引来黄牛
曝PS5 Pro恐因价格高致预购量不佳:甚至没引来黄牛

原标题:曝PS5 Pro恐因价格高致预购量不佳:甚至没引来黄牛 索尼正式公布了

3C数码22小时前

8520万!上交所集采鲲鹏等国产CPU服务器
8520万!上交所集采鲲鹏等国产CPU服务器

原标题:8520万!上交所集采鲲鹏等国产CPU服务器 9月30日消息,上海证券交易所

3C数码22小时前

OPPO Find X8全系50W无线充电:支持反向充电
OPPO Find X8全系50W无线充电:支持反向充电

原标题:OPPO Find X8全系50W无线充电:支持反向充电 9月29日消息,OPPO Find系列产

3C数码2024-09-29 21:49:57

三星推出Galaxy Tab S10系列平板:搭载天玑9300+、优化AI性
三星推出Galaxy Tab S10系列平板:搭载天玑9300+、优化AI性

原标题:三星推出Galaxy Tab S10系列平板:搭载天玑9300+、优化AI性能 9月27日消息

3C数码2024-09-27 20:29:58

三星电子代工业务遇困局:成熟制程水平落后、先进制
三星电子代工业务遇困局:成熟制程水平落后、先进制

原标题:三星电子代工业务遇困局:成熟制程水平落后、先进制程难获订单 9月

3C数码2024-09-25 19:37:48

Redmi Note 14 Pro+首发第三代骁龙7s:4nm旗舰工艺《王者》
Redmi Note 14 Pro+首发第三代骁龙7s:4nm旗舰工艺《王者》

原标题:Redmi Note 14 Pro+首发第三代骁龙7s:4nm旗舰工艺《王者》120fps稳定高帧

3C数码2024-09-24 19:34:04

维修网站iFixit:iPhone 16是史上最容易维修的苹果手机
维修网站iFixit:iPhone 16是史上最容易维修的苹果手机

原标题:维修网站iFixit:iPhone 16是史上最容易维修的苹果手机 9月23日消息,据媒

3C数码2024-09-23 20:52:07