Georgios Batzolis

Research Scientist at Devotion AI Labs

prof_pic.png

I’m a Research Scientist at Devotion AI Labs. Before that, I spent several years at Cambridge, completing my PhD with Prof. Carola-Bibiane Schönlieb in the Department of Applied Mathematics and Theoretical Physics (DAMTP), then working with Prof. Mark Girolami as a Postdoctoral Research Associate in the Department of Engineering.

My current research focuses on diffusion language models (DLMs) and closing the capability gap with autoregressive language models, commonly known as LLMs. In CoBit (Continuous Bitstream Diffusion), we represent text as bitstreams and model it with continuous diffusion. At the time of our arXiv submission, our GenPPL–entropy and GSM8K results suggested an advance in the state of the art among continuous DLMs on those evaluations. The field has moved on rapidly since, so this should be read as a historical result.

The collective progress of the DLM research community makes me optimistic that the capability gap could close sooner than we expect. Matching strong autoregressive models across a broad range of tasks remains an open challenge, but the momentum is exciting. I look forward to sharing new work soon and contributing to that progress.

As the capability gap narrows, I expect speed and inference cost to become increasingly decisive. I’m excited by the prospect of DLMs competing as standalone models in production, and by their potential as parallel draft models within speculative decoding, proposing tokens for an autoregressive model to verify. Both directions offer opportunities for faster generation, provided the gains hold up in practical workloads without compromising quality.

My interest in DLMs also connects to my earlier work on the geometry of continuous data. There, I have studied how trained diffusion models and normalising flows can define Riemannian metrics that reveal intrinsic structure and give meaningful notions of distance and interpolation. I’m particularly interested in how this perspective might extend to discrete distributions, including text and DNA sequences. What notions of geometry are useful for these data, and how can they guide the choice of representations and the design of more efficient diffusion models?

At NeurIPS 2026: I will be presenting CoBit in Sydney and would be happy to meet others working on DLMs. If you will be there, feel free to get in touch.

Research interests

  • Language and multimodal generation: designing unified generative models that capture relationships across modalities and support both joint and conditional generation. Our multimodal CoBit work explores one approach.
  • Fast, capable diffusion models: accelerating inference through caching, parallel generation, and distillation, while understanding the trade-offs between speed and quality.
  • Information theory and optimal sampling: understanding what makes denoising difficult and designing sampling algorithms with guarantees on accuracy and computational cost.
  • Learning the geometry of data: constructing Riemannian metrics from trained generative models and extending geometric methods from continuous data to text and biological sequences.

If any of these topics interest you, feel free to reach out for a chat at g.batz97@gmail.com or Georgios@devotionlabs.ai.

Publications

NeurIPS 2026
CoBit: Language Modeling with Bitstream Diffusion
Georgios Batzolis, Mark Girolami, Luca Ambrogioni
Accepted at NeurIPS 2026; also appeared at ICML SPIGM 2026 and the COLM 2026 workshop on Non-Autoregressive Language Models for Fast & Flexible Text Generation.

[PDF] [Code]
NeurIPS 2026
Compatible Likelihoods for Flow Matching on Manifolds
Lucas Ng, Georgios Batzolis, Mark Girolami
Accepted at NeurIPS 2026.

[arXiv] [PDF]
ICML 2025
Willem Diepeveen*, Georgios Batzolis*, Zakhar Shumaylov, Carola-Bibiane Schönlieb
ICML 2025

[Published Version] [Code]
ICML 2024
FoDS 2024
CAFLOW: conditional autoregressive flows
Georgios Batzolis, Marcello Carioni, Christian Etmann, Soroosh Afyouni, Zoe Kourtzi, Carola-Bibiane Schönlieb
Foundations of Data Science, AIMS, 2024

DOI: 10.3934/fods.2024028
[arXiv] [Official Code]
AAAI 2022
How to distribute data across tasks for meta-learning?
Alexandru Cioba, Michael Bromberg, Qian Wang, Ritwik Niyogi, Georgios Batzolis, Jezabel Garcia, Da-shan Shiu, Alberto Bernacchia
AAAI Conference on Artificial Intelligence, 2022

[PDF]

arXiv Preprints

arXiv 2026
Multimodal Co-Generation With Continuous Bitstream Diffusion
Georgios Batzolis, Georgios Kontos, Bruno Trentini, Pranam Chatterjee, Micha Livne, Mark Girolami, Luca Ambrogioni
Submitted to ICLR 2027.

[arXiv] [PDF]
arXiv 2026
Beyond Unimodal Bases: Pullback Geometry for Multimodal Data
Honglei Brinkmann, Lucas Ng, Georgios Batzolis, Mark Girolami, Carola-Bibiane Schönlieb, Willem Diepeveen
Submitted to ICLR 2027.

[arXiv] [PDF]
arXiv 2026
From Atoms to Entropy: Optimal Noise Allocation for Diffusion Training in the Convex Regime
Luca Ambrogioni, Giulio Franzese, Alberto Foresti, Gabriel Raya, Bac Nguyen, Georgios Batzolis, Yuhta Takida, Naoki Murata, Chieh-Hsin Lai, Yuki Mitsufuji
Submitted to AISTATS.

[arXiv] [PDF]
arXiv 2026
Noise Scheduling as Information-Guided Allocation in Diffusion Training
Gabriel Raya, Bac Nguyen, Georgios Batzolis, Yuhta Takida, Dejan Stancevic, Naoki Murata, Chieh-Hsin Lai, Yuki Mitsufuji, Luca Ambrogioni
Under review; also appeared at ICML SPIGM 2026.

[arXiv]
arXiv 2023
Georgios Batzolis*, Jan Stanczuk*, Carola-Bibiane Schönlieb
arXiv preprint arXiv:2304.12141

[PDF]
arXiv 2022
Non-uniform diffusion models
Georgios Batzolis, Jan Stanczuk, Carola-Bibiane Schönlieb
arXiv preprint arXiv:2207.09786

[PDF]
arXiv 2021
Conditional image generation with score-based diffusion models
Georgios Batzolis, Jan Stanczuk, Carola-Bibiane Schönlieb, Christian Etmann
arXiv preprint arXiv:2111.13606

[PDF]

* Equal contribution.

Contact

Email: g.batz97@gmail.com or Georgios@devotionlabs.ai. You can also find me on Google Scholar or LinkedIn.