arXiv:2505.05513cs.CV2025-05被引 6

用CNN自动分类稻米品种,还解释了模型判断依据。

Exploring Convolutional Neural Networks for Rice Grain Classification: An Explainable AI Approach

  • 基于CNN构建自动稻米分类框架,提升效率与准确性。
  • 模型准确率高,各品类ROC曲线下面积为1.0,误判极少。
  • 结合LIME和SHAP技术,揭示稻米特征如何影响分类决策。

水稻是全球重要的主食作物,在促进国际贸易、经济增长和营养供给方面具有关键作用。中国、印度、巴基斯坦、泰国、越南和印度尼西亚等亚洲国家在水稻种植与利用方面贡献突出,培育出多种稻米品种,如短粒、长粒的巴斯马蒂、茉莉香、凯纳特赛拉、伊普萨拉、阿博里奥等,满足多样化的饮食偏好与文化传统。无论是本地还是国际交易,确保稻米品质以满足客户要求并维护国家声誉至关重要。人工质检耗时费力且易出错,因此亟需自动化解决方案。本文提出一种基于卷积神经网络(CNN)的自动稻米品种分类框架,通过准确率、召回率、精确率和F1分数等指标评估性能。模型经严格训练与验证,取得极高的准确率,并在每个类别的受试者工作特征(ROC)曲线下面积达到1.0,表明分类效果优异。混淆矩阵分析显示各类别间误判极少。此外,融合LIME和SHAP等可解释性技术,揭示了稻米特定形态特征对分类结果的影响,增强了模型可信度。

原文摘要 · Abstract (English)

Rice is an essential staple food worldwide that is important in promoting international trade, economic growth, and nutrition. Asian countries such as China, India, Pakistan, Thailand, Vietnam, and Indonesia are notable for their significant contribution to the cultivation and utilization of rice. These nations are also known for cultivating different rice grains, including short and long grains. These sizes are further classified as basmati, jasmine, kainat saila, ipsala, arborio, etc., catering to diverse culinary preferences and cultural traditions. For both local and international trade, inspecting and maintaining the quality of rice grains to satisfy customers and preserve a country's reputation is necessary. Manual quality check and classification is quite a laborious and time-consuming process. It is also highly prone to mistakes. Therefore, an automatic solution must be proposed for the effective and efficient classification of different varieties of rice grains. This research paper presents an automatic framework based on a convolutional neural network (CNN) for classifying different varieties of rice grains. We evaluated the proposed model based on performance metrics such as accuracy, recall, precision, and F1-Score. The CNN model underwent rigorous training and validation, achieving a remarkable accuracy rate and a perfect area under each class's Receiver Operating Characteristic (ROC) curve. The confusion matrix analysis confirmed the model's effectiveness in distinguishing between the different rice varieties, indicating minimal misclassifications. Additionally, the integration of explainability techniques such as LIME (Local Interpretable Model-agnostic Explanations) and SHAP (SHapley Additive exPlanations) provided valuable insights into the model's decision-making process, revealing how specific features of the rice grains influenced classification outcomes.

图像分类深度学习可解释性农业AI

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