Applying a Pre-trained Language Model to Spanish Twitter Humor Prediction

  • 2019-07-06 21:05:54
  • Bobak Farzin, Piotr Czapla, Jeremy Howard
  • 0

Abstract

Our entry into the HAHA 2019 Challenge placed $3^{rd}$ in the classificationtask and $2^{nd}$ in the regression task. We describe our system andinnovations, as well as comparing our results to a Naive Bayes baseline. Alarge Twitter based corpus allowed us to train a language model from scratchfocused on Spanish and transfer that knowledge to our competition model. Toovercome the inherent errors in some labels we reduce our class confidence withlabel smoothing in the loss function. All the code for our project is includedin a GitHub repository for easy reference and to enable replication by others.

 

Quick Read (beta)

Applying a Pre-trained Language Model to Spanish Twitter Humor Prediction

Bobak Farzin Piotr Czapla and Jeremy Howard
June 2019
1USF Data Institue, WAMRI Visting Scholar, USA

[email protected] 2n-waves, Poland
[email protected] 3University of San Francisco & Fast.ai, USA
[email protected]
Abstract

Our entry into the HAHA 2019 Challenge placed 3rd in the classification task and 2nd in the regression task. We describe our system and innovations, as well as comparing our results to a Naive Bayes baseline. A large Twitter based corpus allowed us to train a language model from scratch focused on Spanish and transfer that knowledge to our competition model. To overcome the inherent errors in some labels we reduce our class confidence with label smoothing in the loss function. All the code for our project is included in a GitHub44 4 https://github.com/bfarzin/haha_2019_final, Accessed on 19 June 2019 repository for easy reference and to enable replication by others.

Keywords:
Natural Language Processing Transfer Learning Sentiment Analysis Humor Classification
footnotetext: Copyright © 2019 for this paper by its authors. Use permitted under Creative Commons License Attribution 4.0 International (CC BY 4.0). IberLEF 2019, 24 September 2019, Bilbao, Spain.

1 Introduction

- !‘Socorro, me ha picado una víbora!
- ?‘Cobra?
- No, gratis.55 5 https://www.fluentin3months.com/spanish-jokes/, Accessed on 19 June 2019

Google Translation:

- Help, I was bitten by a snake!
- Does it charge?
- Not free.

Humor does not translate well because it often relies on double-meaning or a subtle play on word choice, pronunciation, or context. These issues are further exacerbated in areas where space is a premium (as frequent on social media platforms), often leading to usage and development of shorthand, in-jokes, and self-reference. Thus, building a system to classify the humor of tweets is a difficult task. However, with transfer-learning and the Fast.ai library66 6 https://docs.fast.ai/, Accessed on 19 June 2019, we can build a high quality classifier in a foreign language. Our system outperforms a Naive Bayes Support Vector Machine (NBSVM) baseline, which is frequently considered a ”strong baseline” for many Natural Language Processing (NLP) related tasks (see Wang et al [12]).

Rather than hand-crafted language features, we have taken an ”end to end” approach building from the raw text to a final model that achieves the tasks as presented. Our paper lays out the details of the system and our code can be found in a GitHub repository for use by other researchers to extend the state of the art in sentiment analysis.

Contribution

Our contributions are three fold. First, we apply transfer-learning of a language model based on a larger corpus of tweets. Second, we use a label smoothed loss, which provides regularization and allows full training of the final model without gradual unfreezing. Third, we select the best model for each task based on cross-validation and 20 random-seed initialization in the final network training step.

2 Task and Dataset Description

The Humor Analysis based on Human Annotation (HAHA) 2019[4] competition asked for analysis of two tasks in the Spanish language based on a corpus of publicly collected data described in Castro et al. [2]:

  • Task1: Humor Detection:Determine if a tweet is humorous. System ranking is based on F1 score which balances precision and accuracy.

  • Task2: Funniness Score:If humorous, what is the average humor rating of the tweet? System ranking is based on root mean-squared error (RMSE).

