Patent ID: 11921768
Assignee: AMAZON TECHNOLOGIES, INC.
Field: Computer technology (Electrical engineering)
Classification: CPC G | IPC G

Claim 0:
1. A method of discovering topics for text data, comprising:
receiving an input corpus of text data comprising a plurality of documents;
generating a codebook comprising a first number of topics, wherein each topic of the first number of topics is associated by the codebook with a corresponding set of keywords;
determining, for a first topic of the first number of topics, an average embedding of keywords associated with the first topic;
determining a distance in an embedding space between an embedding for a first document of the plurality of documents and the average embedding;
tagging the first document as pertaining to the first topic based at least in part on the distance;
receiving, from a first user, at least one edit to the codebook comprising at least one of adding a keyword, deleting a keyword, modifying a keyword, or re-naming a topic;
generating a modified codebook incorporating the at least one edit to the codebook;
inputting the modified codebook into a teacher model of a student-teacher machine learning model framework;
generating by a student model of the student-teacher machine learning model framework a probability that a second document corresponds to a second topic of the modified codebook; and
tagging the second document as pertaining to the second topic based at least in part on the probability.