transformer
损失函数
交叉熵的原理
公式
xi是true_label,yi是神经网络预测为正确的概率
对比学习loss
对比学习损失函数
InfoNEC Loss(bge中也用的这个)
SimCSE的主要思想:同一句话通过不同的drop out作为正例,其余的作为负例(损失函数还是用的InfoNEC Loss);数据增强用的2次dropout
【【一起读论文】SimCSE: Simple Contrastive Learning of Sentence Embeddings】https://www.bilibili.com/video/BV12g411M7AB?vd_source=22a0d494d7d586e6e37e23570688a816
为什么除以根号dk
- 防止梯度消失
随着dk维度的值变大,点积的大小会增大,如果没有及时对点积的大小进行缩放,那么万一点积的数量级很大,softmax的梯度就会趋向于0,也就会出现梯度消失问题。 - 进行归一化
为什么选择 ,时因为可以使得 Q 和 K 点积 趋向于 期望为0,方差为1的标准正态分布,说白了就是归一化。
标准化
标准化作用
缓解梯度消失 / BN缓解内部协变量偏移 / 有一定正则化作用
内部协变量偏移
在训练深层神经网络时,各层的输入分布会随着参数更新而变化,这种现象被称为内部协变量偏移(Internal Covariate Shift)。具体来说,随着训练的进行,每层的激活值(即层的输出)的均值和方差可能会发生变化。由于每层的输入分布在训练过程中不断变化,优化过程中的梯度更新也会受到影响,这会导致训练过程变得不稳定。批归一化的目标是缓解这一问题,使得网络的每一层的输入分布更加稳定,从而加快训练速度。
为什么用 LayerNorm 不使用 BatchNorm?
- 数据的角度
CV 通常用 BatchNorm,NLP 通常用 LayerNorm。图像数据一个 Channel 内的关联性比较大,不同 Channel 的信息需要保持差异性。文本数据一个 Batch 内的不同样本关联性不大。 - Pad的角度解释
不同句子的长度不同,在句子的末尾归一化会受到 pad 的影响,使得统计量不置信。 - 模型角度解释
Self Attention 中,内积的大小的上界和 q,k 的 L2Norm 有关。LayerNorm 对 L2Norm 限制更加直接。
(3) BN 在训练和推理时的差别
BN 在训练和推理时的差别
训练阶段:在训练过程中,BN 使用当前小批量的数据计算均值和方差。
推理阶段:在推理阶段,BN 使用在训练过程中累积的全局均值和方差进行标准化,以确保模型的稳定性。
BN 与dropout 能否一起用?为什么?
添加链接描述
本论文作者发现理解 Dropout 与 BN 之间冲突的关键是网络状态切换过程中存在神经方差的(neural variance)不一致行为。试想若有图一中的神经响应 X,当网络从训练转为测试时,Dropout 可以通过其随机失活保留率(即 p)来缩放响应,并在学习中改变神经元的方差,而 BN 仍然维持 X 的统计滑动方差。这种方差不匹配可能导致数值不稳定(见下图中的红色曲线)。而随着网络越来越深,最终预测的数值偏差可能会累计,从而降低系统的性能。简单起见,作者们将这一现象命名为「方差偏移」。事实上,如果没有 Dropout,那么实际前馈中的神经元方差将与 BN 所累计的滑动方差非常接近(见下图中的蓝色曲线),这也保证了其较高的测试准确率。
attention的时间复杂度
n2d : n为序列长度,d为隐层维度
如何减少时间复杂度:Sparse Self Attention
instruct_gpt
RM损失函数
RL目标函数
什么阶段达到了所谓 in-context learning(gpt-3)
GPT与transformer decoder的区别
少一层无mask自注意力层(交叉自注意力) / LN层前置 / 模型的输出部分添加一个LN层 / 采用可训练的 postion embedding
分词算法
WordPiece
WordPiece核心思想是将单词拆分成多个前缀符号(比如BERT中的##)最小单元,再通过子词合并规则将最小单元进行合并为子词级别。例如对于单词"word",拆分如下:
w ##o ##r ##d
然后通过合并规则进行合并,从而循环迭代构建出一个词表,以下是核心步骤:
- 计算初始词表:通过训练语料获得或者最初的英文中26个字母加上各种符号以及常见中文字符,这些作为初始词表。
- 计算合并分数:对训练语料拆分的多个子词单元通过合拼规则计算合并分数。
- 合并分数最高的子词对:选择分数最高的子词对,将它们合并成一个新的子词单元,并更新词表。
- 重复合并步骤:不断重复步骤 2 和步骤 3,直到达到预定的词表大小、合并次数,或者直到不再有有意义的合并(即,进一步合并不会显著提高词表的效益)。
- 分词:使用最终得到的词汇表对文本进行分词。
Byte-Pair Encoding (BPE)
核心思想是逐步合并出现频率最高的子词对而不是像Wordpiece计算合并分数,从而构建出一个词汇表。
6. 计算初始词表:通过训练语料获得或者最初的英文中26个字母加上各种符号以及常见中文字符,这些作为初始词表。
7. 构建频率统计:统计所有子词单元对(两个连续的子词)在文本中的出现频率。
合并频率最高的子词对:选择出现频率最高的子词对,将它们合并成一个新的子词单元,并更新词汇表。
8. 重复合并步骤:不断重复步骤 2 和步骤 3,直到达到预定的词汇表大小、合并次数,或者直到不再有有意义的合并(即,进一步合并不会显著提高词汇表的效益)。
9. 分词:使用最终得到的词汇表对文本进行分词。
Byte-level BPE(BBPE)
手撕代码
多头自注意力代码
from math import sqrt
import torch
import torch.nn as nnclass MultiHeadSelfAttention(nn.Module):def __init__(self, dim_in, dim_k, dim_v, num_heads=8):super(MultiHeadSelfAttention, self).__init__()assert dim_k % num_heads == 0 and dim_v % num_heads == 0, "dim_k and dim_v must be multiple of num_heads"self.dim_in = dim_inself.dim_k = dim_kself.dim_v = dim_vself.num_heads = num_heads# 定义线性变换矩阵self.linear_q = nn.Linear(dim_in, dim_k, bias=False)self.linear_k = nn.Linear(dim_in, dim_k, bias=False)self.linear_v = nn.Linear(dim_in, dim_v, bias=False)# 最后通过线性层将输出映射回原来的维度self.fc_out = nn.Linear(dim_v, dim_in)self._norm_fact = 1 / sqrt(dim_k // num_heads)def forward(self, x):batch, n, dim_in = x.shapeassert dim_in == self.dim_innh = self.num_headsdk = self.dim_k // nh # dim_k of each headdv = self.dim_v // nh # dim_v of each head# 线性变换并切分头q = self.linear_q(x).reshape(batch, n, nh, dk).transpose(1, 2) # (batch, nh, n, dk)k = self.linear_k(x).reshape(batch, n, nh, dk).transpose(1, 2) # (batch, nh, n, dk)v = self.linear_v(x).reshape(batch, n, nh, dv).transpose(1, 2) # (batch, nh, n, dv)# 计算注意力dist = torch.matmul(q, k.transpose(2, 3)) * self._norm_fact # batch, nh, n, ndist = torch.softmax(dist, dim=-1) # batch, nh, n, n# 根据注意力分数加权值矩阵att = torch.matmul(dist, v) # batch, nh, n, dvatt = att.transpose(1, 2).reshape(batch, n, self.dim_v) # batch, n, dim_v# 最后通过线性层融合不同头的输出,并映射回原始维度out = self.fc_out(att) # (batch, n, dim_in)return out# test
x = torch.randn(2, 10, 128)
attn = MultiHeadSelfAttention(128, 64, 64, 8)
y = attn(x)
print(y.shape) # torch.Size([2, 10, 64])
反向传播代码
激活函数
激活函数
激活函数
deepspeed(数据并行:zero、模型并行、张量并行(magnetron))
Deepspeed加速/优化方式
单byte-Adam / 稀疏注意力计算内核 / 3D并行
三种并行方式
Data Parallelism、Pipeline Parallelism、Tensor Parallelism
数据并行的5个步骤
通信的术语
all_reduce
不同分布式聚集的时间和带宽复杂度
递归的进行all_reduce
zero1,2,3
magatron
【Megatron LM 论文精读【论文精读】】https://www.bilibili.com/video/BV1nB4y1R7Yz?vd_source=22a0d494d7d586e6e37e23570688a816
petf
常用的轻量级微调方法有什么,异同点,与传统的fine-tuning的区别?
部分参数微调策略仅选择性地更新模型中的某些权重,尤其是在需要保留大部分预训练知识的情况下。
BitFit:Fine-tune only the bias terms
adapter
将较小的神经网络层或模块插入预训练模型的每一层,这些新插入的神经模块称为 adapter(适配器),下游任务微调时也只训练这些适配器参数;
prompt tuning
Prefix-Tuning
prompt tuning和Prefix-Tuning的缺点
在模型的输入或隐层添加个额外可训练的前缀 tokens(这些前缀是连续的伪 tokens,不对应真实的 tokens),只训练这些前缀参数;
lora
通过向模型权重矩阵添加低秩矩阵来进行微调,既允许模型学习新的任务特定模式,又能够保留大部分预训练知识,从而降低过拟合风险并提高训练效率。
lora参数量变化: 原矩阵 dd,lora秩为r (一般很小如4、8、16),则新参数量为 2d*r
qlora
使用一种新颖的高精度技术将预训练模型量化为 4 bit,然后添加一小组可学习的低秩适配器权重,这些权重通过量化权重的反向传播梯度进行微调。
排序算法
插入、交换、选择、归并、基数
稳定性:插入、冒泡、归并、基数
过拟合问题以及如何解决
a) 正则化
b) dropout 在训练和推理阶段分别怎样做
(训练只保留p的神经元激活值,推理激活值除以p恢复数据原始分布)
c) 增大数据规模 / 降低模型复杂度
d) Early-stop策略
L1/L2正则化
为什么加入正则化项只限制参数w,而不限制b
数据类别不均衡问题以及如何解决
- 不均衡时如何评估(类别不均衡时, 准确率意义不大) -> ROC曲线
- 欠采样 (欠采样+集成学习) / 过采样
- 集成学习 + 阈值调整 (如随机森林)
- 一些特殊策略 如:focal loss (对难分类样本附加高权重)
举例三个线性的机器学习方法和三个非线性的机器学习方法
区分线性非线性:决策边界是否线性
线性:线性回归分类器,贝叶斯分类,单层感知机,SVM(线性核)等
非线性:决策树、随机森林、GBDT、SVM(非线性核)、多层感知机、神经网络