The HAHA dataset includes labeled data for 24,000 tweets and a test set of 6,000 tweets (80%/20% train/test split.) Each record includes the raw tweet text (including accents and emoticons), a binary humor label, the number of votes for each of five star ratings and a “Funniness Score” that is the average of the 1 to 5 star votes cast. Examples and data can be found on the CodaLab competition webpage77 7 http://competitions.codalab.org/competitions/22194/ Accessed on 19 June 2019.

3 System Description

We modify the method of Universal Langage Model Fine-tuning for Text Classification (ULMFiT) presented in Howard and Ruder [6]. The primary steps are:

  1. 1.

    Train a language model (LM) on a large corpus of data

  2. 2.

    Fine-tune the LM based on the target task language data

  3. 3.

    Replace the final layer of the LM with a softmax or linear output layer and then fine-tune on the particular task at hand (classification or regression)

Below we will give more detail on each step and the parameters used to generate our system.

3.1 Data, Cleanup & Tokenization

3.2 Additional Data

We collected a corpus for our LM based on Spanish Twitter using tweepy88 8 http://github.com/tweepy/tweepy, Accessed on 19 June 2019 run for three 4-hour sessions and collecting any tweet with any of the terms ’el’,’su’,’lo’,’y’ or ’en’. We excluded retweets to minimize repeated examples in our language model training. In total, we collected 475,143 tweets - a data set is nearly 16 times larger than the text provided by the competition alone. The frequency of terms, punctuation and vocabulary used on Twitter can be quite different from the standard Wikipedia corpus that is often used to train an LM from scratch.

In the fine-tuning step, we combined the train and test text data without labels from the contest data.

3.3 Cleaning

We applied a list of default cleanup functions in sequence (see list below). They are close to the standard clean-up included in the Fast.ai library with the addition of one function for the Twitter dataset. Cleanup of data is key to expressing information in a compact way so that the LM can use the relevant data when trying to predict the next word in a sequence.

  1. 1.

    Replace more than 3 repetitions of the same character (ie. grrrreat becomes g xxrep r 4 eat)

  2. 2.

    Replace repetition at the word level (similar to above)

  3. 3.

    Deal with ALL CAPS words replacing with a token and converting to lower case.

  4. 4.

    Add spaces between special chars (ie. !!! to ! ! !)

  5. 5.

    Remove useless spaces (remove more than 2 spaces in sequence)

  6. 6.

    Addition: Move all text onto a single line by replacing new-lines inside a tweet with a reserved word (ie. \n to xxnl)

The following example shows the application of this data cleaning to a single tweet:

Saber, entender y estar convencides que la frase \
#LaESILaDefendemosEntreTodes es nuestra linea es nuestro eje.\
#AlertaESI!!!!
Vamos por mas!!! e invitamos a todas aquellas personas que quieran \
se parte.
xxbos saber , entender y estar convencides que la frase \
# laesiladefendemosentretodes es nuestra linea es nuestro eje.\
xxnl  # alertaesi xxrep 4 ! xxnl vamos por mas ! ! ! e invitamos a \
todas aquellas personas que quieran se parte.

3.4 Tokenization

We used sentencepiece [7] to parse into sub-word units and reduce the possible out-of-vocabulary terms in the data set. We selected a vocab size of 30,000 and used the byte-pair encoding (BPE) model. To our knowledge this is the first time that the BPE toenization has been used with ULMFiT in a competition model.

4 Training and Results

4.1 LM Training and Fine-tuning

We train the LM using a 90/10 training/validation split, reporting the validation loss and accuracy of next-word prediction on the validation set. For the LM, we selected an ASGD Weight-Dropped Long Short Term Memory (AWD_LSTM, described in Merity et al.[8]) model included in Fast.ai. We replaced the typical Long Short Term Memory (LSTM) units with Quasi Recurrent Neural Network (QRNN, described in Bradbury et al.[1]) units. Our network has 2304 hidden-states, 3 layers and a softmax layer to predict the next-word. We tied the embedding weights[10] on the encoder and decoder for training. We performed some simple tests with LSTM units and a Transformer Language model, finding all models were similar in performance during LM training. We thus chose to use QRNN units due to improved training speed compared to the alternatives. This model has about 60 million trainable parameters.

