R07-marbert: multi-label Arabic dialect identification

Fine-tuned from UBC-NLP/MARBERT for multi-label country-level dialect identification (18 sigmoid outputs, one per country). Bachelor thesis project, German International University.

  • Training data: NADI 2020/2021/2023 training tweets with the LahjatBERT multi-label labels (56,113 unique texts, file nadi_lahjatbert_card1to18.jsonl; links and mentions replaced, no other normalization).
  • Recipe: lower 8 encoder layers frozen, dropout 0.3, AdamW lr 5e-05, 500 warmup steps, batch 24, 2 epochs, fp16, max length 128, seed 42; epoch chosen by macro F1 on a 10% validation split of the training data.
  • Use: sigmoid over the 18 logits, a dialect is valid if its probability is >= 0.3 (MLADI leaderboard method predict_binary_outcomes).
  • Label order: Algeria, Bahrain, Egypt, Iraq, Jordan, Kuwait, Lebanon, Libya, Morocco, Oman, Palestine, Qatar, Saudi_Arabia, Sudan, Syria, Tunisia, UAE, Yemen.
  • MLADI dev set (120 sentences, 8 dialects): macro F1 72.96, precision 74.40, recall 73.23.
  • Training run card1to18_real_only_seed42, code commit c11a0b84f7c6eb8042bbd06b9a01a099dea40645.
Downloads last month
16
Safetensors
Model size
0.2B params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for Ammar-06/mladi-marbert-r07

Base model

UBC-NLP/MARBERT
Finetuned
(29)
this model