Rare and Endangered Plant Leaf Identification Method Based on Transfer Learning and Knowledge Distillation

Abstract

Considering the limited sample size of rare and endangered plant leaves and the issue that leaf identification is mainly conducted using mobile smart devices and other technology with low computing power, this paper proposes a rare and endangered plant leaf identification method based on transfer learning and knowledge distillation. Following the expansion of data sets containing rare and endangered plant leaves, the last fully connected layer was replaced with trained Alexnet, VGG16, GoogLeNet, and ResNet models to conduct transfer learning, and realize a relatively high success rate in identifying images of these species. Then, knowledge distillation was utilized to transfer Alexnet, VGG16, GoogLeNet, and ResNet models into a lightweight model. The experiment results indicate that, compared with other methods, the lightweight rare and endangered plant identification model trained with the methods described in this paper was not only more accurate but also less complex than its alternatives.