GeneFlow Checkpoints

Checkpoints for GeneFlow, a rectified-flow model that generates H&E histology image patches conditioned on gene expression (Xenium spatial transcriptomics). Companion to the NeurIPS 2025 paper.

Code: https://github.com/wangmengbo/GeneFlow

Checkpoints

Folder	Sample	Model type	Epoch	Val loss	SSIM	FID	UNI2-h FID
P1_single/	XeniumP1	single-cell	80	0.0300	0.1991	12.25	25.44
C1_single/	XeniumC1	single-cell	42	0.0238	0.2584	12.40	26.32
C1_multi/	XeniumC1	multi-cell (patch)	80	0.0330	0.2884	39.21	91.06

All three checkpoints match or exceed the metrics reported in the paper for their corresponding configuration (Table 2 / Table 5).

Usage

Load with the training/inference code in the GitHub repo (rectified/rectified_main.py --only_inference or rectified/rectified_evaluate.py), pointing --model_path at the downloaded .pt file.