Joint PGI of Gravity + Magnetic on an Octree mesh without petrophysical information

This tutorial shows through a joint inversion of Gravity and Magnetic data on an Octree mesh how to use the PGI framework introduced in Astic & Oldenburg (2019) and Astic et al. (2021) to make geologic assumptions and learn a suitable petrophysical distribution when no quantitative petrophysical information is available.

Thibaut Astic, Douglas W. Oldenburg, A framework for petrophysically and geologically guided geophysical inversion using a dynamic Gaussian mixture model prior, Geophysical Journal International, Volume 219, Issue 3, December 2019, Pages 1989–2012, DOI: 10.1093/gji/ggz389.

Thibaut Astic, Lindsey J. Heagy, Douglas W Oldenburg, Petrophysically and geologically guided multi-physics inversion using a dynamic Gaussian mixture model, Geophysical Journal International, Volume 224, Issue 1, Januaray 2021, Pages 40-68, DOI: 10.1093/gji/ggaa378.

Import modules

import discretize as ds
import SimPEG.potential_fields as pf
from SimPEG import (
    maps,
    utils,
    simulation,
    inverse_problem,
    inversion,
    optimization,
    regularization,
    data_misfit,
    directives,
)
from SimPEG.utils import io_utils
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.axes_grid1.inset_locator import inset_axes

# Reproducible science
np.random.seed(518936)

Setup

# Load Mesh
mesh_file = io_utils.download(
    "https://storage.googleapis.com/simpeg/pgi_tutorial_assets/mesh_tutorial.ubc"
)
mesh = ds.TreeMesh.read_UBC(mesh_file)

# Load True geological model for comparison with inversion result
true_geology_file = io_utils.download(
    "https://storage.googleapis.com/simpeg/pgi_tutorial_assets/geology_true.mod"
)
true_geology = mesh.read_model_UBC(true_geology_file)

# Plot true geology model
fig, ax = plt.subplots(1, 4, figsize=(20, 4))
ticksize, labelsize = 14, 16
for _, axx in enumerate(ax):
    axx.set_aspect(1)
    axx.tick_params(labelsize=ticksize)
mesh.plotSlice(
    true_geology,
    normal="X",
    ax=ax[0],
    ind=-17,
    clim=[0, 2],
    pcolorOpts={"cmap": "inferno_r"},
    grid=True,
)
mesh.plotSlice(
    true_geology,
    normal="Y",
    ax=ax[1],
    clim=[0, 2],
    pcolorOpts={"cmap": "inferno_r"},
    grid=True,
)
geoplot = mesh.plotSlice(
    true_geology,
    normal="Z",
    ax=ax[2],
    clim=[0, 2],
    ind=-10,
    pcolorOpts={"cmap": "inferno_r"},
    grid=True,
)
geocb = plt.colorbar(geoplot[0], cax=ax[3], ticks=[0, 1, 2])
geocb.set_label(
    "True geology model\n(classification/density/mag. susc.)", fontsize=labelsize
)
geocb.set_ticklabels(
    ["BCKGRD (0 g/cc; 0 SI)", "PK (-0.8 g/cc; 5e-3 SI)", "VK (-0.2 g/cc; 2e-2 SI)"]
)
geocb.ax.tick_params(labelsize=ticksize)
ax[3].set_aspect(10)
plt.show()

# Load geophysical data
data_grav_file = io_utils.download(
    "https://storage.googleapis.com/simpeg/pgi_tutorial_assets/gravity_data.obs"
)
data_grav = io_utils.read_grav3d_ubc(data_grav_file)
data_mag_file = io_utils.download(
    "https://storage.googleapis.com/simpeg/pgi_tutorial_assets/magnetic_data.obs"
)
data_mag = io_utils.read_mag3d_ubc(data_mag_file)

# plot data and mesh
fig, ax = plt.subplots(2, 2, figsize=(15, 10))
ax = ax.reshape(-1)
plt.gca().set_aspect("equal")
plt.gca().set_xlim(
    [
        data_mag.survey.receiver_locations[:, 0].min(),
        data_mag.survey.receiver_locations[:, 0].max(),
    ],
)
plt.gca().set_ylim(
    [
        data_mag.survey.receiver_locations[:, 1].min(),
        data_mag.survey.receiver_locations[:, 1].max(),
    ]
)
mesh.plotSlice(
    np.ones(mesh.nC),
    normal="Z",
    ind=int(-10),
    grid=True,
    pcolorOpts={"cmap": "Greys"},
    ax=ax[0],
)
mm = utils.plot2Ddata(
    data_grav.survey.receiver_locations,
    -data_grav.dobs,
    ax=ax[0],
    level=True,
    nx=20,
    ny=20,
    dataloc=True,
    ncontour=12,
    shade=True,
    contourOpts={"cmap": "Blues_r", "alpha": 0.8},
    levelOpts={"colors": "k", "linewidths": 0.5, "linestyles": "dashed"},
)
ax[0].set_aspect(1)
ax[0].set_title(
    "Gravity data values and locations,\nwith mesh and geology overlays", fontsize=16
)
plt.colorbar(mm[0], cax=ax[2], orientation="horizontal")
ax[2].set_aspect(0.05)
ax[2].set_title("mGal", fontsize=16)
mesh.plotSlice(
    np.ones(mesh.nC),
    normal="Z",
    ind=int(-10),
    grid=True,
    pcolorOpts={"cmap": "Greys"},
    ax=ax[1],
)
mm = utils.plot2Ddata(
    data_mag.survey.receiver_locations,
    data_mag.dobs,
    ax=ax[1],
    level=True,
    nx=20,
    ny=20,
    dataloc=True,
    ncontour=11,
    shade=True,
    contourOpts={"cmap": "Reds", "alpha": 0.8},
    levelOpts={"colors": "k", "linewidths": 0.5, "linestyles": "dashed"},
)
ax[1].set_aspect(1)
ax[1].set_title(
    "Magnetic data values and locations,\nwith mesh and geology overlays", fontsize=16
)
plt.colorbar(mm[0], cax=ax[3], orientation="horizontal")
ax[3].set_aspect(0.05)
ax[3].set_title("nT", fontsize=16)
# overlay true geology model for comparison
indz = -9
indslicezplot = mesh.gridCC[:, 2] == mesh.vectorCCz[indz]
for i in range(2):
    utils.plot2Ddata(
        mesh.gridCC[indslicezplot][:, [0, 1]],
        true_geology[indslicezplot],
        nx=200,
        ny=200,
        contourOpts={"alpha": 0},
        clim=[0, 2],
        ax=ax[i],
        level=True,
        ncontour=2,
        levelOpts={"colors": "k", "linewidths": 2, "linestyles": "--"},
        method="nearest",
    )
