arrow
返回

Incremental Trainable Parameter Selection in Deep Neural Networks

delete2024-05-01
delete4
delete
OA
AI
A
Anshul Thakur *
V
Vinayak Abrol
P
Pulkit Sharma
朱
朱婷婷 (Tingting Zhu)
D
David A. Clifton
DOI:10.1109/TNNLS.2022.3210297delete
delete原文链接
delete分享
delete收藏
摘要

摘要

En 中文
This article explores the utilization of the effective degree-of-freedom (DoF) of a deep learning model to regularize its stochastic gradient descent (SGD)-based training. The effective DoF of a deep learning model is defined only by a subset of its total parameters. This subset is highly responsive or sensitive toward the training loss, and its cardinality can be used to govern the effective DoF of a model during training. To this aim, the incremental trainable parameter selection (ITPS) algorithm is introduced in this article. The proposed ITPS algorithm acts as a wrapper over SGD and incrementally selects the parameters for updation that exhibit the maximum sensitivity toward the training loss. Hence, it gradually increases the DoF of the model during training. In ideal cases, the proposed algorithm arrives at a model configuration (i.e., DoF) optimum for the task at hand. This whole process results in a regularization-like behavior induced by a gradual increment of the DoF. Since the selection and updation of parameters is a function of the training loss, the proposed algorithm can be seen as a task and data-dependent regularization mechanism. This article exhibits the general utility of ITPS by evaluating it on various prominent neural network architectures such as CNNs, transformers, recurrent neural networks (RNNs), and multilayer perceptrons. These models are trained for image classification and healthcare tasks using the publicly available CIFAR-10, SLT-10, and MIMIC-III datasets.
Keyword:
Training
Task analysis
Deep learning
Complexity theory
Behavioral sciences
Medical services
Analytical models
Healthcare informatics
incremental training
regularization

期刊

IEEE Transactions on Neural Networks and Learning Systems 封面图
IEEE Transactions on Neural Networks and Learning Systems
IF:
8.9
论文数:
7.6K
被引数:
7.2W

机构

U
University of Oxford
学者数:
2.2K
论文数: 872
被引数: 0
I
indian institute of technology system (iit system)
学者数:
9.5W
论文数: 9.9W
被引数: 93
引用论文

引用论文

Self-organised path formation in a swarm of robots
err2011-04-21
err0
PREAI
errValerio Sperati; Vito Trianni; Stefano Nolfi
err分享
err收藏
Volatile Substance Misuse Among High School Students in South America
err2011-05-24
err0
errOAAI
errMarya Hynes-Dowell; Pedro Mateu-Gelabert; Helena Maria Taunhauser Barros; Jorge Delva
err分享
err收藏
Self-assembled spongy-like MnO2 electrode materials for supercapacitors自组装海绵状MnO2超级电容器电极材料
err2012-08-01
err0
PREAI
errMeng Dong; Yu Xin Zhang; Hong Fang Song; Xin Qiu; Xiao Dong Hao; Chuan Pu Liu; Yuan Yuan; Xin Lu Li; Jia Mu Huang
err分享
err收藏
Regularizing Multilayer Perceptron for Robustness
err2018-08-01
err31
PREAI
errDey, Prasenjit; Nag, Kaustuv; Pal, Tandra; Pal, Nikhil R.
err分享
err收藏
学者 查看更多内容