Research on graphite ore grade classification based on the integration of multi-level features from ResNet and transformer

Wei Shu; Jionghui Wang; Xueyu Huang

doi:10.1117/12.3051710

18 November 2024 Research on graphite ore grade classification based on the integration of multi-level features from ResNet and transformer

Wei Shu, Jionghui Wang, Xueyu Huang

Author Affiliations +

Proceedings Volume 13403, International Conference on Algorithms, High Performance Computing, and Artificial Intelligence (AHPCAI 2024) ; 134031M (2024) https://doi.org/10.1117/12.3051710
Event: International Conference on Algorithms, High Performance Computing, and Artificial Intelligence, 2024, Zhengzhou, China

Abstract

Traditional graphite ore production typically relies on carbon-sulfur analyzers to determine the carbon grade of graphite ore. This method, however, is cumbersome and lacks timeliness. To address these issues, we propose TRA-FFNet, a graphite carbon grade image recognition and classification model that integrates multi-level features from ResNet and Transformer. Initially, ResNet-50 is employed as the backbone feature extraction network, with model parameters initialized through transfer learning to accelerate convergence. Subsequently, a Transformer module based on spatial feature compression fusion is designed to capture the weight relation-ships among different channels of the graphite ore feature map. Finally, a multi-level feature fusion module is incorporated at the network's terminal position to enhance the joint learning of both global and local features in graphite ore images. Experimental results show that the pro-posed model achieves an accuracy of 93.473% and an F1 score of 94.023% on our self-constructed dataset containing 19 classes of graphite ore, outperforming classic models such as MobileNetv2, EfficientNet-b3, ConvNext, RepVgg, and Swin Transformer. The proposed model achieves high-precision recognition of graphite ore grade in an end-to-end manner, offering a valuable detection solution for smart mining operations.

(2024) Published by SPIE. Downloading of the abstract is permitted for personal use only.

Citation Download Citation

Wei Shu, Jionghui Wang, and Xueyu Huang "Research on graphite ore grade classification based on the integration of multi-level features from ResNet and transformer", Proc. SPIE 13403, International Conference on Algorithms, High Performance Computing, and Artificial Intelligence (AHPCAI 2024) , 134031M (18 November 2024); https://doi.org/10.1117/12.3051710

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available

Members: $17.00

Non-members: $21.00 ADD TO CART

PROCEEDINGS
15 PAGES

DOWNLOAD PAPER SAVE TO MY LIBRARY

GET CITATION

RIGHTS & PERMISSIONS

Get copyright permission Get copyright permission on Copyright Marketplace

KEYWORDS

Graphite

Transformers

Image classification

Machine learning

Visual process modeling

Carbon

Feature extraction

Show All Keywords

Keywords/Phrases

Search In:

Publication Years