Edit model card

Introduction

Emobart-large is part of the EmoLLMs project, the first open-source large language model (LLM) series for comprehensive affective analysis with instruction-following capability. This model is finetuned based on the bart-large foundation model and the full AAID instruction tuning data. The model can be used for affective classification tasks (e.g. sentimental polarity or categorical emotions), and regression tasks (e.g. sentiment strength or emotion intensity).

Ethical Consideration

Recent studies have indicated LLMs may introduce some potential bias, such as gender gaps. Meanwhile, some incorrect prediction results, and over-generalization also illustrate the potential risks of current LLMs. Therefore, there are still many challenges in applying the model to real-scenario affective analysis systems.

Models in EmoLLMs

There are a series of EmoLLMs, including Emollama-7b, Emollama-chat-7b, Emollama-chat-13b, Emoopt-13b, Emobloom-7b, Emot5-large, Emobart-large.

  • Emollama-7b: This model is finetuned based on the LLaMA2-7B.
  • Emollama-chat-7b: This model is finetuned based on the LLaMA2-chat-7B.
  • Emollama-chat-13b: This model is finetuned based on the LLaMA2-chat-13B.
  • Emoopt-13b: This model is finetuned based on the OPT-13B.
  • Emobloom-7b: This model is finetuned based on the Bloomz-7b1-mt.
  • Emot5-large: This model is finetuned based on the T5-large.
  • Emobart-large: This model is finetuned based on the bart-large.

All models are trained on the full AAID instruction tuning data.

Usage

You can use the Emobart-large model in your Python project with the Hugging Face Transformers library. Here is a simple example of how to load the model:

from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained('lzw1008/Emobart-large')
model = AutoModelForSeq2SeqLM.from_pretrained('lzw1008/Emobart-large', device_map='auto')

In this example, AutoTokenizer is used to load the tokenizer, and AutoModelForSeq2SeqLM is used to load the model. The device_map='auto' argument is used to automatically use the GPU if it's available.

Prompt examples

Emotion intensity

Task: Assign a numerical value between 0 (least E) and 1 (most E) to represent the intensity of emotion E expressed in the text.
Text: @CScheiwiller can't stop smiling ๐Ÿ˜†๐Ÿ˜†๐Ÿ˜†
Emotion: joy
Intensity Score:

>>0.896

Sentiment strength

Task: Evaluate the valence intensity of the writer's mental state based on the text, assigning it a real-valued score from 0 (most negative) to 1 (most positive).
Text: Happy Birthday shorty. Stay fine stay breezy stay wavy @daviistuart ๐Ÿ˜˜
Intensity Score:

>>0.879

Sentiment classification

Task: Categorize the text into an ordinal class that best characterizes the writer's mental state, considering various degrees of positive and negative sentiment intensity. 3: very positive mental state can be inferred. 2: moderately positive mental state can be inferred. 1: slightly positive mental state can be inferred. 0: neutral or mixed mental state can be inferred. -1: slightly negative mental state can be inferred. -2: moderately negative mental state can be inferred. -3: very negative mental state can be inferred
Text: Beyoncรฉ resentment gets me in my feelings every time. ๐Ÿ˜ฉ
Intensity Class:

>>-3: very negative emotional state can be inferred

Emotion classification

Task: Categorize the text's emotional tone as either 'neutral or no emotion' or identify the presence of one or more of the given emotions (anger, anticipation, disgust, fear, joy, love, optimism, pessimism, sadness, surprise, trust).
Text: Whatever you decide to do make sure it makes you #happy.
This text contains emotions:

>>joy, love, optimism

The task description can be adjusted according to the specific task.

License

EmoLLMs series are licensed under MIT. For more details, please see the MIT file.

Citation

If you use the series of EmoLLMs in your work, please cite our paper:

@article{liu2024emollms,
  title={EmoLLMs: A Series of Emotional Large Language Models and Annotation Tools for Comprehensive Affective Analysis},
  author={Liu, Zhiwei and Yang, Kailai and Zhang, Tianlin and Xie, Qianqian and Yu, Zeping and Ananiadou, Sophia},
  journal={arXiv preprint arXiv:2401.08508},
  year={2024}
}
Downloads last month
2,176
Safetensors
Model size
406M params
Tensor type
F32
ยท
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.