plt.subplots_adjust(hspace=-0.25, wspace=0.1)
plt.show()

# Load Topo
topo_file = io_utils.download(
    "https://storage.googleapis.com/simpeg/pgi_tutorial_assets/CDED_Lake_warp.xyz"
)
topo = np.genfromtxt(topo_file, skip_header=1)
# find the active cells
actv = utils.surface2ind_topo(mesh, topo, gridLoc="CC")
# Create active map to go from reduce set to full
ndv = np.nan
actvMap = maps.InjectActiveCells(mesh, actv, ndv)
nactv = int(actv.sum())

# Create simulations and data misfits
# Wires mapping
wires = maps.Wires(("den", actvMap.nP), ("sus", actvMap.nP))
gravmap = actvMap * wires.den
magmap = actvMap * wires.sus
idenMap = maps.IdentityMap(nP=nactv)
# Grav problem
simulation_grav = pf.gravity.simulation.Simulation3DIntegral(
    survey=data_grav.survey, mesh=mesh, rhoMap=wires.den, actInd=actv,
)
dmis_grav = data_misfit.L2DataMisfit(data=data_grav, simulation=simulation_grav)
# Mag problem
simulation_mag = pf.magnetics.simulation.Simulation3DIntegral(
    survey=data_mag.survey, mesh=mesh, chiMap=wires.sus, actInd=actv,
)
dmis_mag = data_misfit.L2DataMisfit(data=data_mag, simulation=simulation_mag)
  • Slice -17, X = 557287.50, Slice 16, Y = 7133612.50, Slice -10, Z = 255.00
  • Gravity data values and locations, with mesh and geology overlays, Magnetic data values and locations, with mesh and geology overlays, mGal, nT

Out:

file already exists, new file is called /home/vsts/work/1/s/tutorials/13-pgi/mesh_tutorial.ubc
Downloading https://storage.googleapis.com/simpeg/pgi_tutorial_assets/mesh_tutorial.ubc
   saved to: /home/vsts/work/1/s/tutorials/13-pgi/mesh_tutorial.ubc
Download completed!
file already exists, new file is called /home/vsts/work/1/s/tutorials/13-pgi/geology_true.mod
Downloading https://storage.googleapis.com/simpeg/pgi_tutorial_assets/geology_true.mod
   saved to: /home/vsts/work/1/s/tutorials/13-pgi/geology_true.mod
Download completed!
file already exists, new file is called /home/vsts/work/1/s/tutorials/13-pgi/gravity_data.obs
Downloading https://storage.googleapis.com/simpeg/pgi_tutorial_assets/gravity_data.obs
   saved to: /home/vsts/work/1/s/tutorials/13-pgi/gravity_data.obs
Download completed!
file already exists, new file is called /home/vsts/work/1/s/tutorials/13-pgi/magnetic_data.obs
Downloading https://storage.googleapis.com/simpeg/pgi_tutorial_assets/magnetic_data.obs
   saved to: /home/vsts/work/1/s/tutorials/13-pgi/magnetic_data.obs
Download completed!
file already exists, new file is called /home/vsts/work/1/s/tutorials/13-pgi/CDED_Lake_warp.xyz
Downloading https://storage.googleapis.com/simpeg/pgi_tutorial_assets/CDED_Lake_warp.xyz
   saved to: /home/vsts/work/1/s/tutorials/13-pgi/CDED_Lake_warp.xyz
Download completed!

Create a joint Data Misfit

# Joint data misfit
dmis = 0.5 * dmis_grav + 0.5 * dmis_mag

# initial model
m0 = np.r_[-1e-4 * np.ones(actvMap.nP), 1e-4 * np.ones(actvMap.nP)]

Inversion with no petrophysical information about the means

In this scenario, we do not know the true petrophysical signature of each rock unit. We thus make geologic assumptions to design a coupling term and perform a multi-physics inversion. in addition to a neutral background, we assume that one rock unit is only less dense, and the third one is only magnetic. As we do not know their mean petrophysical values. We start with an initial guess (-1 g/cc) for the updatable mean density-contrast value of the less dense unit (with a fixed susceptibility of 0 SI). The magnetic-contrasting unit’s updatable susceptibility is initialized at a value of 0.1 SI (with a fixed 0 g/cc density contrast). We then let the algorithm learn a suitable set of means under the set constrained (fixed or updatable value), through the kappa argument, denoting our confidences in each initial mean value (high confidence: fixed value; low confidence: updatable value).

