大模型系列炼丹术(二):从离散的token IDs到具有语义信息的embedding

在完成了tokenization之后,我们已经可以将一个个的单词映射到对应的数字,称之为token ID,这些数字已经可以被计算机处理。然而,若直接将这些数字应用于模型训练,仍存在一些问题:

  1. 缺乏语义信息: Token ID 只是一个索引,本身不包含任何语义信息。例如,“cat” 可能被映射为 ID 1254,而 “dog” 是 ID 3920,这两个 ID 之间的数值关系是无意义的。直接使用它们可能会导致模型误解 token 之间的关联性。

  2. 整数之间的数值关系会误导模型: 机器学习模型通常会学习数据之间的模式。如果直接输入 token ID,模型可能会误以为 ID 1254("cat")和 ID 3920("dog")之间存在某种数学关系(如加减乘除),但实际上 ID 只是索引,没有数值上的逻辑关系。

  3. 无法捕捉相似 token 的关系 语义相近的 token 在 embedding 空间中应该具有接近的表示。例如,"king" 和 "queen" 应该在高维空间中比较接近,而 "apple" 和 "computer" 应该相距较远。然而,单纯的 token ID 无法提供这种分布信息。

一、什么是Token Embedding ?

Token Embedding(标记嵌入)是将离散的 token(如单词、子词或字符)转换为连续的向量表示的过程。在自然语言处理(NLP)任务中,神经网络无法直接处理文本,需要将文本转换为有语义特征的数字形式,而 Token Embedding 就是这一转换的核心步骤。

在PyTorch中,使用torch.nn.Embedding来构建embedding层,实际上,这是一个查找表,key是token id,value是token id对应的embedding 向量。

举个例子,假设词汇表大小是6,embedding 维度是3,那么,构建的embedding layer的weight 的shape是6x3

# 构建embedding layer
vocab_size = 6
output_dim = 3

torch.manual_seed(123)
embedding_layer = torch.nn.Embedding(vocab_size, output_dim)

print(embedding_layer.weight)# 6x3

输出:

Parameter containing:
tensor([[ 0.3374, -0.1778, -0.1690],
        [ 0.9178,  1.5810,  1.3010],
        [ 1.2753, -0.2010, -0.1606],
        [-0.4015,  0.9666, -1.1481],
        [-1.1589,  0.3255, -0.6315],
        [-2.8400, -0.7849, -1.4096]], requires_grad=True)

这个就是embedding层的weight,也就是一个查找表,它有6行3列,6是词汇表大小,3是embedding层的维度。

假设输入句子转换到token id后如下:

input_ids = torch.tensor([2, 3, 5, 1])

其中,2,3,5,1其实就是要在当前查找表中查询的索引,因此

print(embedding_layer(input_ids))

的输出如下:

tensor([[ 1.2753, -0.2010, -0.1606],
        [-0.4015,  0.9666, -1.1481],
        [-2.8400, -0.7849, -1.4096],
        [ 0.9178,  1.5810,  1.3010]], grad_fn=<EmbeddingBackward0>)

到这里,我们已经可以将原始的输入文本转换成embedding向量,具体包括两个过程:

  • 1)先通过 Tokenization 将文本拆分为 token ID

  • 2)然后通过 Token Embedding 层 获取 token 的连续表示,即embedding

二、如何训练Embedding层 ?

Embedding 层本质上是一个可学习的查找表,它的核心是一个嵌入矩阵(Embedding Matrix),用于将离散的 token ID 映射到连续的高维向量。这个矩阵的大小通常是:

在训练过程中(比如做文本分类,有文本的类别标签),模型会自动更新 Embedding 矩阵的权重,使得具有相似语义的 token 向量接近,而不相关的 token 向量距离较远。

一般来说,在构建自己的任务时,会加载预训练的一些embedding,这样可以充分利用已有的大规模语料训练的语义信息,提高模型的性能,并加速收敛。

三、nn.Embedding 和 nn.Linear的区别是什么 ?

Embedding 相当于一个固定输入为 one-hot 向量的 Linear 层。

