# End-to-end Sinkhorn Autoencoder with Noise Generator

arXiv: https://arxiv.org/abs/2006.06704 · HTML: https://mlsanity.com/paper/2006.06704/

## Key facts

- Cited by 17 papers in the map (data through 14 Oct 2026).
- First posted on arXiv 11 Jun 2020.

## Built on by (most cited first, 17 of 17)

- [A Living Review of Machine Learning for Particle Physics](https://mlsanity.com/paper/2102.02770/) (2021)
- [Deep Generative Models for Detector Signature Simulation: A Taxonomic Review](https://mlsanity.com/paper/2312.09597/) (2023)
- [Latent Space Refinement for Deep Generative Models](https://mlsanity.com/paper/2106.00792/) (2021)
- [Selectively increasing the diversity of GAN-generated samples](https://mlsanity.com/paper/2207.01561/) (2022)
- [Neural Sinkhorn Gradient Flow](https://mlsanity.com/paper/2401.14069/) (2024)
- [Generative Diffusion Models for Fast Simulations of Particle Collisions at CERN](https://mlsanity.com/paper/2406.03233/) (2024)
- [Multiband VAE: Latent Space Alignment for Knowledge Consolidation in Continual Learning](https://mlsanity.com/paper/2106.12196/) (2021)
- [Machine Learning methods for simulating particle response in the Zero Degree Calorimeter at the ALICE experiment, CERN](https://mlsanity.com/paper/2306.13606/) (2023)
- [Guided Quantum Compression for High Dimensional Data Classification](https://mlsanity.com/paper/2402.09524/) (2024)
- [Inverse Autoregressive Flows for Zero Degree Calorimeter fast simulation](https://mlsanity.com/paper/2512.20346/) (2025)
- [Even Faster Simulations with Flow Matching: A Study of Zero Degree Calorimeter Responses](https://mlsanity.com/paper/2507.18811/) (2025)
- [Particle physics DL-simulation with control over generated data properties](https://mlsanity.com/paper/2405.14049/) (2024)
- [Deep Generative Models for Proton Zero Degree Calorimeter Simulations in ALICE, CERN](https://mlsanity.com/paper/2406.03263/) (2024)
- [SoK: Can Synthetic Images Replace Real Data? A Survey of Utility and Privacy of Synthetic Image Generation](https://mlsanity.com/paper/2506.19360/) (2025)
- [Adapt & Align: Continual Learning with Generative Models Latent Space Alignment](https://mlsanity.com/paper/2312.13699/) (2023)
- [ExpertSim: Fast Particle Detector Simulation Using Mixture-of-Generative-Experts](https://mlsanity.com/paper/2508.20991/) (2025)
- [Fine-tuned Normalizing Flows for ALICE Zero Degree Calorimeter Fast Simulation](https://mlsanity.com/paper/2608.12795/) (2026)

## Abstract (arXiv)

In this work, we propose a novel end-to-end sinkhorn autoencoder with noise generator for efficient data collection simulation. Simulating processes that aim at collecting experimental data is crucial for multiple real-life applications, including nuclear medicine, astronomy and high energy physics. Contemporary methods, such as Monte Carlo algorithms, provide high-fidelity results at a price of high computational cost. Multiple attempts are taken to reduce this burden, e.g. using generative approaches based on Generative Adversarial Networks or Variational Autoencoders. Although such methods are much faster, they are often unstable in training and do not allow sampling from an entire data distribution. To address these shortcomings, we introduce a novel method dubbed end-to-end Sinkhorn Autoencoder, that leverages sinkhorn algorithm to explicitly align distribution of encoded real data examples and generated noise. More precisely, we extend autoencoder architecture by adding a deterministic neural network trained to map noise from a known distribution onto autoencoder latent space representing data distribution. We optimise the entire model jointly. Our method outperforms competing approaches on a challenging dataset of simulation data from Zero Degree Calorimeters of ALICE experiment in LHC. as well as standard benchmarks, such as MNIST and CelebA.