that our proposed RWN consistently produces better performance over these baselines for the tasks of semantic segmentation and scene labeling.
2. Related Work
The recent introduction of fully convolutional networks
(FCNs) [20] has led to remarkable advances in semantic
segmentation. However, due to the large receptive ﬁelds
and many pooling layers, segments predicted by FCNs tend
to be blobby and lack ﬁne object boundary details. Recently
there have been several attempts to address these problems.
These approaches can be divided into several groups.The work in [4, 6, 25, 14, 4] used FCN predictions as
unary potentials in a separate globalization model that reﬁnes the segments using similarity cues based on regions
or boundaries. One disadvantage of these methods is that
the learning of the unary potentials and the training of the
globalization model are completely disjoint. As a result,
these methods often fail to capture semantic relationships
between objects, which produces segmentation results that
are spatially disjoint (see the right side of Fig. 1).
To address these issues, several recent methods [27, 19,
17] have proposed to integrate a CRF or a MRF into the network, thus enabling end-to-end training of the joint model.
Most current semantic segmentation methods rely on fully convolutional
networks (FCNs). However, their use of large receptive fields and many pooling
layers cause low spatial resolution inside the deep layers. This leads to
predictions with poor localization around the boundaries. Prior work has
attempted to address this issue by post-processing predictions with CRFs or
MRFs. But such models often fail to capture semantic relationships between
objects, which causes spatially disjoint predictions. To overcome these
problems, recent methods integrated CRFs or MRFs into an FCN framework. The
downside of these new models is that they have much higher complexity than
traditional FCNs, which renders training and testing more challenging.
  In this work we introduce a simple, yet effective Convolutional Random Walk
Network (RWN) that addresses the issues of poor boundary localization and
spatially fragmented predictions with very little increase in model complexity.
Our proposed RWN jointly optimizes the objectives of pixelwise affinity and
semantic segmentation. It combines these two objectives via a novel random walk
layer that enforces consistent spatial grouping in the deep layers of the
network. Our RWN is implemented using standard convolution and matrix
multiplication. This allows an easy integration into existing FCN frameworks
and it enables end-to-end training of the whole network via standard
back-propagation. Our implementation of RWN requires just $131$ additional
parameters compared to the traditional FCNs, and yet it consistently produces
an improvement over the FCNs on semantic segmentation and scene labeling.
Gedas Bertasius Lorenzo Torresani Stella X. Yu Jianbo Shi