Flaubert Small Cased

🚀 FlauBERT：面向法語的無監督語言模型預訓練

FlauBERT 是一個在非常龐大且多樣化的法語語料庫上訓練的法語 BERT 模型。不同規模的模型藉助法國國家科學研究中心（CNRS）的 Jean Zay 超級計算機進行訓練。

與 FlauBERT 一同推出的還有 FLUE，這是一個類似於流行的 GLUE 基準測試的法語自然語言處理系統評估方案。其目標是在未來推動可復現的實驗，並在法語領域共享模型和研究進展。更多詳情請參考 官方網站。

🚀 快速開始

FlauBERT 是在大規模法語語料上預訓練的語言模型，可用於多種法語自然語言處理任務。下面將介紹如何使用它。

✨ 主要特性

• 多規模模型：提供了不同規模的模型，如 flaubert-small-cased、flaubert-base-uncased 等，可根據具體需求選擇。
• 評估方案：配套 FLUE 評估方案，方便對模型進行評估。

📦 安裝指南

文檔未提及具體安裝步驟，可參考 Hugging Face 的 Transformers 庫安裝相關依賴。

💻 使用示例

基礎用法

import torch
from transformers import FlaubertModel, FlaubertTokenizer

# Choose among ['flaubert/flaubert_small_cased', 'flaubert/flaubert_base_uncased', 
#               'flaubert/flaubert_base_cased', 'flaubert/flaubert_large_cased']
modelname = 'flaubert/flaubert_base_cased' 

# Load pretrained model and tokenizer
flaubert, log = FlaubertModel.from_pretrained(modelname, output_loading_info=True)
flaubert_tokenizer = FlaubertTokenizer.from_pretrained(modelname, do_lowercase=False)
# do_lowercase=False if using cased models, True if using uncased ones

sentence = "Le chat mange une pomme."
token_ids = torch.tensor([flaubert_tokenizer.encode(sentence)])

last_layer = flaubert(token_ids)[0]
print(last_layer.shape)
# torch.Size([1, 8, 768])  -> (batch size x number of tokens x embedding dimension)

# The BERT [CLS] token correspond to the first hidden state of the last layer
cls_embedding = last_layer[:, 0, :]

注意事項

⚠️ 重要提示

如果你的 transformers 版本小於等於 2.10.0，modelname 應取以下值之一：

['flaubert-small-cased', 'flaubert-base-uncased', 'flaubert-base-cased', 'flaubert-large-cased']

📚 詳細文檔

FlauBERT 模型

⚠️ 重要提示

flaubert-small-cased 是部分訓練的，因此不能保證其性能。建議僅將其用於調試目的。

📄 許可證

本項目採用 MIT 許可證。

🔗 參考文獻

如果你在科學出版物中使用了 FlauBERT 或 FLUE 基準測試，或者發現本倉庫中的資源很有用，請引用以下論文之一：

LREC 論文

@InProceedings{le2020flaubert,
  author    = {Le, Hang  and  Vial, Lo\"{i}c  and  Frej, Jibril  and  Segonne, Vincent  and  Coavoux, Maximin  and  Lecouteux, Benjamin  and  Allauzen, Alexandre  and  Crabb\'{e}, Beno\^{i}t  and  Besacier, Laurent  and  Schwab, Didier},
  title     = {FlauBERT: Unsupervised Language Model Pre-training for French},
  booktitle = {Proceedings of The 12th Language Resources and Evaluation Conference},
  month     = {May},
  year      = {2020},
  address   = {Marseille, France},
  publisher = {European Language Resources Association},
  pages     = {2479--2490},
  url       = {https://www.aclweb.org/anthology/2020.lrec-1.302}
}

TALN 論文

@inproceedings{le2020flaubert,
  title         = {FlauBERT: des mod{\`e}les de langue contextualis{\'e}s pr{\'e}-entra{\^\i}n{\'e}s pour le fran{\c{c}}ais},
  author        = {Le, Hang and Vial, Lo{\"\i}c and Frej, Jibril and Segonne, Vincent and Coavoux, Maximin and Lecouteux, Benjamin and Allauzen, Alexandre and Crabb{\'e}, Beno{\^\i}t and Besacier, Laurent and Schwab, Didier},
  booktitle     = {Actes de la 6e conf{\'e}rence conjointe Journ{\'e}es d'{\'E}tudes sur la Parole (JEP, 31e {\'e}dition), Traitement Automatique des Langues Naturelles (TALN, 27e {\'e}dition), Rencontre des {\'E}tudiants Chercheurs en Informatique pour le Traitement Automatique des Langues (R{\'E}CITAL, 22e {\'e}dition). Volume 2: Traitement Automatique des Langues Naturelles},
  pages         = {268--278},
  year          = {2020},
  organization  = {ATALA}
}