MobileBERT-SarcOji-KD-v2
mobilebert student fine-tuned on SarcOji with sarcasm-aware emoji embeddings injected
directly into the token-embedding matrix, distilled from Vandita/Bert-finetuned-Sarc.
Results
| split | accuracy | precision | recall | F1 | MCC | ROC-AUC |
|---|---|---|---|---|---|---|
| validation | 0.8623 | 0.8224 | 0.8242 | 0.8233 | 0.7105 | 0.9381 |
| SarcOjiTest1 | 0.6598 | 0.7212 | 0.5756 | 0.6402 | 0.3326 | 0.7352 |
| SarcOjiTest2 | 0.7249 | 0.4468 | 0.4260 | 0.4362 | 0.2545 | 0.7064 |
SarcOjiTest2 is ~75/25 negative-skewed, so its accuracy is not comparable to Test1's — MCC and ROC-AUC are the meaningful columns there.
Distillation
Hinton et al. (2015): L = α·T²·KL(teacher‖student) + (1−α)·CE, with T = 6,
α = 0.6. The T² factor is applied, so soft-target gradients do not shrink as T
rises.
Emoji handling
1444 emoji tokens added to the tokenizer (vocab
30522 → 31959). Each new row of the
128-d embedding matrix was initialised from a 300-d sarcasm-aware
emoji vector using pca. Remaining dimensions hold the mean of the
original vocabulary embeddings rather than zeros, so new tokens are not anchored to the
origin.
Because the emoji are part of the vocabulary, this is a standard HuggingFace model — no custom architecture, no custom loader.
Usage
from transformers import AutoModelForSequenceClassification, AutoTokenizer
tok = AutoTokenizer.from_pretrained("Vandita/MobileBERT-SarcOji-KD-v2")
model = AutoModelForSequenceClassification.from_pretrained("Vandita/MobileBERT-SarcOji-KD-v2")
Data
SarcOji, 80:20 stratified split. Duplicates retained; identical texts were kept on one side of the split so no text appears in both train and validation. Evaluated on SarcOjiTest1 and SarcOjiTest2 in full.