Feedback Attention for Cell Image Segmentation

Hiroki Tsuda, Eisuke Shibuya, Kazuhiro Hotta

Reviews Visibility: The authors agree that reviews are made publicly visible, if the submission is accepted.

Abstract: In this paper, we address cell image segmentation task by Feedback Attention mechanism like feedback processing. Unlike conventional neural network models of feedforward processing, we focused on the feedback processing in the human brain and assumed that the network learns like a human by connecting feature maps from deep layers to shallow layers. We propose some Feedback Attentions which imitate human brain and feeds back the feature maps of output layer to close layer to the input. U-Net with Feedback Attention showed better result than the conventional methods using only feedforward processing.

Keywords: Cell image, Semantic Segmentation, Attention Mechanism, Feedback Mechanism