Back to Search Start Over

Multi-path x-D recurrent neural networks for collaborative image classification

Authors :
S. Antic
Alexis B. Paulson
Bennett A. Landman
Pierre P. Massion
Yucheng Tang
Shunxing Bao
Riqiang Gao
Emily S. Epstein
Steve Deppen
Yuankai Huo
Kim L. Sandler
Source :
Neurocomputing
Publication Year :
2020
Publisher :
Elsevier BV, 2020.

Abstract

With the rapid development of image acquisition and storage, multiple images per class are commonly available for computer vision tasks (e.g., face recognition, object detection, medical imaging, etc.). Recently, the recurrent neural network (RNN) has been widely integrated with convolutional neural networks (CNN) to perform image classification on ordered (sequential) data. In this paper, by permutating multiple images as multiple dummy orders, we generalize the ordered “RNN+CNN” design (longitudinal) to a novel unordered fashion, called Multi-path x-D Recurrent Neural Network (MxDRNN) for image classification. To the best of our knowledge, few (if any) existing studies have deployed the RNN framework to unordered intra-class images to leverage classification performance. Specifically, multiple learning paths are introduced in the MxDRNN to extract discriminative features by permutating input dummy orders. Eight datasets from five different fields (MNIST, 3D-MNIST, CIFAR, VGGFace2, and lung screening computed tomography) are included to evaluate the performance of our method. The proposed MxDRNN improves the baseline performance by a large margin across the different application fields (e.g., accuracy from 46.40% to 76.54% in VGGFace2 test pose set, AUC from 0.7418 to 0.8162 in NLST lung dataset). Additionally, empirical experiments show the MxDRNN is more robust to category-irrelevant attributes (e.g., expression, pose in face images), which may introduce difficulties for image classification and algorithm generalizability. The code is publicly available.

Details

ISSN :
09252312
Volume :
397
Database :
OpenAIRE
Journal :
Neurocomputing
Accession number :
edsair.doi.dedup.....0d40d3790ca1469a4b2eeb3ad2d43435
Full Text :
https://doi.org/10.1016/j.neucom.2020.02.033