Multi-label visual feature learning with attentional aggregation
Document Type
Conference Proceeding
Publication Date
3-1-2020
Abstract
Today convolutional neural networks (CNNs) have reached out to specialized applications in science communities that otherwise would not be adequately tackled. In this paper, we systematically study a multi-label annotation problem of x-ray scattering images in material science. For this application, we tackle an open challenge with training CNNs - identifying weak scattered patterns with diffuse background interference, which is common in scientific imaging. We articulate an Attentional Aggregation Module (AAM) to enhance feature representations. First, we reweight and highlight important features in the images using data-driven attention maps. We decompose the attention maps into channel and spatial attention components. In the spatial attention component, we design a mechanism to generate multiple spatial attention maps tailored for diversified multi-label learning. Then, we condense the enhanced local features into non-local representations by performing feature aggregation. Both attention and aggregation are designed as network layers with learnable parameters so that CNN training remains fluidly end-to-end, and we apply it in-network a few times so that the feature enhancement is multi-scale. We conduct extensive experiments on CNN training and testing, as well as transfer learning, and empirical studies confirm that our method enhances the discriminative power of visual features of scientific imaging.
Identifier
85085502342 (Scopus)
ISBN
[9781728165530]
Publication Title
Proceedings 2020 IEEE Winter Conference on Applications of Computer Vision Wacv 2020
External Full Text Location
https://doi.org/10.1109/WACV45572.2020.9093311
First Page
2190
Last Page
2198
Grant
1531492
Fund Ref
National Science Foundation
Recommended Citation
Guan, Ziqiao; Yager, Kevin G.; Yu, Dantong; and Qin, Hong, "Multi-label visual feature learning with attentional aggregation" (2020). Faculty Publications. 5440.
https://digitalcommons.njit.edu/fac_pubs/5440
