International Business Machines Corporation
Adaptive control of negative learning for limited reconstruction capability auto encoder

Abstract:

A computer-implemented method is provided for training a classification model. The method includes preparing, by a processor, positive and negative class data. The method further includes iteratively training the classification model, by the processor, using the positive class data and the negative class data such that the positive class data is reconstructed and the negative class data is prevented from being constructed, by the classification model. In response to a selection of a non-integer value as a number of negative learning iterations to be performed to train the classification model, a particular set of the negative class data that is reconstructed best by the classification model from among all of the negative class data is selected to be used for negative learning by the classification model. The training based on the positive class data is performed once before the negative learning iterations and once after each negative learning iteration.