arrow
返回

Tensor Dropout for Robust Learning

delete2021-04-01
delete17
delete
OA
AI
A
Arinbjörn Kolbeinsson *
J
Jean Kossaifi
Y
Yannis Panagakis
A
Adrian Bulat
A
Anandkumar, Animashree
I
Ioanna Tzoulaki
P
Paul M. Matthews
DOI:10.1109/JSTSP.2021.3064182delete
delete原文链接
delete分享
delete收藏
查看原文
摘要

摘要

En 中文
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.
Keyword:
Tensors
Training
Robustness
Magnetic resonance imaging
Perturbation methods
Diseases
Deep learning
Deep learning
randomized tensor regression
robustness
stochastic regularization
tensor dropout
tensor methods
tensor regression
tensor regression layers
AI总结

AI总结

对已上传原文的论文进行重点信息的提取,主要内容包括:简要概述、研究摘要、背景介绍、关键亮点、图文解析、展望与总结。

期刊

IEEE Journal of Selected Topics in Signal Processing 封面图
IEEE Journal of Selected Topics in Signal Processing
IF:
13.7
论文数:
1.9K
被引数:
1.1W

机构

N
nvidia corporation
学者数:
767
论文数: 439
被引数: 1
I
Imperial College London
学者数:
8.3W
论文数: 7.3W
被引数: 11.1W
引用论文

引用论文

Chemical synthesis of titania (TiO2) powder via mixed precursor route for membrane applications
err2001-02-01
err0
PREAI
errA.J Patil; M.H Shinde; H.S Potdar; S.B Deshpande; S.R Sainkar; S Mayadevi; S.K Date
err分享
err收藏
Acidities of Tertiary Alkyl Hydroperoxides
err2003-09-09
err0
PREAI
errWilliam Richardson; Vernon Hodge
err分享
err收藏
Behavioral detection of 60‐Hz electric fields by rats
err2005-12-11
err0
PREAI
errSander Stern; Victor G. Laties; Charles V. Stancampiano; Christopher Cox; John O. de Lorge
err分享
err收藏
A new approach to the neurological assessment of the preterm and full-term newborn infant
err1980-01-01
err0
PREAI
errLilly MS Dubowitz; Victor Dubowitz; Penelope Palmer; M Verghote
err分享
err收藏
err分享
err收藏
Accelerated MRI-predicted brain ageing and its associations with cardiometabolic and brain disorders
err2020-11-17
err40
errOAAI
errKolbeinsson, Arinbjorn; Filippi, Sarah; Panagakis, Yannis; Matthews, Paul M.; Elliott, Paul; Dehghan, Abbas; Tzoulaki, Ioanna
err分享
err收藏
学者 查看更多内容