Method and apparatus for training classification task model, device, and storage medium

Abstract

Provided are an artificial intelligence (AI)-based method and apparatus for training a classification task model, a device, and a storage medium, which relate to the field of machine learning (ML) technologies. The method includes: training an initial feature extractor by using a first dataset to obtain a feature extractor, the first dataset being a class imbalanced dataset; constructing a generative adversarial network, the generative adversarial network including the feature extractor and an initial feature generator; training the generative adversarial network by using second class samples to obtain a feature generator; constructing a classification task model, the classification task model including the feature generator and the feature extractor; and training the classification task model by using the first dataset, the feature generator being configured to augment the second class samples in a feature space in a training process of the classification task model.