Create a petrophysical GMM initial guess

The GMM is our representation of the petrophysical and geological information. Here, we focus on the petrophysical aspect, with the means and covariances of the physical properties of each rock unit. To generate the data above, the PK unit was populated with a density contrast of -0.8 g/cc and a magnetic susceptibility of 0.005 SI. The properties of the HK unit were set at -0.2 g/cc and 0.02 SI. But here, we assume we do not have this information. Thus, we start with initial guess for the means and confidences kappa such that one unit is only less dense and one unit is only magnetic, both embedded in a neutral background. The covariances matrices are set so that we assume petrophysical noise levels of around 0.05 g/cc and 0.001 SI for both unit. The background unit is set at a fixed null contrasts (0 g/cc 0 SI) with a petrophysical noise level of half of the above.

gmmref = utils.WeightedGaussianMixture(
    n_components=3,  # number of rock units: bckgrd, PK, HK
    mesh=mesh,  # inversion mesh
    actv=actv,  # actv cells
    covariance_type="diag",  # diagonal covariances
)
# required: initialization with fit
# fake random samples, size of the mesh
# number of physical properties: 2 (density and mag.susc)
gmmref.fit(np.random.randn(nactv, 2))
# set parameters manually
# set phys. prop means for each unit
gmmref.means_ = np.c_[
    [0.0, 0.0],  # BCKGRD density contrast and mag. susc
    [-1, 0.0],  # PK
    [0, 0.1],  # HK
].T
# set phys. prop covariances for each unit
gmmref.covariances_ = np.array(
    [[6e-04, 3.175e-07], [2.4e-03, 1.5e-06], [2.4e-03, 1.5e-06]]
)
# important after setting cov. manually: compute precision matrices and cholesky
gmmref.compute_clusters_precisions()
# set global proportions; low-impact as long as not 0 or 1 (total=1)
gmmref.weights_ = np.r_[0.9, 0.075, 0.025]

# Plot the 2D GMM
ticksize, labelsize = 10, 12
fig = plt.figure(figsize=(10, 10))
ax1 = plt.subplot2grid((4, 4), (0, 1), colspan=3, rowspan=3)
ax2 = plt.subplot2grid((4, 4), (3, 1), colspan=3)
ax3 = plt.subplot2grid((4, 4), (0, 0), rowspan=3)

ax1.set_xlim(-1.2, 0.1)
ax1.set_ylim(-0.0025, 0.03)
ax2.set_xlim(-1.2, 0.1)
ax3.set_ylim(-0.0025, 0.03)

ax1.set_xticks([-1.2, -1.0, -0.8, -0.6, -0.4, -0.2, 0.0])
ax2.set_xticks([-1.2, -1.0, -0.8, -0.6, -0.4, -0.2, 0.0])

ax1.set_yticks([0, 0.02, 0.04, 0.06, 0.08, 0.1, 0.12])
ax3.set_yticks([0, 0.02, 0.04, 0.06, 0.08, 0.1, 0.12])

x, y = np.mgrid[-1.21:0.1:0.01, -0.1:0.12:0.01]
pos = np.empty(x.shape + (2,))
pos[:, :, 0] = x
pos[:, :, 1] = y
rvm = gmmref.predict(pos.reshape(-1, 2))
rvsmooth = gmmref.score_samples(pos.reshape(-1, 2))
surf = ax1.contourf(x, y, (rvsmooth).reshape(x.shape), 25, cmap="viridis")
ax1.contour(
    x,
    y,
    rvsmooth.reshape(x.shape),
    25,
    colors="k",
    linewidths=1.0,
    linestyles="dashdot",
)
ax1.scatter(
    gmmref.means_[:, 0],
    gmmref.means_[:, 1],
    label="Initial guess\nfor the petrophysical means",
    cmap="inferno_r",
    c=[0, 1, 2],
    marker="o",
    edgecolors="k",
    s=200,
)

ax1.scatter(
    [0, -0.8, -0.02],
    [0, 0.005, 0.02],
    label="True petrophysical means",
    cmap="inferno_r",
    c=[0, 1, 2],
    marker="v",
    edgecolors="k",
    s=200,
)

axbar = inset_axes(ax1, width="40%", height="3%", loc="upper right", borderpad=1,)
cbpetro = fig.colorbar(surf, cax=axbar, orientation="horizontal")
cbpetro.set_ticks([rvsmooth.min(), rvsmooth.max()])
cbpetro.set_ticklabels(["Low", "High"])
cbpetro.set_label("2D Probability Density Distribution", fontsize=labelsize)
cbpetro.ax.tick_params(labelsize=ticksize)
cbpetro.outline.set_edgecolor("k")

# create the 1D GMM profile for density
from sklearn.mixture import GaussianMixture

means_init_grav = gmmref.means_[:, 0].reshape(3, 1)
cov_init_grav = np.array([gmmref.covariances_[:, 0]]).reshape((3, 1, 1))
clfgrav = utils.pgi_utils.GaussianMixture(
    n_components=3,
    means_init=means_init_grav,
    precisions_init=cov_init_grav,
    n_init=1,
    max_iter=2,
    tol=np.inf,
)
# random fit, we set values after.
clfgrav.fit(np.random.randn(10, 1))
clfgrav.means_ = means_init_grav
clfgrav.covariances_ = cov_init_grav
from sklearn.mixture._gaussian_mixture import _compute_precision_cholesky

