polimi-ispl / deep_prior_interpolation

Interpolation of seismic data through Deep Priors
http://ispl.deib.polimi.it/research/geophysics/
33 stars 13 forks source link
3d-seismic-data cnn convolutional-neural-networks interpolation seismic-data-interpolation

Unsupervised 3D Seismic Data Reconstruction Based On Deep Prior

This repository contains codes and examples of the data interpolation schemes that leverages the deep prior paradigm.

Authors

Fantong Kong1, Francesco Picetti2, Vincenzo Lipari2, Paolo Bestagini2, Xiaoming Tang1, and Stefano Tubaro2

1: School of Geosciences - China University of Petroleum (East), Qingdao, China
2: Dipartimento di Elettronica, Informazione e Bioingegneria - Politecnico di Milano, Italy

Abstract

Irregularity and coarse spatial sampling of seismic data strongly affect the performances of processing and imaging algorithms. Therefore, interpolation is a necessary pre-processing step in most of the processing workflows. In this work, we propose a seismic data interpolation method based on the deep prior paradigm: an ad-hoc Convolutional Neural Network is used as a prior to solve the interpolation inverse problem, avoiding any costly and prone-to-overfitting training stage. In particular, the proposed method leverages a multi resolution U-Net with 3D convolution kernels exploiting correlations in 3D seismic data, at different scales in all directions. Numerical examples on different corrupted synthetic and field datasets show the effectiveness and promising features of the proposed approach.

The inverse problem is defined starting from the sampling equation:

Sampling Equation

that is solved using the deep prior paradigm by

Deep Inverse Problem

The estimate of the true model is then obtained as the output of the optimized network:

Output

The architecture we propose is the MultiResolution UNet:

MultiRes UNet

Setup and Code Organization

The code mainly relies on top of pytorch. You can recreate our conda environment named dpi (acronym for "deep prior interpolation") through

conda create env -f environment.yml

Then, activate it with source activate dpi before running any example.
NOTE: if you have initialized conda through conda init, use conda activate dpi instead.

This python project is organized as follows:

Usage Examples

Here we report the example tests on the 3D hyperbolic data included in the paper.

# solve mask 1 saving the CNN weights
python main.py --imgdir ./datasets/hyperbolic3d --imgname original.npy --maskname random66_shot1.npy --datadim 3d --gain 40 --upsample linear --epochs 3000 --savemodel --outdir TL/shot1
# solve mask 2 from scratch
python main.py --imgdir ./datasets/hyperbolic3d --imgname original.npy --maskname random66_shot2.npy --datadim 3d --gain 40 --upsample linear --epochs 3000 --outdir TL/shot2_scratch
# solve mask 2 using as initial guess the CNN weights of mask 1
python main.py --imgdir ./datasets/hyperbolic3d --imgname original.npy --maskname random66_shot2.npy --datadim 3d --gain 40 --upsample linear --epochs 3000 --outdir TL/shot2_transfer --net load --netdir TL/shot1/model.pth

Data preparation

We are glad you want to try our method on your data! To minimize the effort, keep in mind that:

Related Publications

  1. F. Kong, V. Lipari, F. Picetti, P. Bestagini, and S. Tubaro. "A Deep Prior Convolutional Autoencoder for Seismic Data Interpolation", in European Association of Geophysicists and Engineers (EAGE) Annual Meeting, 2020. DOI
  2. F. Kong, F. Picetti, V. Lipari, P. Bestagini, and S. Tubaro. "Deep prior-based seismic data interpolation via multi-res U-net", in Society of Exploration Geophysicists (SEG) Annual Meeting, 2020. DOI
  3. F. Kong, F. Picetti, V. Lipari, P. Bestagini, X. Tang, and S. Tubaro. "Deep Prior Based Unsupervised Reconstruction of Irregularly Sampled Seismic Data", in IEEE Geoscience and Remote Sensing Letters (GRSL), 2020. DOI
  4. F. Picetti, V. Lipari, P. Bestagini, and S. Tubaro. "Anti-Aliasing Add-On For Deep Prior Seismic Data Interpolation", in IEEE International Conference on Image Processing (ICIP), 2021. DOI