DECORE: Deep Compression with Reinforcement Learning

no code implementations • CVPR 2022 • Manoj Alwani, Yang Wang, Vashisht Madhavan

For a larger dataset like ImageNet with just 30 epochs of training, it can compress the ResNet-50 architecture by 44. 7% and reduce FLOPs by 42. 3%, with just a 0. 69% drop on Top-5 accuracy of the uncompressed model.

reinforcement-learning Reinforcement Learning (RL)