Saved in:
Bibliographic Details
Main Authors: Lee, An-Sheng, Pao, Yu-Wen, Lin, Hsuan-Tien, Liou, Sofia Ya Hsuan
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2410.12330
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911146258202624
author Lee, An-Sheng
Pao, Yu-Wen
Lin, Hsuan-Tien
Liou, Sofia Ya Hsuan
author_facet Lee, An-Sheng
Pao, Yu-Wen
Lin, Hsuan-Tien
Liou, Sofia Ya Hsuan
contents Pre-training foundation models has become the de-facto procedure for deep learning approaches, yet its application remains limited in the geological studies, where in needs of the model transferability to break the shackle of data scarcity. Here we target on the X-ray fluorescence (XRF) scanning data, a standard high-resolution measurement in extensive scientific drilling projects. We propose a scalable self-supervised learner, masked autoencoders on XRF spectra (MAX), to pre-train a foundation model covering geological records from multiple regions of the Pacific and Southern Ocean. In pre-training, we find that masking a high proportion of the input spectrum (50\%) yields a nontrivial and meaningful self-supervisory task. For downstream tasks, we select the quantification of XRF spectra into two costly geochemical measurements, CaCO$_3$ and total organic carbon, due to their importance in understanding the paleo-oceanic carbon system. Our results show that MAX, requiring only one-third of the data, outperforms models without pre-training in terms of quantification accuracy. Additionally, the model's generalizability improves by more than 60\% in zero-shot tests on new materials, with explainability further ensuring its robustness. Thus, our approach offers a promising pathway to overcome data scarcity in geological discovery by leveraging the self-supervised foundation model and fast-acquired XRF scanning data.
format Preprint
id arxiv_https___arxiv_org_abs_2410_12330
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle MAX: Masked Autoencoder for X-ray Fluorescence in Geological Investigation
Lee, An-Sheng
Pao, Yu-Wen
Lin, Hsuan-Tien
Liou, Sofia Ya Hsuan
Machine Learning
Pre-training foundation models has become the de-facto procedure for deep learning approaches, yet its application remains limited in the geological studies, where in needs of the model transferability to break the shackle of data scarcity. Here we target on the X-ray fluorescence (XRF) scanning data, a standard high-resolution measurement in extensive scientific drilling projects. We propose a scalable self-supervised learner, masked autoencoders on XRF spectra (MAX), to pre-train a foundation model covering geological records from multiple regions of the Pacific and Southern Ocean. In pre-training, we find that masking a high proportion of the input spectrum (50\%) yields a nontrivial and meaningful self-supervisory task. For downstream tasks, we select the quantification of XRF spectra into two costly geochemical measurements, CaCO$_3$ and total organic carbon, due to their importance in understanding the paleo-oceanic carbon system. Our results show that MAX, requiring only one-third of the data, outperforms models without pre-training in terms of quantification accuracy. Additionally, the model's generalizability improves by more than 60\% in zero-shot tests on new materials, with explainability further ensuring its robustness. Thus, our approach offers a promising pathway to overcome data scarcity in geological discovery by leveraging the self-supervised foundation model and fast-acquired XRF scanning data.
title MAX: Masked Autoencoder for X-ray Fluorescence in Geological Investigation
topic Machine Learning
url https://arxiv.org/abs/2410.12330