This is the official Git repository page for the paper:
Gaurish Thakkar, Sherzod Hakimov and Marko Tadić. 2024.
"M2SA: Multimodal and Multilingual Model for Sentiment Analysis of Tweets".
The 2024 Joint International Conference on Computational Linguistics, Language Resources and Evaluation (LREC-COLING 2024), Torino (Italia)
- Paper
- Dataset with tweetid and label only
- Models multimodal, text-only
- Demo multimodal, text-only
For access to images and tweets, send an email with organization (university/institute) and purpose/usage details to [email protected]
Create a virtual environment in Python 3.10 and run the following scripts
pip install -r requirements.txt
python twitter_curation/curate.py
2. Language translation using https://huggingface.co/facebook/nllb-200-3.3B model
python twitter_curation/translation.py
1.1 To train model with DINOv2 and mBERT run
# Define an array of seeds
SEEDS=(42 123 777 2020 31337)
LANGUAGES=("ar" "bg" "bs" "da" "de" "en" "es" "fr" "hr" "hu" "it" "mt" "pl" "pt" "ru" "sr" "sv" "tr" "zh" "lv" "sq" "bg_mt" "bs_mt" "da_mt" "fr_mt" "hr_mt" "hu_mt" "mt_mt" "pt_mt" "ru_mt" "sr_mt" "sv_mt" "tr_mt" "zh_mt")
# Use a for loop to iterate over the alphabets
for LANG in "${LANGUAGES[@]}"
do
echo "Language: $LANG"
for seed in "${SEEDS[@]}"
do
echo "Seed: $seed"
cuda-wrapper.sh python src_VTDEM/train-dino-mbert.py \
--output_dir ./mbert-dino-finetuned-"$LANG"-"$seed" \
--model_name_or_path bert-base-multilingual-uncased \
--image_processor_name facebook/dinov2-base \
--dataset_name "$LANG" \
--data_dir ./data \
--image_column image_paths \
--caption_column normalized_text \
--label_column label \
--remove_unused_columns False \
--do_train \
--do_eval \
--do_predict \
--num_train_epochs="50" \
--per_device_train_batch_size 8 \
--per_device_eval_batch_size 8 \
--auto_find_batch_size \
--learning_rate 5e-5 \
--warmup_steps 0 \
--weight_decay 0.1 \
--overwrite_output_dir yes \
--seed "$seed" \
--load_best_model_at_end yes \
--evaluation_strategy steps \
--fp16 \
--optim adafactor \
--disable_tqdm="True" \
--save_total_limit 2
done
done
To train with CLIP-mBERT run
SEEDS=(42 123 777 2020 31337)
LANGUAGES=("ar" "bg" "bs" "da" "de" "en" "es" "fr" "hr" "hu" "it" "mt" "pl" "pt" "ru" "sr" "sv" "tr" "zh" "lv" "sq" "bg_mt" "bs_mt" "da_mt" "fr_mt" "hr_mt" "hu_mt" "mt_mt" "pt_mt" "ru_mt" "sr_mt" "sv_mt" "tr_mt" "zh_mt")
for LANG in "${LANGUAGES[@]}"
do
echo "Language: $LANG"
for seed in "${SEEDS[@]}"
do
echo "Seed: $seed"
cuda-wrapper.sh python src_VTDEM/train.py \
--output_dir ./clip-mbert-finetuned-"$LANG"-"$seed" \
--model_name_or_path ./clip-mbert \
--dataset_name "$LANG" \
--data_dir ./data \
--image_column image_paths \
--caption_column normalized_text \
--label_column label \
--remove_unused_columns False \
--do_train \
--do_eval \
--do_predict \
--num_train_epochs="50" \
--per_device_train_batch_size 8 \
--per_device_eval_batch_size 8 \
--auto_find_batch_size \
--learning_rate 5e-5 \
--warmup_steps 0 \
--weight_decay 0.1 \
--overwrite_output_dir yes \
--seed "$seed" \
--load_best_model_at_end yes \
--evaluation_strategy steps \
--fp16 \
--optim adafactor
done
done
- To train text only models run
# Define an array of seeds
SEEDS=(42 123 777 2020 31337)
LANGUAGES=("ar" "bg" "bs" "da" "de" "en" "es" "fr" "hr" "hu" "it" "mt" "pl" "pt" "ru" "sr" "sv" "tr" "zh" "lv" "sq" "bg_mt" "bs_mt" "da_mt" "fr_mt" "hr_mt" "hu_mt" "mt_mt" "pt_mt" "ru_mt" "sr_mt" "sv_mt" "tr_mt" "zh_mt")
# Use a for loop to iterate over the alphabets
for LANG in "${LANGUAGES[@]}"
do
echo "Language: $LANG"
for seed in "${SEEDS[@]}"
do
echo "Seed: $seed"
cuda-wrapper.sh python src_VTDEM/train-text-only.py \
--output_dir ./cardiffnlp-twitter-xlmr-finetuned-txtnly-"$LANG"-"$seed" \
--model_name_or_path cardiffnlp/twitter-xlm-roberta-base-sentiment-multilingual \
--dataset_name "$LANG" \
--data_dir ./data \
--image_column image_paths \
--caption_column normalized_text \
--label_column label \
--remove_unused_columns False \
--do_train \
--do_eval \
--do_predict \
--num_train_epochs="50" \
--per_device_train_batch_size 8 \
--per_device_eval_batch_size 8 \
--auto_find_batch_size \
--learning_rate 5e-5 \
--warmup_steps 0 \
--weight_decay 0.1 \
--overwrite_output_dir yes \
--seed "$seed" \
--load_best_model_at_end yes \
--evaluation_strategy steps \
--fp16 \
--optim adafactor \
--disable_tqdm="True" \
--save_total_limit 2
done
done
- To train the Twitter-XLMR and CLIP model run
# Define an array of seeds
SEEDS=(42 123 777 2020 31337)
LANGUAGES=("ar" "bg" "bs" "da" "de" "en" "es" "fr" "hr" "hu" "it" "mt" "pl" "pt" "ru" "sr" "sv" "tr" "zh" "lv" "sq" "bg_mt" "bs_mt" "da_mt" "fr_mt" "hr_mt" "hu_mt" "mt_mt" "pt_mt" "ru_mt" "sr_mt" "sv_mt" "tr_mt" "zh_mt")
# Use a for loop to iterate over the alphabets
for LANG in "${LANGUAGES[@]}"
do
echo "Language: $LANG"
for seed in "${SEEDS[@]}"
do
echo "Seed: $seed"
cuda-wrapper.sh python src_VTDEM/train-twitter-xlmr-clip.py \
--output_dir ./twitter-xlmr-clip-finetuned-"$LANG"-"$seed" \
--model_name_or_path cardiffnlp/twitter-xlm-roberta-base-sentiment-multilingual \
--image_processor_name openai/clip-vit-base-patch32 \
--dataset_name "$LANG" \
--data_dir ./data \
--image_column image_paths \
--caption_column normalized_text \
--label_column label \
--remove_unused_columns False \
--do_train \
--do_eval \
--do_predict \
--num_train_epochs="50" \
--per_device_train_batch_size 8 \
--per_device_eval_batch_size 8 \
--auto_find_batch_size \
--learning_rate 5e-5 \
--warmup_steps 0 \
--weight_decay 0.1 \
--overwrite_output_dir yes \
--seed "$seed" \
--load_best_model_at_end yes \
--evaluation_strategy steps \
--fp16 \
--optim adafactor \
--disable_tqdm="True" \
--save_total_limit 2
done
done
To eval a model run the following code
conda activate environment
# Define an array of seeds # 123 777 2020 31337
SEEDS=(42 123 777 2020 31337)
LANGUAGES=("ar" "bg" "bs" "da" "de" "en" "es" "fr" "hr" "hu" "it" "mt" "pl" "pt" "ru" "sr" "sv" "tr" "zh" "lv" "sq" "bg_mt" "bs_mt" "da_mt" "fr_mt" "hr_mt" "hu_mt" "mt_mt" "pt_mt" "ru_mt" "sr_mt" "sv_mt" "tr_mt" "zh_mt")
# Use a for loop to iterate over the alphabets
for LANG in "${LANGUAGES[@]}"
do
echo Language: $LANG
for seed in ${SEEDS[@]}
do
echo Seed: $seed
python src_VTDEM/eval.py \
--output_dir ./twitter-xlmr-txtonly-finetuned-evaluation-all-"$seed"/"$LANG" \
--model_name_or_path cardiffnlp/twitter-xlm-roberta-base-sentiment-multilingual \
--seed "$seed" \
--image_processor_name openai/clip-vit-base-patch32 \
--dataset_name "$LANG" \
--data_dir ./data \
--image_column image_paths \
--caption_column normalized_text \
--label_column label \
--remove_unused_columns False \
--do_eval \
--do_predict \
--per_device_eval_batch_size 128\
--overwrite_output_dir yes \
--evaluation_strategy steps \
--fp16 \
--optim adafactor
done
done
To use the model use the following script. Kindly set the device based on availability of the GPU.
from transformers import (pipeline)
analyzer = pipeline(
"sentiment-analysis", model="FFZG-cleopatra/M2SA-text-only"
)
input_text = "I feel amazing today."
print(analyzer(input_text)[0]["label"])
To use the model use the following script. Kindly refer to the app.py or for the Transform and VisionTextDualEncoderModel class definitions.
import torch
import torch.nn as nn
import torchvision
from torchvision.transforms import CenterCrop, ConvertImageDtype, Normalize, Resize
from torchvision.transforms.functional import InterpolationMode
from torchvision import transforms
from torchvision.io import ImageReadMode, read_image
from transformers import CLIPModel, AutoModel
from huggingface_hub import hf_hub_download
from safetensors.torch import load_model
from datasets import load_dataset, load_metric
from transformers import (
AutoConfig,
AutoImageProcessor,
AutoModelForSequenceClassification,
AutoTokenizer,
logging,
)
id2label = {0: "negative", 1: "neutral", 2: "positive"}
label2id = {"negative": 0, "neutral": 1, "positive": 2}
tokenizer = AutoTokenizer.from_pretrained("cardiffnlp/twitter-xlm-roberta-base-sentiment-multilingual")
model = VisionTextDualEncoderModel(num_classes=3)
config = model.vision_encoder.config
# https://huggingface.co/FFZG-cleopatra/M2SA/blob/main/model.safetensors
sf_filename = hf_hub_download("FFZG-cleopatra/M2SA", filename="model.safetensors")
load_model(model, sf_filename)
image_processor = AutoImageProcessor.from_pretrained("openai/clip-vit-base-patch32")
def predict_sentiment(text, image):
# read the image file
image = read_image(image, mode=ImageReadMode.RGB)
text_inputs = tokenizer(
text,
max_length=512,
padding="max_length",
truncation=True,
return_tensors="pt"
)
image_transformations = Transform(
config.vision_config.image_size,
image_processor.image_mean,
image_processor.image_std,
)
image_transformations = torch.jit.script(image_transformations)
pixel_values = image_transformations(image)
text_inputs["pixel_values"] = pixel_values.unsqueeze(0)
prediction = None
with torch.no_grad():
outputs = model(**text_inputs)
print(outputs)
prediction = np.argmax(outputs["logits"], axis=-1)
print(id2label[prediction[0].item()])
return id2label[prediction[0].item()]
text = "I feel good today"
image = "link-to-image"
predict_sentiment(text, image)
If you find the resources or the code useful, please cite us:
@inproceedings{thakkar-etal-2024-m2sa-multimodal,
title = "{M}2{SA}: Multimodal and Multilingual Model for Sentiment Analysis of Tweets",
author = "Thakkar, Gaurish and
Hakimov, Sherzod and
Tadi{\'c}, Marko",
editor = "Calzolari, Nicoletta and
Kan, Min-Yen and
Hoste, Veronique and
Lenci, Alessandro and
Sakti, Sakriani and
Xue, Nianwen",
booktitle = "Proceedings of the 2024 Joint International Conference on Computational Linguistics, Language Resources and Evaluation (LREC-COLING 2024)",
month = may,
year = "2024",
address = "Torino, Italia",
publisher = "ELRA and ICCL",
url = "https://aclanthology.org/2024.lrec-main.946",
pages = "10833--10845",
abstract = "In recent years, multimodal natural language processing, aimed at learning from diverse data types, has garnered significant attention. However, there needs to be more clarity when it comes to analysing multimodal tasks in multi-lingual contexts. While prior studies on sentiment analysis of tweets have predominantly focused on the English language, this paper addresses this gap by transforming an existing textual Twitter sentiment dataset into a multimodal format through a straightforward curation process. Our work opens up new avenues for sentiment-related research within the research community. Additionally, we conduct baseline experiments utilising this augmented dataset and report the findings. Notably, our evaluations reveal that when comparing unimodal and multimodal configurations, using a sentiment-tuned large language model as a text encoder performs exceptionally well.",
}