End-To-End Learned Random Walker for Seeded Image Segmentation

Lorenzo Cerrone; Alexander Zeilmann; Fred A. Hamprecht

CVPR 2019poster32 citations

End-To-End Learned Random Walker for Seeded Image Segmentation

Lorenzo Cerrone, Alexander Zeilmann, Fred A. Hamprecht

Abstract

We present an end-to-end learned algorithm for seeded segmentation. Our method is based on the Random Walker algorithm, where we predict the edge weights of the un- derlying graph using a convolutional neural network. This can be interpreted as learning context-dependent diffusiv- ities for a linear diffusion process. After calculating the exact gradient for optimizing these diffusivities, we pro- pose simplifications that sparsely sample the gradient while still maintaining competitive results. The proposed method achieves the currently best results on the seeded CREMI neuron segmentation challenge.

BibTeX

@inproceedings{cvpr2019_endtoendlearnedr,
  title = {End-To-End Learned Random Walker for Seeded Image Segmentation},
  author = {Lorenzo Cerrone and Alexander Zeilmann and Fred A. Hamprecht},
  booktitle = {CVPR 2019},
  year = {2019}
}