链载Ai

标题: LLM中的Token和Embedding到底是啥? [打印本页]

作者: 链载Ai    时间: 昨天 20:40
标题: LLM中的Token和Embedding到底是啥?


01


引言



GPT4 Turbo 的上下文长度为 128K 个Tokens。

Claude 2.1 的上下文长度为 200K 个Tokens。

那么问题就来了。。。

上文中的Tokens到底是什么?

我们来看个简单的示例:输入句子 “It’s over 9000!”

ingFang SC", system-ui, -apple-system, "Segoe UI", Rototo, Helvetica, Arial, sans-serif;font-size: 15px;font-style: normal;font-variant-ligatures: normal;font-variant-caps: normal;font-weight: 400;letter-spacing: normal;orphans: 2;text-align: start;text-indent: 0px;text-transform: none;widows: 2;word-spacing: 0px;-webkit-text-stroke-width: 0px;text-decoration-thickness: initial;text-decoration-style: initial;text-decoration-color: initial;">我们可以将其表示为 ["It's", "over", "9000!"] 每个数组中的单个元素都可以称为一个Token。

ingFang SC", system-ui, -apple-system, "Segoe UI", Rototo, Helvetica, Arial, sans-serif;font-size: 15px;font-style: normal;font-variant-ligatures: normal;font-variant-caps: normal;font-weight: 400;letter-spacing: normal;orphans: 2;text-align: start;text-indent: 0px;text-transform: none;widows: 2;word-spacing: 0px;-webkit-text-stroke-width: 0px;text-decoration-thickness: initial;text-decoration-style: initial;text-decoration-color: initial;">在自然语言处理领域,Token是我们定义的最小分析单位。如何称呼Token取决于大家采用的tokenization的方法;此类方法目前有很多,创建Token基本上是大多数 NLP 任务的第一步。





02

NLP中Tokenization化的方法

让我们直接通过以下代码示例,来了解一些常用的输入文本Tokenization化的方法。

#Examplestringfortokenizationexample_string="It'sover9000!"#Method1:WhiteSpaceTokenization#Thismethodsplitsthetextbasedonwhitespaceswhite_space_tokens=example_string.split()#Method2:WordPunctTokenization#Thismethodsplitsthetextintowordsandpunctuationfromnltk.tokenizeimportWordPunctTokenizerwordpunct_tokenizer=WordPunctTokenizer()wordpunct_tokens=wordpunct_tokenizer.tokenize(example_string)#Method3:TreebankWordTokenization#ThismethodusesthestandardwordtokenizationofthePennTreebankfromnltk.tokenizeimportTreebankWordTokenizertreebank_tokenizer=TreebankWordTokenizer()treebank_tokens=treebank_tokenizer.tokenize(example_string)white_space_tokens,wordpunct_tokens,treebank_tokens

ingFang SC", system-ui, -apple-system, "Segoe UI", Rototo, Helvetica, Arial, sans-serif;font-size: 15px;font-style: normal;font-variant-ligatures: normal;font-variant-caps: normal;font-weight: 400;letter-spacing: normal;orphans: 2;text-align: start;text-indent: 0px;text-transform: none;widows: 2;word-spacing: 0px;-webkit-text-stroke-width: 0px;text-decoration-thickness: initial;text-decoration-style: initial;text-decoration-color: initial;">上述代码运行后的输出如下:

(["It's",'over','9000!'],['It',"'",'s','over','9000','!'],['It',"'s",'over','9000','!'])

ingFang SC", system-ui, -apple-system, "Segoe UI", Rototo, Helvetica, Arial, sans-serif;font-size: 15px;font-style: normal;font-variant-ligatures: normal;font-variant-caps: normal;font-weight: 400;letter-spacing: normal;orphans: 2;text-align: start;text-indent: 0px;text-transform: none;widows: 2;word-spacing: 0px;-webkit-text-stroke-width: 0px;text-decoration-thickness: initial;text-decoration-style: initial;text-decoration-color: initial;">上述不同方法都有将句子分解为Token的独特方式。如果大家愿意,可以创建自己的方法,但基本要领是相同的。

