Do Neural Networks for Segmentation Understand Insideness?

Journal: Neural computation
Published Date:

Abstract

The insideness problem is an aspect of image segmentation that consists of determining which pixels are inside and outside a region. Deep neural networks (DNNs) excel in segmentation benchmarks, but it is unclear if they have the ability to solve the insideness problem as it requires evaluating long-range spatial dependencies. In this letter, we analyze the insideness problem in isolation, without texture or semantic cues, such that other aspects of segmentation do not interfere in the analysis. We demonstrate that DNNs for segmentation with few units have sufficient complexity to solve the insideness for any curve. Yet such DNNs have severe problems with learning general solutions. Only recurrent networks trained with small images learn solutions that generalize well to almost any curve. Recurrent networks can decompose the evaluation of long-range dependencies into a sequence of local operations, and learning with small images alleviates the common difficulties of training recurrent networks with a large number of unrolling steps.

Authors

  • Kimberly Villalobos
    Center for Brains, Minds and Machines, MIT, Cambridge, MA 02139, U.S.A. kimvc@mit.edu.
  • Vilim Štih
    Center for Brains, Minds and Machines, MIT, Cambridge, MA 02139, U.S.A., and Max Planck Institute of Neurobiology, 82152 Martinsried, Germany vilim@neuro.mpg.de.
  • Amineh Ahmadinejad
    Center for Brains, Minds and Machines, MIT, Cambridge, MA 02139, U.S.A. amineh@mit.edu.
  • Shobhita Sundaram
    Center for Brains, Minds and Machines, MIT, Cambridge, MA 02139, U.S.A. shobhita@mit.edu.
  • Jamell Dozier
    Center for Brains, Minds and Machines, MIT, Cambridge, MA 02139, U.S.A. jamell@mit.edu.
  • Andrew Francl
    Center for Brains, Minds and Machines, MIT, Cambridge, MA 02139, U.S.A. francl@mit.edu.
  • Frederico Azevedo
    Center for Brains, Minds and Machines, MIT, Cambridge, MA 02139, U.S.A. fazevedo@mit.edu.
  • Tomotake Sasaki
    Fujitsu Laboratories, Kawasaki 211-8588, Japan, and Center for Brains, Minds and Machines, MIT, Cambridge, MA 02139, U.S.A. tomotake.sasaki@fujitsu.com.
  • Xavier Boix
    Center for Brains, Minds and Machines, MIT, Cambridge, MA 02139, U.S.A. xboix@mit.edu.