Tensor dropout for robust learning
File(s)R_TRL_for_IEEE_special_issue-5.pdf (3.96 MB)
Accepted version
Author(s)
Type
Journal Article
Abstract
CNNs achieve high levels of performance by leveraging deep, over-parametrized neural architectures, trained on large datasets. However, they exhibit limited generalization abilities outside their training domain and lack robustness to corruptions such as noise and adversarial attacks. To improve robustness and obtain more computationally and memory efficient models, better inductive biases are needed. To provide such inductive biases, tensor layers have been successfully proposed to leverage multi-linear structure through higher-order computations. In this paper, we propose tensor dropout, a randomization technique that can be applied to tensor factorizations, such as those parametrizing tensor layers. In particular, we study tensor regression layers, parametrized by low-rank weight tensors and augmented with our proposed tensor dropout. We empirically show that our approach improves generalization for image classification on ImageNet and CIFAR-100. We also establish state-of-the-art accuracy for phenotypic trait prediction on the largest available dataset of brain MRI (U.K. Biobank), where multi-linear structure is paramount. In all cases, we demonstrate superior performance and significantly improved robustness, both to noisy inputs and to adversarial attacks. We establish the theoretical validity of our approach and the regularizing effect of tensor dropout by demonstrating the link between randomized tensor regression with tensor dropout and deterministic regularized tensor regression.
Date Issued
2021-04-01
Date Acceptance
2021-02-01
Citation
IEEE Journal of Selected Topics in Signal Processing, 2021, 15 (3), pp.630-640
ISSN
1932-4553
Publisher
Institute of Electrical and Electronics Engineers
Start Page
630
End Page
640
Journal / Book Title
IEEE Journal of Selected Topics in Signal Processing
Volume
15
Issue
3
Copyright Statement
© 2021 IEEE. Personal use of this material is permitted. Permission from IEEE must be obtained for all other uses, in any current or future media, including reprinting/republishing this material for advertising or promotional purposes, creating new collective works, for resale or redistribution to servers or lists, or reuse of any copyrighted component of this work in other works.
Sponsor
Engineering & Physical Science Research Council (EPSRC)
Medical Research Council (MRC)
Medical Research Council (MRC)
UK DRI Ltd
UK DRI Ltd
UK DRI Ltd
MRC-PHE Centre for Environment and Health
Grant Number
EP/N014529/1
MRC/MR/M024962/1
HQR00720
N/A
4050641385
N/A
1792496
Subjects
cs.LG
cs.LG
stat.ML
Networking & Telecommunications
0801 Artificial Intelligence and Image Processing
0906 Electrical and Electronic Engineering
1005 Communications Technologies
Publication Status
Published
Date Publish Online
2021-03-18