gilramos's picture
Update README.md
190e59b verified
|
raw
history blame
7.36 kB
metadata
license: cc
language:
  - pt
tags:
  - Hate Speech
  - kNOwHATE
widget:
  - text: >-
      as pessoas tem que perceber que ser 'panasca' não é deixar de ser homem, é
      deixar de ser humano 😂😂
pipeline_tag: text-classification

    This is the model card for HateBERTimbau. You may be interested in some of the other models from the kNOwHATE project.


HateBERTimbau

HateBERTimbau is a foundation, large language model for European Portuguese from Portugal for Hate Speech content.

It is an encoder of the BERT family, based on the neural architecture Transformer and developed over the BERTimbau model, retrained on a dataset of 229,103 tweets specifically focused on potential hate speech.

Model Description

Several models were developed by fine-tuning Base HateBERTimbau for Hate Speech detection present in the table bellow:

Uses

You can use this model directly with a pipeline for masked language modeling:

from transformers import pipeline
unmasker = pipeline('fill-mask', model='knowhate/HateBERTimbau')

unmasker("Os [MASK] são todos uns animais, deviam voltar para a sua terra.")

[{'score': 0.6771652698516846,
  'token': 12714,
  'token_str': 'africanos',
  'sequence': 'Os africanos são todos uns animais, deviam voltar para a sua terra.'},
 {'score': 0.08679857850074768,
  'token': 15389,
  'token_str': 'homossexuais',
  'sequence': 'Os homossexuais são todos uns animais, deviam voltar para a sua terra.'},
 {'score': 0.03806231543421745,
  'token': 4966,
  'token_str': 'portugueses',
  'sequence': 'Os portugueses são todos uns animais, deviam voltar para a sua terra.'},
 {'score': 0.035253893584012985,
  'token': 16773,
  'token_str': 'Portugueses',
  'sequence': 'Os Portugueses são todos uns animais, deviam voltar para a sua terra.'},
 {'score': 0.023521048948168755,
  'token': 8618,
  'token_str': 'brancos',
  'sequence': 'Os brancos são todos uns animais, deviam voltar para a sua terra.'}]

Or this model can be used by fine-tuning it for a specific task/dataset:

from transformers import AutoTokenizer, AutoModelForSequenceClassification, TrainingArguments, Trainer
from datasets import load_dataset

tokenizer = AutoTokenizer.from_pretrained("knowhate/HateBERTimbau")
model = AutoModelForSequenceClassification.from_pretrained("knowhate/HateBERTimbau")
dataset = load_dataset("knowhate/youtube-train")

def tokenize_function(examples):
    return tokenizer(examples["sentence1"], examples["sentence2"], padding="max_length", truncation=True)

tokenized_datasets = dataset.map(tokenize_function, batched=True)

training_args = TrainingArguments(output_dir="hatebertimbau", evaluation_strategy="epoch")
trainer = Trainer(
    model=model,
    args=training_args,
    train_dataset=tokenized_datasets["train"],
    eval_dataset=tokenized_datasets["validation"],
)

trainer.train()

Training

Data

229,103 tweets associated with offensive content were used to retrain the base model.

Training Hyperparameters

  • Batch Size: 4 samples
  • Epochs: 100
  • Learning Rate: 5e-5 with Adam optimizer
  • Maximum Sequence Length: 512 sentence pieces

Testing

Data

We used two different datasets for testing, one for YouTube comments here and another for Tweets here.

Hate Speech Classification Results (with no fine-tuning)

Dataset Precision Recall F1-score
YouTube 0.928 0.108 0.193
Twitter 0.686 0.211 0.323

BibTeX Citation

@mastersthesis{Matos-Automatic-Hate-Speech-Detection-in-Portuguese-Social-Media-Text,
title = {{Automatic Hate Speech Detection in Portuguese Social Media Text}},
author = {Matos, Bernardo Cunha},
month = nov,
year = {2022},
abstract = {{Online Hate Speech (HS) has been growing dramatically on social media and its uncontrolled spread has motivated researchers to develop a diversity of methods for its automated detection. However, the detection of online HS in Portuguese still merits further research. To fill this gap, we explored different models that proved to be successful in the literature to address this task. In particular, we have explored models that use the BERT architecture. Beyond testing single-task models we also explored multitask models that use the information on other related categories to learn HS. To better capture the semantics of this type of texts, we developed HateBERTimbau, a retrained version of BERTimbau more directed to social media language including potential HS targeting African descent, Roma, and LGBTQI+ communities. The performed experiments were based on CO-HATE and FIGHT, corpora of social media messages posted by the Portuguese online community that were labelled regarding the presence of HS among other categories.
The results achieved show the importance of considering the annotator's agreement on the data used to develop HS detection models. Comparing different subsets of data used for the training of the models it was shown that, in general, a higher agreement on the data leads to better results.
HATEBERTimbau consistently outperformed BERTimbau on both datasets confirming that further pre-training of BERTimbau was a successful strategy to obtain a language model more suitable for online HS detection in Portuguese. 
The implementation of target-specific models, and multitask learning have shown potential in obtaining better results.}},
language = {eng},
copyright = {embargoed-access},
}

Acknowledgements

This work was funded in part by the European Union under Grant CERV-2021-EQUAL (101049306). However the views and opinions expressed are those of the author(s) only and do not necessarily reflect those of the European Union or Knowhate Project. Neither the European Union nor the Knowhate Project can be held responsible.