假设 num_embeddings=10, embedding_dim=4,那么 Embedding 其实是一个形状为 [10, 4] 的权重矩阵 E。

如果用 Linear 模拟 Embedding:

先把 token ID 转换成 one-hot 向量(形状 [10])。 然后进行 矩阵乘法(one-hot 只会选中对应的行,相当于输入的X是1,与weight matrix行相乘的结果就是weight matrix行)。

代码如下:

# 创建一个相当于 nn.Embedding 的 Linear 层
embedding_as_linear = nn.Linear(in_features=10, out_features=4, bias=False)

# 生成一个 one-hot 形式的输入(假设词汇表大小 10)
one_hot_input = torch.eye(10)[[1, 3, 5]]  # 选中索引 1、3、5
print(one_hot_input.shape)  # (3, 10)

# 进行计算
word_vectors = embedding_as_linear(one_hot_input)
print(word_vectors.shape)  # (3, 4)

其中:

embedding_as_linear.weight.T:
Parameter containing:
tensor([[ 0.2629, -0.0249, -0.0409, -0.1259],
        [-0.0418,  0.0100, -0.2939,  0.0127],
        [-0.2674, -0.0493, -0.1956, -0.0738],
        [-0.0907,  0.0497,  0.2699, -0.0348],
        [-0.2227,  0.2818,  0.0189, -0.3083],
        [ 0.0209,  0.1934, -0.2562,  0.1481],
        [-0.0590,  0.1122,  0.0499,  0.2776],
        [-0.1696,  0.0687,  0.2613,  0.1933],
        [-0.0288,  0.0746, -0.2988, -0.2239],
        [ 0.2996,  0.1222, -0.2129, -0.2549]], grad_fn=<PermuteBackward0>)


one_hot_input:
tensor([[0., 1., 0., 0., 0., 0., 0., 0., 0., 0.],
        [0., 0., 0., 1., 0., 0., 0., 0., 0., 0.],
        [0., 0., 0., 0., 0., 1., 0., 0., 0., 0.]])


word_vectors:
tensor([[-0.0418,  0.0100, -0.2939,  0.0127],
        [-0.0907,  0.0497,  0.2699, -0.0348],
        [ 0.0209,  0.1934, -0.2562,  0.1481]], grad_fn=<MmBackward0>)

四、为什么还需要Positinal Embedding?

位置编码(Positional Encoding,PE)是一种用于在无序输入(如 Transformer)中引入位置信息的技术,其编码得到的向量称之为Positinal Embedding。由于 Transformer 不使用循环(RNN)或卷积(CNN),它无法捕捉序列顺序,因此需要额外的信息来表示单词的顺序。

Transformer 采用自注意力机制(Self-Attention),它本质上是对输入进行加权求和,不考虑输入的顺序。例如:

句子1:“小猫在沙发上睡觉。”
句子2:“沙发在小猫上睡觉。”

在 LSTM/RNN 中,单词的顺序 会通过循环网络的隐状态传递体现出来,而在 Transformer 中,所有单词是同时处理的,所以:

  • 没有位置编码:模型会认为两句话的含义是一样的!

  • 有位置编码:模型能够识别单词的顺序,从而理解句子的语义。

五、常见的位置编码方式

1. 按照参数是否可以学习来分类

按照参数是否是可学习的,可以分为:可学习的(Learnable)位置编码,固定的(Sinusoidal)位置编码。

可学习的(Learnable)位置编码

直接为每个位置训练一个可学习的向量,类似于 nn.Embedding,让模型自动学习最佳的位置信息。

# 假设最大序列长度为 100,每个位置用 512 维向量表示
max_len = 100
d_model = 512  # 词向量维度

# 可学习的位置编码
positional_embedding = nn.Embedding(max_len, d_model)

# 生成 10 个 token 的位置信息
positions = torch.arange(10).unsqueeze(0)  # shape: (1, 10)
pe = positional_embedding(positions)  # shape: (1, 10, 512)
  • 优点:训练时可以自动调整位置编码的权重;适用于特定任务(如 NLP),如果训练数据的序列长度固定,可以使用这种方法。

  • 缺点:不能推广到比训练时更长的序列。