Parameters used for training and fine-tuning are shown in Table 1. For all networks we applied a dropout multiplier which scales the dropout used throughout the network. We used the Adam optimizer with weight decay as indicated in the table.

Following the work of Smith[11] we found the largest learning-rate that we could apply and then ran a one-cycle policy for a single epoch. This largest weight is shown in Table 1 under ”Learning Rate.” Subsequent training epochs were run with one-cycle and lower learning rates indicated in Table 1 under ”Continued Training.”

Table 1: LM Training Parameters
Param LM Fine-Tune LM
Weight Decay 0.1 0.1
Dropout Mult 1.0 1.0
Learning Rate 1 epoch at 5*10-3 5 epochs at 3*10-3
Continued Training 15 epochs at 1*10-3 10 epochs at 1*10-4

4.2 Classification and Regression Fitting

Again, following the play-book from Howard and Ruder[6], we change the pre-trained network head to a softmax or linear output layer (as appropriate for the transfer task) and then load the LM weights for the layers below. We train just the new head from random initialization, then unfreeze the entire network and train with differential learning rates. We layout our training parameters in Table 2.

With the same learning rate and weight decay we apply a 5-fold cross-validation on the outputs and take the mean across the folds as our ensemble. We sample 20 random seeds (see more in section 4.3) to find the best initialization for our gradient descent search. From these samples, we select the best validation F1 metric or Mean Squared Error (MSE) for use in our test submission.

4.2.1 Classifier setup

For the classifier, we have a hidden layer and softmax head. We over-sample the minority class to balance the outcomes for better training using Synthetic Minority Oversampling Technique (SMOTE, described in Chawla et al.[3]). Our loss is label smoothing as described in Pereyra et al.[9] of the flattened cross-entropy loss. In ULMFiT, gradual unfreezing allows us to avoid catastropic forgetting, focus each stage of training and preventing over-fitting of the parameters to the training cases. We take an alternative approach to regularization and in our experiments found that we got similar results with label smoothing but without the separate steps and learning rate refinement required of gradual unfreezing.

4.2.2 Regression setup

For the regression task, we fill all #N/A labels with scores of 0. We add a hidden layer and linear output head and MSE loss function.

Table 2: Classification and Regression Training Parameters
Param Value
Weight Decay 0.1
Dropout Mult 0.7
Learning Rate (Head) 2 epochs at 1*10-2
Cont. Training 15 epochs with diff lr:(1*10-3/(2.64), 5*10-3)

4.3 Random Seed as a Hyperparamter

For classification and regression, the random seed sets the initial random weights of the head layer. This initialization affects the final F1 metric achievable.

Across each of the 20 random seeds, we average the 5-folds and obtain a single F1 metric on the validation set. The histogram of 20-seed outcomes is shown in Figure 1 and covers a range from 0.820 to 0.825 over the validation set. We selected our single best random seed for the test submission. With more exploration, a better seed could likely be found. Though we only use a single seed for the LM training, one could do a similar search with random seeds for LM pre-training, and further select the best down-stream seed similar to Czapla et al [5].

Figure 1: Histogram of F1 metric averaged across 5-fold metric

4.4 Results

Table 3 gives three results from our submissions in the competition. The first is the baseline NBSVM solution, with an F1 of 0.7548. Second is our first random seed selected for the classifier which produces a 0.8083 result. While better than the NBSVM solution, we pick the best validation F1 from the 20 seeds we tried. This produced our final submission of 0.8099. Our best model achieved an five-fold average F1 of 0.8254 on the validation set shown in Figure 1 but a test set F1 of 0.8099 - a drop of 0.0155 in F1 for the true out-of-sample data. Also note that our third place entry was 1.1% worse in F1 score than first place but 1.2% better in F1 than the 4th place entry.

Table 3: Comparative Results
Accuracy Precision Recall F1
NBSVM 0.8223 0.8180 0.7007 0.7548
First Seed 0.8461 0.7869 0.8309 0.8083
Best Seed 0.8458 0.7806 0.8416 0.8099

5 Conclusion

