Transformer - 李宏毅笔记

简介: Transformer - 李宏毅笔记

1.Sequence-to-sequence简介


前一章提到过,模型在处理文本或者语音的数据时,是输入一个序列,输出一个class或者一个序

Seq2seq表示的不是一个模型的名称而是一种模型的类型的统称(即输入序列输出也是序列),在有了self-attention后,通过self-attention的变体模型结构transformer,可以很好的应用到Seq2seq的各种模型应用中


在输入一个序列输出一个序列的模型应用,有如下几种:


语音识别,机器翻译,语音翻译8c276591f81c4369ac549abd146aa831.png


聊天机器人


17022a87dba4479ebfa90e15731a0952.png在深度学习中还有很多的人类语言处理模型

71cf5cde3f644644bcb7ade976c5c9ad.pngSeq2seq也可以用于多标签分类


2.transformer结构


先从seq2seq的结构来看:


b1358fe3aa544aa9a36e97219517a44e.png


seq2seq这一类模型的结构,通常都是由一个编码器(Encoder)和一个解码器(Decode)组成。

那对应到transformer也是如此如下图所示:


9f777a7a61c74ab39b86845358a0630d.png

下文将详细对transformer结构展开解释

2.1 transformer的Encoder


2.1.1 Encode结构



注意这里是由N个上图的结构组成的哦,N个组合的方式如下图:

这里Block就有N个,每一个Block就对应上图里的结构


8119a2f7583c4d9aac6f7bbe492fc66f.png

2.1.2 Encoder具体结构分解


1.多头注意力+残差,再进行layer norm


725bb4456af04451bac224c2b899e01e.png


具体计算过程



2.前馈神经网络+残差,进行layer norm

30befecf2aa444afb84e03a784d385ac.png


具体计算过程


9411bc00d66040cd81660fcf75f38418.png


3.layer norm


在神经网络中norm有不同的比如batch norm和layer norm具体区别

layer norm具体步骤,就是对每一蓝色层,一层层求解,先计算出这一层的平均值与标准差,在进行标准化,转换即可。

9136043787fb4d56b26b85bd632fb779.png


2.2 transformer的Decoder


2.2.1 Decode结构

2fe206a0db864db3a6abe67674721f28.png

880b28f4b0e440ff8d378b80ded867ec.png


这里详细介绍一下在self-attention使用mask的含义的,实现过程


2.2.2 Mask


self-attention里mask是一个参数设置,mask有二个作用:


1. key_padding_mask



我们在输入文本的时候,每一个batch的每一个句子的长度一般是不可能完全相同的,所以我们会使用padding把一些空缺补上。而这里的这个key_padding_mask是用来“遮挡”这些padding的。

这个mask是二元(binary)的,也就是说,它是一个矩阵和我们key的大小是一样的,里面的值是1或0,我们先取得key中有padding的位置,然后把mask里相应位置的数字设置为1,这样attention就会把key相应的部分变为"-inf". (为什么变为-inf我们稍后再说)


保持输入文本的每个句子长度一致。用mask填充


2.attn_mask


这个mask经常是用来遮挡“正确答案”的:

假如你想要用这个模型每次预测下一个单词,我们每一个位置的attention输出是怎么得来的?是不是要看一遍整个序列,然后每一个单词都计算一个attention weight?那也就是说,你在预测第5个词的时候,你其实会看到整个序列,这样的话你在预测之前不就已经知道第5个单词是什么了,这就是作弊了。

我们不想让模型作弊,因为在真实使用这个模型去预测的时候,我们是没有整个序列的信息的。那么怎么办?那就让第5个单词的attention weight=0吧,即声明:我不想看这个单词,我的注意力一点也别分给它。


保持输出预测的文本句子长度一致


2.2.3 Cross attention

4f733315477d409a84bf1779d6b01567.png

这里蓝色和绿色输入的部分,以及后续三者如何结合的过程如下图所示:


1cde924d2d454ae2b586eb2557e998ce.png

248d18644e4c4153bcd939f7aa63f561.png

3.self-attention(pytorch实现)


