Published on Wed Mar 31 2021

FANet: A Feedback Attention Network for Improved Biomedical Image Segmentation

Nikhil Kumar Tomar, Debesh Jha, Michael A. Riegler, Håvard D. Johansen, Dag Johansen, Jens Rittscher, Pål Halvorsen, Sharib Ali
0
0
0
Abstract

With the increase in available large clinical and experimental datasets, there has been substantial amount of work being done on addressing the challenges 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 learnt 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 feedback attention model provides a substantial improvement on most segmentation metrics tested on seven publicly available biomedical imaging datasets demonstrating the effectiveness of the proposed FANet.

Fri Feb 08 2019
Computer Vision
FocusNet: An attention-based Fully Convolutional Network for Medical Image Segmentation
We propose a novel technique to incorporate attention within convolutional networks. We evaluate our model on benchmark segmentation datasets in skin cancer segmentation and lung lesion segmentation. Results show a highly competitive performance when compared with U-Net.
0
0
0
Wed Apr 17 2019
Computer Vision
Automated Design of Deep Learning Methods for Biomedical Image Segmentation
nnU-Net surpasses most specialised deep learning pipelines in 19 public international competitions. The results demonstrate a vast hidden potential in the systematic adaptation of deep learning methods to different datasets.
0
0
0
Mon Sep 28 2020
Computer Vision
Medical Image Segmentation Using Deep Learning: A Survey
Deep learning has been widely used for medical image segmentation. This paper focuses on supervised and weakly supervised learning approaches. We analyze literatures in three aspects: the selection of backbone networks, the design of network blocks, and the improved loss functions.
0
0
0
Sun Feb 21 2021
Computer Vision
Medical Transformer: Gated Axial-Attention for Medical Image Segmentation
The proposed Medical Transformer (MedT) is evaluated on three different medical image segmentation datasets. It is shown that it achieves better performance than the convolutional and other related transformer-based architectures.
0
0
0
Fri Jul 02 2021
Computer Vision
UTNet: A Hybrid Transformer Architecture for Medical Image Segmentation
UTNet is a simple yet powerful hybrid Transformer architecture. It integrates self-attention into aconvolutional neural network for enhancing medical image segmentation. UTNet demonstrates superior segmentation performance and robustness against the state-of-the-art approaches.
0
0
0
Wed Dec 05 2018
Computer Vision
Brain Segmentation from k-space with End-to-end Recurrent Attention Network
The proposed method outperforms state-of-the-art methods. The end-to-end framework consists of a unique task-driven attention module that recurrently utilizes intermediate segmentation estimation.
0
0
0