Mohammed Al-Jaff

PhD student in Machine Learning · KTH Royal Institute of Technology · Stockholm


I am a PhD student in machine learning at KTH Royal Institute of Technology in Stockholm, supervised by Prof. Danica Kragic Jensfelt, Prof. Hossein Azizpour, and Prof. Gustav Eje Henter.

My research focuses on generative modelling and learning dynamics, with an emphasis on how objectives, parameterization, and data structure shape model behavior under finite data and limited capacity.

I am particularly interested in the boundary between memorization and generalization, the emergence of compositional structure, and how modelling choices affect robustness and downstream performance.

generative models flow matching representation learning compositionality anomaly detection learning dynamics
A Non-Adversarial Approach to Idempotent Generative Modelling
ECAI 2025

Loose thoughts, hypotheses, and ongoing ideas.