clfgrav.precisions_cholesky_ = _compute_precision_cholesky(
    clfgrav.covariances_, clfgrav.covariance_type
)
clfgrav.weights_ = gmmref.weights_
testXplot_grav = np.linspace(-1.2, 0.1, 1000)[:, np.newaxis]
score_grav = clfgrav.score_samples(testXplot_grav)
ax2.plot(
    testXplot_grav,
    np.exp(score_grav),
    linewidth=3.0,
    label="1D Probability Density Distribution",
    c="k",
)
ax2.set_ylim([0.0, 2])
ax2.legend(fontsize=ticksize)

# create the 1D GMM profile for mag. susc.
means_init_mag = gmmref.means_[:, 1].reshape(3, 1)
cov_init_mag = np.array([gmmref.covariances_[:, 1]]).reshape((3, 1, 1))
clfmag = GaussianMixture(
    n_components=3,
    means_init=means_init_mag,
    precisions_init=cov_init_mag,
    n_init=1,
    max_iter=2,
    tol=np.inf,
)
# random fit, we set values after.
clfmag.fit(np.random.randn(10, 1))
clfmag.means_ = means_init_mag
clfmag.covariances_ = cov_init_mag
clfmag.precisions_cholesky_ = _compute_precision_cholesky(
    clfmag.covariances_, clfmag.covariance_type
)
clfmag.weights_ = gmmref.weights_
testXplot_mag = np.linspace(-0.01, 1.2, 1000)[:, np.newaxis]
score_mag = clfmag.score_samples(testXplot_mag)
ax3.plot(np.exp(score_mag), testXplot_mag, linewidth=3.0, c="k")

ax3.set_xlim([0.0, 50])
ax3.set_xlabel(
    "1D Probability Density values", fontsize=labelsize, rotation=-45, labelpad=0, x=0.5
)
ax2.set_xlabel("Density (g/cc)", fontsize=labelsize)
ax3.set_ylabel("Magnetic Susceptibility (SI)", fontsize=labelsize)
ax2.tick_params(labelsize=ticksize)
ax3.tick_params(labelsize=ticksize)
ax1.legend(fontsize=labelsize, loc=2)
ax1.tick_params(labelleft=False)
ax1.tick_params(labelbottom=False)
ax1.set_ylabel("")
ax1.set_xlabel("")
ax2.tick_params(axis="both", which="both", labelsize=ticksize)
ax3.tick_params(axis="both", which="both", labelsize=ticksize)
plt.show()
plot inv 2 joint pf pgi no info tutorial

Inverse problem with no mean information

# Create PGI regularization
# Sensitivity weighting
wr_grav = np.sum(simulation_grav.G ** 2.0, axis=0) ** 0.5
wr_grav = wr_grav / np.max(wr_grav)

wr_mag = np.sum(simulation_mag.G ** 2.0, axis=0) ** 0.5
wr_mag = wr_mag / np.max(wr_mag)

# create joint PGI regularization with smoothness
reg = utils.make_SimplePGI_regularization(
    gmmref=gmmref,
    mesh=mesh,
    wiresmap=wires,
    maplist=[idenMap, idenMap],
    mref=m0,
    indActive=actv,
    alpha_s=1.0,
    alpha_x=1.0,
    alpha_y=1.0,
    alpha_z=1.0,
    alpha_xx=0.0,
    alpha_yy=0.0,
    alpha_zz=0.0,
    cell_weights_list=[wr_grav, wr_mag],  # weights each phys. prop. by each sensW
)

# Directives
# Add directives to the inversion
# ratio to use for each phys prop. smoothness in each direction:
# roughly the ratio of range of each phys. prop.
alpha0_ratio = np.r_[
    np.zeros(len(reg.objfcts[0].objfcts)),
    1e-2 * np.ones(len(reg.objfcts[1].objfcts)),
    1e-2 * 100.0 * np.ones(len(reg.objfcts[2].objfcts)),
]
Alphas = directives.AlphasSmoothEstimate_ByEig(alpha0_ratio=alpha0_ratio, verbose=True)
# initialize beta and beta/alpha_s schedule
beta = directives.BetaEstimate_ByEig(beta0_ratio=1e-4)
betaIt = directives.PGI_BetaAlphaSchedule(
    verbose=True, coolingFactor=2.0, tolerance=0.2, progress=0.2,
)
# geophy. and petro. target misfits
targets = directives.MultiTargetMisfits(
    verbose=True, chiSmall=0.5,  # ask for twice as much clustering (target value is /2)
)
# add learned mref in smooth once stable
MrefInSmooth = directives.PGI_AddMrefInSmooth(wait_till_stable=True, verbose=True,)
# update the parameters in smallness (L2-approx of PGI)
update_smallness = directives.PGI_UpdateParameters(
    update_gmm=True,  # update the GMM each iteration
    kappa=np.c_[  # confidences in each mean phys. prop. of each cluster
        1e10
        * np.ones(
            2
        ),  # fixed background at 0 density, 0 mag. susc. (high confidences of 1e10)
        [
            0,
            1e10,
        ],  # density-contrasting cluster: updatable density mean, fixed mag. susc.
        [
            1e10,
            0,
        ],  # magnetic-contrasting cluster: fixed density mean, updatable mag. susc.
    ].T,
)
# pre-conditioner
update_Jacobi = directives.UpdatePreconditioner()
# iteratively balance the scaling of the data misfits
scaling_init = directives.ScalingMultipleDataMisfits_ByEig(chi0_ratio=[1.0, 100.0])
scale_schedule = directives.JointScalingSchedule(verbose=True)

