Conditional Sampling via Wasserstein Autoencoders and Triangular Transport
Fuente:
arXiv
Gespeichert in:
| Hauptverfasser: | , , , , |
|---|---|
| Format: | Preprint |
| Veröffentlicht: |
2026
|
| Schlagworte: | |
| Online-Zugang: | |
| Tags: |
Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
|
| _version_ | 1866910099467927552 |
|---|---|
| author | Al-Jarrah, Mohammad Martino, Michele Yim, Marcus Hosseini, Bamdad Taghvaei, Amirhossein |
| author_facet | Al-Jarrah, Mohammad Martino, Michele Yim, Marcus Hosseini, Bamdad Taghvaei, Amirhossein |
| contents | We present Conditional Wasserstein Autoencoders (CWAEs), a framework for conditional simulation that exploits low-dimensional structure in both the conditioned and the conditioning variables. The key idea is to modify a Wasserstein autoencoder to use a (block-) triangular decoder and impose an appropriate independence assumption on the latent variables. We show that the resulting model gives an autoencoder that can exploit low-dimensional structure while simultaneously the decoder can be used for conditional simulation. We explore various theoretical properties of CWAEs, including their connections to conditional optimal transport (OT) problems. We also present alternative formulations that lead to three architectural variants forming the foundation of our algorithms. We present a series of numerical experiments that demonstrate that our different CWAE variants achieve substantial reductions in approximation error relative to the low-rank ensemble Kalman filter (LREnKF), particularly in problems where the support of the conditional measures is truly low-dimensional. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2604_02644 |
| institution | arXiv |
| publishDate | 2026 |
| record_format | arxiv |
| spellingShingle | Conditional Sampling via Wasserstein Autoencoders and Triangular Transport Al-Jarrah, Mohammad Martino, Michele Yim, Marcus Hosseini, Bamdad Taghvaei, Amirhossein Machine Learning Optimization and Control We present Conditional Wasserstein Autoencoders (CWAEs), a framework for conditional simulation that exploits low-dimensional structure in both the conditioned and the conditioning variables. The key idea is to modify a Wasserstein autoencoder to use a (block-) triangular decoder and impose an appropriate independence assumption on the latent variables. We show that the resulting model gives an autoencoder that can exploit low-dimensional structure while simultaneously the decoder can be used for conditional simulation. We explore various theoretical properties of CWAEs, including their connections to conditional optimal transport (OT) problems. We also present alternative formulations that lead to three architectural variants forming the foundation of our algorithms. We present a series of numerical experiments that demonstrate that our different CWAE variants achieve substantial reductions in approximation error relative to the low-rank ensemble Kalman filter (LREnKF), particularly in problems where the support of the conditional measures is truly low-dimensional. |
| title | Conditional Sampling via Wasserstein Autoencoders and Triangular Transport |
| topic | Machine Learning Optimization and Control |
| url | https://arxiv.org/abs/2604.02644 |