当前位置：首页 > article >正文

【LLM】一文了解 NLP 里程碑模型 BERT

article 2025/2/28 19:48:27

一、BERT 的基本介绍

BERT（Bidirectional Encoder Representations from Transformers）是一种用于自然语言处理（NLP）的深度学习模型，由谷歌于 2018 年提出，是一个深刻改变 NLP 领域的里程碑模型，通过其双向上下文表示和预训练-微调的策略，在文本理解任务中取得了显著成果。
BERT 是 Transformer 架构的一个变体，尤其擅长处理语言理解任务，是许多 NLP 应用的核心技术。尽管其计算需求较高，优化版本（如 DistilBERT）已解决部分问题，广泛应用于搜索引擎、问答系统等实际场景。

1. BERT 的核心特点

双向上下文表示
与传统的单向语言模型（如 GPT）不同，BERT 在对文本进行编码时，同时关注上下文中的前后单词，生成更完整的语义表示。这种双向性使 BERT 能够捕捉句子中单词与上下文之间的深层联系。
Transformer 架构
BERT 的核心是 Transformer 的 Encoder 部分，完全基于自注意力机制（Self-Attention）。它通过堆叠多个编码层（通常为 12 层或 24 层）来提取句子的深层语义信息。
预训练 + 微调
- 预训练：BERT 在大规模语料库（如 Wikipedia 和 BookCorpus）上进行两种任务的预训练：
  1. Masked Language Model (MLM)：随机遮掩部分单词，让模型预测它们。
  2. Next Sentence Prediction (NSP)：判断两段文本是否连续。
- 微调：将预训练后的 BERT 模型应用到具体的下游任务（如分类、问答），通过较少的数据快速适配。

2. BERT 的适用场景

BERT 广泛应用于多种 NLP 任务：

文本分类：情感分析、垃圾邮件检测等。
问答系统：如 SQuAD 数据集上的答案抽取任务。
命名实体识别 (NER)：提取文本中的关键实体（如人名、地名）。
机器翻译和文本生成：尽管 BERT 更专注于语言理解，改进版本（如 T5）也应用于生成任务。
信息检索：谷歌搜索已经将 BERT 集成到搜索引擎中，用于更准确地理解用户查询。

3. BERT 的技术优势

语言理解能力强
双向性和深层语义建模使其在语言理解任务中远超传统方法。
模块化结构
预训练模型可轻松迁移到不同任务，仅需少量微调。
高效利用大规模数据
预训练阶段充分利用大语料库，能在下游任务上实现更好的泛化。

4. BERT 的局限性

计算开销大：由于双向注意力机制，BERT 的训练和推理需要大量计算资源。
难以处理长序列：基础版 BERT 的最大输入长度为 512 个词，长文本处理可能需要分段。
生成能力有限：原版 BERT 专注于语言理解，无法直接用于文本生成任务。

5. Hugging Face 调用 BERT 的代码

以下代码展示了如何使用预训练的 BERT 模型进行文本分类：

from transformers import BertTokenizer, BertForSequenceClassification
import torch

# 加载预训练的 BERT 模型和分词器
tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")
model = BertForSequenceClassification.from_pretrained("bert-base-uncased", num_labels=2)

# 输入文本
text = "BERT is amazing!"
inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True)

# 模型推理
outputs = model(**inputs)
logits = outputs.logits
predicted_class = torch.argmax(logits, dim=1).item()

print(f"Predicted class: {predicted_class}")

二、BERT 和 Transformer 之间的关系

在谷歌的官方课程中，BERT 和 Transformer 会被联合讲解，故在此归纳 BERT 和 Transformer 的关系：

1. BERT 是 Transformer 的具体实现

Transformer 是通用的序列建模架构，而 BERT 是其在 NLP 任务上的典型应用实例。Transformer 是一种深度学习架构，专注于处理序列数据（如文本），由 Vaswani 等人在 2017 年提出，基于 自注意力机制（Self-Attention）。
Transformer 包含两个主要部分：Encoder（编码器），即处理输入序列，生成上下文嵌入表示；Decoder（解码器），即基于上下文表示生成输出序列。
而 BERT 是 Transformer 的 Encoder 部分，并去除了 Decoder，因为它的设计目标是处理语言理解任务，而不是生成任务。BERT 的双向上下文建模能力完全依赖 Transformer Encoder 的自注意力机制。
所以，学习 Transformer 可以掌握构建序列模型的通用方法。而通过 BERT，可以看到 Transformer 的实际效果，理解如何通过预训练和微调解决特定的 NLP 问题。

2. 技术理论关联

核心机制相同：BERT 的工作原理和 Transformer Encoder 的原理完全一致，都依赖 多头自注意力机制 和 前馈神经网络（FFN）。
技术继承关系：BERT 在训练阶段使用了改进的预训练策略（如 Masked Language Model 和 Next Sentence Prediction），但架构和 Transformer Encoder 几乎完全一致。因此，理解 Transformer 是理解 BERT 的基础。