Plants (Nov 2024)

AIpollen: An Analytic Website for Pollen Identification Through Convolutional Neural Networks

  • Xingchen Yu,
  • Jiawen Zhao,
  • Zhenxiu Xu,
  • Junrong Wei,
  • Qi Wang,
  • Feng Shen,
  • Xiaozeng Yang,
  • Zhonglong Guo

DOI
https://doi.org/10.3390/plants13223118
Journal volume & issue
Vol. 13, no. 22
p. 3118

Abstract

Read online

With the rapid development of artificial intelligence, deep learning has been widely applied to complex tasks such as computer vision and natural language processing, demonstrating its outstanding performance. This study aims to exploit the high precision and efficiency of deep learning to develop a system for the identification of pollen. To this end, we constructed a dataset across 36 distinct genera. In terms of model selection, we employed a pre-trained ResNet34 network and fine-tuned its architecture to suit our specific task. For the optimization algorithm, we opted for the Adam optimizer and utilized the cross-entropy loss function. Additionally, we implemented ELU activation function, data augmentation, learning rate decay, and early stopping strategies to enhance the training efficiency and generalization capability of the model. After training for 203 epochs, our model achieved an accuracy of 97.01% on the test set and 99.89% on the training set. Further evaluation metrics, such as an F1 score of 95.9%, indicate that the model exhibits good balance and robustness across all categories. To facilitate the use of the model, we develop a user-friendly web interface. Users can upload images of pollen grains through the URL link provided in this article) and immediately receive predicted results of their genus names. Altogether, this study has successfully trained and validated a high-precision pollen grain identification model, providing a powerful tool for the identification of pollen.

Keywords