Projects per year
Abstract
Data-Free Knowledge Distillation (DFKD) has made significant recent strides by transferring knowledge from a teacher neural network to a student neural network without accessing the original data. Nonetheless, existing approaches encounter a significant challenge when attempting to generate samples from random noise inputs, which inherently lack meaningful information. Consequently, these models struggle to effectively map this noise to the ground-truth sample distribution, resulting in prolonging training times and low-quality outputs. In this paper, we propose a novel Noisy Layer Generation method (NAYER) which re-locates the random source from the input to a noisy layer and utilizes the meaningful constant label-text embedding (LTE) as the input. LTE is generated by using the language model once, and then it is stored in memory for all subsequent training processes. The significance of LTE lies in its ability to contain substantial meaningful inter-class information, enabling the generation of high-quality samples with only a few training steps. Simultaneously, the noisy layer plays a key role in addressing the issue of diversity in sample generation by preventing the model from overemphasizing the constrained label information. By reinitializing the noisy layer in each iteration, we aim to facilitate the generation of diverse samples while still retaining the method's efficiency, thanks to the ease of learning provided by LTE. Experiments carried out on multiple datasets demonstrate that our NAYER not only outperforms the state-of-the-art methods but also achieves speeds 5 to 15 times faster than previous approaches. The code is available at https://github.com/tmtuan1307/nayer.
Original language | English |
---|---|
Title of host publication | Proceedings - 2024 IEEE/CVF Conference on Computer Vision and Pattern Recognition, CVPR 2024 |
Editors | Eric Mortensen |
Place of Publication | Piscataway NJ USA |
Publisher | IEEE, Institute of Electrical and Electronics Engineers |
Pages | 23860-23869 |
Number of pages | 10 |
ISBN (Electronic) | 9798350353006 |
ISBN (Print) | 9798350353013 |
DOIs | |
Publication status | Published - 2024 |
Event | IEEE Conference on Computer Vision and Pattern Recognition 2024 - Seattle, United States of America Duration: 17 Jun 2024 → 21 Jun 2024 https://openaccess.thecvf.com/CVPR2024 (Proceedings) https://cvpr.thecvf.com/Conferences/2024 (Website) https://ieeexplore.ieee.org/xpl/conhome/10654794/proceeding (Proceedings) |
Conference
Conference | IEEE Conference on Computer Vision and Pattern Recognition 2024 |
---|---|
Abbreviated title | CVPR 2024 |
Country/Territory | United States of America |
City | Seattle |
Period | 17/06/24 → 21/06/24 |
Internet address |
|
Keywords
- data-free
- knowledge distillation
- knowledge transfer
- text embedding
Projects
- 1 Active
-
Exploiting Geometries of Learning for Fast, Adaptive and Robust AI
Phung, D. (Primary Chief Investigator (PCI)), Tafazzoli Harandi, M. (Chief Investigator (CI)), Hartley, R. I. (Chief Investigator (CI)), Le, T. (Chief Investigator (CI)) & Koniusz, P. (Partner Investigator (PI))
ARC - Australian Research Council
8/05/23 → 7/05/26
Project: Research