Recognition of food images based on transfer learning and ensemble learning.

PLoS One

Department of Computer Engineering, Jinling Institute of Technology, Nanjing Jiangsu, China.

Published: January 2024

AI Article Synopsis

  • - The study addresses the challenges in accurately recognizing food images, which are important for nutrition monitoring and food recommendations, due to issues like complex backgrounds and varying characteristics within food categories.
  • - The authors propose a new method that combines transfer learning, using pre-trained convolutional neural network models (like VGG19 and ResNet50), with ensemble learning for improved accuracy in recognizing food images.
  • - Experimental results show that the ensemble model achieved the highest accuracy of 96.88%, outperforming individual models, indicating that this approach is effective and practical for food image recognition.

Article Abstract

The recognition of food images is of great significance for nutrition monitoring, food retrieval and food recommendation. However, the accuracy of recognition had not been high enough due to the complex background of food images and the characteristics of small inter-class differences and large intra-class differences. To solve these problems, this paper proposed a food image recognition method based on transfer learning and ensemble learning. Firstly, generic image features were extracted by using the convolutional neural network models (VGG19, ResNet50, MobileNet V2, AlexNet) pre-trained on the ImageNet dataset. Secondly, the 4 pre-trained models were transferred to the food image dataset for model fine-tuning. Finally, different basic learner combination strategies were adopted to establish the ensemble model and classify feature information. In this paper, several kinds of experiments were performed to compare the results of food image recognition between single models and ensemble models on food-11 dataset. The experimental results demonstrated that the accuracy of the ensemble model was the highest, reaching 96.88%, which was superior to any base learner. Therefore, the convolutional neural network model based on transfer learning and ensemble learning has strong learning ability and generalization ability, and it is feasible and practical to apply the method to food image recognition.

Download full-text PDF

Source
http://www.ncbi.nlm.nih.gov/pmc/articles/PMC10798480PMC
http://journals.plos.org/plosone/article?id=10.1371/journal.pone.0296789PLOS

Publication Analysis

Top Keywords

food image
16
food images
12
based transfer
12
transfer learning
12
learning ensemble
12
ensemble learning
12
image recognition
12
recognition food
8
food
8
convolutional neural
8

Similar Publications

Want AI Summaries of new PubMed Abstracts delivered to your In-box?

Enter search terms and have AI summaries delivered each week - change queries or unsubscribe any time!