Improve Fine-grained Visual Classification Accuracy by Controllable Location Knowledge Distillation

You Lin Tsai, Cheng Hung Lin, Po Yung Chou*

*此作品的通信作者

研究成果: 書貢獻/報告類型會議論文篇章

摘要

The current state-of-the-art network models have achieved remarkable performance. However, they often face an issue of having excessively large architectures, making them challenging to deploy on edge devices. In response to this challenge, a groundbreaking solution known as knowledge distillation has been introduced. The concept of knowledge distillation involves transferring information from a complex teacher model to a simpler student model, effectively reducing the model's complexity. Prior approach has demonstrated promising transfer effects through this technique. Nevertheless, in the field of fine-grained image classification, there has been limited exploration of distillation methods custom-tailored specifically for this domain. In this paper, we focus on knowledge distillation specifically designed for fine-grained image recognition. Notably, this strategy is inspired by the Class Activation Maps (CAM). We first train a complex model and use it generated feature maps with spatial information, which is call hint maps. Furthermore, we propose an adjustment strategy for this hint map, which can control local distribution of information. Referring to it as Controllable Size CAM (CTRLS-CAM). Use it as a guide allowing the student model to effectively learn from the teacher model's behavior and concentrate on discriminate details. In contrast to conventional distillation models, this strategy proves particularly advantageous for fine-grained recognition, enhancing learning outcomes and enabling the student model to achieve superior performance. In CTRLS-CAM method, we refine the hint maps value distribution, redefining the relative relationships between primary and secondary feature areas. We conducted experiments using the CUB200-2011 dataset, and our results demonstrated a significant accuracy improvement of about 5% compared to the original non-distilled student model. Moreover, our approach achieved a 1.23% enhancement over traditional knowledge distillation methods.

原文英語
主出版物標題2024 IEEE International Conference on Consumer Electronics, ICCE 2024
發行者Institute of Electrical and Electronics Engineers Inc.
ISBN(電子)9798350324136
DOIs
出版狀態已發佈 - 2024
事件2024 IEEE International Conference on Consumer Electronics, ICCE 2024 - Las Vegas, 美国
持續時間: 2024 1月 62024 1月 8

出版系列

名字Digest of Technical Papers - IEEE International Conference on Consumer Electronics
ISSN(列印)0747-668X
ISSN(電子)2159-1423

會議

會議2024 IEEE International Conference on Consumer Electronics, ICCE 2024
國家/地區美国
城市Las Vegas
期間2024/01/062024/01/08

ASJC Scopus subject areas

  • 工業與製造工程
  • 電氣與電子工程

指紋

深入研究「Improve Fine-grained Visual Classification Accuracy by Controllable Location Knowledge Distillation」主題。共同形成了獨特的指紋。

引用此