Entity level data augmentation in chatbots for robust named entity recognition
Techniques for data augmentation for training chatbot systems in natural language processing. In one particular aspect, a method is provided that includes generating a list of values to cover for an entity, selecting utterances from a set of data that have context for the entity, converting the utterances into templates, where each template of the templates comprises a slot that maps to the list of values for the entity, selecting a template from the templates, selecting a value from the list of values based on the mapping between the slot within the selected template and the list of values for the entity; and creating an artificial utterance based on the selected template and the selected value, where the creating the artificial utterance comprises inserting the selected value into the slot of the selected template that maps to the list of values for the entity.
1 . A method comprising:
accessing an utterance received by a chatbot system comprising one or more chatbots;
using a trained machine learning model to extract entity information from the utterance, wherein the trained machine learning model is trained by:
training a machine learning model using a set of training data selected from a set of data comprising utterances, at least one utterance of the utterances comprising a given entity;
evaluating performance of the machine learning model with respect to the given entity using a set of testing data selected from the set of data;
in response to evaluating the performance of the machine learning model, determining that the set of training data should be augmented;
in response to determining that the set of training data should be augmented, generating an augmented set of training data by:
selecting a template from a plurality of templates associated with the given entity;
identifying a plurality of slots in the template, wherein the plurality of slots comprises a first slot that represents a value element for the given entity and a second slot that represents a function element for the given entity;
generating, based on an augmentation rate, a plurality of artificial utterances, wherein the plurality of artificial utterances is generated based on: (i) inserting a value selected from a list of values for the given entity into the first slot; and (ii) inserting a function output obtained by executing a function selected from a plurality of functions associated with the given entity into the second slot, wherein each value of the list of values representing a variation of the given entity, and wherein a number of the plurality of artificial utterances to be generated is determined based on the augmentation rate; and
adding the plurality of artificial utterances to the set of training data; and
using the augmented set of training data to further train the machine learning model to result in the trained machine learning model;
using the entity information identify a chatbot of the one or more chatbots; and
routing the utterance to the chatbot.
2 . The method of claim 1 , further comprising:
converting training utterances selected from the set of training data that comprise the given entity and context for the given entity into the plurality of templates.
3 . The method of claim 2 , wherein converting the training utterances into the plurality of templates comprises maintaining the context for the given entity and replacing the given entity in each training utterance of the training utterances with a slot associated with one or more elements for the given entity, and wherein an element of the one or more elements maps to the list of values for the given entity.
4 . The method of claim 1 , wherein the template is selected based on a mapping between the given entity and the plurality of templates, and wherein the plurality of artificial utterances is generated by performing the following iteratively:
selecting a selected value from the list of values for the given entity;
inserting the selected value into the first slot;
executing a function selected from the plurality of functions to obtain a selected function output; and
inserting the selected function output into the second slot.
5 . The method of claim 4 , wherein the template is selected based on heuristics.
6 . The method of claim 1 , further comprising:
evaluating performance of the machine learning model trained using the augmented set of training data using an augmented set of testing data selected from the augmented set of training data; and
in response to evaluating the performance of the machine learning model trained using the augmented set of training data, determining that the augmented set of training data should be augmented to include at least one additional variation of the given entity.
7 . The method of claim 1 , wherein the augmentation rate is selected based on the performance of the machine learning model, a random selection method, a predefined schema, a distribution of entities in the set of training data selected from the set of data comprising utterances, or any combination thereof.
8 . A system comprising:
one or more processors; and
a memory coupled to the one or more processors, the memory storing a plurality of instructions executable by the one or more processors, the plurality of instructions comprising instructions that when executed by the one or more processors cause the one or more processors to perform processing comprising:
accessing an utterance received by a chatbot system comprising one or more chatbots;
using a trained machine learning model to extract entity information from the utterance, wherein the trained machine learning model is trained by:
training a machine learning model using a set of training data selected from a set of data comprising utterances, at least one utterance of the utterances comprising a given entity;
evaluating performance of the machine learning model with respect to the given entity using a set of testing data selected from the set of data;
in response to evaluating the performance of the machine learning model, determining that the set of training data should be augmented;
in response to determining that the set of training data should be augmented, generating an augmented set of training data by:
selecting a template from a plurality of templates associated with the given entity;
identifying a plurality of slots in the template, wherein the plurality of slots comprises a first slot that represents a value element for the given entity and a second slot that represents a function element for the given entity;
generating, based on an augmentation rate, a plurality of artificial utterances, wherein the plurality of artificial utterances is generated based on: (i) inserting a value selected from a list of values for the given entity into the first slot; and (ii) inserting a function output obtained by executing a function selected from a plurality of functions associated with the given entity into the second slot, wherein each value of the list of values representing a variation of the given entity, and wherein a number of the plurality of artificial utterances to be generated is determined based on the augmentation rate; and
adding the plurality of artificial utterances to the set of training data; and
using the augmented set of training data to further train the machine learning model to result in the trained machine learning model;
using the entity information identify a chatbot of the one or more chatbots; and
routing the utterance to the chatbot.
9 . The system of claim 8 , further comprising:
converting training utterances selected from the set of training data that comprise the given entity and context for the given entity into the plurality of templates.
10 . The system of claim 9 , wherein converting the training utterances into the plurality of templates comprises maintaining the context for the given entity and replacing the given entity in each training utterance of the training utterances with a slot associated with one or more elements for the given entity, and wherein an element of the one or more elements maps to the list of values for the given entity.
11 . The system of claim 8 , wherein the template is selected based on a mapping between the given entity and the plurality of templates, and wherein the plurality of artificial utterances is generated by performing the following iteratively:
selecting a selected value from the list of values for the given entity;
inserting the selected value into the first slot;
executing a function selected from the plurality of functions to obtain a selected function output; and
inserting the selected function output into the second slot.
12 . The system of claim 11 , wherein the template is selected based on heuristics.
13 . The system of claim 8 , further comprising:
evaluating performance of the machine learning model trained using the augmented set of training data using an augmented set of testing data selected from the augmented set of training data; and
in response to evaluating the performance of the machine learning model trained using the augmented set of training data, determining that the augmented set of training data should be augmented to include at least one additional variation of the given entity.
14 . The system of claim 8 , wherein the augmentation rate is selected based on the performance of the machine learning model, a random selection method, a predefined schema, a distribution of entities in the set of training data selected from the set of data comprising utterances, or any combination thereof.
15 . A non-transitory computer-readable memory storing a plurality of instructions executable by one or more processors, the plurality of instructions comprising instructions that when executed by the one or more processors cause the one or more processors to perform processing comprising:
accessing an utterance received by a chatbot system comprising one or more chatbots;
using a trained machine learning model to extract entity information from the utterance, wherein the trained machine learning model is trained by:
training a machine learning model using a set of training data selected from a set of data comprising utterances, at least one utterance of the utterances comprising a given entity;
evaluating performance of the machine learning model with respect to the given entity using a set of testing data selected from the set of data;
in response to evaluating the performance of the machine learning model, determining that the set of training data should be augmented;
in response to determining that the set of training data should be augmented, generating an augmented set of training data by:
selecting a template from a plurality of templates associated with the given entity;
identifying a plurality of slots in the template, wherein the plurality of slots comprises a first slot that represents a value element for the given entity and a second slot that represents a function element for the given entity;
generating, based on an augmentation rate, a plurality of artificial utterances, wherein the plurality of artificial utterances is generated based on: (i) inserting a value selected from a list of values for the given entity into the first slot; and (ii) inserting a function output obtained by executing a function selected from a plurality of functions associated with the given entity into the second slot, wherein each value of the list of values representing a variation of the given entity, and wherein a number of the plurality of artificial utterances to be generated is determined based on the augmentation rate; and
adding the plurality of artificial utterances to the set of training data; and
using the augmented set of training data to further train the machine learning model to result in the trained machine learning model;
using the entity information identify a chatbot of the one or more chatbots; and
routing the utterance to the chatbot.
16 . The non-transitory computer-readable memory of claim 15 , further comprising:
converting training utterances selected from the set of training data that comprise the given entity and context for the given entity into the plurality of templates.
17 . The non-transitory computer-readable memory of claim 16 , wherein converting the training utterances into the plurality of templates comprises maintaining the context for the given entity and replacing the given entity in each training utterance of the training utterances with a slot associated with one or more elements for the given entity, and wherein an element of the one or more elements maps to the list of values for the given entity.
18 . The non-transitory computer-readable memory of claim 15 , wherein the template is selected based on a mapping between the given entity and the plurality of templates, and wherein the plurality of artificial utterances is generated by performing the following iteratively:
selecting a selected value from the list of values for the given entity;
inserting the selected value into the first slot;
executing a function selected from the plurality of functions to obtain a selected function output; and
inserting the selected function output into the second slot.
19 . The non-transitory computer-readable memory of claim 15 , further comprising:
evaluating performance of the machine learning model trained using the augmented set of training data using an augmented set of testing data selected from the augmented set of training data; and
in response to evaluating the performance of the machine learning model trained using the augmented set of training data, determining that the augmented set of training data should be augmented to include at least one additional variation of the given entity.
20 . The non-transitory computer-readable memory of claim 15 , wherein the augmentation rate is selected based on the performance of the machine learning model, a random selection method, a predefined schema, a distribution of entities in the set of training data selected from the set of data comprising utterances, or any combination thereof.