Robust Neural Pruning with Gradient Sampling Optimization for Residual Neural Networks

Fuente: arXiv
Saved in:
Bibliographic Details
Main Author: Yun, Juyoung
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914772981645312
author Yun, Juyoung
author_facet Yun, Juyoung
contents This research embarks on pioneering the integration of gradient sampling optimization techniques, particularly StochGradAdam, into the pruning process of neural networks. Our main objective is to address the significant challenge of maintaining accuracy in pruned neural models, critical in resource-constrained scenarios. Through extensive experimentation, we demonstrate that gradient sampling significantly preserves accuracy during and after the pruning process compared to traditional optimization methods. Our study highlights the pivotal role of gradient sampling in robust learning and maintaining crucial information post substantial model simplification. The results across CIFAR-10 datasets and residual neural architectures validate the versatility and effectiveness of our approach. This work presents a promising direction for developing efficient neural networks without compromising performance, even in environments with limited computational resources.
format Preprint
id arxiv_https___arxiv_org_abs_2312_16020
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Robust Neural Pruning with Gradient Sampling Optimization for Residual Neural Networks
Yun, Juyoung
Machine Learning
This research embarks on pioneering the integration of gradient sampling optimization techniques, particularly StochGradAdam, into the pruning process of neural networks. Our main objective is to address the significant challenge of maintaining accuracy in pruned neural models, critical in resource-constrained scenarios. Through extensive experimentation, we demonstrate that gradient sampling significantly preserves accuracy during and after the pruning process compared to traditional optimization methods. Our study highlights the pivotal role of gradient sampling in robust learning and maintaining crucial information post substantial model simplification. The results across CIFAR-10 datasets and residual neural architectures validate the versatility and effectiveness of our approach. This work presents a promising direction for developing efficient neural networks without compromising performance, even in environments with limited computational resources.
title Robust Neural Pruning with Gradient Sampling Optimization for Residual Neural Networks
topic Machine Learning
url https://arxiv.org/abs/2312.16020