Mondaic%matplotlib inline
%config Completer.use_jedi = False
import os
SALVUS_FLOW_SITE_NAME = os.environ.get("SITE_NAME", "local")In this mini-tutorial, we demonstrate how the mapping function can be used to invert for homogeneous material parameters. Here, the adjoint operation of the mapping class maps the space-dependent physical gradient into the subspace of homogeneous model parameters.
Selecting
scaling="homogeneous" in the mapping function of an InverseProblemConfiguration enables inversion for homogeneous model parameters with full-waveform inversion.sn.Mapping( scaling="homogeneous", inversion_parameters=[...], ... )
import matplotlib.pyplot as plt
import numpy as np
import pathlib
import time
import xarray as xr
import salvus.namespace as snWe consider a simplistic setup with a 2D elastic box domain with a single source-receiver pair.
Check out the full FWI tutorial for more details on the individual steps.
p = sn.Project.from_domain(
path="project",
domain=sn.domain.dim2.BoxDomain(
x0=0.0,
x1=1.0,
y0=0.0,
y1=1.5,
),
load_if_exists=True,
)
src = sn.simple_config.source.cartesian.VectorPoint2D(
x=0.33, y=0.0, fx=0.0, fy=1.0e9
)
rec = sn.simple_config.receiver.cartesian.Point2D(
x=0.67,
y=1.5,
fields=["displacement"],
network_code="REC",
station_code="000",
)
p += sn.Event(event_name="event", sources=src, receivers=rec)ec = sn.EventConfiguration(
waveform_simulation_configuration=sn.WaveformSimulationConfiguration(
end_time_in_seconds=0.0006
),
wavelet=sn.simple_config.stf.Ricker(center_frequency=10000.0),
)p.viz.nb.domain()
We consider two different homogeneous isotropic media. The "true" model we seek to reconstruct varies from the initial model in
vp, vs and rho. Note that the deviations in the parameters have opposite signs and different magnitudes.true_model = sn.model.background.homogeneous.IsotropicElastic(
vp=5800.0, vs=3200.0, rho=2600.0
)
initial_model = sn.model.background.homogeneous.IsotropicElastic(
vp=5600.0, vs=3300.0, rho=2400.0
)for name, model in zip(["true", "init"], [true_model, initial_model]):
p.add_to_project(
sn.SimulationConfiguration(
name=name,
elements_per_wavelength=2,
tensor_order=1,
max_frequency_in_hertz=20000.0,
model_configuration=sn.ModelConfiguration(background_model=model),
event_configuration=ec,
),
overwrite=True,
)
p.simulations.launch(
simulation_configuration=name,
events=p.events.list(),
site_name=SALVUS_FLOW_SITE_NAME,
ranks_per_job=1,
)[2025-02-07 20:41:05,319] INFO: Creating mesh. Hang on. [2025-02-07 20:41:05,387] INFO: Submitting job ... Uploading 1 files... 🚀 Submitted job_2502072041522925_f3b12c3fb0@local [2025-02-07 20:41:05,609] INFO: Creating mesh. Hang on. [2025-02-07 20:41:05,654] INFO: Submitting job ... Uploading 1 files... 🚀 Submitted job_2502072041659138_8892666328@local
p.simulations.query(block=True)
True
This is of course not a realistic setup, but the waveforms differ in both components.
p.viz.nb.waveforms(data=["true", "init"], receiver_field="displacement")
Next, we setup the inverse problem. We are using
scaling="homogeneous" in the mapping function, which enforces a homogeneous medium in every iteration. Note that a smoothing preconditioner is not necessary here, because the mapped gradient is already homogeneous.p += sn.MisfitConfiguration(
name="L2",
observed_data="true",
misfit_function="L2",
receiver_field="displacement",
)p += sn.InverseProblemConfiguration(
name="inversion",
prior_model="init",
events=p.events.list(),
mapping=sn.Mapping(
scaling="homogeneous",
inversion_parameters=["VP", "VS", "RHO"],
),
method=sn.TrustRegion(initial_trust_region_linf=100.0),
misfit_configuration="L2",
job_submission=sn.SiteConfig(
site_name=SALVUS_FLOW_SITE_NAME, ranks_per_job=1
),
)for i in range(10):
p.inversions.iterate(
inverse_problem_configuration="inversion",
timeout_in_seconds=360,
ping_interval_in_seconds=1,
)[2025-02-07 20:41:07,372] INFO: Adding new iteration #0. [2025-02-07 20:41:07,382] INFO: Resuming iteration #0. [2025-02-07 20:41:07,383] INFO: 1 new tasks have been issued. [2025-02-07 20:41:07,383] INFO: Processing task `misfit_and_gradient` [2025-02-07 20:41:07,404] INFO: The following events have been simulated before, but checkpoints are not available for this combination of `site_name` and `ranks_per_job` and wavefield compression settings. They will be run again: ['event'] [2025-02-07 20:41:07,417] INFO: Submitting job ... [2025-02-07 20:41:07,812] INFO: Launched simulations for 1 events. Please check again to see if they are finished. [2025-02-07 20:41:07,812] INFO: Some tasks of iteration #0 are still running. Please check again later. [2025-02-07 20:41:08,814] INFO: Processing task `misfit_and_gradient` [2025-02-07 20:41:08,864] INFO: Some tasks of iteration #0 are still running. Please check again later. [2025-02-07 20:41:09,866] INFO: Processing task `misfit_and_gradient` [2025-02-07 20:41:10,083] INFO: Submitting job ... Uploading 1 files... 🚀 Submitted job_2502072041085964_9472cab58f@local [2025-02-07 20:41:10,139] INFO: Launched adjoint simulations for 1 events. Please check again to see if they are finished. [2025-02-07 20:41:10,140] INFO: Some tasks of iteration #0 are still running. Please check again later. [2025-02-07 20:41:11,141] INFO: Processing task `misfit_and_gradient` [2025-02-07 20:41:11,212] INFO: 1 events have already been submitted. They will not be submitted again. [2025-02-07 20:41:11,381] INFO: Iteration 0: Number of events: 1 chi = 3.7156536966723097e-10 ||g|| = 3.3910359372571976e-12 pred = --- ared = --- norm_update = --- tr_radius = --- [2025-02-07 20:41:11,386] INFO: 1 new tasks have been issued. [2025-02-07 20:41:11,387] INFO: Processing task `misfit` [2025-02-07 20:41:11,426] INFO: Submitting job ... [2025-02-07 20:41:11,478] INFO: Launched simulations for 1 events. Please check again to see if they are finished. [2025-02-07 20:41:11,479] INFO: Some tasks of iteration #0 are still running. Please check again later. [2025-02-07 20:41:12,495] INFO: Processing task `misfit` [2025-02-07 20:41:12,542] INFO: Some tasks of iteration #0 are still running. Please check again later. [2025-02-07 20:41:13,557] INFO: Processing task `misfit` [2025-02-07 20:41:13,662] INFO: old misfit control group: 3.7156536966723097e-10 new misfit control group: 1.073889989952079e-10 predicted reduction control group: -2.1653942143243586e-10 actual reduction control group: -2.6417637067202306e-10 1 out of 1 event(s) improved the misfit. [2025-02-07 20:41:13,663] INFO: Model update accepted. [2025-02-07 20:41:13,663] INFO: 1 new tasks have been issued. [2025-02-07 20:41:13,664] INFO: Processing task `finalize_iteration` [2025-02-07 20:41:13,690] INFO: Succesfully completed iteration #0. [2025-02-07 20:41:13,692] INFO: Adding new iteration #1. [2025-02-07 20:41:13,696] INFO: Resuming iteration #1. [2025-02-07 20:41:13,697] INFO: 1 new tasks have been issued. [2025-02-07 20:41:13,697] INFO: Processing task `gradient` [2025-02-07 20:41:13,749] INFO: Submitting job ... Uploading 1 files... 🚀 Submitted job_2502072041751713_83eeeb8586@local [2025-02-07 20:41:13,815] INFO: Launched adjoint simulations for 1 events. Please check again to see if they are finished. [2025-02-07 20:41:13,815] INFO: Some tasks of iteration #1 are still running. Please check again later. [2025-02-07 20:41:14,817] INFO: Processing task `gradient` [2025-02-07 20:41:14,858] INFO: 1 events have already been submitted. They will not be submitted again. [2025-02-07 20:41:15,006] INFO: Iteration 1: Number of events: 1 chi = 1.073889989952079e-10 ||g|| = 1.037229670389857e-12 pred = -2.1653942143243586e-10 ared = -2.6417637067202306e-10 norm_update = 127.71284376748982 tr_radius = 127.71284376748997 [2025-02-07 20:41:15,020] INFO: 1 new tasks have been issued. [2025-02-07 20:41:15,021] INFO: Processing task `misfit` [2025-02-07 20:41:15,065] INFO: Submitting job ... [2025-02-07 20:41:15,125] INFO: Launched simulations for 1 events. Please check again to see if they are finished. [2025-02-07 20:41:15,125] INFO: Some tasks of iteration #1 are still running. Please check again later. [2025-02-07 20:41:16,143] INFO: Processing task `misfit` [2025-02-07 20:41:16,250] INFO: old misfit control group: 1.073889989952079e-10 new misfit control group: 7.296709075757565e-11 predicted reduction control group: -1.886436638060351e-11 actual reduction control group: -3.442190823763225e-11 1 out of 1 event(s) improved the misfit. [2025-02-07 20:41:16,251] INFO: Model update accepted. [2025-02-07 20:41:16,252] INFO: 1 new tasks have been issued. [2025-02-07 20:41:16,252] INFO: Processing task `finalize_iteration` [2025-02-07 20:41:16,279] INFO: Succesfully completed iteration #1. [2025-02-07 20:41:16,282] INFO: Adding new iteration #2. [2025-02-07 20:41:16,289] INFO: Resuming iteration #2. [2025-02-07 20:41:16,290] INFO: 1 new tasks have been issued. [2025-02-07 20:41:16,291] INFO: Processing task `gradient` [2025-02-07 20:41:16,358] INFO: Submitting job ... Uploading 1 files... 🚀 Submitted job_2502072041360857_7b763b6f1a@local [2025-02-07 20:41:16,419] INFO: Launched adjoint simulations for 1 events. Please check again to see if they are finished. [2025-02-07 20:41:16,420] INFO: Some tasks of iteration #2 are still running. Please check again later. [2025-02-07 20:41:17,423] INFO: Processing task `gradient` [2025-02-07 20:41:17,461] INFO: 1 events have already been submitted. They will not be submitted again. [2025-02-07 20:41:17,593] INFO: Iteration 2: Number of events: 1 chi = 7.296709075757565e-11 ||g|| = 8.841255720378636e-13 pred = -1.886436638060351e-11 ared = -3.442190823763225e-11 norm_update = 38.055155588990125 tr_radius = 255.42568753497994 [2025-02-07 20:41:17,610] INFO: 1 new tasks have been issued. [2025-02-07 20:41:17,611] INFO: Processing task `misfit` [2025-02-07 20:41:17,653] INFO: Submitting job ... [2025-02-07 20:41:17,713] INFO: Launched simulations for 1 events. Please check again to see if they are finished. [2025-02-07 20:41:17,713] INFO: Some tasks of iteration #2 are still running. Please check again later. [2025-02-07 20:41:18,730] INFO: Processing task `misfit` [2025-02-07 20:41:18,780] INFO: Some tasks of iteration #2 are still running. Please check again later. [2025-02-07 20:41:19,795] INFO: Processing task `misfit` [2025-02-07 20:41:19,906] INFO: old misfit control group: 7.296709075757565e-11 new misfit control group: 7.29921706188182e-12 predicted reduction control group: -5.3947833613705984e-11 actual reduction control group: -6.566787369569383e-11 1 out of 1 event(s) improved the misfit. [2025-02-07 20:41:19,907] INFO: Model update accepted. [2025-02-07 20:41:19,907] INFO: 1 new tasks have been issued. [2025-02-07 20:41:19,907] INFO: Processing task `finalize_iteration` [2025-02-07 20:41:19,931] INFO: Succesfully completed iteration #2. [2025-02-07 20:41:19,934] INFO: Adding new iteration #3. [2025-02-07 20:41:19,940] INFO: Resuming iteration #3. [2025-02-07 20:41:19,941] INFO: 1 new tasks have been issued.
[2025-02-07 20:41:19,941] INFO: Processing task `gradient` [2025-02-07 20:41:20,005] INFO: Submitting job ... Uploading 1 files... 🚀 Submitted job_2502072041010574_2b3024da5a@local [2025-02-07 20:41:20,059] INFO: Launched adjoint simulations for 1 events. Please check again to see if they are finished. [2025-02-07 20:41:20,060] INFO: Some tasks of iteration #3 are still running. Please check again later. [2025-02-07 20:41:21,062] INFO: Processing task `gradient` [2025-02-07 20:41:21,098] INFO: 1 events have already been submitted. They will not be submitted again. [2025-02-07 20:41:21,130] INFO: Some simulations are still running. Please check again to see if they are finished. [2025-02-07 20:41:21,131] INFO: Some tasks of iteration #3 are still running. Please check again later. [2025-02-07 20:41:22,134] INFO: Processing task `gradient` [2025-02-07 20:41:22,171] INFO: 1 events have already been submitted. They will not be submitted again. [2025-02-07 20:41:22,315] INFO: Iteration 3: Number of events: 1 chi = 7.29921706188182e-12 ||g|| = 2.1080709923504721e-13 pred = -5.3947833613705984e-11 ared = -6.566787369569383e-11 norm_update = 127.71284376748984 tr_radius = 127.71284376748997 [2025-02-07 20:41:22,338] INFO: 1 new tasks have been issued. [2025-02-07 20:41:22,342] INFO: Processing task `misfit` [2025-02-07 20:41:22,378] INFO: Submitting job ... [2025-02-07 20:41:22,442] INFO: Launched simulations for 1 events. Please check again to see if they are finished. [2025-02-07 20:41:22,443] INFO: Some tasks of iteration #3 are still running. Please check again later. [2025-02-07 20:41:23,457] INFO: Processing task `misfit` [2025-02-07 20:41:23,554] INFO: old misfit control group: 7.29921706188182e-12 new misfit control group: 3.853990185690102e-12 predicted reduction control group: -3.625527296947774e-12 actual reduction control group: -3.4452268761917177e-12 1 out of 1 event(s) improved the misfit. [2025-02-07 20:41:23,555] INFO: Model update accepted. [2025-02-07 20:41:23,555] INFO: 1 new tasks have been issued. [2025-02-07 20:41:23,555] INFO: Processing task `finalize_iteration` [2025-02-07 20:41:23,586] INFO: Succesfully completed iteration #3. [2025-02-07 20:41:23,590] INFO: Adding new iteration #4. [2025-02-07 20:41:23,601] INFO: Resuming iteration #4. [2025-02-07 20:41:23,601] INFO: 1 new tasks have been issued. [2025-02-07 20:41:23,602] INFO: Processing task `gradient` [2025-02-07 20:41:23,656] INFO: Submitting job ... Uploading 1 files... 🚀 Submitted job_2502072041659757_7fa15f3cbd@local [2025-02-07 20:41:23,722] INFO: Launched adjoint simulations for 1 events. Please check again to see if they are finished. [2025-02-07 20:41:23,723] INFO: Some tasks of iteration #4 are still running. Please check again later. [2025-02-07 20:41:24,726] INFO: Processing task `gradient` [2025-02-07 20:41:24,765] INFO: 1 events have already been submitted. They will not be submitted again. [2025-02-07 20:41:24,815] INFO: Some simulations are still running. Please check again to see if they are finished. [2025-02-07 20:41:24,816] INFO: Some tasks of iteration #4 are still running. Please check again later. [2025-02-07 20:41:25,819] INFO: Processing task `gradient` [2025-02-07 20:41:25,856] INFO: 1 events have already been submitted. They will not be submitted again. [2025-02-07 20:41:25,901] INFO: Some simulations are still running. Please check again to see if they are finished. [2025-02-07 20:41:25,902] INFO: Some tasks of iteration #4 are still running. Please check again later. [2025-02-07 20:41:26,905] INFO: Processing task `gradient` [2025-02-07 20:41:26,952] INFO: 1 events have already been submitted. They will not be submitted again. [2025-02-07 20:41:27,104] INFO: Iteration 4: Number of events: 1 chi = 3.853990185690102e-12 ||g|| = 5.491251362114438e-14 pred = -3.625527296947774e-12 ared = -3.4452268761917177e-12 norm_update = 37.634593986576924 tr_radius = 255.42568753497994 [2025-02-07 20:41:27,134] INFO: 1 new tasks have been issued. [2025-02-07 20:41:27,135] INFO: Processing task `misfit` [2025-02-07 20:41:27,167] INFO: Submitting job ... [2025-02-07 20:41:27,222] INFO: Launched simulations for 1 events. Please check again to see if they are finished. [2025-02-07 20:41:27,222] INFO: Some tasks of iteration #4 are still running. Please check again later. [2025-02-07 20:41:28,240] INFO: Processing task `misfit` [2025-02-07 20:41:28,284] INFO: Some tasks of iteration #4 are still running. Please check again later. [2025-02-07 20:41:29,297] INFO: Processing task `misfit` [2025-02-07 20:41:29,399] INFO: old misfit control group: 3.853990185690102e-12 new misfit control group: 3.660798440136839e-12 predicted reduction control group: -1.1394903979292904e-13 actual reduction control group: -1.9319174555326333e-13 1 out of 1 event(s) improved the misfit. [2025-02-07 20:41:29,400] INFO: Model update accepted. [2025-02-07 20:41:29,400] INFO: 1 new tasks have been issued. [2025-02-07 20:41:29,401] INFO: Processing task `finalize_iteration` [2025-02-07 20:41:29,436] INFO: Succesfully completed iteration #4. [2025-02-07 20:41:29,440] INFO: Adding new iteration #5. [2025-02-07 20:41:29,451] INFO: Resuming iteration #5. [2025-02-07 20:41:29,451] INFO: 1 new tasks have been issued. [2025-02-07 20:41:29,452] INFO: Processing task `gradient` [2025-02-07 20:41:29,511] INFO: Submitting job ... Uploading 1 files... 🚀 Submitted job_2502072041514564_f6ca8c7a38@local [2025-02-07 20:41:29,569] INFO: Launched adjoint simulations for 1 events. Please check again to see if they are finished. [2025-02-07 20:41:29,569] INFO: Some tasks of iteration #5 are still running. Please check again later. [2025-02-07 20:41:30,572] INFO: Processing task `gradient` [2025-02-07 20:41:30,621] INFO: 1 events have already been submitted. They will not be submitted again. [2025-02-07 20:41:30,674] INFO: Some simulations are still running. Please check again to see if they are finished. [2025-02-07 20:41:30,675] INFO: Some tasks of iteration #5 are still running. Please check again later. [2025-02-07 20:41:31,678] INFO: Processing task `gradient` [2025-02-07 20:41:31,717] INFO: 1 events have already been submitted. They will not be submitted again. [2025-02-07 20:41:31,759] INFO: Some simulations are still running. Please check again to see if they are finished. [2025-02-07 20:41:31,760] INFO: Some tasks of iteration #5 are still running. Please check again later. [2025-02-07 20:41:32,763] INFO: Processing task `gradient` [2025-02-07 20:41:32,800] INFO: 1 events have already been submitted. They will not be submitted again. [2025-02-07 20:41:32,924] INFO: Iteration 5: Number of events: 1 chi = 3.660798440136839e-12 ||g|| = 3.682525298193585e-14 pred = -1.1394903979292904e-13 ared = -1.9319174555326333e-13 norm_update = 4.998974676234866 tr_radius = 127.71284376748997 [2025-02-07 20:41:32,959] INFO: 1 new tasks have been issued. [2025-02-07 20:41:32,960] INFO: Processing task `misfit` [2025-02-07 20:41:32,991] INFO: Submitting job ... [2025-02-07 20:41:33,046] INFO: Launched simulations for 1 events. Please check again to see if they are finished.
[2025-02-07 20:41:33,046] INFO: Some tasks of iteration #5 are still running. Please check again later. [2025-02-07 20:41:34,059] INFO: Processing task `misfit` [2025-02-07 20:41:34,161] INFO: old misfit control group: 3.660798440136839e-12 new misfit control group: 2.9810284223434555e-12 predicted reduction control group: -4.4323683818356934e-13 actual reduction control group: -6.797700177933832e-13 1 out of 1 event(s) improved the misfit. [2025-02-07 20:41:34,161] INFO: Model update accepted. [2025-02-07 20:41:34,162] INFO: 1 new tasks have been issued. [2025-02-07 20:41:34,162] INFO: Processing task `finalize_iteration` [2025-02-07 20:41:34,197] INFO: Succesfully completed iteration #5. [2025-02-07 20:41:34,201] INFO: Adding new iteration #6. [2025-02-07 20:41:34,213] INFO: Resuming iteration #6. [2025-02-07 20:41:34,216] INFO: 1 new tasks have been issued. [2025-02-07 20:41:34,216] INFO: Processing task `gradient` [2025-02-07 20:41:34,272] INFO: Submitting job ... Uploading 1 files... 🚀 Submitted job_2502072041274476_ef5d62a33d@local [2025-02-07 20:41:34,327] INFO: Launched adjoint simulations for 1 events. Please check again to see if they are finished. [2025-02-07 20:41:34,327] INFO: Some tasks of iteration #6 are still running. Please check again later. [2025-02-07 20:41:35,331] INFO: Processing task `gradient` [2025-02-07 20:41:35,367] INFO: 1 events have already been submitted. They will not be submitted again. [2025-02-07 20:41:35,411] INFO: Some simulations are still running. Please check again to see if they are finished. [2025-02-07 20:41:35,411] INFO: Some tasks of iteration #6 are still running. Please check again later. [2025-02-07 20:41:36,415] INFO: Processing task `gradient` [2025-02-07 20:41:36,461] INFO: 1 events have already been submitted. They will not be submitted again. [2025-02-07 20:41:36,595] INFO: Iteration 6: Number of events: 1 chi = 2.9810284223434555e-12 ||g|| = 1.1334072418867954e-13 pred = -4.4323683818356934e-13 ared = -6.797700177933832e-13 norm_update = 25.945609247270596 tr_radius = 63.856421883744986 [2025-02-07 20:41:36,637] INFO: 1 new tasks have been issued. [2025-02-07 20:41:36,638] INFO: Processing task `misfit` [2025-02-07 20:41:36,668] INFO: Submitting job ... [2025-02-07 20:41:36,720] INFO: Launched simulations for 1 events. Please check again to see if they are finished. [2025-02-07 20:41:36,720] INFO: Some tasks of iteration #6 are still running. Please check again later. [2025-02-07 20:41:37,736] INFO: Processing task `misfit` [2025-02-07 20:41:37,773] INFO: Some tasks of iteration #6 are still running. Please check again later. [2025-02-07 20:41:38,791] INFO: Processing task `misfit` [2025-02-07 20:41:38,894] INFO: old misfit control group: 2.9810284223434555e-12 new misfit control group: 1.9555860418200896e-12 predicted reduction control group: -6.673080610254194e-13 actual reduction control group: -1.025442380523366e-12 1 out of 1 event(s) improved the misfit. [2025-02-07 20:41:38,895] INFO: Model update accepted. [2025-02-07 20:41:38,895] INFO: 1 new tasks have been issued. [2025-02-07 20:41:38,896] INFO: Processing task `finalize_iteration` [2025-02-07 20:41:38,926] INFO: Succesfully completed iteration #6. [2025-02-07 20:41:38,932] INFO: Adding new iteration #7. [2025-02-07 20:41:38,944] INFO: Resuming iteration #7. [2025-02-07 20:41:38,944] INFO: 1 new tasks have been issued. [2025-02-07 20:41:38,945] INFO: Processing task `gradient` [2025-02-07 20:41:39,002] INFO: Submitting job ... Uploading 1 files... 🚀 Submitted job_2502072041005235_283422cd2e@local [2025-02-07 20:41:39,062] INFO: Launched adjoint simulations for 1 events. Please check again to see if they are finished. [2025-02-07 20:41:39,062] INFO: Some tasks of iteration #7 are still running. Please check again later. [2025-02-07 20:41:40,066] INFO: Processing task `gradient` [2025-02-07 20:41:40,115] INFO: 1 events have already been submitted. They will not be submitted again. [2025-02-07 20:41:40,156] INFO: Some simulations are still running. Please check again to see if they are finished. [2025-02-07 20:41:40,157] INFO: Some tasks of iteration #7 are still running. Please check again later. [2025-02-07 20:41:41,161] INFO: Processing task `gradient` [2025-02-07 20:41:41,202] INFO: 1 events have already been submitted. They will not be submitted again. [2025-02-07 20:41:41,249] INFO: Some simulations are still running. Please check again to see if they are finished. [2025-02-07 20:41:41,250] INFO: Some tasks of iteration #7 are still running. Please check again later. [2025-02-07 20:41:42,253] INFO: Processing task `gradient` [2025-02-07 20:41:42,293] INFO: 1 events have already been submitted. They will not be submitted again. [2025-02-07 20:41:42,435] INFO: Iteration 7: Number of events: 1 chi = 1.9555860418200896e-12 ||g|| = 1.8082278362551017e-13 pred = -6.673080610254194e-13 ared = -1.025442380523366e-12 norm_update = 51.891218494541064 tr_radius = 51.89121849454119 [2025-02-07 20:41:42,485] INFO: 1 new tasks have been issued. [2025-02-07 20:41:42,486] INFO: Processing task `misfit` [2025-02-07 20:41:42,516] INFO: Submitting job ... [2025-02-07 20:41:42,605] INFO: Launched simulations for 1 events. Please check again to see if they are finished. [2025-02-07 20:41:42,605] INFO: Some tasks of iteration #7 are still running. Please check again later. [2025-02-07 20:41:43,621] INFO: Processing task `misfit` [2025-02-07 20:41:43,668] INFO: Some tasks of iteration #7 are still running. Please check again later. [2025-02-07 20:41:44,685] INFO: Processing task `misfit` [2025-02-07 20:41:44,800] INFO: old misfit control group: 1.9555860418200896e-12 new misfit control group: 7.195492692978477e-13 predicted reduction control group: -8.887018361461577e-13 actual reduction control group: -1.2360367725222419e-12 1 out of 1 event(s) improved the misfit. [2025-02-07 20:41:44,800] INFO: Model update accepted. [2025-02-07 20:41:44,801] INFO: 1 new tasks have been issued. [2025-02-07 20:41:44,801] INFO: Processing task `finalize_iteration` [2025-02-07 20:41:44,840] INFO: Succesfully completed iteration #7. [2025-02-07 20:41:44,844] INFO: Adding new iteration #8. [2025-02-07 20:41:44,861] INFO: Resuming iteration #8. [2025-02-07 20:41:44,861] INFO: 1 new tasks have been issued. [2025-02-07 20:41:44,862] INFO: Processing task `gradient` [2025-02-07 20:41:44,920] INFO: Submitting job ... Uploading 1 files... 🚀 Submitted job_2502072041924730_74cee26fff@local [2025-02-07 20:41:44,981] INFO: Launched adjoint simulations for 1 events. Please check again to see if they are finished. [2025-02-07 20:41:44,981] INFO: Some tasks of iteration #8 are still running. Please check again later. [2025-02-07 20:41:45,986] INFO: Processing task `gradient` [2025-02-07 20:41:46,017] INFO: 1 events have already been submitted. They will not be submitted again. [2025-02-07 20:41:46,049] INFO: Some simulations are still running. Please check again to see if they are finished. [2025-02-07 20:41:46,050] INFO: Some tasks of iteration #8 are still running. Please check again later. [2025-02-07 20:41:47,054] INFO: Processing task `gradient`
[2025-02-07 20:41:47,092] INFO: 1 events have already been submitted. They will not be submitted again. [2025-02-07 20:41:47,242] INFO: Iteration 8: Number of events: 1 chi = 7.195492692978477e-13 ||g|| = 1.668305656509431e-13 pred = -8.887018361461577e-13 ared = -1.2360367725222419e-12 norm_update = 83.04605740553336 tr_radius = 103.78243698908238 [2025-02-07 20:41:47,300] INFO: 1 new tasks have been issued. [2025-02-07 20:41:47,300] INFO: Processing task `misfit` [2025-02-07 20:41:47,332] INFO: Submitting job ... [2025-02-07 20:41:47,390] INFO: Launched simulations for 1 events. Please check again to see if they are finished. [2025-02-07 20:41:47,390] INFO: Some tasks of iteration #8 are still running. Please check again later. [2025-02-07 20:41:48,409] INFO: Processing task `misfit` [2025-02-07 20:41:48,528] INFO: old misfit control group: 7.195492692978477e-13 new misfit control group: 3.428681080907987e-14 predicted reduction control group: -5.160693460554207e-13 actual reduction control group: -6.852624584887678e-13 1 out of 1 event(s) improved the misfit. [2025-02-07 20:41:48,529] INFO: Model update accepted. [2025-02-07 20:41:48,530] INFO: 1 new tasks have been issued. [2025-02-07 20:41:48,530] INFO: Processing task `finalize_iteration` [2025-02-07 20:41:48,577] INFO: Succesfully completed iteration #8. [2025-02-07 20:41:48,583] INFO: Adding new iteration #9. [2025-02-07 20:41:48,605] INFO: Resuming iteration #9. [2025-02-07 20:41:48,606] INFO: The total number of iterations in this inversion tree has been reached and no new iteration will be added. If you want to continue with this inversion, you need to adjust `max_iterations_global` using `set_stopping_criteria`.
Finally, we load the final model and check that it indeed converged to the true model.
The reconstruction matches the true model with less than 0.1% error for compressional and shear wave speed and less than 0.2% for density.
p.viz.nb.inversion(inverse_problem_configuration="inversion")
solution = p.simulations.get_mesh(
p.inversions.get_simulation_name("inversion", 9)
)
np.testing.assert_allclose(solution.elemental_fields["VP"], 5800.0, 1e-3)
np.testing.assert_allclose(solution.elemental_fields["VS"], 3200.0, 1e-3)
np.testing.assert_allclose(solution.elemental_fields["RHO"], 2600.0, 4e-3)Another useful extension deals with piecewise constant material parameters. We assume that the the partition of the domain into two or more materials is known and captured by the mesh, but the exact material parameters are unknown.
Discontinuous material interfaces can be indicated using an elemental field
block with integer IDs labeling the different parts of the model. For some meshing routines, this field is automatically added. In this example, we manually assign different IDs to distinguish a small square at [0.4, 0.6] x [0.5, 0.7] from the background media.mesh = p.simulations.get_mesh("true")
blocks = np.zeros(mesh.nelem, dtype=int)
mask1 = mesh.get_element_centroid()[:, 0] > 0.4
mask2 = mesh.get_element_centroid()[:, 0] < 0.6
mask3 = mesh.get_element_centroid()[:, 1] > 0.5
mask4 = mesh.get_element_centroid()[:, 1] < 0.7
mask = mask1 & mask2 & mask3 & mask4
blocks[mask] = 1
mesh.attach_field("block", blocks)
mesh
<salvus.mesh.data_structures.unstructured_mesh.unstructured_mesh.UnstructuredMesh object at 0x7eda11513910>
To demonstrate inverting for piece-wise constant material, we add two simulation configurations, a homogeneous model with the block IDs as initial model, and the "true model" with different properties in the selected square.
p.add_to_project(
sn.UnstructuredMeshSimulationConfiguration(
name="two_blocks_initial",
unstructured_mesh=mesh,
event_configuration=ec,
),
)
mesh.elemental_fields["VP"][blocks == 1] *= 1.1
mesh.elemental_fields["VS"][blocks == 1] *= 1.1
mesh.elemental_fields["RHO"][blocks == 1] *= 1.1
p.add_to_project(
sn.UnstructuredMeshSimulationConfiguration(
name="two_blocks",
unstructured_mesh=mesh,
event_configuration=ec,
),
)for name in ["two_blocks_initial", "two_blocks"]:
p.simulations.launch(
simulation_configuration=name,
events=p.events.list(),
site_name=SALVUS_FLOW_SITE_NAME,
ranks_per_job=1,
)
p.simulations.query(block=True)[2025-02-07 20:41:50,353] INFO: Submitting job ... Uploading 1 files... 🚀 Submitted job_2502072041356637_d5744d8567@local
[2025-02-07 20:41:51,275] INFO: Submitting job ... Uploading 1 files... 🚀 Submitted job_2502072041280555_3a9e32521f@local
A new misfit configuration is required to reference the data from te block model.
p += sn.MisfitConfiguration(
name="L2-block",
observed_data="two_blocks",
misfit_function="L2",
receiver_field="displacement",
)The key element to bring in the discontinuous model parameterization into the inversion is in the mapping function. Here, we specify
discontinuous_model_blocks, which refers to the values defined in the field block and indicates the piece-wise constant regions of the model.
In this example, only the area with block = 1 will be updated. The background model is kept constant. Note that in this example no region of interest or preconditioner is required.p += sn.InverseProblemConfiguration(
name="inversion_two",
prior_model="two_blocks_initial",
events=p.events.list(),
mapping=sn.Mapping(
scaling="homogeneous",
discontinuous_model_blocks=[[1]],
inversion_parameters=["VP", "VS", "RHO"],
),
method=sn.TrustRegion(initial_trust_region_linf=100.0),
misfit_configuration="L2-block",
job_submission=sn.SiteConfig(
site_name=SALVUS_FLOW_SITE_NAME, ranks_per_job=1
),
)We perform a single iteration to validate that the update only affects the block inclusion.
p.inversions.iterate(
inverse_problem_configuration="inversion_two",
timeout_in_seconds=360,
ping_interval_in_seconds=5,
)[2025-02-07 20:41:52,608] INFO: Adding new iteration #0. [2025-02-07 20:41:52,616] INFO: Resuming iteration #0. [2025-02-07 20:41:52,616] INFO: 1 new tasks have been issued. [2025-02-07 20:41:52,617] INFO: Processing task `misfit_and_gradient` [2025-02-07 20:41:52,636] INFO: The following events have been simulated before, but checkpoints are not available for this combination of `site_name` and `ranks_per_job` and wavefield compression settings. They will be run again: ['event'] [2025-02-07 20:41:52,650] INFO: Submitting job ... [2025-02-07 20:41:52,715] INFO: Launched simulations for 1 events. Please check again to see if they are finished. [2025-02-07 20:41:52,715] INFO: Some tasks of iteration #0 are still running. Please check again later. [2025-02-07 20:41:57,717] INFO: Processing task `misfit_and_gradient` [2025-02-07 20:41:57,875] INFO: Submitting job ... Uploading 1 files... 🚀 Submitted job_2502072041877647_2ace727405@local [2025-02-07 20:41:57,939] INFO: Launched adjoint simulations for 1 events. Please check again to see if they are finished. [2025-02-07 20:41:57,940] INFO: Some tasks of iteration #0 are still running. Please check again later. [2025-02-07 20:42:02,941] INFO: Processing task `misfit_and_gradient` [2025-02-07 20:42:02,999] INFO: 1 events have already been submitted. They will not be submitted again. [2025-02-07 20:42:03,130] INFO: Iteration 0: Number of events: 1 chi = 1.8063817169917176e-11 ||g|| = 3.4633490526471955e-13 pred = --- ared = --- norm_update = --- tr_radius = --- [2025-02-07 20:42:03,134] INFO: 1 new tasks have been issued. [2025-02-07 20:42:03,134] INFO: Processing task `misfit` [2025-02-07 20:42:03,167] INFO: Submitting job ... [2025-02-07 20:42:03,227] INFO: Launched simulations for 1 events. Please check again to see if they are finished. [2025-02-07 20:42:03,228] INFO: Some tasks of iteration #0 are still running. Please check again later. [2025-02-07 20:42:08,244] INFO: Processing task `misfit` [2025-02-07 20:42:08,348] INFO: old misfit control group: 1.8063817169917176e-11 new misfit control group: 1.0995271385440556e-11 predicted reduction control group: -4.165941036285124e-12 actual reduction control group: -7.06854578447662e-12 1 out of 1 event(s) improved the misfit. [2025-02-07 20:42:08,349] INFO: Model update accepted. [2025-02-07 20:42:08,349] INFO: 1 new tasks have been issued. [2025-02-07 20:42:08,349] INFO: Processing task `finalize_iteration` [2025-02-07 20:42:08,375] INFO: Succesfully completed iteration #0. [2025-02-07 20:42:08,377] INFO: Adding new iteration #1.
p.simulations.get_mesh(p.inversions.get_simulation_name("inversion_two", 1))
<salvus.mesh.data_structures.unstructured_mesh.unstructured_mesh.UnstructuredMesh object at 0x7eda10e8b010>
PAGE CONTENTS