ingFang SC", system-ui, -apple-system, "Segoe UI", Rototo, Helvetica, Arial, sans-serif;font-size: 15px;font-style: normal;font-variant-ligatures: normal;font-variant-caps: normal;font-weight: 400;letter-spacing: normal;orphans: 2;text-align: start;text-indent: 0px;text-transform: none;widows: 2;word-spacing: 0px;-webkit-text-stroke-width: 0px;text-decoration-thickness: initial;text-decoration-style: initial;text-decoration-color: initial;">那么为什么需要将输入文本进行Tokenization呢?可以归纳为以下原因:

将复杂的文本分解为易于处理的单元。

以更易于分析或操作的格式呈现文本。

适用于特定的语言任务,如句法分析和命名实体识别。

在 NLP 应用中统一预处理文本,并创建结构化的训练数据。

ingFang SC", system-ui, -apple-system, "Segoe UI", Rototo, Helvetica, Arial, sans-serif;font-size: 15px;font-style: normal;font-variant-ligatures: normal;font-variant-caps: normal;font-weight: 400;letter-spacing: normal;orphans: 2;text-align: start;text-indent: 0px;text-transform: none;widows: 2;word-spacing: 0px;-webkit-text-stroke-width: 0px;text-decoration-thickness: initial;text-decoration-style: initial;text-decoration-color: initial;">大多数 NLP 系统都会对这些Tokens执行一些操作,以完成特定任务。例如,我们可以设计一个系统来处理一串Tokens并预测下一个Token。我们还可以将Tokens转换为语音表示,作为文本到语音系统的一部分。我们还可以完成许多其他 NLP 任务,如关键词提取、翻译任务等。

ingFang SC", system-ui, -apple-system, "Segoe UI", Rototo, Helvetica, Arial, sans-serif;font-size: 15px;font-style: normal;font-variant-ligatures: normal;font-variant-caps: normal;font-weight: 400;letter-spacing: normal;orphans: 2;text-align: start;text-indent: 0px;text-transform: none;widows: 2;word-spacing: 0px;-webkit-text-stroke-width: 0px;text-decoration-thickness: initial;text-decoration-style: initial;text-decoration-color: initial;">

ingFang SC", system-ui, -apple-system, "Segoe UI", Rototo, Helvetica, Arial, sans-serif;font-size: 15px;font-style: normal;font-variant-ligatures: normal;font-variant-caps: normal;font-weight: 400;letter-spacing: normal;orphans: 2;text-align: start;text-indent: 0px;text-transform: none;widows: 2;word-spacing: 0px;-webkit-text-stroke-width: 0px;text-decoration-thickness: initial;text-decoration-style: initial;text-decoration-color: initial;">



03


如何利用这些Tokens?‍

在上一小节,我们介绍了使用不同方法将文本拆解为Tokens的方法。那么接下来的问题是我们要如何利用好这些Tokens呢?

如果您是新手,请不要担心您刚刚读到的这些关键词。简单来说,我们将文本输入转换成独立的单位,称为 "Tokens"。这样,稍后将它们转换成计算机能理解的 "数字 "就更容易了。

ingFang SC", system-ui, -apple-system, "Segoe UI", Rototo, Helvetica, Arial, sans-serif;font-size: 15px;font-style: normal;font-variant-ligatures: normal;font-variant-caps: normal;font-weight: 400;letter-spacing: normal;orphans: 2;text-align: start;text-indent: 0px;text-transform: none;widows: 2;word-spacing: 0px;-webkit-text-stroke-width: 0px;text-decoration-thickness: initial;text-decoration-style: initial;text-decoration-color: initial;">




04


ChatGPT中的Tokens‍

