MUMINS: Metadata-conditioned Uncertainty-aware Medical Image Next-state Synthesis

arXiv:2609.17169 · cs.CV, cs.AI · Submitted 2026-09-15 · Read on arXiv

cs.CV, cs.AI

Submitted: 2026-09-15

Updated: 2026-09-15

Comments: Supplementary material to follow in future versions

Code: https://github.com/aolivtous/MUMINS

License: http://creativecommons.org/licenses/by-nc-nd/4.0/

The gist: Forecasting anatomical changes such as tumor growth and neurodegeneration is a challenging generative vision task.

Terminology

Abstract

Forecasting anatomical changes such as tumor growth and neurodegeneration is a challenging generative vision task. Morphological evolution is subtle relative to static anatomy, highly patient-specific, and inherently stochastic. Existing methods struggle with several issues: deterministic networks ignore biological stochasticity, while standard diffusion models require computationally prohibitive multi-pass sampling to quantify uncertainty. We propose MUMINS (Metadata-conditioned Uncertainty-aware Medical Image Next-state Synthesis), an efficient diffusion framework that jointly diffuses a baseline scan and its follow-up residual, summed to synthesize the follow-up scan, while concurrently predicting a spatial uncertainty map, in a single reverse diffusion process. Conditioned on the time interval and relevant metadata, it preserves fine-grained anatomy by dynamically re-injecting the baseline as a soft anchor at every denoising step, and a negative-log-likelihood head learns the uncertainty map to explicitly flag error-prone regions. Designed without organ-specific heuristics, the same architecture is reused across anatomies via separate, dataset-specific retraining. Extensive evaluations demonstrate that dataset-specific retraining of MUMINS matches or outperforms dedicated, domain-specific state-of-the-art methods on lung CT (PNG) and brain MRI (OASIS-3). Project page: https://github.com/aolivtous/MUMINS.

Sources

Related papers