arXiv:2410.02304cs.CV2024-10被引 6

用高效模型+注意力机制,实现快速高精度食物分类。

A Novel Method for Accurate & Real-time Food Classification: The Synergistic Integration of EfficientNetB7, CBAM, Transfer Learning, and Data Augmentation

  • 融合EfficientNetB7与CBAM注意力,提升特征捕捉能力。
  • 在Food11数据集上达96.40%平均准确率,每秒处理超60张图。
  • 适合需要实时精准识别的智慧餐饮、营养管理场景。

将人工智能融入现代社会正深刻变革日常任务效率。在食品领域,AI识别可改善营养追踪、减少浪费并提升生产消费效率。准确的食物分类是后续应用的关键前提,需兼顾高精度与快速处理。尽管已有研究,但在性能与速度间仍存短板。本研究采用先进EfficientNetB7架构,结合迁移学习、数据增强与CBAM注意力模块,在Kaggle的Food11数据集上实现96.40%平均准确率,该数据集含11类共16643张图像,类别间差异小、内部变化大。模型推理时可在1秒内完成超过60张新图像的分类,具备实际部署潜力,显著提升食物识别与后续流程效率。

原文摘要 · Abstract (English)

Integrating artificial intelligence into modern society is profoundly transformative, significantly enhancing productivity by streamlining various daily tasks. AI-driven recognition systems provide notable advantages in the food sector, including improved nutrient tracking, tackling food waste, and boosting food production and consumption efficiency. Accurate food classification is a crucial initial step in utilizing advanced AI models, as the effectiveness of this process directly influences the success of subsequent operations; therefore, achieving high accuracy at a reasonable speed is essential. Despite existing research efforts, a gap persists in improving performance while ensuring rapid processing times, prompting researchers to pursue cost-effective and precise models. This study addresses this gap by employing the state-of-the-art EfficientNetB7 architecture, enhanced through transfer learning, data augmentation, and the CBAM attention module. This methodology results in a robust model that surpasses previous studies in accuracy while maintaining rapid processing suitable for real-world applications. The Food11 dataset from Kaggle was utilized, comprising 16643 imbalanced images across 11 diverse classes with significant intra-category diversities and inter-category similarities. Furthermore, the proposed methodology, bolstered by various deep learning techniques, consistently achieves an impressive average accuracy of 96.40%. Notably, it can classify over 60 images within one second during inference on unseen data, demonstrating its ability to deliver high accuracy promptly. This underscores its potential for practical applications in accurate food classification and enhancing efficiency in subsequent processes.

食物识别高效网络注意力机制

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。