This paper describes our implementation of a neural net model for classification and regression in the HAHA 2019 challenge. Our solution placed 3rd in Task 1 and 2nd in Task 2 in the final competition standings. We describe the data collection, pre-training, and final model building steps for this contest. Twitter has slang and abbreviations that are unique to the short-format as well as generous use of emoticons. To capture these features, we collected our own dataset based on Spanish Tweets that is 16 times larger than the competition data set and allowed us to pre-train a language model. Humor is subtle and using a label smoothed loss prevented us from becoming overconfident in our predictions and train more quickly without the gradual unfreezing required by ULMFiT. We have open-sourced all code used in this contest to further enable research on this task in the future.

6 Author Contributions

BF was the primary researcher. PC contributed with suggestions for the random seeds as a hyper-parameters and label smoothing to speed up training. JH contributed with suggestion for higher dropout throughout the network for more generalization.

7 Acknowledgements

The author would like to thank all the participants on the fast.ai forums 99 9 http://forums.fasta.ai for their ideas and suggestions. Also, Kyle Kastner for his edits, suggestions and recommendations in writing up these results.

References

  • [1] Bradbury, J., Merity, S., Xiong, C., Socher, R.: Quasi-recurrent neural networks. CoRR abs/1611.01576 (2016), http://arxiv.org/abs/1611.01576
  • [2] Castro, S., Chiruzzo, L., Rosá, A., Garat, D., Moncecchi, G.: A crowd-annotated spanish corpus for humor analysis. In: Proceedings of the Sixth International Workshop on Natural Language Processing for Social Media. pp. 7–11 (2018)
  • [3] Chawla, N.V., Bowyer, K.W., Hall, L.O., Kegelmeyer, W.P.: Smote: Synthetic minority over-sampling technique. J. Artif. Int. Res. 16(1), 321–357 (Jun 2002), http://dl.acm.org/citation.cfm?id=1622407.1622416
  • [4] Chiruzzo, L., Castro, S., Etcheverry, M., Garat, D., Prada, J.J., Rosá, A.: Overview of HAHA at IberLEF 2019: Humor Analysis based on Human Annotation. In: Proceedings of the Iberian Languages Evaluation Forum (IberLEF 2019). CEUR Workshop Proceedings, CEUR-WS, Bilbao, Spain (9 2019)
  • [5] Czapla, P., Howard, J., Kardas, M.: Universal language model fine-tuning with subword tokenization for polish. CoRR abs/1810.10222 (2018), http://arxiv.org/abs/1810.10222
  • [6] Howard, J., Ruder, S.: Universal language model fine-tuning for text classification. CoRR abs/1801.06146 (2018), http://arxiv.org/abs/1801.06146
  • [7] Kudo, T., Richardson, J.: Sentencepiece: A simple and language independent subword tokenizer and detokenizer for neural text processing. CoRR abs/1808.06226 (2018), http://arxiv.org/abs/1808.06226
  • [8] Merity, S., Keskar, N.S., Socher, R.: Regularizing and optimizing LSTM language models. CoRR abs/1708.02182 (2017), http://arxiv.org/abs/1708.02182
  • [9] Pereyra, G., Tucker, G., Chorowski, J., Kaiser, L., Hinton, G.E.: Regularizing neural networks by penalizing confident output distributions. CoRR abs/1701.06548 (2017), http://arxiv.org/abs/1701.06548
  • [10] Press, O., Wolf, L.: Using the output embedding to improve language models. CoRR abs/1608.05859 (2016), http://arxiv.org/abs/1608.05859
  • [11] Smith, L.N.: A disciplined approach to neural network hyper-parameters: Part 1 - learning rate, batch size, momentum, and weight decay. CoRR abs/1803.09820 (2018), http://arxiv.org/abs/1803.09820
  • [12] Wang, S., Manning, C.D.: Baselines and bigrams: Simple, good sentiment and topic classification. In: Proceedings of the 50th Annual Meeting of the Association for Computational Linguistics: Short Papers - Volume 2. pp. 90–94. ACL ’12, Association for Computational Linguistics, Stroudsburg, PA, USA (2012), http://dl.acm.org/citation.cfm?id=2390665.2390688