Funny how times have changed.
Why? Because data were confidential, I could not use the Cloud to get a nice GPU, and I needed almost one model per enterprise. Training DistilBert for 3 epochs was like 12 hours and it was a domain where data/concept drift happen often which means re-training was mandatory on weekly basis, at least.
After some feature engineering, I was able to get almost the same performance (<1% difference) with a model that was able to train in less than 5 minutes and could be use in production easily.
I love the fact that SVM, at least in scikit-learn, have a built-in early stopping mechanism. Really handy.
Another important thing to consider is that a lot of the actual work in setting up a machine learning pipeline has nothing to do with machine learning (making sure that data is clean, exceptions are handled, etc). On a small data set an SVM can achieve good performance out of the box. Starting with an SVM can be a really helpful in validating the basic approach and understand the nature of the problem before you sink in a bunch to time/money collecting data and training a complicated model.
The system has a compound set of machine learning models and parsers to finally extract from the government official public news (http://in.gov.br) documents with the following structured info:
- who was/will hired and fired (PERSON entity)
- which job role it will/did have. (JOB entity)
- when will happens¹ (DATE entity)
Each entity is extracted individually using a custom trained NER and each sentence is passed to the Relation Extraction system, which is built using SVM. Features are concatenated word vectors² compound by the slices of the text in the form (entity1, entity2, before, between, after).
The system is being alive for almost two years. It produces great results. Just did need retrain the entity recognizer twice in all that time (built using spacy which uses a averaged perceptron).
The SVM part (Relation Extraction) was not retrained since the first day deployed and it still works gracefully :D
¹this info is on the text as natural lang, sometimes is different from the post date ²gensim.Word2Vec custom model trained on this corpus.
"Hire Bill Gates as Front-End Developer at 05/05/2020"
Suppose the NER extracts the following entities from it:
- Bill Gates (PERSON)
- Front-End Developer (JOB)
- 05/05/2020 (DATE)
In my domain problem, I need the relation PERSON-JOB-DATE, which can be decomposed in two binary relations: PERSON-JOB and JOB-DATE. Each binary relation it's a model by itself with the following class outcomes: invalid, hiring, firing. If two binary relations has the same job and outcome classes, I build the triple PERSON-JOB-DATE.
At feature engineering level, which is you asked for, I build a tuple of local attention from entities perspective based on slices of the text: (entity1, entity2, before, between, after). Each part it's transformed into a vector by using average word vector and finally each part it's concatenated in a final vector that will be used by SVM. Since word vector dimension I choose in my experiments was n=100, my model will have n=500 features.
An example about the slices for PERSON-JOB it will be:
("Bill Gates", "Front-End Developer", "Hire", "as", "at 05/05/2020")
1. Each element of the tuple is tokenized
2. With tokens available, use word vector model to transform each one.
3. For each element of the tuple take the average of the vectors.
4. Concatenate each averaged vector into a final vector.
Using that structure the model is biased strongly by how the sentence is written with the words around of the entities. For my domain where the sentences are very regular it worked well. I am not sure if will work in more general domain, like social media.
I hope this clarify your question in some level.
I've even used DNNs when I couldn't make SVMs work, but someone more experienced came along and showed me a kernel method that for the trick with an SVM.
In short, it's a great tool if it works.