CANCER · Open position
Learning World Models of the Tumor Microenvironment
Can a self-supervised world model (JEPA) predict the future of a coupled Cellular Potts + reaction-diffusion tumour simulation, and skip the expensive solver entirely?
- CANCER
- ML/AI
- CSM
- Supervisor
- V. S. Muniraj
- Host
- University of Amsterdam · AmsterdamUMC
- Level
- MSc
- Contact
- [email protected]
Read the full project description
The tumor microenvironment (TME) is a dynamic ecosystem where tumor cells, immune cells, and stromal cells interact through physical contact and diffusible signals like cytokines and chemokines. Simulating this system typically requires computationally expensive hybrid models that couple a Cellular Potts Model (CPM) for cellular dynamics with reaction-diffusion partial differential equations for chemical signals. These simulations are slow, which limits their use in parameter sweeps, uncertainty quantification, and therapy design. This project asks a simple question: can we learn a fast surrogate that predicts the future of such a system directly from its current state, skipping the expensive solver entirely?
Recent work has shown that deep learning surrogates, including Physics-Informed Neural Networks (PINNs), can approximate these kinds of spatiotemporal systems. Meanwhile, state-of-the-art approaches elsewhere in biology now model tissues as spatially coupled systems whose future states can be predicted and even virtually perturbed, as seen in frameworks such as GraphTME and TMEformer. This project pushes that idea further by training a Joint Embedding Predictive Architecture (JEPA), a self-supervised world model, to predict the evolution of coupled reaction-diffusion and Cellular Potts systems without reconstructing every pixel. Instead of learning in raw pixel space, the model learns a hidden representation where prediction happens in latent space, making it far more efficient and scalable than PINN-style pixel-level reconstruction.
Objectives
- Build a hybrid ground-truth simulator coupling a Cellular Potts Model with reaction-diffusion PDEs, and use it to generate a large, diverse dataset of simulated TME trajectories.
- Train a JEPA-style world model that learns to predict future latent states of the system from current states, with an optional conditioning signal for external perturbations such as drug delivery or cytokine gradients.
- Benchmark the learned world model against PINN surrogates from prior work, evaluating both predictive accuracy and computational speedup over the original solver.
- Explore whether the model enables long-horizon rollouts, uncertainty estimation, and in silico perturbation experiments that are infeasible with the full simulation.
References
- Papapanagiotou, I., et al. "From simulations to surrogates: Neural networks enhancing burn wound healing predictions" (2025).
- Sheraton, M. V., et al. "Emergence of spatio-temporal variations in chemotherapeutic drug efficacy: in-vitro and in-silico 3D tumour spheroid studies." BMC Cancer 20 (2020).
- LeCun, Y. "A path towards autonomous machine intelligence." (2022) — JEPA position paper.
Expectations
Prior experience with physics simulation, graph neural networks, or self-supervised learning is a plus but not required.
Work environment
The student will join a diverse team of researchers from the Informatics Institute (IvI) and AmsterdamUMC, with expertise in AI development, physical simulation, and experimental validation.
These descriptions are condensed summaries of the full project proposals and may contain minor errors or be superseded by later revisions. Please confirm the details with the supervisor before applying.