DiffPINN: Generative diffusion-initialized physics-informed neural networks for accelerating seismic wavefield representation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Cheng, Shijun, Alkhalifah, Tariq
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913868786171904
author Cheng, Shijun
Alkhalifah, Tariq
author_facet Cheng, Shijun
Alkhalifah, Tariq
contents Physics-informed neural networks (PINNs) offer a powerful framework for seismic wavefield modeling, yet they typically require time-consuming retraining when applied to different velocity models. Moreover, their training can suffer from slow convergence due to the complexity of of the wavefield solution. To address these challenges, we introduce a latent diffusion-based strategy for rapid and effective PINN initialization. First, we train multiple PINNs to represent frequency-domain scattered wavefields for various velocity models, then flatten each trained network's parameters into a one-dimensional vector, creating a comprehensive parameter dataset. Next, we employ an autoencoder to learn latent representations of these parameter vectors, capturing essential patterns across diverse PINN's parameters. We then train a conditional diffusion model to store the distribution of these latent vectors, with the corresponding velocity models serving as conditions. Once trained, this diffusion model can generate latent vectors corresponding to new velocity models, which are subsequently decoded by the autoencoder into complete PINN parameters. Experimental results indicate that our method significantly accelerates training and maintains high accuracy across in-distribution and out-of-distribution velocity scenarios.
format Preprint
id arxiv_https___arxiv_org_abs_2506_00471
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle DiffPINN: Generative diffusion-initialized physics-informed neural networks for accelerating seismic wavefield representation
Cheng, Shijun
Alkhalifah, Tariq
Geophysics
Machine Learning
Computational Physics
Physics-informed neural networks (PINNs) offer a powerful framework for seismic wavefield modeling, yet they typically require time-consuming retraining when applied to different velocity models. Moreover, their training can suffer from slow convergence due to the complexity of of the wavefield solution. To address these challenges, we introduce a latent diffusion-based strategy for rapid and effective PINN initialization. First, we train multiple PINNs to represent frequency-domain scattered wavefields for various velocity models, then flatten each trained network's parameters into a one-dimensional vector, creating a comprehensive parameter dataset. Next, we employ an autoencoder to learn latent representations of these parameter vectors, capturing essential patterns across diverse PINN's parameters. We then train a conditional diffusion model to store the distribution of these latent vectors, with the corresponding velocity models serving as conditions. Once trained, this diffusion model can generate latent vectors corresponding to new velocity models, which are subsequently decoded by the autoencoder into complete PINN parameters. Experimental results indicate that our method significantly accelerates training and maintains high accuracy across in-distribution and out-of-distribution velocity scenarios.
title DiffPINN: Generative diffusion-initialized physics-informed neural networks for accelerating seismic wavefield representation
topic Geophysics
Machine Learning
Computational Physics
url https://arxiv.org/abs/2506.00471