import torch
import torch.nn as nn
import numpy as np
import math
class SelfAttention(nn.Module):
    def __init__(self, hidden_size, num_attention_heads, dropout_prob):   
        """
        假设 hidden_size = 128, num_attention_heads = 8, dropout_prob = 0.2
        即隐层维度为128,注意力头设置为8个
        """
        super(SelfAttention, self).__init__()
        if hidden_size % num_attention_heads != 0:   # 整除
            raise ValueError(
                "The hidden size (%d) is not a multiple of the number of attention "
                "heads (%d)" % (hidden_size, num_attention_heads))
        # 参数定义
        self.num_attention_heads = num_attention_heads    # 8
        self.attention_head_size = int(hidden_size / num_attention_heads)  # 16  每个注意力头的维度
        self.all_head_size = int(self.num_attention_heads * self.attention_head_size)   
        # all_head_size = 128 即等于hidden_size, 一般自注意力输入输出前后维度不变
        # query, key, value 的线性变换(上述公式2)
        self.query = nn.Linear(hidden_size, self.all_head_size)    # 128, 128
        self.key = nn.Linear(hidden_size, self.all_head_size)
        self.value = nn.Linear(hidden_size, self.all_head_size)
        # dropout
        self.dropout = nn.Dropout(dropout_prob)
    def transpose_for_scores(self, x):
        # INPUT:  x'shape = [bs, seqlen, hid_size]  假设hid_size=128
        new_x_shape = x.size()[:-1] + (self.num_attention_heads, self.attention_head_size) # [bs, seqlen, 8, 16]
        x = x.view(*new_x_shape)   # 
        return x.permute(0, 2, 1, 3)   # [bs, 8, seqlen, 16]
    def forward(self, hidden_states, attention_mask):
        # eg: attention_mask = torch.LongTensor([[1, 1, 1], [1, 1, 0]])  shape=[bs, seqlen]
        attention_mask = attention_mask.unsqueeze(1).unsqueeze(2)   # [bs, 1, 1, seqlen] 增加维度
        attention_mask = (1.0 - attention_mask) * -10000.0   # padding的token置为-10000,exp(-1w)=0
        # 线性变换
        mixed_query_layer = self.query(hidden_states)   # [bs, seqlen, hid_size]
        mixed_key_layer = self.key(hidden_states)       # [bs, seqlen, hid_size]
        mixed_value_layer = self.value(hidden_states)   # [bs, seqlen, hid_size]
        query_layer = self.transpose_for_scores(mixed_query_layer)    # [bs, 8, seqlen, 16]
        key_layer = self.transpose_for_scores(mixed_key_layer)
        value_layer = self.transpose_for_scores(mixed_value_layer)   # [bs, 8, seqlen, 16]
        # Take the dot product between "query" and "key" to get the raw attention scores.
        # 计算query与title之间的点积注意力分数,还不是权重(个人认为权重应该是和为1的概率分布)
        attention_scores = torch.matmul(query_layer, key_layer.transpose(-1, -2))
        # [bs, 8, seqlen, 16]*[bs, 8, 16, seqlen]  ==> [bs, 8, seqlen, seqlen]
        attention_scores = attention_scores / math.sqrt(self.attention_head_size)   # [bs, 8, seqlen, seqlen]
        # 除以根号注意力头的数量,可看原论文公式,防止分数过大,过大会导致softmax之后非0即1
        attention_scores = attention_scores + attention_mask
        # 加上mask,将padding所在的表示直接-10000
        # 将注意力转化为概率分布,即注意力权重
        attention_probs = nn.Softmax(dim=-1)(attention_scores)    # [bs, 8, seqlen, seqlen]
        # This is actually dropping out entire tokens to attend to, which might
        # seem a bit unusual, but is taken from the original Transformer paper.
        attention_probs = self.dropout(attention_probs)
        # 矩阵相乘,[bs, 8, seqlen, seqlen]*[bs, 8, seqlen, 16] = [bs, 8, seqlen, 16]
        context_layer = torch.matmul(attention_probs, value_layer)   # [bs, 8, seqlen, 16]
        context_layer = context_layer.permute(0, 2, 1, 3).contiguous()   # [bs, seqlen, 8, 16]
        new_context_layer_shape = context_layer.size()[:-2] + (self.all_head_size,)   # [bs, seqlen, 128]
        context_layer = context_layer.view(*new_context_layer_shape)
        return context_layer    # [bs, seqlen, 128] 得到输出
attention=SelfAttention(4,2,0.2)
x_in=torch.randn(3,5,4)
x_mask=torch.Tensor([[1,1,1,0,0],
                    [1,1,0,0,0],
                    [1,1,1,1,1],])
