Title
Select, Attend, and Transfer: Light, Learnable Skip Connections.
Abstract
Skip connections in deep networks have improved both segmentation and classification performance by facilitating the training of deeper network architectures and reducing the risks for vanishing gradients. The skip connections equip encoder-decoder like networks with richer feature representations, but at the cost of higher memory usage, computation, and possibly resulting in transferring non-discriminative feature maps. In this paper, we focus on improving the skip connections used in segmentation networks. We propose light, learnable skip connections which learn to first select the most discriminative channels, and then aggregate the selected ones as single channel attending to the most discriminative regions of input. We evaluate the proposed method on 3 different 2D and volumetric datasets and demonstrate that the proposed skip connections can outperform the traditional heavy skip connections of 4 different models in terms of segmentation accuracy (2% Dice), memory usage (at least 50%), and the number of network parameters (up to 70%).
Year
DOI
Venue
2018
10.1007/978-3-030-32692-0_48
Lecture Notes in Computer Science
Keywords
Field
DocType
Deep neural networks,Skip connections,Image segmentation
Pattern recognition,Computer science,Segmentation,Communication channel,Network architecture,Artificial intelligence,Discriminative model,Machine learning,Computation
Journal
Volume
ISSN
Citations 
11861
0302-9743
1
PageRank 
References 
Authors
0.37
24
11
Name
Order
Citations
PageRank
Saeid Asgari Taghanaki1203.44
Aïcha BenTaieb2261.47
Anmol Sharma310.71
Zhou S. Kevin447441.40
Yefeng Zheng51391114.67
Bogdan Georgescu61638138.49
Puneet Sharma727138.61
Sasa Grbic88213.77
Zhoubing Xu912012.56
Dorin Comaniciu108389601.83
Ghassan Hamarneh111353110.14