SWEEP (Seismic Wave Equation Exploration Platform): A Unified Solver Framework for Differentiable Wave Physics

SWEEP (Seismic Wave Equation Exploration Platform) is een uitgebreid en automatisch differentieerbaar framework voor golfveldmodellering en inversie dat diverse golfvoortplantingsmechanismen ondersteunt en naadloze integratie van aangepaste loss-functies, multi-GPU-berekening en neurale netwerken mogelijk maakt voor geavanceerde seismische inverse problemen.

De kern

SWEEP: De Zwitserse Zakmes voor de Aardwetenschappen

Stel je voor dat je een enorme, onzichtbare röntgenfoto wilt maken van de aarde. Je wilt weten wat er diep onder de grond zit: waar olie zit, waar de aardkorst breekt, of waar de grond stevig is. Maar je kunt niet boren tot in de kern; je moet luisteren naar de trillingen die je zelf veroorzaakt.

Dit is wat seismologen doen: ze slaan op de grond (met een explosie of een trilmotor) en luisteren naar het echo-systeem. De manier waarop die trillingen terugkomen, vertelt hen wat er onder de grond zit.

Het probleem? De wiskunde om die trillingen te berekenen is ontzettend complex. Het is alsof je probeert te voorspellen hoe een golf zich gedraagt in een badkuip, maar dan met een badkuip die uit honderden lagen verschillende soep bestaat, en je moet dit doen voor elke seconde van de trilling.

Wat is SWEEP dan?

SWEEP (Seismic Wave Equation Exploration Platform) is een nieuwe, slimme software die deze hele ingewikkelde wiskunde voor je regelt. Je kunt het zien als een universele "bouwkit" voor aard-trillingen.

Hier is hoe het werkt, vertaald naar alledaagse termen:

1. De "Zwarte Doos" die alles snapt

Vroeger moesten wetenschappers de regels voor deze trillingen met de hand uitschrijven. Ze moesten een "tegen-golf" bedenken om de fouten te corrigeren. Dit was als het handmatig oplossen van een Sudoku-puzzel van 10.000 vakjes: saai, foutgevoelig en tijdrovend.

SWEEP gebruikt een trucje genaamd automatische differentiatie. Stel je voor dat je een robot hebt die niet alleen de Sudoku oplost, maar ook automatisch ziet waar je een fout hebt gemaakt en precies weet hoe je die moet verbeteren. SWEEP doet dit voor de golven. Je geeft de computer de grondsoorten, en hij berekent niet alleen hoe de golf gaat, maar ook precies hoe je de grondsoorten moet aanpassen om de golf beter te laten passen bij de werkelijkheid.

2. De Lego-blokken (Modulair ontwerp)

Het mooie aan SWEEP is dat het uit losse blokken bestaat, net als Lego.

• De Propagator: Dit is de motor die de golf door de tijd laat bewegen.
• De Bron: Dit is de "hamer" die op de grond slaat.
• De Ontvanger: Dit zijn de "oren" die de echo opvangen.

Je kunt deze blokken op verschillende manieren组合eren. Wil je een golf door een simpele zee? Gebruik het 'akoestische' blok. Wil je door een stevige rotslaag met verschillende richtingen? Dan wissel je naar het 'elastisch' of 'anisotroop' blok. Het systeem past zich automatisch aan, zonder dat je de hele code hoeft te herschrijven.

3. De "Koffiezetapparaat" voor computers (Snelheid)

Een ander groot probleem is snelheid. Het berekenen van deze golven kost enorm veel tijd.

