Diffusion-Based Inference for Heterogeneous Multichannel Fluorescence Imaging

Abstract

Fluorescence microscopy is a central tool in biomedical image analysis, but its practical use is often constrained by incomplete and heterogeneous data. This is especially true in rare disease research, where patient cohorts are small, biological material is limited, and complex study designs frequently yield images with missing, degraded, or low-quality channels. We introduce Reliability-Mask-Conditioned DDPM (RM-DDPM), a conditional diffusion model that formulates multichannel image generation, missing-channel imputation, and degraded-channel restoration as a single reliability-guided inference problem. The model is trained on a curated subset of complete, high-quality multichannel images using a self-supervised masking objective to learn rich cross-channel dependencies from the available channels. At inference time, reliable observations condition the reverse diffusion process, allowing the same trained model to solve multiple reconstruction tasks without task-specific retraining. We evaluate RM-DDPM against task-specific baselines on one public and two in-house fluorescence microscopy datasets, comprising a rare-disease preclinical model study and a clinical patient cohort. Across these settings, RM-DDPM matches the specialized baselines in reconstruction quality, hallucinates fewer false structures, and remains more robust as degradation severity increases, while repeated sampling provides per-pixel uncertainty maps that flag unreliable reconstructions.