作者
Wanqing Xie, Yubin Ge, Mingzhen Li, Xuyang Li, Zhenhua Guo, Jane You, Xiaofeng Liu
发表日期
2023/4/14
期刊
Neurocomputing
卷号
530
页码范围
1-10
出版商
Elsevier
简介
The cross-entropy (CE) loss is arguably the most important empirical risk minimization objective for deep discriminative models for classification, and has achieved notable success in numerous applications. Though the CE loss is widely adopted, it essentially ignores the correlation between categories. For example, predicting a shepherd dog to husky is more acceptable than a tiger for the subsequent decision processes, while these two misclassifications result in the same CE loss. Therefore, the usually used CE loss does not incorporate the risk of misclassification of different categories, which can be measured by the distance between the predicted category and ground-truth category in a semantic hierarchical tree (SHT). In this work, to explicitly take the SHT-defined risk-aware inter-categorical correlation into consideration, by proposing a discrete optimal transport (DOT) training framework via configuring its …


