Edit model card


This model is an adversarially fine-tuned version of aubmindlab/bert-base-arabertv2 on the Hotel Arabic Reviews Dataset (HARD) dataset, augmented by our generated adversarial examples. It achieves the following results on

  • Loss: 0.4285
  • Accuracy: 0.8267

Model description

BibTeX Citation:

    title = "{Arabic Synonym BERT-based Adversarial Examples for Text Classification}",
    author = "Alshahrani, Norah  and Alshahrani, Saied  and Wali, Esma  and Matthews, Jeanna ",
    booktitle = "Proceedings of the 18th Conference of the European Chapter of the Association for Computational Linguistics: Student Research Workshop",
    month = March,
    year = "2024",
    address = "Malta",
    publisher = "Association for Computational Linguistics",
    abstract = "Text classification systems have been proven vulnerable to adversarial text examples, modified versions of the original text examples that are often unnoticed by human eyes, yet can force text classification models to alter their classification. Often, research works quantifying the impact of adversarial text attacks have been applied only to models trained in English. In this paper, we introduce the first word-level study of adversarial attacks in Arabic. Specifically, we use a synonym (word-level) attack using a Masked Language Modeling (MLM) task with a BERT model in a black-box setting to assess the robustness of the state-of-the-art text classification models to adversarial attacks in Arabic. To evaluate the grammatical and semantic similarities of the newly produced adversarial examples using our synonym BERT-based attack, we invite four human evaluators to assess and compare the produced adversarial examples with their original examples. We also study the transferability of these newly produced Arabic adversarial examples to various models and investigate the effectiveness of defense mechanisms against these adversarial examples on the BERT models. We find that fine-tuned BERT models were more susceptible to our synonym attacks than the other Deep Neural Networks (DNN) models like WordCNN and WordLSTM we trained. We also find that fine-tuned BERT models were more susceptible to transferred attacks. We, lastly, find that fine-tuned BERT models successfully regain at least 2% in accuracy after applying adversarial training as an initial defense mechanism.",

Training procedure

We have trained this model using the PaperSpace GPU-Cloud service. We used a machine with 8 CPUs, 45GB RAM, and A6000 GPU with 48GB RAM.

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 2e-05
  • train_batch_size: 16
  • eval_batch_size: 8
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • num_epochs: 3

Training results

Training Loss Epoch Step Validation Loss Accuracy
0.4562 1.0 5978 0.4058 0.8257
0.3904 2.0 11956 0.4075 0.8289
0.3508 3.0 17934 0.4285 0.8267

Framework versions

  • Transformers 4.32.1
  • Pytorch 1.12.1+cu116
  • Datasets 2.4.0
  • Tokenizers 0.12.1
Downloads last month
Inference Examples
Inference API (serverless) has been turned off for this model.

Finetuned from

Dataset used to train NorahAlshahrani/Adv_BERT_Hard