Extraction of genealogy data from obituaries

A computer-implemented method for extracting data from obituaries, the method comprising:

• receiving an image;
• segmenting the image into a plurality of sections;
• determining that a section of the plurality of sections contains an obituary containing a plurality of words, the obituary corresponding to a deceased individual;
• and
• assigning, using an entity tagging machine learning (ML) model, an entity tag from a set of entity tags to each of one or more words of the plurality of words, wherein each particular entity tag from the set of entity tags includes a relationship component and a category component, wherein the relationship component indicates a relationship between a particular word of the plurality of words to which the particular entity tag is assigned and the deceased individual, and wherein the category component indicates a categorization of the particular word to a particular category from a set of categories;
• wherein, prior to assigning the entity tag, the entity tagging ML model is trained by:
• receiving an input obituary containing a plurality of input words;
• creating a first training set based on the plurality of input words;
• training the entity tagging ML model in a first stage using the first training set;
• creating a second training set including a subset of the plurality of input words to which entity tags were incorrectly assigned after the first stage;
• and
• training the entity tagging ML model in a second stage using the second training set.

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:

• receiving an obituary containing a plurality of words, the obituary corresponding to a deceased individual;
• and
• assigning, using an entity tagging machine learning (ML) model, an entity tag from a set of entity tags to each of one or more words of the plurality of words, wherein each particular entity tag from the set of entity tags includes a relationship component and a category component, wherein the relationship component indicates a relationship between a particular word of the plurality of words to which the particular entity tag is assigned and the deceased individual, and wherein the category component indicates a categorization of the particular word to a particular category from a set of categories;
• wherein, prior to assigning the entity tag, the entity tagging ML model is trained by:
• receiving an input obituary containing a plurality of input words;
• creating a first training set based on the plurality of input words;
• training the entity tagging ML model in a first stage using the first training set;
• creating a second training set including a subset of the plurality of input words to which entity tags were incorrectly assigned after the first stage;
• and
• training the entity tagging ML model in a second stage using the second training set.