Impact of dataset size and convolutional neural network architecture on transfer learning for carbonate rock classification
File(s)1-s2.0-S0098300422002333-main.pdf (6.54 MB)
Published version
Author(s)
Dawson, Harriet L
Dubrule, Olivier
John, Cedric M
Type
Journal Article
Abstract
Modern geological practices, in both industry and academia, rely largely on a legacy of observational data at a range of scales. However, widespread ambiguities in the petrographic description of rock facies reduce the reliability of descriptive data. Previous studies have demonstrated a great potential for the use of convolutional neural networks (CNNs) in the classification of facies from digital images; however, it remains to be determined which of the available CNN architectures performs best for a geological classification task. We evaluate the ability of top-performing CNNs to classify carbonate core images using transfer learning, systematically developing a performance comparison between these architectures on a complex geological dataset. Three datasets with orders of magnitude difference in data quantity (7000–104,000 samples) were created that contain images across seven classes from the modified Dunham Classification for carbonate rocks. Following training of nine different CNNs of four architectures on these datasets, we find the Inception-v3 architecture to be most suited to this classification task, achieving 92% accuracy when trained on the larger dataset. Furthermore, we show that even when using transfer learning the size of the dataset plays a key role in the performance of the models, with those trained on the smaller datasets showing a strong tendency to overfit. This has direct implications for the application of deep learning in geosciences as many papers currently published use very small datasets of less than 5000 samples. Application of the framework developed in this research could aid the future of deep learning based carbonate classification, with further potential to be easily modified to suit the classification of cores originating from different formations and lithologies.
Date Issued
2023-02-01
Date Acceptance
2022-12-07
Citation
Computers and Geosciences, 2023, 171, pp.1-11
ISSN
0098-3004
Publisher
Elsevier
Start Page
1
End Page
11
Journal / Book Title
Computers and Geosciences
Volume
171
Copyright Statement
© 2022 The Authors. Published by Elsevier Ltd. This is an open access article under the CC BY license (http://creativecommons.org/licenses/by/4.0/)
License URL
Identifier
https://www.webofscience.com/api/gateway?GWVersion=2&SrcApp=PARTNER_APP&SrcAuth=LinksAMR&KeyUT=WOS:000906651400001&DestLinkType=FullRecord&DestApp=ALL_WOS&UsrCustomerID=a2bf6146997ec60c407a63945d4e92bb
Subjects
Science & Technology
Technology
Physical Sciences
Computer Science, Interdisciplinary Applications
Geosciences, Multidisciplinary
Computer Science
Geology
Deep learning
Machine learning
Dunham classification
Geological images
SEISMIC FACIES ANALYSIS
CITIZEN SCIENCE
VALIDATION
SYSTEM
Publication Status
Published
Article Number
ARTN 105284
Date Publish Online
2022-12-08