We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.CV

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo ScienceWISE logo

Computer Science > Computer Vision and Pattern Recognition

Title: FANet: A Feedback Attention Network for Improved Biomedical Image Segmentation

Abstract: The increase of available large clinical and experimental datasets has contributed to a substantial amount of important contributions in the area of biomedical image analysis. Image segmentation, which is crucial for any quantitative analysis, has especially attracted attention. Recent hardware advancement has led to the success of deep learning approaches. However, although deep learning models are being trained on large datasets, existing methods do not use the information from different learning epochs effectively. In this work, we leverage the information of each training epoch to prune the prediction maps of the subsequent epochs. We propose a novel architecture called feedback attention network (FANet) that unifies the previous epoch mask with the feature map of the current training epoch. The previous epoch mask is then used to provide a hard attention to the learned feature maps at different convolutional layers. The network also allows to rectify the predictions in an iterative fashion during the test time. We show that our proposed \textit{feedback attention} model provides a substantial improvement on most segmentation metrics tested on seven publicly available biomedical imaging datasets demonstrating the effectiveness of FANet. The source code is available at \url{this https URL}.
Subjects: Computer Vision and Pattern Recognition (cs.CV); Image and Video Processing (eess.IV)
Journal reference: IEEE Transactions on Neural Networks and Learning Systems, 2022
Cite as: arXiv:2103.17235 [cs.CV]
  (or arXiv:2103.17235v3 [cs.CV] for this version)

Submission history

From: Debesh Jha [view email]
[v1] Wed, 31 Mar 2021 17:34:20 GMT (8860kb,D)
[v2] Mon, 31 Jan 2022 03:25:47 GMT (37652kb,D)
[v3] Fri, 25 Mar 2022 18:17:11 GMT (27098kb)

Link back to: arXiv, form interface, contact.