arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2002.07965cs.LGstat.ML

Being Bayesian about Categorical Probability

ESTsoft

Taejong Joo, Uijung Chung, Min-Gwan Seo

英文摘要:

Neural networks utilize the softmax as a building block in classification tasks, which contains an overconfidence problem and lacks an uncertainty representation ability. As a Bayesian alternative to the softmax, we consider a random variable of a categorical probability over class labels. In this framework, the prior distribution explicitly models the presumed noise inherent in the observed label, which provides consistent gains in generalization performance in multiple challenging tasks. The proposed method inherits advantages of Bayesian approaches that achieve better uncertainty estimation and model calibration. Our method can be implemented as a plug-and-play loss function with negligible computational overhead compared to the softmax with the cross-entropy loss function.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