Regularized Frank-Wolfe for Dense CRFs: Generalizing Mean Field and Beyond

D. Khuê Lê-Huu, Karteek Alahari

Keywords: mean field, frank-wolfe, conditional gradient, crfs, mrfs, conditional random fields, markov random fields, map inference, semantic segementation

TL;DR: This paper proposes a new class of CRF inference algorithms called Regularized Frank-Wolfe that includes existing algorithms, such as mean field or concave-convex procedure, as special cases.

Abstract: We introduce regularized Frank-Wolfe, a general and effective algorithm for inference and learning of dense conditional random fields (CRFs). The algorithm optimizes a nonconvex continuous relaxation of the CRF inference problem using vanilla Frank-Wolfe with approximate updates, which are equivalent to minimizing a regularized energy function. Our proposed method is a generalization of existing algorithms such as mean field or concave-convex procedure. This perspective not only offers a unified analysis of these algorithms, but also allows an easy way of exploring different variants that potentially yield better performance. We illustrate this in our empirical results on standard semantic segmentation datasets, where several instantiations of our regularized Frank-Wolfe outperform mean field inference, both as a standalone component and as an end-to-end trainable layer in a neural network. We also show that dense CRFs, coupled with our new algorithms, produce significant improvements over strong CNN baselines.

Code Of Conduct: I certify that all co-authors of this work have read and commit to adhering to the NeurIPS Statement on Ethics, Fairness, Inclusivity, and Code of Conduct.

Code: https://github.com/netw0rkf10w/CRF