print(x_mask.shape)
x_out=attention(x_in,x_mask)
相关文章
|
自然语言处理 安全 虚拟化
如何查看Windows所有软件的快捷键?
【2月更文挑战第12天】本文介绍在Windows电脑中,基于OpenArk工具,查看电脑操作系统与所有软件的快捷键,并对快捷键冲突加以处理的方法~
2171 9
如何查看Windows所有软件的快捷键?
|
4月前
|
机器学习/深度学习 传感器 人工智能
构建AI智能体:九十五、YOLO视觉大模型入门指南:从零开始掌握目标检测
本文介绍了视觉大模型及YOLO目标检测技术,重点讲解YOLOv8在CPU上的部署与应用。涵盖模型选择、图像检测、实时摄像头识别及性能优化,适合初学者快速上手。
738 2
构建AI智能体:九十五、YOLO视觉大模型入门指南:从零开始掌握目标检测
|
7月前
|
存储 缓存 负载均衡
TensorRT LLM 中的并行策略
TensorRT LLM提供多种GPU并行策略,支持大模型在显存与性能受限时的高效部署。涵盖张量、流水线、数据、专家及上下文并行,并推出宽专家并行(Wide-EP)应对大规模MoE模型的负载不均与通信挑战,结合智能负载均衡与优化通信核心,提升推理效率与可扩展性。
1110 154
|
4月前
|
存储 机器学习/深度学习 人工智能
构建AI智能体:八十一、SVD模型压缩的艺术:如何科学选择K值实现最佳性能
本文探讨了SVD(奇异值分解)在深度学习模型压缩中的应用。随着模型规模不断扩大,面临的存储、计算和能耗问题日益突出。SVD通过将大型矩阵分解为三个特殊矩阵(U、Σ、Vᵀ),并根据奇异值大小进行截断,实现模型的高效压缩。文章详细分析了SVD的数学原理、压缩机制和参数选择策略,重点讨论了如何通过能量保留、目标压缩率和拐点检测等方法确定最佳截断秩k。以一个500×300的用户-物品矩阵为例,实证显示k=32时能实现82.93%的压缩率,99.97%的能量保留和5.86倍计算加速,达到了精度损失与压缩收益的最佳平衡
378 5
|
存储 人工智能 自然语言处理
LangChain RAG入门教程:构建基于私有文档的智能问答助手
本文介绍如何利用检索增强生成(RAG)技术与LangChain框架构建基于特定文档集合的AI问答系统。通过结合检索系统和生成机制,RAG能有效降低传统语言模型的知识局限与幻觉问题,提升回答准确性。文章详细展示了从环境配置、知识库构建到系统集成的全流程,并提供优化策略以改进检索与响应质量。此技术适用于专业领域信息检索与生成,为定制化AI应用奠定了基础。
3999 5
LangChain RAG入门教程:构建基于私有文档的智能问答助手
|
7月前
netlify信用额度急剧消耗,300免费信用分到底谁占用了?
作者使用Netlify托管Publii生成的静态网站,原以为仅GitHub同步耗积分,结果频繁更新导致300积分快速消耗。发现每次部署竟扣15分,数日即用掉75%额度,方知高频更新不现实,访问量大时更受限,服务模式不适合持续更新的站点。
1086 1
|
11月前
|
安全 Java API
2025 年 Java 校招面试常见问题及详细答案汇总
本资料涵盖Java校招常见面试题,包括Java基础、并发编程、JVM、Spring框架、分布式与微服务等核心知识点,并提供详细解析与实操代码,助力2025校招备战。
484 1
|
人工智能 IDE 开发工具
JetBrains Rider 2025.1 发布 - 快速且强大的跨平台 .NET IDE
JetBrains Rider 2025.1 (macOS, Linux, Windows) - 快速且强大的跨平台 .NET IDE
1126 29
JetBrains Rider 2025.1 发布 - 快速且强大的跨平台 .NET IDE
|
存储 人工智能 PyTorch
【AI系统】张量并行
在大模型训练中,单个设备难以满足需求,模型并行技术应运而生。其中,张量并行(Tensor Parallelism, TP)将模型内部的参数和计算任务拆分到不同设备上,特别适用于大规模模型。本文介绍了张量并行的基本概念、实现方法及其在矩阵乘法、Transformer、Embedding和Cross Entropy Loss等场景中的应用,以及通过PyTorch DeviceMesh实现TP的具体步骤。
1609 11
【AI系统】张量并行