CoolFace
Modelpublic

KM4STfulltext/CSSCI_ABS_roberta_wwm

sourceHugging Faceapache-2.0updated 4y agoView on Hugging Face
0likes58downloads
Model Card

Pre-trained Language Model for the Humanities and Social Sciences in Chinese

Introduction

The research for social science texts in Chinese needs the support natural language processing tools.

The pre-trained language model has greatly improved the accuracy of text mining in general texts. At present, there is an urgent need for a pre-trained language model specifically for the automatic processing of scientific texts in Chinese social science.

We used the abstract of social science research as the training set. Based on the deep language model framework of BERT, we constructed CSSCIABSBERT, CSSCIABSroberta and CSSCIABSroberta-wwm pre-training language models by transformers/run_mlm.py and transformers/mlm_wwm.

We designed four downstream tasks of Text Classification on different Chinese social scientific article corpus to verify the performance of the model.

  • CSSCIABSBERT , CSSCIABSroberta and CSSCIABSroberta-wwm are trained on the abstract of articles published in CSSCI journals. The training set involved in the experiment included a total of 510,956,094 words.
  • Based on the idea of Domain-Adaptive Pretraining, CSSCI_ABS_BERT and CSSCI_ABS_roberta combine a large amount of abstracts of scientific articles in Chinese based on the BERT structure, and continue to train the BERT and Chinese-RoBERTa models respectively to obtain pre-training models for the automatic processing of Chinese Social science research texts.

News

  • 2022-06-15 : CSSCIABSBERT, CSSCIABSroberta and CSSCIABSroberta-wwm has been put forward for the first time.

How to use

Huggingface Transformers

The from_pretrained method based on Huggingface Transformers can directly obtain CSSCIABSBERT, CSSCIABSroberta and CSSCIABSroberta-wwm models online.

  • CSSCIABSBERT
python
from transformers import AutoTokenizer, AutoModel

tokenizer = AutoTokenizer.from_pretrained("KM4STfulltext/CSSCI_ABS_BERT")

model = AutoModel.from_pretrained("KM4STfulltext/CSSCI_ABS_BERT")
  • CSSCIABSroberta
python
from transformers import AutoTokenizer, AutoModel

tokenizer = AutoTokenizer.from_pretrained("KM4STfulltext/CSSCI_ABS_roberta")

model = AutoModel.from_pretrained("KM4STfulltext/CSSCI_ABS_roberta")
  • CSSCIABSroberta-wwm
python
from transformers import AutoTokenizer, AutoModel

tokenizer = AutoTokenizer.from_pretrained("KM4STfulltext/CSSCI_ABS_roberta_wwm")

model = AutoModel.from_pretrained("KM4STfulltext/CSSCI_ABS_roberta_wwm")

Download Models

  • The version of the model we provide is PyTorch.

From Huggingface

Evaluation & Results

  • We useCSSCIABSBERT, CSSCIABSroberta and CSSCIABSroberta-wwm to perform Text Classificationon different social science research corpus. The experimental results are as follows.
Discipline classification experiments of articles published in CSSCI journals

https://github.com/S-T-Full-Text-Knowledge-Mining/CSSCI-BERT

Movement recognition experiments for data analysis and knowledge discovery abstract
Tagbert-base-Chinesechinese-roberta-wwm,extCSSCI_ABS_BERTCSSCI_ABS_robertaCSSCI_ABS_roberta_wwmsupport
Abstract55.2362.4456.857.9658.26223
Location61.6154.3861.8361.461.942866
Metric45.084145.2746.7447.13622
Organization46.8535.2945.7245.4444.65327
Person88.6682.7988.2188.2988.514850
Thing71.6865.3471.8871.6871.815993
Time65.3560.3864.1565.2666.031272
avg72.6966.6272.5972.6172.8916153
Chinese literary entity recognition
Tagbert-base-Chinesechinese-roberta-wwm,extCSSCI_ABS_BERTCSSCI_ABS_robertaCSSCI_ABS_roberta_wwmsupport
Abstract55.2362.4456.857.9658.26223
Location61.6154.3861.8361.461.942866
Metric45.084145.2746.7447.13622
Organization46.8535.2945.7245.4444.65327
Person88.6682.7988.2188.2988.514850
Thing71.6865.3471.8871.6871.815993
Time65.3560.3864.1565.2666.031272
avg72.6966.6272.5972.6172.8916153

Cited

Disclaimer

  • The experimental results presented in the report only show the performance under a specific data set and hyperparameter combination, and cannot represent the essence of each model. The experimental results may change due to random number seeds and computing equipment.
  • Users can use the model arbitrarily within the scope of the license, but we are not responsible for the direct or indirect losses caused by using the content of the project.

Acknowledgment

  • CSSCIABSBERT was trained based on BERT-Base-Chinese](https://github.com/google-research/bert)).
  • CSSCIABSroberta and CSSCIABSroberta-wwm was trained based on RoBERTa-wwm-ext, Chinese](https://github.com/ymcui/Chinese-BERT-wwm)).