2025

1. arXiv

   FEAT: Free energy Estimators with Adaptive Transport   Monte Carlo

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

   arXiv Bib

   @unpublished{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},
     year = {2025},
   }

2. arXiv

   Accelerated Parallel Tempering via Neural Transports   Monte Carlo

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

   arXiv
3. ICML

   Progressive Tempering Sampler with Diffusion   Monte Carlo

   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},
   }

4. FPI @ ICLR

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

   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},
   }

5. AISTATS

   Training Neural Samplers with Reverse Diffusive KL Divergence   Monte Carlo

   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   coding

   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   coding

   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   coding

   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   coding

   Jiajun He

   University of Cambridge

   PDF