在像 ChatGPT 这样的 LLM 中,Tokens是什么样子的?用于 LLM 的tokenization 方法与用于一般 NLP 任务中的方法不同。
从广义上讲,我们可以称之为 "subword tokenization",我们创建的tokens不一定是完整的单词,就像我们在基于空格Token化中看到的那样。这正是一个单词不等于一个Token的原因。
当他们说 GPT-4 Turbo 的上下文长度为 128K 个Tokens时,其实并不完全是 128K个单词,而是一个接近的数字。

为什么要使用如此不同和复杂的tokenization方法?

可以归纳为以下原因:

这些Tokens是比完整单词更复杂的语言表征。

它们有助于处理大量词汇,包括罕见和未知词汇。

使用较小的子单元计算效率更高。

它们有助于更好地理解上下文。

它对不同语言的适应性更强,而这些语言可能与英语大相径庭。







05


字节对编码‍

许多开源模型,如 Meta 的 LLAMA-2 和较早的 GPT 模型,都使用这种字节对编码(Byte-Pair-Encoding)方法的一个版本。在现实应用中,BPE 会对大量文本进行分析,以确定最常见的配对。
让我们用 GPT-2 中的Tokenizer来举一个简单的例子。
fromtransformersimportGPT2Tokenizer#Initializethetokenizertokenizer=GPT2Tokenizer.from_pretrained("gpt2")text="It'sover9000!"#Tokenizethetexttoken_ids=tokenizer.encode(text,add_special_tokens=True)#OutputthetokenIDsprint("TokenIDs:",token_ids)#ConverttokenIDsbacktorawtokensandoutputthemraw_tokens=[tokenizer.decode([token_id])fortoken_idintoken_ids]print("Rawtokens:",raw_tokens)
输出如下:
TokenIDs:[1026,338,625,50138,0]Rawtokens:['It',"'s",'over','9000','!']

那么什么是 Token ID ? 它为什么会是一个数字呢?

让我们来分析一下这个过程是如何进行的。
建立词汇表是BPE算法的一部分,操作步骤如下:
为每个Token分配对应的Token ID常用方法如下:
关键的一点是,Tokens ID 的分配不是任意的,而是基于模型所训练的语言数据的出现频率和组合模式。这样,GPT-2 和类似的模型就能使用可管理的、有代表性的Tokens集来有效地处理和生成人类语言。

在这里,词汇表指的是模型可以识别和处理的所有独特的Token的集合。从本质上讲,它是在训练数据的帮助下,使用给定的tokenization 方法所创建的Tokens集合。

目前的大多数 LLM 都使用某种不同的 BPE。例如,Mistral 模型使用的方法为byte fallback BPE tokenizer。
除 BPE 外,还有一些其他方法,如单字词法、句片法和词片法。
别担心这些了。
现在,我们需要知道的是,创建Tokens是处理 NLP 或 LLM 的第一步。有不同的方法来创建Tokens,这些Tokens也被分配了一些Token ID, 表示该Token在词汇表中的索引。




06


什么是Embedding?‍

我们经常接触这个词。在讨论这个词之前,我们先来消除一些困惑。
  • Token ID 是tokens的直接数字表示。事实上,它们并不能捕捉Tokens之间任何更深层次的关系或模式
  • 标准的矢量化技术(如 TF-IDF)包括根据某些逻辑创建更复杂的数字表示。
  • 嵌入(Embeddings)是Tokens的高级向量表示。它们试图捕捉Tokens之间最细微的差别、联系和语义。每个嵌入通常是神经网络计算的向量空间上的一系列实数。
简而言之,输入文本被转换为Tokens。每个Token被赋予一个Token ID。这些Token ID 可用于创建Embedding,以便在复杂的模型中进行更细致的数字表示。

为什么要这样做?

因为计算机能理解数字,并通过数字进行运算。而Embeddings才是 LLM 的 "真正输入"。


ingFang SC", system-ui, -apple-system, "Segoe UI", Rototo, Helvetica, Arial, sans-serif;font-size: 15px;font-style: normal;font-variant-ligatures: normal;font-variant-caps: normal;font-weight: 400;letter-spacing: normal;orphans: 2;text-align: start;text-indent: 0px;text-transform: none;widows: 2;word-spacing: 0px;-webkit-text-stroke-width: 0px;text-decoration-thickness: initial;text-decoration-style: initial;text-decoration-color: initial;">





