ImageNet Classification Using WordNet Hierarchy
Ankita Chatterjee, Jayanta Mukherjee, Partha Pratim Das · IEEE Transactions on Artificial Intelligence · 2023
Convolutional neural networks (ConvNets) have become increasingly popular for image classification tasks. All contemporary computer vision problems are being dominated by ConvNets. Conventional training methods using cross-entropy loss for training have constantly outperformed the state-of-the-art technique to set a new standard in theImageNetclassification challenge. However, growing accuracy come at the cost of enormous number of parameters and computations. Further, classical learning algorithms do not utilize the semantic relationship between the classes present in the dataset. Thus, interpreting the behavior of the model become difficult even though the results may be desirable. Hence, we demonstrate a classification method by leveraging theWordNethierarchy on theImageNetdataset to establish class relationships and label embedding. The model is trained using cross entropy with soft labels based on the semantic similarity between the generated output and the ground truth. Unlike categorical cross entropy, it does not treat every predicted label as equally erroneous. The method generates meaningful neighboring classes in the feature space of the true label.