基于改进ConvNeXt与可解释人工智能的金相组织图像分类研究

    Research on Metallographic Structure Image Classification Based on Improved ConvNeXt and Explainable Artificial Intelligence

    • 摘要: 采用以深度学习为代表的机器学习方法对金相组织图像进行智能分类是材料信息学研究的重要内容。本工作以火电用钢金相检验图像建立数据集,在深度卷积中添加逐点卷积、引入Ghost模块和注意力机制改进ConvNeXt模型,结合数据增强与迁移学习建立金相组织分类模型,通过消融实验和对比实验评价模型的性能。结果表明,所建模型的平均精度均值、精确率和召回率分别达到97%、96%和96%,能够满足金相组织分类任务需求。为提升模型可解释性,进一步构建了局部可解释性—全局可解释性双层解释框架:在局部层面,通过基于梯度的显著性分析量化特征通道对分类结果的贡献,并定位关键响应区域;在全局层面,采用结构化特征映射、决策树建模及桑基图可视化解析“特征阈值—组织类别”的判别逻辑。本研究在验证模型分类精度的同时,提高了模型决策过程的透明度与可信度,为金相组织智能分类提供了兼具性能与可解释性的技术方案。

       

      Abstract: The intelligent classification of metallographic microstructure images using machine learning methods represented by deep learning constitutes an important research area in materials informatics. In this study, a metallographic inspection image dataset for steels used in thermal power plants was established, and an improved ConvNeXt-based framework was developed for microstructure classification. The proposed model integrated pointwise convolution, Ghost modules and an attention mechanism, and it is further enhanced by data augmentation and transfer learning. Its performance was systematically evaluated through ablation and comparative experiments. The results show that the proposed framework achieves an mean average precision of 97%, a precision of 96%, and a recall of 96%, demonstrating its effectiveness for metallographic microstructure classification. To improve the model interpretability, a dual-level explanation framework was further constructed. At the local level, gradient-based saliency analysis was employed to quantify the contribution of feature channels and identify the regions most relevant to classification. At the global level, structured feature mapping, decision-tree modeling, and Sankey-diagram visualization was used to reveal the decision logic linking feature thresholds to microstructural categories. While verifying the classification accuracy of the model, this study improves the transparency and reliability of the model's decision-making process, and provides a technical solution with both favorable performance and interpretability for the intelligent classification of metallographic microstructures.

       

    /

    返回文章
    返回