SampleRank: Training Factor Graphs with Atomic Gradients

SampleRank: Training Factor Graphs with Atomic Gradients

Michael Wick, Khashayar Rohanimanesh, Kedare Bellare, Aron Culotta, Andrew McCallum

01 July 2011

We present SampleRank, an alternative to con- trastive divergence (CD) for estimating param- eters in complex graphical models. SampleR- ank harnesses a user-provided loss function to distribute stochastic gradients across an MCMC chain. As a result, parameter updates can be computed between arbitrary MCMC states. Sam- pleRank is not only faster than CD, but also achieves better accuracy in practice (up to 23% error reduction on noun-phrase coreference).


Venue : International Conference on Machine Learning (ICML)

External Link: http://ciir-publications.cs.umass.edu/getpdf.php?id=990