STREAM: Stochastic Riemannian Flow Matching with Anisotropic Decoder for Digital Histopathology Image Generation
This work addresses the issue of “condition collapse” in digital histopathology image generation, which arises when using pretrained vision foundation models as conditioning signals. The authors propose a Riemannian flow matching–based generative framework that models normalized patch-token features as latent variables on the unit hypersphere. Their approach introduces, for the first time, a spherical-aware bridging stochastic perturbation mechanism and an anisotropic decoder, integrated within a Diffusion Transformer (DiT) architecture. Generation is further guided by a decoding strategy based on the directional energy of the Jacobian matrix of the velocity field. Evaluated on breast and colorectal cancer datasets, the method achieves state-of-the-art performance in both reconstruction fidelity and generation diversity, significantly outperforming existing approaches.