Abstract:Lithology classification is important in mineral resource exploration, engineering geological exploration, and disaster monitoring. Traditional laboratory methods for the qualitative analysis of rocks are limited by sampling conditions and analytical techniques, resulting in high costs, low efficiency, and the inability to quickly obtain large-scale geological information. Hyperspectral remote sensing technology can classify and identify lithology using the spectral characteristics of rock, and is characterized by fast detection, large coverage area, and environmental friendliness, which provide the application potential for lithological mapping at a large regional scale. In this study, ZY1-02D hyperspectral images were used as data sources to construct a new two-layer extreme gradient boosting (XGBoost) lithology classification model based on the XGBoost decision tree and an improved greedy search algorithm. A total of 153 spectral bands of the preprocessed hyperspectral images were input into the first layer of the XGBoost model. Based on the tree traversal structural characteristics of the leaf nodes in the XGBoost model, three built-in XGBoost importance indexes were split and combined. The improved greedy search algorithm was used to extract the spectral band variables, which were imported into the second layer of the XGBoost model, and the bat algorithm was used to optimize the modeling parameters of XGBoost. The extraction model of rock classification information was constructed, and the classification map of regional surface rock types was drawn. Field verification was performed for the two-layer XGBoost rock classification model, and its accuracy and reliability were evaluated based on four indexes, namely, accuracy, precision, recall, and F1 score. The results showed that the two-layer XGBoost model had a good lithological classification effect, robustness, and adaptability to small sample datasets. Compared with the traditional machine learning model, the two-layer XGBoost model shows superior performance. The accuracy, precision, recall, and F1 score of the verification set were 0.8343, 0.8406, 0.8350, and 0.8157, respectively. The variable extraction ability of the constructed two-layer XGBoost model was significantly improved. Compared with traditional feature selection methods, the GREED-GFC method, when applied to the two-layer XGBoost model, contributes to more stable rock classification performance and higher lithology prediction accuracy, and the smallest number of extracted features. The lithological distribution information identified by the model was in good agreement with the lithology information verified in the field.

Lithologic classification using multilevel spectral characteristics

Rapid lithological mapping using multi-source remote sensing data fusion and automatic sample generation strategy

Geological mapping of basalt using stream sediment geochemical data: Case study of covered areas in Jining, Inner Mongolia, China

Towards lithology mapping in semi-arid areas using time-series Landsat-8 data

Unsupervised Machine Learning for Lithological Mapping Using Geochemical Data in Covered Areas of Jining, China

Application Research of Hyperspectral Remote Sensing in the Identification of Stratigraphic Boundaries Alongside Growing Tunnels

Influence of GLCM texture parameters on lithological mapping using Sentinel-1 imagery

Lithological Classification by Hyperspectral Images Based on a Two-Layer XGBoost Model, Combined with a Greedy Algorithm

Lithology classification in semi-arid area combining multi-source remote sensing images using support vector machine optimized by improved particle swarm algorithm

Classification of geological materials on geochemical lithogenes: Illustration on a case study in Gejiu area of Yunnan Province, China

Detailed Land Use Classification in a Rare Earth Mining Area Using Hyperspectral Remote Sensing Data for Sustainable Agricultural Development

A multi-scale classification method for rocky desertification mapping in the red-bed area of northwestern, Jiangxi, China

Hyperspectral Band Selection for Lithologic Discrimination and Geological Mapping

Lithological Mapping from Hyperspectral Data by Improved Use of Spectral Angle Mapper

Lithological Mapping In The Eastern Section Of Gangdise, Tibet Using Aster And Field Spectroscopy Data

Towards automatic lithological classification from remote sensing data using support vector machines

Hyperspectral remote sensing in lithological mapping, mineral exploration, and environmental geology: an updated review

The use of multiscale texture in image classification for lithologic mapping

A semi-automatic approach for generating geological profiles by integrating multi-source data

Lithologic Mapping Using ASTER Imagery and Multivariate Texture

Application of Lithological Mapping Based on GF-5 AHSI Imagery