固定的(Sinusoidal)位置编码

Transformer 论文提出了一种固定的位置编码方法,利用正弦(sin)和余弦(cos)函数,使不同位置的编码具有唯一性,并且可以推广到更长的序列,其代码实现如下:

import numpy as np
import torch

def positional_encoding(seq_len, d_model):
    """
    生成 Transformer 位置编码
    :param seq_len: 序列长度
    :param d_model: 词向量维度
    :return: 位置编码矩阵 (seq_len, d_model)
    """
    pe = np.zeros((seq_len, d_model))
    position = np.arange(seq_len)[:, np.newaxis]  # (seq_len, 1)
    div_term = np.exp(np.arange(0, d_model, 2) * (-np.log(10000.0) / d_model))

    pe[:, 0::2] = np.sin(position * div_term)  # 偶数维度使用 sin
    pe[:, 1::2] = np.cos(position * div_term)  # 奇数维度使用 cos

    return torch.tensor(pe, dtype=torch.float32)

# 生成 10 个 token 的位置编码,词向量维度 512
pe = positional_encoding(10, 512)
print(pe.shape)  #  torch.Size([10, 512])
  • 优点:无需训练,可直接计算;可以推广到更长的序列,不会因为训练长度限制而失效。

  • 缺点:对特定任务可能没有可学习的位置编码效果好。

2. 按照位置关系分类

在位置编码的讨论中,"绝对" 和 "相对" 是根据编码方式的依赖关系来区分的。它们的核心区别在于 位置编码所表示的内容:是单独表示每个 token 在序列中的具体位置(绝对),还是表示 token 之间的相对位置关系(相对)。

绝对位置编码

绝对位置编码为每个 token 分配一个唯一的、固定的标识符,这个标识符表示该 token 在序列中的位置,其特点归纳如下:

  • 位置编码是根据每个 token 在序列中的位置生成的,不考虑 token 之间的相对距离或顺序。

  • 每个位置都有一个固定的编码值,即每个位置的编码与其绝对位置直接相关。

上面介绍的根据正弦(sin)和余弦(cos)函数设计的固定位置编码就是一种绝对位置编码。

相对位置编码

相对位置编码表示token 之间的相对位置,即某个 token 与其他 token 之间的相对位置关系,而不是每个 token 的绝对位置,其特点归纳如下:

  • 位置编码关注的是 token 之间的相对关系,而不依赖于它们的绝对位置。这样就能将模型从单一的固定序列长度中解耦,提供更大的灵活性。

  • 相对位置编码通常不为每个位置定义一个唯一的标识符,而是为每对 token 之间的相对距离生成一个编码(如相对位置的偏置或因子)。

六、GPT使用哪种位置编码方式 ?

GPT 系列使用的是三角函数的绝对位置编码。

GPT 系列模型是基于自回归的 Transformer 架构,目的是生成序列(如文本)。使用绝对位置编码有助于模型理解 token 在序列中的顺序,以便在生成文本时考虑上下文信息。

相较于相对位置编码,绝对位置编码较为简单,适用于很多任务,尤其是在模型输入序列长度相对固定或有限的情况下。由于 GPT 系列通常处理的输入长度相对较短,绝对位置编码是足够有效的。

此外,Llama 系列模型和一些其他的模型采用了 旋转位置编码(RoPE, Rotary Position Embedding)。这种位置编码方式不同于传统的绝对位置编码和相对位置编码,它通过旋转的方式来处理序列中的位置信息。具体的实现细节这里暂时不进行讲解。

ok,现在已经将句子映射到了embedding,那么embedding在后续将会被如何处理呢?它在神经网络中的数据流向又是怎样的呢?所有内容将在下篇文章中继续介绍,欢迎持续关注。

 如何系统的去学习大模型LLM ?

大模型时代,火爆出圈的LLM大模型让程序员们开始重新评估自己的本领。 “AI会取代那些行业?”“谁的饭碗又将不保了?”等问题热议不断。

