This study aims to understand whether a word2vec embedding coupled with a neural network for classification can achieve better performance than simpler embeddings like bag-of-words or TF-IDF vectorizers. Our study is divided into two main parts,
-
PART 1: We train a word2vec embedder using ~20,000 abstracts collected via the Nature metadata dataset, filtered by the keyword field theory.
-
PART 2: We consider a restricted dataset, produced in a previous study (see the link) where the papers have labels from 9 possible classes, and we train different neural network architectures for classification.
We use a dataset that we have extracted, analyzed, and cleaned in a different study (see link above). The dataset is obtained via the Nature metadata API. From the previous analysis, we know that ensemble learners and linear methods achieve an accuracy of around 73-77% on this dataset, with observations from 3 of the 9 classes being particularly difficult to correctly assign. The word embedding used in the previous study is a TF-IDF vectorizer.
In the first part of the project, we train the gensim word2vec model to embed words commonly appearing in scientific papers, and particularly in papers concerning quantum field theory. We decide to train the embedder instead of using a pre-trained model since abstracts of papers published in scientific journals often contain lots of jargon.
The second stage of this project consists of training different architectures of neural networks for classifying the abstracts. Specifically, we first use a Convolutional Neural Network (CNN) searching for specific features involving a given number of words in the text. Later, we use a Recursive Neural Network (RNN) to obtain information from the whole abstract.
For what concerns the word2vec embedding we trained, we can only judge its performance by demanding similar words to a given set of words, that we consider meaningful for the task. While the model we trained is by no means general enough to be used in production, it seems to be sufficient for our purposes. Indeed, almost all the word associations made by the model seem to be reasonable.
For example, the model suggests that the most similar words to "eth" are "eigenstate" and "thermalization", which indeed are the first two terms in the acronym. For "AdS-CFT", the model associate "holography", that indeed is a primary feature of the AdS-CFT correspondence.
We then train abstract classifiers using different Neural Network architectures, see below for a detailed description of the architectures. Interestingly, the two architectures used perform very differently, with the CNN achieving an accuracy of ~81% (thus outperforming all previous methods used on this dataset), and the RNN obtaining an accuracy of ~ 74% (close to the one achieved by Gradient Boosting in the previous study).
Furthermore, we find that the RNN requires a higher number of weights (equivalently, hidden nodes/layers) than the CNN, and the training takes considerably more time.
Multiple aspects could be interesting to study.
-
For the word2vec embedding, we trained the model using CBOW (continuous bag-of-words), but an alternative could be Skip-Gram. To achieve better embeddings, one should also probably use a much bigger corpus than the one used here.
-
We could also try to use pre-trained embeddings, although these are (reasonably) trained on non-scientific texts, and therefore would struggle with the jargon used in scientific journal abstracts. We could use the embedding as a baseline.
-
The two NN architectures used here are quite simple, and it might be interesting to use a more advanced one (e.g. Bidirectional Encoder Representations from Transformers, BERT)
-
Word2vec embedding using gensim
-
Abstract classifiers using TensorFlow/Keras
- Convolutional Neural Network
- Recursive Neural Network
-
Model evaluation (Mc Nemar Test)
In this repo, we provide a small notebook serving_model.ipynb
where we show how a model deployed with TensorFlow Serving can be accessed and used for predictions.