Nonlinear Stein Variational Gradient Descent for Learning Diversified Mixture Models
Nonlinear Stein Variational Gradient Descent for Learning Diversified Mixture Models
复制标题
DOI:
--
复制
发表时间:
2019-05
期刊:
影响因子:
--
通讯作者:
Dilin Wang;Qiang Liu
中科院分区:
文献类型:
--
作者:
Dilin Wang;Qiang Liu
Diversification has been shown to be a powerful mechanism for learning robust models in nonconvex settings. A notable example is learning mixture models, in which enforcing diversity between the different mixture components allows us to prevent the model collapsing phenomenon and capture more patterns from the observed data. In this work, we present a variational approach for diversity-promoting learning, which leverages the entropy functional as a natural mechanism for enforcing diversity. We develop a simple and efficient functional gradientbased algorithm for optimizing the variational objective function, which provides a significant generalization of Stein variational gradient descent (SVGD). We test our method on various challenging real world problems, including deep embedded clustering and deep anomaly detection. Empirical results show that our method provides an effective mechanism for diversitypromoting learning, achieving substantial improvement over existing methods.