Title
Learning Affinity via Spatial Propagation Networks.
Abstract
In this paper, we propose a spatial propagation networks for learning affinity matrix. We show that by constructing a row/column linear propagation model, the spatially variant transformation matrix constitutes an affinity matrix that models dense, global pairwise similarities of an image. Specifically, we develop a three-way connection for the linear propagation model, which (a) formulates a sparse transformation matrix where all elements can be the output from a deep CNN, but (b) results in a dense affinity matrix that is effective to model any task-specific pairwise similarity. Instead of designing the similarity kernels according to image features of two points, we can directly output all similarities in a pure data-driven manner. The spatial propagation network is a generic framework that can be applied to numerous tasks, which traditionally benefit from designed affinity, e.g., image matting, colorization, and guided filtering, to name a few. Furthermore, the model can also learn semantic-aware affinity for high-level vision tasks due to the learning capability of the deep model. We validate the proposed framework by refinement of object segmentation. Experiments on the HELEN face parsing and PASCAL VOC-2012 semantic segmentation tasks show that the spatial propagation network provides general, effective and efficient solutions for generating high-quality segmentation results.
Year
Venue
DocType
2017
neural information processing systems
Journal
Volume
Citations 
PageRank 
abs/1710.01020
11
0.47
References 
Authors
19
6
Name
Order
Citations
PageRank
Sifei Liu122717.54
Shalini Gupta229920.42
Jinwei Gu368739.49
Guangyu Zhong4342.83
Yang Ming-Hsuan515303620.69
Jan Kautz63615198.77