2026

1. ICLR

   CREPE: Controlling Diffusion with Replica Exchange   Monte Carlo and Generative Models

   Jiajun He, Paul Jeha, Peter Potaptchik, and 5 more authors

   In International Conference on Learning Representations

   arXiv Bib Code

   @inproceedings{he2025crepe,
     title = {CREPE: Controlling Diffusion with Replica Exchange},
     author = {He, Jiajun and Jeha, Paul and Potaptchik, Peter and Zhang, Leo and Hern{\'a}ndez-Lobato, Jos{\'e} Miguel and Du, Yuanqi and Syed, Saifuddin and Vargas, Francisco},
     booktitle = {International Conference on Learning Representations},
     year = {2026},
   }

2. ICLR

   RNE: a plug-and-play framework for diffusion density estimation and inference-time control   Monte Carlo and Generative Models

   Jiajun He, José Miguel Hernández-Lobato, Yuanqi Du, and 1 more author

   In International Conference on Learning Representations

   arXiv Bib

   @inproceedings{he2025rne,
     title = {RNE: a plug-and-play framework for diffusion density estimation and inference-time control},
     author = {He, Jiajun and Hern{\'a}ndez-Lobato, Jos{\'e} Miguel and Du, Yuanqi and Vargas, Francisco},
     booktitle = {International Conference on Learning Representations},
     year = {2026},
   }

3. ICLR

   Accelerated Parallel Tempering via Neural Transports   Monte Carlo and Sampling

   Leo Zhang^*, Peter Potaptchik^*, Jiajun He^*, and 5 more authors

   In International Conference on Learning Representations

   arXiv Bib Slides

   @inproceedings{zhang2025accelerated,
     title = {Accelerated Parallel Tempering via Neural Transports},
     author = {Zhang, Leo and Potaptchik, Peter and He, Jiajun and Du, Yuanqi and Doucet, Arnaud and Vargas, Francisco and Dau, Hai-Dang and Syed, Saifuddin},
     booktitle = {International Conference on Learning Representations},
     year = {2026},
   }

2025

1. NeurIPS

   FEAT: Free energy Estimators with Adaptive Transport   Monte Carlo and Generative Models

   Jiajun He^*, Yuanqi Du^*, Francisco Vargas, and 4 more authors

   In Advances in Neural Information Processing Systems

   arXiv Bib Code

   @inproceedings{he2025feat,
     title = {FEAT: Free energy Estimators with Adaptive Transport},
     author = {He, Jiajun and Du, Yuanqi and Vargas, Francisco and Wang, Yuanqing and Gomes, Carla P and Hern{\'a}ndez-Lobato, Jos{\'e} Miguel and Vanden-Eijnden, Eric},
     booktitle = {Advances in Neural Information Processing Systems},
     year = {2025},
   }

2. ICML

   Progressive Tempering Sampler with Diffusion   Monte Carlo and Sampling

   Severi Rissanen^*, RuiKang OuYang^*, Jiajun He, and 4 more authors

   In International Conference on Machine Learning

   arXiv Bib Code

   @inproceedings{rissanen2025progressive,
     title = {Progressive Tempering Sampler with Diffusion},
     author = {Rissanen, Severi and OuYang, RuiKang and He, Jiajun and Chen, Wenlin and Heinonen, Markus and Solin, Arno and Hern{\'a}ndez-Lobato, Jos{\'e} Miguel},
     year = {2025},
     booktitle = {International Conference on Machine Learning},
   }

3. FPI @ ICLR

   No Trick, No Treat: Pursuits and Challenges Towards Simulation-free Training of Neural Samplers   Monte Carlo and Sampling

   Jiajun He^*, Yuanqi Du^*, Francisco Vargas, and 5 more authors

   In Workshop on Frontiers in Probabilistic Inference: Sampling Meets Learning @ ICLR 2025

   arXiv Bib Poster Slides

   @inproceedings{he2025no,
     title = {No Trick, No Treat: Pursuits and Challenges Towards Simulation-free Training of Neural Samplers},
     author = {He, Jiajun and Du, Yuanqi and Vargas, Francisco and Zhang, Dinghuai and Padhy, Shreyas and OuYang, RuiKang and Gomes, Carla and Hern{\'a}ndez-Lobato, Jos{\'e} Miguel},
     year = {2025},
     booktitle = {Workshop on Frontiers in Probabilistic Inference: Sampling Meets Learning @ ICLR 2025},
   }

4. AISTATS

   Training Neural Samplers with Reverse Diffusive KL Divergence   Monte Carlo and Sampling

   Jiajun He^*, Wenlin Chen^*, Mingtian Zhang^*, and 2 more authors

   In International Conference on Artificial Intelligence and Statistics

   arXiv Bib Code Poster

   @inproceedings{he2024training,
     title = {Training Neural Samplers with Reverse Diffusive KL Divergence},
     author = {He, Jiajun and Chen, Wenlin and Zhang, Mingtian and Barber, David and Hern{\'a}ndez-Lobato, Jos{\'e} Miguel},
     year = {2025},
     booktitle = {International Conference on Artificial Intelligence and Statistics},
   }

2024

1. NeurIPS

   Accelerating Relative Entropy Coding with Space Partitioning   Information Theory and Compression

   Jiajun He, Gergely Flamich, and José Miguel Hernández-Lobato

   In Advances in Neural Information Processing Systems

   arXiv Bib Poster

   @inproceedings{he2024accelerating,
     title = {Accelerating Relative Entropy Coding with Space Partitioning},
     author = {He, Jiajun and Flamich, Gergely and Hern{\'a}ndez-Lobato, Jos{\'e} Miguel},
     year = {2024},
     booktitle = {Advances in Neural Information Processing Systems},
   }

2. ICLR

   RECOMBINER: Robust and Enhanced Compression with Bayesian Implicit Neural Representations   Information Theory and Compression

   Jiajun He^*, Gergely Flamich^*, Zongyu Guo, and 1 more author

   In International Conference on Learning Representations

   arXiv Bib Code Poster Slides

   @inproceedings{he2024recombiner,
     title = {RECOMBINER: Robust and Enhanced Compression with Bayesian Implicit Neural Representations},
     author = {He, Jiajun and Flamich, Gergely and Guo, Zongyu and Hern{\'a}ndez-Lobato, Jos{\'e} Miguel},
     booktitle = {International Conference on Learning Representations},
     year = {2024},
   }

2023

1. NeurIPS

   Compression with bayesian implicit neural representations   Information Theory and Compression

   Zongyu Guo^*, Gergely Flamich^*, Jiajun He, and 2 more authors

   In Advances in Neural Information Processing Systems

   Spotlight at NeurIPS 2023

   arXiv Bib Code Poster Slides

   @inproceedings{guo2023compression,
     title = {Compression with bayesian implicit neural representations},
     author = {Guo, Zongyu and Flamich, Gergely and He, Jiajun and Chen, Zhibo and Hern{\'a}ndez-Lobato, Jos{\'e} Miguel},
     booktitle = {Advances in Neural Information Processing Systems},
     year = {2023},
   }

2. Mphil Thesis

   Data Compression with Variational Implicit Neural Representations   Information Theory and Compression

   Jiajun He

   University of Cambridge

   PDF