# Create inverse problem
# Optimization
# set lower and upper bounds
lowerbound = np.r_[-2.0 * np.ones(actvMap.nP), 0.0 * np.ones(actvMap.nP)]
upperbound = np.r_[0.0 * np.ones(actvMap.nP), 1e-1 * np.ones(actvMap.nP)]
opt = optimization.ProjectedGNCG(
    maxIter=30,
    lower=lowerbound,
    upper=upperbound,
    maxIterLS=20,
    maxIterCG=100,
    tolCG=1e-4,
)
# create inverse problem
invProb = inverse_problem.BaseInvProblem(dmis, reg, opt)
inv = inversion.BaseInversion(
    invProb,
    # directives: evaluate alphas (and data misfits scales) before beta
    directiveList=[
        Alphas,
        scaling_init,
        beta,
        update_smallness,
        targets,
        scale_schedule,
        betaIt,
        MrefInSmooth,
        update_Jacobi,
    ],
)
# Invert
pgi_model_no_info = inv.run(m0)


# Plot the result with full petrophysical information
density_model_no_info = gravmap * pgi_model_no_info
magsus_model_no_info = magmap * pgi_model_no_info
learned_gmm = reg.objfcts[0].gmm
quasi_geology_model_no_info = actvMap * reg.objfcts[0].membership(reg.objfcts[0].mref)

fig, ax = plt.subplots(3, 4, figsize=(15, 10))
for _, axx in enumerate(ax):
    for _, axxx in enumerate(axx):
        axxx.set_aspect(1)
        axxx.tick_params(labelsize=ticksize)

indx = 15
indy = 17
indz = -9
# geology model
mesh.plotSlice(
    quasi_geology_model_no_info,
    normal="X",
    ax=ax[0, 0],
    clim=[0, 2],
    ind=indx,
    pcolorOpts={"cmap": "inferno_r"},
)
mesh.plotSlice(
    quasi_geology_model_no_info,
    normal="Y",
    ax=ax[0, 1],
    clim=[0, 2],
    ind=indy,
    pcolorOpts={"cmap": "inferno_r"},
)
geoplot = mesh.plotSlice(
    quasi_geology_model_no_info,
    normal="Z",
    ax=ax[0, 2],
    clim=[0, 2],
    ind=indz,
    pcolorOpts={"cmap": "inferno_r"},
)
geocb = plt.colorbar(geoplot[0], cax=ax[0, 3], ticks=[0, 1, 2])
geocb.set_ticklabels(["BCK", "PK", "VK"])
geocb.set_label("Quasi-Geology model\n(Rock units classification)", fontsize=16)
ax[0, 3].set_aspect(10)

# gravity model
mesh.plotSlice(
    density_model_no_info,
    normal="X",
    ax=ax[1, 0],
    clim=[-1, 0],
    ind=indx,
    pcolorOpts={"cmap": "Blues_r"},
)
mesh.plotSlice(
    density_model_no_info,
    normal="Y",
    ax=ax[1, 1],
    clim=[-1, 0],
    ind=indy,
    pcolorOpts={"cmap": "Blues_r"},
)
denplot = mesh.plotSlice(
    density_model_no_info,
    normal="Z",
    ax=ax[1, 2],
    clim=[-1, 0],
    ind=indz,
    pcolorOpts={"cmap": "Blues_r"},
)
dencb = plt.colorbar(denplot[0], cax=ax[1, 3])
dencb.set_label("Density contrast\nmodel (g/cc)", fontsize=16)
ax[1, 3].set_aspect(10)

# magnetic model
mesh.plotSlice(
    magsus_model_no_info,
    normal="X",
    ax=ax[2, 0],
    clim=[0, 0.025],
    ind=indx,
    pcolorOpts={"cmap": "Reds"},
)
mesh.plotSlice(
    magsus_model_no_info,
    normal="Y",
    ax=ax[2, 1],
    clim=[0, 0.025],
    ind=indy,
    pcolorOpts={"cmap": "Reds"},
)
susplot = mesh.plotSlice(
    magsus_model_no_info,
    normal="Z",
    ax=ax[2, 2],
    clim=[0, 0.025],
    ind=indz,
    pcolorOpts={"cmap": "Reds"},
)
suscb = plt.colorbar(susplot[0], cax=ax[2, 3])
suscb.set_label("Magnetic susceptibility\nmodel (SI)", fontsize=16)
ax[2, 3].set_aspect(10)

# overlay true geology model for comparison
indslicexplot = mesh.gridCC[:, 0] == mesh.vectorCCx[indx]
indsliceyplot = mesh.gridCC[:, 1] == mesh.vectorCCy[indy]
indslicezplot = mesh.gridCC[:, 2] == mesh.vectorCCz[indz]
for i in range(3):
    for j, (plane, indd) in enumerate(
        zip([[1, 2], [0, 2], [0, 1]], [indslicexplot, indsliceyplot, indslicezplot])
    ):
        utils.plot2Ddata(
            mesh.gridCC[indd][:, plane],
            true_geology[indd],
            nx=100,
            ny=100,
            contourOpts={"alpha": 0},
            clim=[0, 2],
            ax=ax[i, j],
            level=True,
            ncontour=2,
            levelOpts={"colors": "grey", "linewidths": 2, "linestyles": "--"},
            method="nearest",
        )

