In order to improve the prediction accuracy of lithology affected by imbalanced geological data, an ECA-MSCB ResNet model was proposed. The model integrates ECA (efficient channel attention) and MSCB (multi-scale convolutional block) into the traditional ResNet architecture to achieve efficient extraction and representation of lithological data features. For the issue of imbalanced lithology categories, prior probability-balanced logit bias was introduced during model training, and the focal loss function was modified to enhance the recognition of minority lithology classes. Experimental results show that the model based on ECA-MSCB ResNet performs well on the imbalanced geological lithology dataset, achieving an average prediction accuracy improvement of approximately 7.45% compared to the original ResNet model and 27.33% compared to the random forest method. Notably, the recognition of minority lithology classes improves by an average of 17.9%. Furthermore, the model demonstrates strong lithology classification ability on public datasets, achieving an F1-score of 75.77%. In addition, the recognition accuracy of the proposed model outperformed both traditional and mainstream methods. The ECA-MSCB ResNet method holds significant application value in the field of imbalanced geological lithology recognition.
| 科 Family | 属数 Number of genus | 种数 Number of species | 占总种数比例 Percentage of total species (%) | 属 Genus | 种数 Number of species | 占总种数比例 Percentage of total species (%) |
|---|---|---|---|---|---|---|
| 鹅膏菌科Amanitaceae | 2 | 11 | 5.26 | 鹅膏菌属 Amanita | 10 | 4.78 |
| 小菇科 Mycenaceae | 2 | 12 | 5.74 | 丝盖伞属 Inocybe | 5 | 2.39 |
| 多孔菌科 Polyporaceae | 8 | 14 | 6.70 | 蜡蘑属 Laccaria | 5 | 2.39 |
| 红菇科 Russulaceae | 3 | 23 | 11.00 | 小皮伞属 Marasmius | 6 | 2.87 |
| 小菇属 Mycena | 11 | 5.26 | ||||
| 光柄菇属 Pluteus | 5 | 2.39 | ||||
| 红菇属 Russula | 17 | 8.13 | ||||
| 栓菌属 Trametes | 5 | 2.39 |