Model pruning using hypernetwork with partial weight training and semisupervised multitask learning (Conference Presentation)
In model compression, filter pruning stands out as a pivotal technique. Its significance becomes particularly crucial as the present deep learning models are developed into larger and more complicated architectures, which leads to massive parameters and high floating-point operations per second (FLOPs). Challenges have appeared due to the high computational demands associated with these advanced model structures. In this work, we introduce two novel methods aimed at addressing the challenges above: innovative automatic filter pruning methods via semi-supervised multi-task learning (SSMTL) hypernetwork and partial weight training hypernetwork, respectively. Both methods effectively train the hypernetwork and enhance the precision of the neural architecture search with reinforcement learning. Compared to other filter pruning methods, our approach achieves higher model accuracy at similar pruning ratios.
Authors
- Albert Lin
- Po‐Han Chen (ORCID: https://orcid.org/0000-0002-9471-4471)
- Shih-Teng Yang
Institutions
- National Yang Ming Chiao Tung University (TW)
Publication Details
- Published
- 2024-10-03
- DOI
- https://doi.org/10.1117/12.3027322
- Primary Topic
- Machine Learning and Data Classification
- Type
- article
- Field-Weighted Citation Impact
- 0.00