GitHub

Please use 'Bert' related functions to load this model!

This repository contains the resources in our paper "Revisiting Pre-trained Models for Chinese Natural Language Processing", which will be published in "Findings of EMNLP". You can read our camera-ready paper through ACL Anthology or arXiv pre-print.

Revisiting Pre-trained Models for Chinese Natural Language Processing
Yiming Cui, Wanxiang Che, Ting Liu, Bing Qin, Shijin Wang, Guoping Hu

You may also interested in,

More resources by HFL: https://github.com/ymcui/HFL-Anthology

Introduction

MacBERT is an improved BERT with novel MLM as correction pre-training task, which mitigates the discrepancy of pre-training and fine-tuning.

Instead of masking with [MASK] token, which never appears in the fine-tuning stage, we propose to use similar words for the masking purpose. A similar word is obtained by using Synonyms toolkit (Wang and Hu, 2017), which is based on word2vec (Mikolov et al., 2013) similarity calculations. If an N-gram is selected to mask, we will find similar words individually. In rare cases, when there is no similar word, we will degrade to use random word replacement.

Here is an example of our pre-training task. | | Example | | -------------- | ----------------- | | Original Sentence | we use a language model to predict the probability of the next word. | | MLM | we use a language [M] to [M] ##di ##ct the pro [M] ##bility of the next word . | | Whole word masking | we use a language [M] to [M] [M] [M] the [M] [M] [M] of the next word . | | N-gram masking | we use a [M] [M] to [M] [M] [M] the [M] [M] [M] [M] [M] next word . | | MLM as correction | we use a text system to ca ##lc ##ulate the po ##si ##bility of the next word . |

Except for the new pre-training task, we also incorporate the following techniques.

  • Whole Word Masking (WWM)
  • N-gram masking
  • Sentence-Order Prediction (SOP)

Note that our MacBERT can be directly replaced with the original BERT as there is no differences in the main neural architecture.

For more technical details, please check our paper: Revisiting Pre-trained Models for Chinese Natural Language Processing

Citation

If you find our resource or paper is useful, please consider including the following citation in your paper.

  • https://arxiv.org/abs/2004.13922
    @inproceedings{cui-etal-2020-revisiting,
      title = "Revisiting Pre-Trained Models for {C}hinese Natural Language Processing",
      author = "Cui, Yiming  and
        Che, Wanxiang  and
        Liu, Ting  and
        Qin, Bing  and
        Wang, Shijin  and
        Hu, Guoping",
      booktitle = "Proceedings of the 2020 Conference on Empirical Methods in Natural Language Processing: Findings",
      month = nov,
      year = "2020",
      address = "Online",
      publisher = "Association for Computational Linguistics",
      url = "https://www.aclweb.org/anthology/2020.findings-emnlp.58",
      pages = "657--668",
    }
    
New

Select AutoNLP in the “Train” menu to fine-tune this model automatically.

Downloads last month
8,024
Hosted inference API
Fill-Mask
Mask token: [MASK]
Examples
Examples
This model can be loaded on the Inference API on-demand.