JOINT EXTRACTION OF NAMED ENTITIES AND RELATIONS FROM TEXT USING MACHINE LEARNING MODELS
Described herein are systems, methods, and other techniques for training a machine learning (ML) model to jointly perform named entity recognition (NER) and relation extraction (RE) on an input text. A set of hyperparameters for the ML model are set to a first set of values. The ML model is trained using a training dataset and is evaluated to produce a first result. The set of hyperparameters are modified from the first set of values to a second set of values. The ML model is trained using the training dataset and is evaluated to produce a second result. Either the first set of values or the second set of values are selected and used for the set of hyperparameters for the ML model based on a comparison between the first result and the second result.
1 . A method of training a machine learning (ML) model to jointly perform named entity recognition (NER) and relation extraction (RE) on an input text, the method comprising:
setting a set of hyperparameters for the ML model to a first set of values, the set of hyperparameters including a quantity of shared layers in the ML model, a quantity of NER-specific layers in the ML model, and a quantity of RE-specific layers in the ML model, wherein the shared layers precede each of the NER-specific layers and the RE-specific layers in the ML model;
training the ML model having the first set of values for the set of hyperparameters using a training dataset;
evaluating the ML model having the first set of values for the set of hyperparameters using an evaluation dataset to produce a first evaluation result;
modifying the set of hyperparameters from the first set of values to a second set of values;
training the ML model having the second set of values for the set of hyperparameters using the training dataset;
evaluating the ML model having the second set of values for the set of hyperparameters using the evaluation dataset to produce a second evaluation result; and
selecting either the first set of values or the second set of values for the set of hyperparameters for the ML model based on a comparison between the first evaluation result and the second evaluation result.
2 . The method of claim 1 , wherein the ML model is a neural network.
3 . The method of claim 1 , wherein an output of the NER-specific layers is provided to an intermediate layer of the RE-specific layers.
4 . The method of claim 1 , wherein the ML model having the first set of values for the set of hyperparameters and the ML model having the second set of values for the set of hyperparameters are evaluated using an evaluation dataset.
5 . The method of claim 1 , wherein the shared layers include one or more shared bidirectional recurrent neural network (BiRNN) layers, and wherein the quantity of the shared layers corresponds to a quantity of the shared BiRNN layers.
6 . The method of claim 1 , wherein the NER-specific layers include one or more NER-specific bidirectional recurrent neural network (BiRNN) layers, and wherein the quantity of the NER-specific layers corresponds to a quantity of the NER-specific BiRNN layers.
7 . The method of claim 1 , wherein the RE-specific layers include one or more RE-specific bidirectional recurrent neural network (BiRNN) layers, and wherein the quantity of the RE-specific layers corresponds to a quantity of the RE-specific BiRNN layers.
8 . A non-transitory computer-readable medium comprising instructions that, when executed by one or more processors, cause the one or more processors to perform operations comprising:
setting a set of hyperparameters for a machine learning (ML) model to a first set of values, the set of hyperparameters including a quantity of shared layers in the ML model, a quantity of named entity recognition (NER)-specific layers in the ML model, and a quantity of relation extraction (RE)-specific layers in the ML model, wherein the shared layers precede each of the NER-specific layers and the RE-specific layers in the ML model;
training the ML model having the first set of values for the set of hyperparameters using a training dataset;
evaluating the ML model having the first set of values for the set of hyperparameters using an evaluation dataset to produce a first evaluation result;
modifying the set of hyperparameters from the first set of values to a second set of values;
training the ML model having the second set of values for the set of hyperparameters using the training dataset;
evaluating the ML model having the second set of values for the set of hyperparameters using the evaluation dataset to produce a second evaluation result; and
selecting either the first set of values or the second set of values for the set of hyperparameters for the ML model based on a comparison between the first evaluation result and the second evaluation result.
9 . The non-transitory computer-readable medium of claim 8 , wherein the ML model is a neural network.
10 . The non-transitory computer-readable medium of claim 8 , wherein an output of the NER-specific layers is provided to an intermediate layer of the RE-specific layers.
11 . The non-transitory computer-readable medium of claim 8 , wherein the ML model having the first set of values for the set of hyperparameters and the ML model having the second set of values for the set of hyperparameters are evaluated using an evaluation dataset.
12 . The non-transitory computer-readable medium of claim 8 , wherein the shared layers include one or more shared bidirectional recurrent neural network (BiRNN) layers, and wherein the quantity of the shared layers corresponds to a quantity of the shared BiRNN layers.
13 . The non-transitory computer-readable medium of claim 8 , wherein the NER-specific layers include one or more NER-specific bidirectional recurrent neural network (BiRNN) layers, and wherein the quantity of the NER-specific layers corresponds to a quantity of the NER-specific BiRNN layers.
14 . The non-transitory computer-readable medium of claim 8 , wherein the RE-specific layers include one or more RE-specific bidirectional recurrent neural network (BiRNN) layers, and wherein the quantity of the RE-specific layers corresponds to a quantity of the RE-specific BiRNN layers.
15 . A system for training a machine learning (ML) model to jointly perform named entity recognition (NER) and relation extraction (RE) on an input text, the system comprising:
one or more processors; and
a computer-readable medium comprising instructions that, when executed by the one or more processors, cause the one or more processors to perform operations comprising:
setting a set of hyperparameters for the ML model to a first set of values, the set of hyperparameters including a quantity of shared layers in the ML model, a quantity of NER-specific layers in the ML model, and a quantity of RE-specific layers in the ML model, wherein the shared layers precede each of the NER-specific layers and the RE-specific layers in the ML model;
training the ML model having the first set of values for the set of hyperparameters using a training dataset;
evaluating the ML model having the first set of values for the set of hyperparameters using an evaluation dataset to produce a first evaluation result;
modifying the set of hyperparameters from the first set of values to a second set of values;
training the ML model having the second set of values for the set of hyperparameters using the training dataset;
evaluating the ML model having the second set of values for the set of hyperparameters using the evaluation dataset to produce a second evaluation result; and
selecting either the first set of values or the second set of values for the set of hyperparameters for the ML model based on a comparison between the first evaluation result and the second evaluation result.
16 . The system of claim 15 , wherein the ML model is a neural network.
17 . The system of claim 15 , wherein an output of the NER-specific layers is provided to an intermediate layer of the RE-specific layers.
18 . The system of claim 15 , wherein the ML model having the first set of values for the set of hyperparameters and the ML model having the second set of values for the set of hyperparameters are evaluated using an evaluation dataset.
19 . The system of claim 15 , wherein the shared layers include one or more shared bidirectional recurrent neural network (BiRNN) layers, and wherein the quantity of the shared layers corresponds to a quantity of the shared BiRNN layers.
20 . The system of claim 15 , wherein the NER-specific layers include one or more NER-specific bidirectional recurrent neural network (BiRNN) layers, and wherein the quantity of the NER-specific layers corresponds to a quantity of the NER-specific BiRNN layers.