事实上,抢你饭碗的不是AI,而是会利用AI的人。

科大讯飞、阿里、华为等巨头公司发布AI产品后,很多中小企业也陆续进场!超高年薪,挖掘AI大模型人才! 如今大厂老板们,也更倾向于会AI的人,普通程序员,还有应对的机会吗?

与其焦虑……

不如成为「掌握AI工具的技术人」,毕竟AI时代,谁先尝试,谁就能占得先机!

但是LLM相关的内容很多,现在网上的老课程老教材关于LLM又太少。所以现在小白入门就只能靠自学,学习成本和门槛很高。

基于此,我用做产品的心态来打磨这份大模型教程,深挖痛点并持续修改了近70次后,终于把整个AI大模型的学习门槛,降到了最低!

在这个版本当中:

第一您不需要具备任何算法和数学的基础
第二不要求准备高配置的电脑
第三不必懂Python等任何编程语言

您只需要听我讲,跟着我做即可,为了让学习的道路变得更简单,这份大模型教程已经给大家整理并打包,现在将这份 LLM大模型资料 分享出来:包括LLM大模型书籍、640套大模型行业报告、LLM大模型学习视频、LLM大模型学习路线、开源大模型学习教程等, 😝有需要的小伙伴,可以 扫描下方二维码领取🆓↓↓↓

一、LLM大模型经典书籍

AI大模型已经成为了当今科技领域的一大热点,那以下这些大模型书籍就是非常不错的学习资源。

在这里插入图片描述

二、640套LLM大模型报告合集

这套包含640份报告的合集,涵盖了大模型的理论研究、技术实现、行业应用等多个方面。无论您是科研人员、工程师,还是对AI大模型感兴趣的爱好者,这套报告合集都将为您提供宝贵的信息和启示。(几乎涵盖所有行业)
在这里插入图片描述

三、LLM大模型系列视频教程

在这里插入图片描述

四、LLM大模型开源教程(LLaLA/Meta/chatglm/chatgpt)

在这里插入图片描述

五、AI产品经理大模型教程

在这里插入图片描述

LLM大模型学习路线 

阶段1:AI大模型时代的基础理解

  • 目标:了解AI大模型的基本概念、发展历程和核心原理。

  • 内容

    • L1.1 人工智能简述与大模型起源
    • L1.2 大模型与通用人工智能
    • L1.3 GPT模型的发展历程
    • L1.4 模型工程
    • L1.4.1 知识大模型
    • L1.4.2 生产大模型
    • L1.4.3 模型工程方法论
    • L1.4.4 模型工程实践
    • L1.5 GPT应用案例

阶段2:AI大模型API应用开发工程

  • 目标:掌握AI大模型API的使用和开发,以及相关的编程技能。

  • 内容

    • L2.1 API接口
    • L2.1.1 OpenAI API接口
    • L2.1.2 Python接口接入
    • L2.1.3 BOT工具类框架
    • L2.1.4 代码示例
    • L2.2 Prompt框架
    • L2.3 流水线工程
    • L2.4 总结与展望

阶段3:AI大模型应用架构实践

  • 目标:深入理解AI大模型的应用架构,并能够进行私有化部署。

  • 内容

    • L3.1 Agent模型框架
    • L3.2 MetaGPT
    • L3.3 ChatGLM
    • L3.4 LLAMA
    • L3.5 其他大模型介绍

阶段4:AI大模型私有化部署

  • 目标:掌握多种AI大模型的私有化部署,包括多模态和特定领域模型。

  • 内容

    • L4.1 模型私有化部署概述
    • L4.2 模型私有化部署的关键技术
    • L4.3 模型私有化部署的实施步骤
    • L4.4 模型私有化部署的应用场景

这份 LLM大模型资料 包括LLM大模型书籍、640套大模型行业报告、LLM大模型学习视频、LLM大模型学习路线、开源大模型学习教程等, 😝有需要的小伙伴,可以 扫描下方二维码领取🆓↓↓↓

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值