• Oude methode: Je zet één koffiezetapparaat aan, wacht tot het klaar is, en doet dan de volgende kop.
• SWEEP-methode: SWEEP heeft een apparaat dat 100 koppen koffie tegelijk zet. Het kan duizenden simulaties tegelijk draaien op krachtige computers (zoals GPU's).

Dit betekent dat wat vroeger een week duurde, nu misschien in een uur klaar is.

4. Waarom is dit belangrijk?

Met SWEEP kunnen wetenschappers:

• Beter zoeken naar energie: Ze vinden olie en gas sneller en met minder boringen.
• Veiligere gebouwen: Ze kunnen zien waar de grond instabiel is voor aardbevingen.
• Nieuwe ontdekkingen: Omdat het zo makkelijk is om te experimenteren, kunnen ze nieuwe manieren bedenken om de aarde te "scannen" die voorheen te moeilijk waren om te berekenen.

Kortom:
SWEEP is de tool die de complexe wiskunde van aardtrillingen verandert in een gebruiksvriendelijk, snel en flexibel systeem. Het is alsof je van het handmatig bouwen van een auto bent gegaan naar het besturen van een zelfrijdende fabriek die elke soort auto kan bouwen die je maar wilt. Het maakt het mogelijk om de geheimen van de aarde dieper en sneller te ontrafelen dan ooit tevoren.

Technische samenvatting

Titel

SWEEP: Een Unificeerend Solver Framework voor Differentieerbare Golfphysica

1. Het Probleem

Binnen de seismische exploratie zijn het bouwen van snelheidsmodellen en het maken van afbeeldingen van de ondergrond gebaseerd op het oplossen van golfvergelijkingen. Voor inverse problemen (zoals Full Waveform Inversion - FWI) is het ook noodzakelijk om de bijbehorende geadjungeerde golfvergelijkingen op te lossen.

• Huidige uitdagingen: Traditioneel vereist dit het handmatig afleiden en implementeren van deze geadjungeerde vergelijkingen en gradiëntexpressies via de adjoint-state method. Dit proces is complex, vatbaar voor fouten (bijv. inconsistenties in tijd, randvoorwaarden) en moeilijk te onderhouden.
• Automatische Differentiatie (AD) dilemma: AD biedt een alternatief om gradiënten automatisch te berekenen, maar er bestaat een afweging tussen implementatiegemak en rekenkracht:
  • PyTorch: Hoog ontwikkelgemak, maar vaak lagere runtime-prestaties.
  • CUDA (handgeschreven): Hoge prestaties, maar complexe ontwikkeling en minder flexibiliteit.
  • Er is behoefte aan een framework dat de flexibiliteit van AD combineert met de prestaties van handgeschreven code.

2. Methodologie

SWEEP (Seismic Wave Equation Exploration Platform) is een unificeerend en uitbreidbaar bibliotheekframework dat is ontwikkeld om deze uitdagingen aan te pakken.

• Architectuur en Backends:
  • SWEEP ondersteunt zowel PyTorch als JAX als backends.
  • Door gebruik te maken van JAX (met Just-In-Time compilatie via XLA), bereikt het framework prestaties die vergelijkbaar zijn met handgeschreven CUDA-code, terwijl het de ontwikkelgemak van Python behoudt.
  • De code is modulair opgebouwd met drie kerncomponenten:
    1. Propagator: Verantwoordelijk voor het tijdsstappenproces, geheugenallocatie en updates van het golfveld.
    2. Source: Beheert de injectie van golven (flexibele configuraties).
    3. Receiver: Registreert golfvelden op specifieke locaties.
• Abstraktie van Tijdsstappen:
  • De tijdsstap-implementation is geabstraheerd in een structuur die lijkt op een Recurrent Neural Network (RNN). Dit is een conceptuele abstractie; de propagatie gebeurt niet via neurale netwerken, maar deze vorm biedt een uniforme interface voor verschillende golfvergelijkingen.
  • Gebruikers hoeven alleen de update-beweging voor één tijdstap te definiëren; het framework infereert automatisch modelvormen en geheugen.
• Differentieerbaarheid:
  • Alle operaties zijn volledig differentieerbaar, wat naadloze implementatie van gradiënt-gebaseerde optimalisatie (FWI, LSRTM) mogelijk maakt zonder handmatige afleiding van adjoint-vergelijkingen.
• Schalbaarheid:
  • Ondersteuning voor batch-modeling: Meerdere schoten (sources) worden simultaan berekend om kernel-launch overhead te verminderen.
  • High-Performance Computing (HPC): Naadloze uitbreiding naar meerdere GPU's/TPU's via DistributedDataParallel (PyTorch) of pmap (JAX).

3. Belangrijkste Bijdragen

• Unificatie van Golfvergelijkingen: Het framework ondersteunt een breed scala aan golfvergelijkingen, waaronder:
  • Akoestisch (1e en 2e orde), elastisch, visko-akoestisch.
  • Anisotrope media: VTI (Vertical Transverse Isotropy) en TTI (Tilted Transverse Isotropy), inclusief ontkoppelde versies om artefacten te verminderen.
  • Gekoppelde systemen: Akoestisch-elastisch gekoppelde golfvergelijkingen.
  • Born-approximaties voor Least-Squares Reverse Time Migration (LSRTM).
• Plug-and-Play Architectuur: Gebruikers kunnen eenvoudig aangepaste verliesfuncties, neurale netwerken voor snelheidsrepresentatie en multi-schaal strategieën integreren.
• Codekwaliteit en Flexibiliteit: Een schone, beknopte codebase die complexe low-level operaties (zoals Laplace-operatoren en Fourier-transformaties) verbergt, waardoor onderzoekers zich kunnen focussen op hoog-niveau workflow-ontwerp.

4. Resultaten en Demonstraties

Het paper presenteert diverse numerieke experimenten die de functionaliteit van SWEEP aantonen:

• Golfveldvisualisatie: Succesvolle simulaties van golfvelden voor verschillende vergelijkingen (akoestisch, elastisch, VTI, TTI, visko-akoestisch).
  • VTI/TTI: Demonstratie van het verminderen van artefacten (zoals het "diamond pattern" in qP-golven) door gebruik van ontkoppelde vergelijkingen.
  • Visko-akoestisch: Visualisatie van faseverschuiving en amplitude-demping door absorptie (Q-factor).
• Gekoppelde Systemen: Simulatie van akoestisch-elastisch gekoppelde golven in een meerlagig model, waarbij de P-component correct wordt gegenereerd uit de som van de normaalspanningen.
• Pseudo-elastisch: Vergelijking tussen elastisch, akoestisch en pseudo-elastisch, waarbij het pseudo-elastisch model dispersie in lagen met lage S-golf snelheid effectief simuleert.
• LSRTM: Implementatie van zowel akoestische als elastische LSRTM workflows, waarbij de perturbed (verstoorde) golfvelden correct worden berekend voor inversie.

5. Betekenis en Impact

SWEEP vertegenwoordigt een significante doorbraak in computationele geofysica door:

1. Complexiteitsreductie: Het elimineert de noodzaak voor handmatige afleiding van geadjungeerde vergelijkingen, wat de kans op fouten verkleint en de ontwikkelingstijd verkort.
2. Prestaties: Het overbrugt de kloof tussen snelle prototyping (PyTorch) en high-performance computing (CUDA/JAX), waardoor het geschikt is voor grote schaal problemen.
3. Toekomstbestendigheid: Door de naadloze integratie met neurale netwerken en differentieerbare programmering, positioneert SWEEP zich als een ideaal platform voor de opkomende velden van machine learning in seismische inversie en data-gedreven geofysica.

Samenvattend biedt SWEEP een krachtig, flexibel en unificeerend platform dat zowel traditionele als nieuwe benaderingen voor seismische inverse problemen ondersteunt, met name gericht op het versnellen van wetenschappelijke ontdekkingen door het automatiseren van complexe fysica-implementaties.