An Optimization-based Framework to Learn Conditional Random Fields for Multi-label Classification.
An Optimization-based Framework to Learn Conditional Random Fields for Multi-label Classification.
复制标题
DOI:
10.1137/1.9781611973440.113
复制
发表时间:
2014
期刊:
影响因子:
--
通讯作者:
Hauskrecht M
中科院分区:
文献类型:
--
作者:
Naeini MP;Batal I;Liu Z;Hong C;Hauskrecht M
This paper studies multi-label classification problem in which data instances are associated with multiple, possibly high-dimensional, label vectors. This problem is especially challenging when labels are dependent and one cannot decompose the problem into a set of independent classification problems. To address the problem and properly represent label dependencies we propose and study a pairwise conditional random Field (CRF) model. We develop a new approach for learning the structure and parameters of the CRF from data. The approach maximizes the pseudo likelihood of observed labels and relies on the fast proximal gradient descend for learning the structure and limited memory BFGS for learning the parameters of the model. Empirical results on several datasets show that our approach outperforms several multi-label classification baselines, including recently published state-of-the-art methods.