自适应少样本地震相位分割,支持不同数据集间相位数量变化。
AdaSemSeg: An Adaptive Few-shot Semantic Segmentation of Seismic Facies
- 提出可适应不同相位数量的少样本分割框架
- 在三个公开地震数据集上实现优于基线的泛化性能
- 适用于跨数据集迁移,适合地质解释场景
利用深度学习自动解释地震图像面临训练数据稀缺的挑战。少样本学习因其能在少量标注下适应新任务而成为合适范式。现有少样本语义分割方法固定目标类别数,无法在类别数不同的多个数据集上联合训练。在地震相位解释中,固定类别数会限制模型在不同相位数量数据集间的泛化能力。为此,本文提出自适应少样本语义分割方法 AdaSemSeg,可适应不同数据集中相位数量的变化。由于缺乏类似 ImageNet 的大规模标注地震数据集,采用自监督算法对地震数据进行骨干网络预训练。在三个具有不同相位数量的公开地震相位数据集上训练并评估 AdaSemSeg,其在未见数据集上的表现优于基于原型的少样本方法和各类基线。代码已开源:https://github.com/Surojit-Utah/AdaSemSeg。
原文摘要 · Abstract (English)
Automated interpretation of seismic images using deep learning methods is challenging because of the limited availability of training data. Few-shot learning is a suitable learning paradigm in such scenarios due to its ability to adapt to a new task with limited supervision (small training budget). Existing few-shot semantic segmentation (FSSS) methods fix the number of target classes. Therefore, they do not support joint training on multiple datasets varying in the number of classes. In the context of the interpretation of seismic facies, fixing the number of target classes inhibits the generalization capability of a model trained on one facies dataset to another, which is likely to have a different number of facies. To address this shortcoming, we propose a few-shot semantic segmentation method for interpreting seismic facies that can adapt to the varying number of facies across the dataset, dubbed the AdaSemSeg. In general, the backbone network of FSSS methods is initialized with the statistics learned from the ImageNet dataset for better performance. The lack of such a huge annotated dataset for seismic images motivates using a self-supervised algorithm on seismic datasets to initialize the backbone network. We have trained the AdaSemSeg on three public seismic facies datasets with different numbers of facies and evaluated the proposed method on multiple metrics. The performance of the AdaSemSeg on unseen datasets (not used in training) is better than the prototype-based few-shot method and baselines. Code: https://github.com/Surojit-Utah/AdaSemSeg
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。