University of Pittsburgh

An Optimization-based Framework to Learn Conditional Random Fields for Multi-label Classification

ISP Graduate Student
Friday, January 10, 2014 - 12:30pm - 1:00pm

In this paper we study 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.

Copyright 2009–2019 | Send feedback about this site