# plot the locations of the cross-sections
for i in range(3):
    ax[i, 0].plot(
        mesh.vectorCCy[indy] * np.ones(2), [-300, 500], c="k", linestyle="dotted"
    )
    ax[i, 0].plot(
        [
            data_mag.survey.receiver_locations[:, 1].min(),
            data_mag.survey.receiver_locations[:, 1].max(),
        ],
        mesh.vectorCCz[indz] * np.ones(2),
        c="k",
        linestyle="dotted",
    )
    ax[i, 0].set_xlim(
        [
            data_mag.survey.receiver_locations[:, 1].min(),
            data_mag.survey.receiver_locations[:, 1].max(),
        ],
    )

    ax[i, 1].plot(
        mesh.vectorCCx[indx] * np.ones(2), [-300, 500], c="k", linestyle="dotted"
    )
    ax[i, 1].plot(
        [
            data_mag.survey.receiver_locations[:, 0].min(),
            data_mag.survey.receiver_locations[:, 0].max(),
        ],
        mesh.vectorCCz[indz] * np.ones(2),
        c="k",
        linestyle="dotted",
    )
    ax[i, 1].set_xlim(
        [
            data_mag.survey.receiver_locations[:, 0].min(),
            data_mag.survey.receiver_locations[:, 0].max(),
        ],
    )

    ax[i, 2].plot(
        mesh.vectorCCx[indx] * np.ones(2),
        [
            data_mag.survey.receiver_locations[:, 1].min(),
            data_mag.survey.receiver_locations[:, 1].max(),
        ],
        c="k",
        linestyle="dotted",
    )
    ax[i, 2].plot(
        [
            data_mag.survey.receiver_locations[:, 0].min(),
            data_mag.survey.receiver_locations[:, 0].max(),
        ],
        mesh.vectorCCy[indy] * np.ones(2),
        c="k",
        linestyle="dotted",
    )
    ax[i, 2].set_xlim(
        [
            data_mag.survey.receiver_locations[:, 0].min(),
            data_mag.survey.receiver_locations[:, 0].max(),
        ],
    )
    ax[i, 2].set_ylim(
        [
            data_mag.survey.receiver_locations[:, 1].min(),
            data_mag.survey.receiver_locations[:, 1].max(),
        ],
    )

plt.tight_layout()
plt.show()

# Plot the 2D GMM
ticksize, labelsize = 10, 12
fig = plt.figure(figsize=(10, 10))
ax1 = plt.subplot2grid((4, 4), (0, 1), colspan=3, rowspan=3)
ax2 = plt.subplot2grid((4, 4), (3, 1), colspan=3)
ax3 = plt.subplot2grid((4, 4), (0, 0), rowspan=3)

ax1.set_xlim(-1.2, 0.1)
ax1.set_ylim(-0.0025, 0.03)
ax2.set_xlim(-1.2, 0.1)
ax3.set_ylim(-0.0025, 0.03)

ax1.set_xticks([-1.2, -1.0, -0.8, -0.6, -0.4, -0.2, 0.0])
ax2.set_xticks([-1.2, -1.0, -0.8, -0.6, -0.4, -0.2, 0.0])

ax1.set_yticks([0, 0.005, 0.01, 0.015, 0.02, 0.025])
ax3.set_yticks([0, 0.005, 0.01, 0.015, 0.02, 0.025])

x, y = np.mgrid[-1.21:0.1:0.01, -0.01:0.031:0.001]
pos = np.empty(x.shape + (2,))
pos[:, :, 0] = x
pos[:, :, 1] = y
rvm = learned_gmm.predict(pos.reshape(-1, 2))
rvsmooth = learned_gmm.score_samples(pos.reshape(-1, 2))
surf = ax1.contourf(
    x, y, (rvsmooth).reshape(x.shape), 25, cmap="viridis"
)  # linestyles='dashdot', linewidths=2.,
ax1.contour(
    x,
    y,
    rvsmooth.reshape(x.shape),
    25,
    colors="k",
    linewidths=1.0,
    linestyles="dashdot",
)
ax1.scatter(
    density_model_no_info[actv],
    magsus_model_no_info[actv],
    c=quasi_geology_model_no_info[actv],
    cmap="inferno_r",
    edgecolors="k",
    label="recovered PGI model",
    alpha=0.5,
)
ax1.scatter(
    np.r_[0, -0.8, -0.2],
    np.r_[0, 0.005, 0.02],
    label="True petrophysical means",
    cmap="inferno_r",
    c=[0, 1, 2],
    marker="v",
    edgecolors="k",
    s=200,
)
ax1.scatter(
    learned_gmm.means_[:, 0],
    learned_gmm.means_[:, 1],
    label="learned petrophysical means",
    cmap="inferno_r",
    c=[0, 1, 2],
    marker="o",
    edgecolors="k",
    s=200,
)

axbar = inset_axes(
    ax1,
    width="40%",  # width = 50% of parent_bbox width
    height="3%",  # height : 5%
    loc="upper right",
    borderpad=1,
)
cbpetro = fig.colorbar(surf, cax=axbar, orientation="horizontal")
cbpetro.set_ticks([rvsmooth.min(), rvsmooth.max()])
cbpetro.set_ticklabels(["Low", "High"])
cbpetro.set_label("Probability Density", fontsize=labelsize)
cbpetro.ax.tick_params(labelsize=ticksize)
cbpetro.outline.set_edgecolor("k")

