Kirill Korolev

Research scientist in HDI Lab at HSE University, Moscow.

About

M.Sc. in Mathematics of Machine Learning and B.Sc. in Applied Mathematics and Computer Science from HSE University, Moscow. Machine learning researcher working at the intersection of generative modeling, reinforcement learning, and probabilistic inference. My research focuses on developing scalable algorithms for sampling, generative modeling, with recent work on entropy-regularized reinforcement learning and diffusion-based samplers.

I am broadly interested in generative models, reinforcement learning, diffusion models, Monte Carlo methods.

Education

Publications

  1. Submissions at ICLR 2027 and AISTATS 2027 under review
  2. Numerical Integration on Bounded Domains via Learned Reflected Diffusions Kirill Korolev*, Artur Goldman, Timofei Gritsaev, Tigran Ramazyan, Nikita Morozov Oral at NeurIPS 2026 STODY workshop Source code
    Visualization of reflected diffusion
  3. Stop the Sampler! Classifier-Based Adaptive Stopping for Sampling Kernels Kirill Korolev*, Nikita Morozov, Stepan Pavlenko, Esmeralda S. Whitammer, Sergey Samsonov Accepted at NeurIPS 2026 main conference and at ICML 2026 SPIGM workshop Source code
    Visualization of the state space

Teaching