APPARATUS AND METHOD FOR TRAINING CLASSIFICATION MODEL AND APPARATUS FOR PERFORMING CLASSIFICATION BY USING CLASSIFICATION MODEL

摘要

An apparatus and method for training a classification model and an apparatus for performing classification by using a classification model are provided. The apparatus for training a classification model includes: a feature extraction unit configured to set, with respect to each training set of a first predetermined number of training sets, feature extraction layers respectively, and extract features of a sample image, where at least two of the training sets at least partially overlap; a feature fusion unit configured to set, with respect to each training set, feature fusion layers respectively, and perform a fusion on the extracted features of the sample image; and a loss determination unit configured to set, with respect to each training set, a loss determination layer respectively, calculate a loss function of the sample image based on the fused feature of the sample image, and train a classification model based on the loss function. The first predetermined number of training sets share at least one layer of feature fusion layers and feature extraction layers set with respect to each training set.