ONNX 模型库
返回模型

说明文档

XLM-RoBERTa(基础版模型)

XLM-RoBERTa 模型是在包含 100 种语言的 2.5TB 过滤后 CommonCrawl 数据上预训练得到的。该模型由 Conneau 等人在论文 Unsupervised Cross-lingual Representation Learning at Scale 中提出,并首次在此仓库中发布。

免责声明:发布 XLM-RoBERTa 的团队并未为该模型编写模型卡片,因此本模型卡片由 Hugging Face 团队编写。

模型描述

XLM-RoBERTa 是 RoBERTa 的多语言版本。它在包含 100 种语言的 2.5TB 过滤后 CommonCrawl 数据上进行了预训练。

RoBERTa 是一种以自监督方式在大规模语料库上预训练的 transformers 模型。这意味着它仅在原始文本上进行预训练,无需人工标注(这也是它能够使用大量公开数据的原因),通过自动流程从文本中生成输入和标签。

更准确地说,它是通过掩码语言建模(Masked Language Modeling,MLM)目标进行预训练的。给定一个句子,模型会随机掩码输入中 15% 的单词,然后将整个被掩码的句子传入模型,让其预测被掩码的单词。这与传统的循环神经网络(RNN)不同——RNN 通常按顺序逐个处理单词,也不同于 GPT 等自回归模型(内部会掩码未来的 token)。这种方式使模型能够学习句子的双向表示。

通过这种方式,模型学习了 100 种语言的内部表示,随后可用于提取对下游任务有用的特征:例如,如果你有已标注的句子数据集,可以使用 XLM-RoBERTa 模型产生的特征作为输入,训练一个标准的分类器。

预期用途与限制

你可以直接使用原始模型进行掩码语言建模,但它主要用于在下游任务上进行微调。请访问模型中心查找你感兴趣的任务上已微调的版本。

请注意,该模型主要针对使用整个句子(可能包含掩码)进行决策的任务进行微调,例如序列分类、token 分类或问答任务。对于文本生成等任务,你应该查看 GPT2 等模型。

使用方法

你可以直接使用流水线进行掩码语言建模:

>>> from transformers import pipeline
>>> unmasker = pipeline('fill-mask', model='xlm-roberta-base')
>>> unmasker("Hello I'm a <mask> model.")

[{'score': 0.10563907772302628,
  'sequence': "Hello I'm a fashion model.",
  'token': 54543,
  'token_str': 'fashion'},
 {'score': 0.08015287667512894,
  'sequence': "Hello I'm a new model.",
  'token': 3525,
  'token_str': 'new'},
 {'score': 0.033413201570510864,
  'sequence': "Hello I'm a model model.",
  'token': 3299,
  'token_str': 'model'},
 {'score': 0.030217764899134636,
  'sequence': "Hello I'm a French model.",
  'token': 92265,
  'token_str': 'French'},
 {'score': 0.026436051353812218,
  'sequence': "Hello I'm a sexy model.",
  'token': 17473,
  'token_str': 'sexy'}]

以下是如何在 PyTorch 中使用该模型获取给定文本特征的方法:

from transformers import AutoTokenizer, AutoModelForMaskedLM

tokenizer = AutoTokenizer.from_pretrained('xlm-roberta-base')
model = AutoModelForMaskedLM.from_pretrained("xlm-roberta-base")

# 准备输入
text = "Replace me by any text you'd like."
encoded_input = tokenizer(text, return_tensors='pt')

# 前向传播
output = model(**encoded_input)

BibTeX 条目和引用信息

@article{DBLP:journals/corr/abs-1911-02116,
  author    = {Alexis Conneau and
               Kartikay Khandelwal and
               Naman Goyal and
               Vishrav Chaudhary and
               Guillaume Wenzek and
               Francisco Guzm{\'{a}}n and
               Edouard Grave and
               Myle Ott and
               Luke Zettlemoyer and
               Veselin Stoyanov},
  title     = {Unsupervised Cross-lingual Representation Learning at Scale},
  journal   = {CoRR},
  volume    = {abs/1911.02116},
  year      = {2019},
  url       = {http://arxiv.org/abs/1911.02116},
  eprinttype = {arXiv},
  eprint    = {1911.02116},
  timestamp = {Mon, 11 Nov 2019 18:38:09 +0100},
  biburl    = {https://dblp.org/rec/journals/corr/abs-1911-02116.bib},
  bibsource = {dblp computer science bibliography, https://dblp.org}
}

<a href="https://huggingface.co/exbert/?model=xlm-roberta-base"> <img width="300px" src="https://cdn-media.huggingface.co/exbert/button.png"> </a>

Ewel/FacebookAI_xlm_roberta_base

作者 Ewel

↓ 1 ♥ 0

创建时间: 2025-05-22 20:24:41+00:00

更新时间: 2025-05-22 20:26:14+00:00

在 Hugging Face 上查看

文件 (12)

.gitattributes
.ipynb_checkpoints/config-checkpoint.json
README.md
config.json
flax_model.msgpack
model.onnx ONNX
model.safetensors
pytorch_model.bin
sentencepiece.bpe.model
tf_model.h5
tokenizer.json
tokenizer_config.json