Domain Adaptive Pre-trained Model for Mushroom Image Classification

Yifei SHEN*, Zhuo LI, Yu YANG, Jiaxing SHEN

*Corresponding author for this work

Research output: Book Chapters | Papers in Conference ProceedingsConference paper (refereed)Researchpeer-review

Abstract

Mushroom is highly diverse in morphology and colors, and difficult for ordinary people to discriminate between them. The lack of high-quality labeled mushroom datasets is one of the bottlenecks restricting cutting-edge image recognition models to achieve high performance in mushroom image recognition. To address the limitation, in this paper, we will introduce a large mushroom dataset, called Mushroom-23, constructed by us. It collects over 35000 labeled mushroom images belonging to 203 popular mushroom species in Hong Kong. Along with constructing the new mushroom dataset, we also propose the domain adaptive pre-trained (DAPT) model to make the state-of-the-art Vision Transformer (ViT) adaptive to specific mushroom recognition. The DAPT is first pre-trained on ImageNet and Danish Fungi (DF20) datasets, then fine-tuned on the collected Mushroom-23 dataset to gain the capability of different categories of mushrooms. Extensive experimental results show DAPT outperforms all baseline models by a large margin in terms of Accuracy and Macro F1.

Original languageEnglish
Title of host publicationAdvanced Data Mining and Applications :19th International Conference, ADMA 2023, Shenyang, China, August 21–23, 2023, Proceedings, Part IV
EditorsXiaochun YANG, Heru SUHARTANTO, Guoren WANG, Bin WANG, Jing JIANG, Bing LI, Huaijie ZHU, Ningning CUI
PublisherSpringer
Chapter20
Pages284-296
Number of pages13
ISBN (Electronic)9783031466748
ISBN (Print)9783031466731
DOIs
Publication statusPublished - 2023
Event19th International Conference on Advanced Data Mining and Applications, ADMA 2023 - Shenyang, China
Duration: 21 Aug 202323 Aug 2023

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume14179
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

Conference19th International Conference on Advanced Data Mining and Applications, ADMA 2023
Country/TerritoryChina
CityShenyang
Period21/08/2323/08/23

Bibliographical note

Publisher Copyright:
© The Author(s), under exclusive license to Springer Nature Switzerland AG 2023.

Funding

This research was partially conducted in Research Institute for Artificial Intelligence of Things (RIAIoT) at PolyU and supported by PolyU Research and Innovation Office (No. BD4A), Hong Kong, China, and the Ph.D. Scientific Research Foundation, The Chongqing University of Posts and Telecommunications (No. E012A2022026), Chongqing, China, and benefited from the financial support of the Hong Kong Institute of Business Studies, Faculty of Business, Lingnan University, Hong Kong, China.

Keywords

  • Mushroom Recognition
  • Domain Adaption
  • Pre-trained Model
  • Vision Transformer

Fingerprint

Dive into the research topics of 'Domain Adaptive Pre-trained Model for Mushroom Image Classification'. Together they form a unique fingerprint.

Cite this