Up to 100x Faster Data-free Knowledge Distillation
Fang, Gongfan, Mo, Kanya, Wang, Xinchao, Song, Jie, Bei, Shitao, Zhang, Haofei, Song, Mingli
–arXiv.org Artificial Intelligence
Data-free knowledge distillation (DFKD) has recently been attracting increasing attention from research communities, attributed to its capability to compress a model only using synthetic data. Despite the encouraging results achieved, state-of-the-art DFKD methods still suffer from the inefficiency of data synthesis, making the data-free training process extremely time-consuming and thus inapplicable for large-scale tasks. In this work, we introduce an efficacious scheme, termed as FastDFKD, that allows us to accelerate DFKD by a factor of orders of magnitude. At the heart of our approach is a novel strategy to reuse the shared common features in training data so as to synthesize different data instances. Unlike prior methods that optimize a set of data independently, we propose to learn a meta-synthesizer that seeks common features as the initialization for the fast data synthesis. As a result, FastDFKD achieves data synthesis within only a few steps, significantly enhancing the efficiency of data-free training. Experiments over CIFAR, NYUv2, and ImageNet demonstrate that the proposed FastDFKD achieves 10$\times$ and even 100$\times$ acceleration while preserving performances on par with state of the art.
arXiv.org Artificial Intelligence
Dec-12-2021
- Country:
- Asia
- Singapore (0.04)
- China > Zhejiang Province (0.04)
- Asia
- Genre:
- Research Report (0.64)
- Industry:
- Education (0.70)
- Technology:
- Information Technology > Artificial Intelligence
- Natural Language (1.00)
- Vision (0.94)
- Machine Learning > Neural Networks (0.46)
- Information Technology > Artificial Intelligence