Deep Learning for Refined Lithology Identification of Sandstone Microscopic Images

Refined lithology identification is an essential task, often constrained by the subjectivity and low efficiency of classical methods. Computer-aided automatic identification, while useful, has seldom been specifically geared toward refined lithology identification. In this study, we introduce Rock-ViT, an innovative machine learning approach. Its architecture, enhanced with supervised contrastive loss and rooted in visual Transformer principles, markedly improves accuracy in identifying complex lithological patterns. To this end, we have collected public datasets and implemented data augmentation, aiming to validate our method using sandstone as a focal point. The results demonstrate that Rock-ViT achieves superior accuracy and effectiveness in the refined lithology identification of sandstone. Rock-ViT presents a new perspective and a feasible approach for detailed lithological analysis, offering fresh insights and innovative solutions in geological analysis.