IDEAS home Printed from https://ideas.repec.org/a/gam/jmathe/v10y2022i13p2317-d854380.html
   My bibliography  Save this article

Experimental Studies on Rock Thin-Section Image Classification by Deep Learning-Based Approaches

Author

Listed:
  • Diyuan Li

    (School of Resources and Safety Engineering, Central South University, Changsha 410083, China)

  • Junjie Zhao

    (School of Resources and Safety Engineering, Central South University, Changsha 410083, China)

  • Jinyin Ma

    (School of Resources and Safety Engineering, Central South University, Changsha 410083, China)

Abstract

Experimental studies were carried out to analyze the impact of optimizers and learning rate on the performance of deep learning-based algorithms for rock thin-section image classification. A total of 2634 rock thin-section images including three rock types—metamorphic, sedimentary, and volcanic rocks—were acquired from an online open-source science data bank. Four CNNs using three different optimizer algorithms (Adam, SGD, RMSprop) under two learning-rate decay schedules (lambda and cosine decay modes) were trained and validated. Then, a systematic comparison was conducted based on the performance of the trained model. Precision, f1-scores, and confusion matrix were adopted as the evaluation indicators. Trials revealed that deep learning-based approaches for rock thin-section image classification were highly effective and stable. Meanwhile, the experimental results showed that the cosine learning-rate decay mode was the better option for learning-rate adjustment during the training process. In addition, the performance of the four neural networks was confirmed and ranked as VGG16, GoogLeNet, MobileNetV2, and ShuffleNetV2. In the last step, the influence of optimization algorithms was evaluated based on VGG16 and GoogLeNet, and the results demonstrated that the capabilities of the model using Adam and RMSprop optimizers were more robust than that of SGD. The experimental study in this paper provides important practical value for training a high-precision rock thin-section image classification model, which can also be transferred to other similar image classification tasks.

Suggested Citation

  • Diyuan Li & Junjie Zhao & Jinyin Ma, 2022. "Experimental Studies on Rock Thin-Section Image Classification by Deep Learning-Based Approaches," Mathematics, MDPI, vol. 10(13), pages 1-28, July.
  • Handle: RePEc:gam:jmathe:v:10:y:2022:i:13:p:2317-:d:854380
    as

    Download full text from publisher

    File URL: https://www.mdpi.com/2227-7390/10/13/2317/pdf
    Download Restriction: no

    File URL: https://www.mdpi.com/2227-7390/10/13/2317/
    Download Restriction: no
    ---><---

    Citations

    Citations are extracted by the CitEc Project, subscribe to its RSS feed for this item.
    as


    Cited by:

    1. Zilong Zhou & Hang Yuan & Xin Cai, 2023. "Rock Thin Section Image Identification Based on Convolutional Neural Networks of Adaptive and Second-Order Pooling Methods," Mathematics, MDPI, vol. 11(5), pages 1-27, March.
    2. Dachang Zhu, 2023. "Underwater Image Enhancement Based on the Improved Algorithm of Dark Channel," Mathematics, MDPI, vol. 11(6), pages 1-11, March.

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:gam:jmathe:v:10:y:2022:i:13:p:2317-:d:854380. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    We have no bibliographic references for this item. You can help adding them by using this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: MDPI Indexing Manager (email available below). General contact details of provider: https://www.mdpi.com .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.