We propose an approach to build and sample surrogate stochastic models leveraging state-of-the-art score-based diffusion approaches, either abstracting a known stochastic process or learning directly the model from data. In particular, we propose a method for efficient conditional sampling from such surrogate models, enforcing logical and consistency constraints on generated samples in a soft fashion. As a preliminary case study, we consider a surrogate SIR model, in both its ergodic and non-ergodic formulations. Using the aforementioned method, we are able to sample trajectories from such models that exhibit desirable features having low probability in the unconstrained models, allowing us to explore epidemiologically relevant scenarios. Although the proposed approach is still a work-in-progress, it has significant potential for applications in epidemiology and other fields. The method is also efficient in the sense that retraining is not needed to generate samples satisfying different constraints.

Model Abstraction and Conditional Sampling with Score-Based Diffusion Models

Bortolussi, Luca
Primo
;
Cairoli, Francesca
Secondo
;
Giacomarra, Francesco
Penultimo
;
Scassola, Davide
Ultimo
2023-01-01

Abstract

We propose an approach to build and sample surrogate stochastic models leveraging state-of-the-art score-based diffusion approaches, either abstracting a known stochastic process or learning directly the model from data. In particular, we propose a method for efficient conditional sampling from such surrogate models, enforcing logical and consistency constraints on generated samples in a soft fashion. As a preliminary case study, we consider a surrogate SIR model, in both its ergodic and non-ergodic formulations. Using the aforementioned method, we are able to sample trajectories from such models that exhibit desirable features having low probability in the unconstrained models, allowing us to explore epidemiologically relevant scenarios. Although the proposed approach is still a work-in-progress, it has significant potential for applications in epidemiology and other fields. The method is also efficient in the sense that retraining is not needed to generate samples satisfying different constraints.
File in questo prodotto:
File Dimensione Formato  
QEST23_ConditionalSampling.pdf

Accesso chiuso

Tipologia: Documento in Versione Editoriale
Licenza: Copyright Editore
Dimensione 626.87 kB
Formato Adobe PDF
626.87 kB Adobe PDF   Visualizza/Apri   Richiedi una copia
Pubblicazioni consigliate

I documenti in IRIS sono protetti da copyright e tutti i diritti sono riservati, salvo diversa indicazione.

Utilizza questo identificativo per citare o creare un link a questo documento: https://hdl.handle.net/11368/3099283
Citazioni
  • ???jsp.display-item.citation.pmc??? ND
  • Scopus 0
  • ???jsp.display-item.citation.isi??? 0
social impact