07


Token到Embedding的转换


就像不同的tokenization 化方法一样,我们也有多种Token到Embedding转换的方法。下面是一些常用的方法:

我们暂且不用担心每种方法的内部工作原理。大家只需知道,我们可以使用它们来创建计算机可以理解的文本数字表示。

我们以 BERT 创建Embedding为例来讲解:

fromtransformersimportBertTokenizer,BertModelimporttorch#Loadpre-trainedmodeltokenizertokenizer=BertTokenizer.from_pretrained('bert-base-uncased')#Loadpre-trainedmodelmodel=BertModel.from_pretrained('bert-base-uncased')#Texttobetokenizedtext="It'sover9000!"#Encodetextinput_ids=tokenizer.encode(text,add_special_tokens=True)#OutputthetokenIDsprint("TokenIDs:",input_ids)#ConverttokenIDsbacktorawtokensandoutputthemraw_tokens=[tokenizer.decode([token_id])fortoken_idininput_ids]print("Rawtokens:",raw_tokens)#ConvertlistofIDstoatensorinput_ids_tensor=torch.tensor([input_ids])#Passtheinputthroughthemodelwithtorch.no_grad()utputs=model(input_ids_tensor)#Extracttheembeddingsembeddings=outputs.last_hidden_state#Printtheembeddingsprint("Embeddings:",embeddings)

输出如下:

TokenIDs:[101,2009,1005,1055,2058,7706,2692,999,102]Rawtokens:['[CLS]','it',"'",'s','over','900','##0','!','[SEP]']Embeddings:tensor([[[0.1116,0.0722,0.3173,...,-0.0635,0.2166,0.3236],[-0.4159,-0.5147,0.5690,...,-0.2577,0.5710,0.4439],[-0.4893,-0.8719,0.7343,...,-0.3001,0.6078,0.3938],...,[-0.2746,-0.6479,0.2702,...,-0.4827,0.1755,-0.3939],[0.0846,-0.3420,0.0216,...,0.6648,0.3375,-0.2893],[0.6566,0.2011,0.0142,...,0.0786,-0.5767,-0.4356]]])

观察上述代码,可以知道:


如大家所见,Embeddings基本上就是数字数组。









08


Embeddings的作用


Embeddings处理过程如此庞大和复杂,那么它到底意味着什么呢?

简而言之,Embeddigns是使 LLMs 运行良好的秘诀。如果能找到更好的嵌入方法,就有可能创建出更好的模型。

当这些数字被训练有素的人工智能模型架构处理时,它就会以相同的格式计算出新的值,代表模型所训练任务的答案。在 LLM 中,这就是对下一个标记的预测。

在训练 LLM 时,我们基本上是在尝试优化模型中与输入嵌入相关的所有数学计算,以创建所需的输出。

所有此类计算都包含在一些称为模型权重的参数中。它们决定了模型如何处理输入数据以产生输出。


嵌入实际上是LLM模型权重的一个子集。它们通常是与嵌入层(在Transformer等模型中)(通常是第一层)相关的权重。
模型权重和嵌入可以作为随机变量初始化(或计算),也可以从预先训练好的模型中提取。然后在训练阶段更新这些值。
我们的目标是为模型权重找到合适的值,以便在输入的情况下,模型所做的计算能在给定的环境下产生最准确的输出。



09


结论


大模型在处理文本输入时,遵循Text -> Tokens -> Token IDs -> Embeddings处理过程,而Embeddings是 LLM 理解输入上下文语义的秘诀。此外,有许多不同的技术可以创建Token和Embedding,这对模型的工作方式有很大影响。


希望可以加深大家对大语言模型的理解。






欢迎光临 链载Ai (https://www.lianzai.com/) Powered by Discuz! X3.5