Jiajun He

PhD student in machine learning,
University of Cambridge. Curriculum Vitae

Hello there! I’m Jiajun He (何佳峻)! I am a first-year PhD in the Machine Learning Group at the University of Cambridge under the supervision of Prof. José Miguel Hernández-Lobato. My research interests lie in probabilistic models, Bayesian methods, and their applications in generation and compression.

Previously, I completed my MPhil in Machine Learning and Machine Intelligence from University of Cambridge, where I graduated with distinction as the first of the cohort. For my MPhil dissertation, I worked with Prof. José Miguel Hernández-Lobato and Gergely Flamich on data compression with Implicit Neural Representations. Prior to that, I obained an MSc in Bioinformatics at University of Copenhagen in Denmark, and a BSc in Biological Science at Tsinghua University in China.

I am an landscape photographer in my spare time. Feel free to explore my works on Instagram. I will do my best to keep it updated!

Selected Publications [full list]

Training Neural Samplers with Reverse Diffusive KL Divergence

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

2024

arXiv Bib

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

Accelerating Relative Entropy Coding with Space Partitioning

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

In Advances in Neural Information Processing Systems (NeurIPS), 2024

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 (NeurIPS)},
}

RECOMBINER: Robust and Enhanced Compression with Bayesian Implicit Neural Representations

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

In International Conference on Learning Representations (ICLR), 2024

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 (ICLR)},
  year = {2024},
}