# create the 1D GMM profile for density
from sklearn.mixture import GaussianMixture

means_init_grav = learned_gmm.means_[:, 0].reshape(3, 1)
cov_init_grav = np.array([learned_gmm.covariances_[:, 0]]).reshape((3, 1, 1))
clfgrav = utils.pgi_utils.GaussianMixture(
    n_components=3,
    means_init=means_init_grav,
    precisions_init=cov_init_grav,
    n_init=1,
    max_iter=2,
    tol=np.inf,
)
# random fit, we set values after.
clfgrav.fit(np.random.randn(10, 1))
clfgrav.means_ = means_init_grav
clfgrav.covariances_ = cov_init_grav
from sklearn.mixture._gaussian_mixture import _compute_precision_cholesky

clfgrav.precisions_cholesky_ = _compute_precision_cholesky(
    clfgrav.covariances_, clfgrav.covariance_type
)
clfgrav.weights_ = learned_gmm.weights_
testXplot_grav = np.linspace(-1.2, 0.1, 1000)[:, np.newaxis]
score_grav = clfgrav.score_samples(testXplot_grav)
ax2.plot(
    testXplot_grav, np.exp(score_grav), linewidth=3.0, label="proba.\ndensity", c="k"
)
ax2.set_ylim([0.0, 2])
ax2.legend(fontsize=ticksize)

# create the 1D GMM profile for mag. susc.
means_init_mag = learned_gmm.means_[:, 1].reshape(3, 1)
cov_init_mag = np.array([learned_gmm.covariances_[:, 1]]).reshape((3, 1, 1))
clfmag = GaussianMixture(
    n_components=3,
    means_init=means_init_mag,
    precisions_init=cov_init_mag,
    n_init=1,
    max_iter=2,
    tol=np.inf,
)
# random fit, we set values after.
clfmag.fit(np.random.randn(10, 1))
clfmag.means_ = means_init_mag
clfmag.covariances_ = cov_init_mag
clfmag.precisions_cholesky_ = _compute_precision_cholesky(
    clfmag.covariances_, clfmag.covariance_type
)
clfmag.weights_ = learned_gmm.weights_
testXplot_mag = np.linspace(-0.025, 0.03, 1000)[:, np.newaxis]
score_mag = clfmag.score_samples(testXplot_mag)
ax3.plot(np.exp(score_mag), testXplot_mag, linewidth=3.0, c="k")

ax3.set_xlim([0.0, 50])
ax3.set_xlabel(
    "Probability\nDensity", fontsize=labelsize, rotation=-45, labelpad=0, x=0.5
)
ax2.set_xlabel("Density (g/cc)", fontsize=labelsize)
ax3.set_ylabel("Magnetic Susceptibility (SI)", fontsize=labelsize)
ax2.tick_params(labelsize=ticksize)
ax3.tick_params(labelsize=ticksize)
ax1.text(-0.9, 0.0025, "PK/VK", fontsize=labelsize)
ax1.text(-0.175, 0.02, "HK", fontsize=labelsize)
ax1.text(-0.2, 0.002, "BCKGRD", fontsize=labelsize)  # , color='white')
ax1.tick_params(labelleft=False)
ax1.tick_params(labelbottom=False)
ax1.set_ylabel("")
ax1.set_xlabel("")
ax2.tick_params(axis="both", which="both", labelsize=ticksize)
ax3.tick_params(axis="both", which="both", labelsize=ticksize)

ax1.legend(fontsize=labelsize, loc=2)
ax2.hist(density_model_no_info[actv], density=True, bins=50)
ax3.hist(magsus_model_no_info[actv], density=True, bins=50, orientation="horizontal")
plt.show()
  • Slice 15, X = 557287.50, Slice 17, Y = 7133637.50, Slice -9, Z = 280.00, Slice 15, X = 557287.50, Slice 17, Y = 7133637.50, Slice -9, Z = 280.00, Slice 15, X = 557287.50, Slice 17, Y = 7133637.50, Slice -9, Z = 280.00
  • plot inv 2 joint pf pgi no info tutorial

Out:

        SimPEG.InvProblem is setting bfgsH0 to the inverse of the eval2Deriv.
        ***Done using same Solver and solver_opts as the Simulation3DIntegral problem***
Alpha scales:  [1, 1, 1]
Alpha scales:  [1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0]
Alpha scales:  [0.0, 19324.30079347746, 11727.414966407947, 32570.083086375373]
Alpha scales:  [0.0, 1241123.0214763389, 872625.8274458644, 2696321.1696726256]
Initial data misfit scales:  [0.96658918 0.03341082]
model has any nan: 0
=============================== Projected GNCG ===============================
  #     beta     phi_d     phi_m       f      |proj(x-g)-x|  LS    Comment
-----------------------------------------------------------------------------
x0 has any nan: 0
   0  2.39e-04  2.11e+06  0.00e+00  2.11e+06    2.17e+02      0
geophys. misfits: 223846.4 (target 288.0 [False]); 62367.9 (target 288.0 [False]) | smallness misfit: 69753.4 (target: 5863.5 [False])
Beta cooling evaluation: progress: [223846.4  62367.9] ; minimum progress targets: [1735115.6  286362. ]
mref changed in  4575  places
   1  2.39e-04  2.18e+05  7.76e+04  2.18e+05    2.14e+01      0
