1
$\begingroup$

I'm looking for a supervized system/approach, that could learn how to categorize incoming texts/documents, where new categories can be added over time and the training set will be small. The trained model should not be static and should be able to evolve with adding new categories or evaluating new documents.

For each document it should first give it's suggestion that can be then corrected.

$\endgroup$
1
$\begingroup$

Transfer Learning allows you to add new categories to be predicted to the output layer without needing to re-train the entire model every time a new category needs to be classified. Rather, the weights of all initial layers up to the last few layers of the network can be frozen and only the last one or two layers can be made trainable for fine-tuning the classification rather than training from scratch. This approach would be relatively fast than other methods since the dataset in your case is small.

In order for model to flag the need of a new output class by inferring on a test document, you could include a class "Unknown" in addition to existing N classes (hence, the output layer now contains N+1 classes). If the model predicts "Unknown" with highest probability, you could add a new class to the output layer after examining the test data.

| improve this answer | |
$\endgroup$
  • $\begingroup$ nice idea with the Unknown, thank you very much $\endgroup$ – Matej Tymes Jan 21 at 19:00

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service, privacy policy and cookie policy

Not the answer you're looking for? Browse other questions tagged or ask your own question.