BIHEADS / README.md
T0KII's picture
Create README.md
5659656 verified

BIHEADS: Multi-Task BiLSTM+BiGRU for Egyptian Arabic

BIHEADS is a recurrent neural network ensemble designed to provide a complementary perspective to transformer-based models for Egyptian Arabic. It uses FastText embeddings and a parallel architecture to capture temporal nuances in dialectal speech.

Model Details

  • Architecture: Parallel BiLSTM + BiGRU shared backbone
  • Embeddings: FastText facebook/fasttext-arz-vectors (300-dim)
  • Hidden Size: 256 per direction
  • Layers: 2-layer LSTM, 2-layer GRU

Task Heads & Labels

  • Emotion (8 classes): none, anger, joy, sadness, love, sympathy, surprise, fear
  • Sentiment (3 classes): negative, neutral, positive
  • Sarcasm (2 classes): not sarcastic, sarcastic

Training Data

The model utilizes the same Egyptian-filtered training splits as MASRIHEADS to ensure feature alignment for downstream ensemble learners.

Performance (Test Set F1)

Task Macro-F1
Sarcasm 0.6293
Sentiment 0.6284
Emotion 0.5514
Mean 0.6030