geophys. misfits: 101546.5 (target 288.0 [False]); 30522.2 (target 288.0 [False]) | smallness misfit: 60595.3 (target: 5863.5 [False])
Beta cooling evaluation: progress: [101546.5  30522.2] ; minimum progress targets: [179077.1  49894.3]
mref changed in  2450  places
   2  2.39e-04  9.92e+04  1.74e+05  9.92e+04    2.11e+01      0
geophys. misfits: 40716.4 (target 288.0 [False]); 5840.0 (target 288.0 [False]) | smallness misfit: 46984.3 (target: 5863.5 [False])
Beta cooling evaluation: progress: [40716.4  5840. ] ; minimum progress targets: [81237.2 24417.8]
mref changed in  1080  places
   3  2.39e-04  3.96e+04  1.84e+05  3.96e+04    2.01e+01      0   Skip BFGS
geophys. misfits: 1323.2 (target 288.0 [False]); 237.8 (target 288.0 [True]) | smallness misfit: 35380.9 (target: 5863.5 [False])
Updating scaling for data misfits by  1.2112115890741197
New scales: [0.97225372 0.02774628]
Beta cooling evaluation: progress: [1323.2  237.8] ; minimum progress targets: [32573.1  4672. ]
mref changed in  581  places
   4  2.39e-04  1.29e+03  1.93e+05  1.34e+03    1.84e+01      0   Skip BFGS
geophys. misfits: 87.7 (target 288.0 [True]); 26.0 (target 288.0 [True]) | smallness misfit: 32587.2 (target: 5863.5 [False])
Beta cooling evaluation: progress: [87.7 26. ] ; minimum progress targets: [1058.6  345.6]
Warming alpha_s to favor clustering:  7.170695011546249
mref changed in  163  places
   5  2.39e-04  8.60e+01  2.08e+05  1.36e+02    2.63e+01      0   Skip BFGS
geophys. misfits: 87.4 (target 288.0 [True]); 15.7 (target 288.0 [True]) | smallness misfit: 30335.7 (target: 5863.5 [False])
Beta cooling evaluation: progress: [87.4 15.7] ; minimum progress targets: [345.6 345.6]
Warming alpha_s to favor clustering:  77.55554130211962
mref changed in  10  places
   6  2.39e-04  8.54e+01  3.37e+05  1.66e+02    2.98e+01      0   Skip BFGS
geophys. misfits: 65.2 (target 288.0 [True]); 295.0 (target 288.0 [False]) | smallness misfit: 20845.0 (target: 5863.5 [False])
Updating scaling for data misfits by  4.414940893507613
New scales: [0.88810414 0.11189586]
Beta cooling evaluation: progress: [ 65.2 295. ] ; minimum progress targets: [345.6 345.6]
mref changed in  31  places
   7  2.39e-04  9.09e+01  3.10e+05  1.65e+02    1.92e+01      0
geophys. misfits: 72.3 (target 288.0 [True]); 71.6 (target 288.0 [True]) | smallness misfit: 19131.0 (target: 5863.5 [False])
Beta cooling evaluation: progress: [72.3 71.6] ; minimum progress targets: [345.6 345.6]
Warming alpha_s to favor clustering:  310.56278715085574
mref changed in  25  places
   8  2.39e-04  7.22e+01  6.22e+05  2.21e+02    1.67e+01      0
geophys. misfits: 66.2 (target 288.0 [True]); 147.5 (target 288.0 [True]) | smallness misfit: 10880.0 (target: 5863.5 [False])
Beta cooling evaluation: progress: [ 66.2 147.5] ; minimum progress targets: [345.6 345.6]
Warming alpha_s to favor clustering:  978.3249299271946
mref changed in  1  places
   9  2.39e-04  7.53e+01  1.07e+06  3.31e+02    2.83e+01      0
geophys. misfits: 91.2 (target 288.0 [True]); 285.1 (target 288.0 [True]) | smallness misfit: 7030.6 (target: 5863.5 [False])
Beta cooling evaluation: progress: [ 91.2 285.1] ; minimum progress targets: [345.6 345.6]
Warming alpha_s to favor clustering:  2038.1177502533358
mref changed in  0  places
Add mref to Smoothness. Changes in mref happened in 0.0 % of the cells
  10  2.39e-04  1.13e+02  1.13e+06  3.84e+02    3.41e+01      0
geophys. misfits: 165.4 (target 288.0 [True]); 74.1 (target 288.0 [True]) | smallness misfit: 5501.0 (target: 5863.5 [True])
All targets have been reached
Beta cooling evaluation: progress: [165.4  74.1] ; minimum progress targets: [345.6 345.6]
Warming alpha_s to favor clustering:  5737.463189621767
mref changed in  0  places
Add mref to Smoothness. Changes in mref happened in 0.0 % of the cells
------------------------- STOP! -------------------------
1 : |fc-fOld| = 0.0000e+00 <= tolF*(1+|f0|) = 2.1084e+05
0 : |xc-x_last| = 1.7023e+00 <= tolX*(1+|x0|) = 1.0153e-01
0 : |proj(x-g)-x|    = 3.3654e+01 <= tolG          = 1.0000e-01
0 : |proj(x-g)-x|    = 3.3654e+01 <= 1e3*eps       = 1.0000e-02
0 : maxIter   =      30    <= iter          =     11
------------------------- DONE! -------------------------

Total running time of the script: ( 5 minutes 0.285 seconds)

Estimated memory usage: 107 MB

Gallery generated by Sphinx-Gallery