当前位置:   article > 正文

【昇思MindSpore技术公开课】请查收第五讲LLaMA知识点回顾_mindspore llama量化

mindspore llama量化

昇思MindSpore公开课大模型专题第二季课程火爆来袭!未报名的小伙伴抓紧时间扫描下方二维码参与课程,并同步加入课程群,有免费丰富的课程资源在等着你。课程同步赋能华为ICT大赛2023-2024,助力各位选手取得理想成绩!

image.png

在学习了GLM系列模型后,我们开启另一个系列模型——LLaMA。上周六我们围绕LLaMA讲解了LLaMA模型的背景,其子孙模型,并拆解了LLaMA在模型结构上的创新点,最后基于MindSpore Transformers演示了LLaMA的推理代码。下面我们对本周公开课内容进行总结,迎接下一节的深入:

一、

课程回顾

1、LLaMA背景介绍

1.1 大模型特点

Transfomer-based结构,模型参数量大,训练数据大,训练数据大;目前很多开发者苦于没有足够的算力资源支撑个人大模型的开发工作。

1.2 LLaMA聚焦

在模型规模小,以及资源有限的情况下,选择合适的训练方法,用足够多高质量的数据,训练足够长的时间,依旧可以得到一个能力很好的模型。

1.3 LLaMA通过增加数据,进行微调或二次预训练的方法,发展出了一系列羊驼家族大模型。

其中主要分为以下三条脉络:

(1)基于合成数据,进行指令微调得到的Alpaca模型,及其作为基座模型衍生的大模型

(2)基于对话数据,进行指令微调得到的Vicuna模型,及其作为基座模型衍生的大模型

(3)基于中文数据,在原有词表上进行扩充,二次预训练得到的中文LLaMA,及其作为基座模型衍生的大模型

image.png

1.4 LLaMA目前遇到的bug

reversal curse,即如果模型用<name>is<description>形式的句子进行训练,那么模型将不会自动预测反方向的<description>is<name>

image.png

2、LLaMA模型结构解析

2.1 Transfomer-based常见的创新点(模块角度)

(1)位置编码(针对模型外推能力):绝对位置编码,相对位置编码,...

(2)Attention结构 (针对优化训推性能):sparse attention,low-rank attention,multi-query attention,grouped-query attention,...

(3)Add&Norm(针对模型训练稳定性和效果):一般为Norm的改动

(4)Feed-forward Network(针对模型效果):一般为激活函数的改动

image.png

2.2 LLaMA在Add&Norm模块中的改动:Pre-Norm和RMSNorm

(1)Pre-Norm:normalization发生在输入进入子层之前,在避免梯度爆炸/梯度消失、训练更容易的角度效果更好。

image.png

(2)RMSNorm:LayerNorm的变体,去除了LayerNorm中减均值的部分,减少计算量。

2.3 LLaMA在Feed-forward Network模块中的改动:SwiGLU

image.png

2.4 LLaMA在位置编码中的改动:rotary positional embedding

(1)大模型的外推能力:大模型输入长度超过预训练文本长度时,输出表现变化情况;外推能力不强的模型在输入长度超出预训练文本长度时,模型就会“一本正经地胡说八道。”

(2)绝对位置编码记录token的索引,因为其长度固定,训练出的大模型生成长度无法超过预训练文本长度。

(3)相对位置编码记录token的索引差,训练出的大模型具有外推性。

(4)旋转位置编码(rotary positional embedding,RoPE)用绝对位置编码来表征相对位置编码,其使用旋转矩阵对绝对位置进行编码,作用于query和key,在自注意力公式中引入了显示的相对位置依赖。

image.png

3、使用MindSpore Transformers试用LLaMA推理

3.1 使用MindSpore Transformer进行文本生成任务

增加高概率单词的似然并降低低概率单词的似然,值越大结果随机性越高。

(1)选择对应Config类,进行模型超参数配置

(2)选择对应Model类,进行模型实例化

(3)选择对应tokenizer

  1. # set model config
  2. model_config = AutoConfig.from_pretrained(model_type)
  3. # if use parallel, data_parallel * model_parallel = device_num
  4. model_config.parallel_config.data_parallel = 1
  5. model_config.parallel_config.model_parallel = 1
  6. model_config.batch_size = len(inputs)
  7. model_config.use_past = use_past
  8. if checkpoint_path and not use_parallel:
  9. model_config.checkpoint_name_or_path = checkpoint_path
  10. print(f"config is: {model_config}")
  11. # build tokenizer
  12. tokenizer = AutoTokenizer.from_pretrained(model_type)
  13. # build model from config
  14. network = AutoModel.from_config(model_config)
复制

(4)调用tokenizer对输入进行tokenize

(5)调用generate函数,进行推理

(6)调用tokenizer,对输出进行decode,将数字索引转换回文本

  1. inputs_ids = tokenizer(inputs, max_length=model_config.seq_length, padding="max_length")["input_ids"]
  2. outputs = network.generate(inputs_ids, max_length=model_config.max_decode_length)
  3. for output in outputs:
  4. print(tokenizer.decode(output))
复制

3.2 MindSpore Transformer的pipeline组件解析:面向任务领域设计pipeline推理服务

(1)选择模型,tokenizer,进行实例化

  1. # set model config
  2. model_config = AutoConfig.from_pretrained(model_type)
  3. model_config.use_past = use_past
  4. # if use parallel, data_parallel * model_parallel = device_num
  5. model_config.parallel_config.data_parallel = 1
  6. model_config.parallel_config.model_parallel = 1
  7. if checkpoint_path and not use_parallel:
  8. model_config.checkpoint_name_or_path = checkpoint_path
  9. print(f"config is: {model_config}")
  10. # build tokenizer
  11. tokenizer = AutoTokenizer.from_pretrained(model_type)
  12. # build model from config
  13. network = AutoModel.from_config(model_config)
复制

(2)选择任务,实例化pipeline,输入数据进行推理

  1. text_generation_pipeline = pipeline(task="text_generation", model=network, tokenizer=tokenizer)
  2. outputs = text_generation_pipeline(inputs)
复制

二、

下节课预告

下节课我们继续LLaMA系列模型的学习之旅,讲解LLaMA2系列模型,这里稍稍地“剧透”下课程的内容:

1、LLaMA2介绍及推理部署代码演示

2、业界SOTA模型介绍

课程直播时间为12月23日(下周六)14:00-15:30,欢迎关注各大直播平台,我们不见不散!

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/你好赵伟/article/detail/509534
推荐阅读
相关标签
  

闽ICP备14008679号