02/02/2021

Efficient Folded Attention for Medical Image Reconstruction and Segmentation

Hang Zhang, Jinwei Zhang, Rongguang Wang, Qihao Zhang, Pascal Spincemaille, Thanh D. Nguyen, Yi Wang

Keywords:

Abstract Paper Similar Papers

Abstract: Recently, 3D medical image reconstruction (MIR) and segmentation (MIS) based on deep neural networks have been developed with promising results, and attention mechanism has been further designed for performance enhancement. However, the large size of 3D volume images poses a great computational challenge to traditional attention methods. In this paper, we propose a folded attention (FA) approach to improve the computational efficiency of traditional attention methods on 3D medical images. The main idea is that we apply tensor folding and unfolding operations to construct four small sub-affinity matrices to approximate the original affinity matrix. Through four consecutive sub-attention modules of FA, each element in the feature tensor can aggregate spatial-channel information from all other elements. Compared to traditional attention methods, with the moderate improvement of accuracy, FA can substantially reduce the computational complexity and GPU memory consumption. We demonstrate the superiority of our method on two challenging tasks for 3D MIR and MIS, which are quantitative susceptibility mapping and multiple sclerosis lesion segmentation.

The video of this talk cannot be embedded. You can watch it here:
https://slideslive.com/38948450
(Link will open in new window)
 0
 0
 0
 0
  Share    
This is an embedded video. Talk and the respective paper are published at AAAI 2021 virtual conference. If you are one of the authors of the paper and want to manage your upload, see the question "My papertalk has been externally embedded..." in the FAQ section.

Comments

Post Comment
no comments yet
code of conduct: tbd Characters remaining: 140

Similar Papers

 0
 0
 0
 0
 5:00 
 0
 0
 0
 0
 0:50 
 0
 0
 0
 0
 5:51 
 0
 0
 0
 0
 1:01 
 0
 0
 0
 0
 2:57 
 0
 0
 0
 0
 2:59 
 0
 0
 0
 0
 0:59 
 0
 0
 0
 0
 5:00 
 0
 0
 0
 0
 13:38 
 0
 0
 0
 0
 15:49 
 0
 0
 0
 0
 1:00 
 0
 0
 0
 0
 5:01 
 0
 0
 0
 0
 14:56 
 0
 0
 0
 0
 14:15 
 0
 0
 0
 0
 5:00 
 0
 0
 0
 0
 3:19 
 0
 0
 0
 0
 2:56 
 0
 0
 0
 0
 9:26 
 0
 0
 0
 0
 10:57 
 0
 0
 0
 0
 4:56 
 0
 0
 0
 0
 5:00 
 0
 0
 0
 0
 3:23 
 0
 0
 0
 0
 13:29 
 0
 0
 0
 0
 9:59 
 0
 0
 0
 0
 1:01 
 0
 0
 0
 0
 8:45 
 0
 0
 0
 0
 15:20 
 0
 0
 0
 0
 5:00 
 0
 0
 0
 0
 3:14 
 0
 0
 0
 0
 4:59 
 0
 0
 0
 0
 4:59 
 0
 0
 0
 0
 3:29 
 0
 0
 0
 0
 11:04 
 0
 0
 0
 0
 5:00 
 0
 0
 0
 0
 12:21 
 0
 0
 0
 0
 3:37 
 0
 0
 0
 0
 9:22 
 0
 0
 0
 0
 4:50 
 0
 0
 0
 0
 11:28 
 0
 0
 0
 0
 3:50