Skip to main content

Vision-integrated spatial transcriptomics SVC reconstruction and Sim2Real-ST benchmarking.

Project description

REVISE

PyPI Documentation Status License: MIT

REVISE (REconstruction via Vision-integrated Spatial Estimation) reconstructs Spatially-inferred Virtual Cells (SVCs) from spatial transcriptomics data by integrating ST measurements, histological images, and matched single-cell RNA-seq references.

The current codebase is organized around one configuration-driven engine, REVISEPipeline, and two user-facing modes:

Mode Goal Main entry points Primary outputs
benchmark Reproduce Sim2Real-ST evaluations across six confounding factors benchmark_main.py, benchmark_main.sh, reproduce/benchmark/*.ipynb metrics_normalized.csv with PCC, SSIM, MSE, and NRMSE
application Reconstruct SVCs and run downstream real-data analysis application_sp_SVC_recon.py, application_sc_SVC_recon.py, reproduce/case/*.ipynb sp_SVC.h5ad, sc_SVC_expr.h5ad, sc_SVC_spatial.h5ad, notebook figures

Documentation: https://revise-svc.readthedocs.io/en/latest/

Dataset and reproduced results: https://zenodo.org/records/17705737

What REVISE Covers

Sim2Real-ST benchmarks six confounding factors across three spatial transcriptomics platform types:

  • Spatially heterogeneous factors: image segmentation artifacts and bin-to-cell assignment errors.
  • Spatially homogeneous factors: spot size, batch effect, gene panel limitation, and gene dropout.

REVISE reconstructs two complementary SVC types:

  • sp-SVC: spatial refinement for hST platforms such as Visium HD.
  • sc-SVC: molecular completion and cell-state refinement for iST/sST platforms such as Xenium and Visium.

Architecture

Modern runs flow through:

  1. revise.framework.REVISEPipeline
  2. revise/revise.yaml profiles and runtime/io overrides
  3. revise.recon.pipeline.UnifiedReconstructionPipeline
  4. backend strategy and plugin registries in revise/backend/

UnifiedReconstructionPipeline owns the fixed lifecycle: input validation, global anchoring, local unit preparation, graph construction, OT problem construction, OT solving, expression update, SVC finalization, and optional benchmark evaluation.

Legacy-style runner classes are kept under revise/backend/runners/ for notebook compatibility and parity checks. New code should prefer REVISEPipeline or the root wrapper scripts.

Installation

Install the package from PyPI:

pip install revise-svc

Optional annotation support:

pip install "revise-svc[annotation]"

Development install:

git clone https://github.com/wuys13/REVISE.git
cd REVISE
pip install -e ".[dev]"

Download Sim2Real-ST benchmark data and real application data from Zenodo, then place them under raw_data/ if you want to reproduce the paper results.

Quick Start

Benchmark Mode

benchmark_main.py runs Sim2Real-ST cases and writes per-gene benchmark metrics. The paper-facing metrics are PCC, SSIM, and MSE; NRMSE is also retained in the CSV for legacy compatibility.

python benchmark_main.py \
  --cf segmentation \
  --raw_data_path raw_data/Sim2Real-ST \
  --sample_name P2CRC/cut_part1 \
  --task segmentation \
  --save_path output/benchmark

Supported --cf values:

  • segmentation
  • bin2cell
  • batch_effect
  • spot_size
  • gene_panel
  • gene_dropout

Use the merged launcher for multi-case reproduction:

bash benchmark_main.sh

Application Mode

Application scripts default to output/ subdirectories so notebook analysis can load the reconstructed SVC files directly.

For hST / Visium HD style sp-SVC reconstruction:

python application_sp_SVC_recon.py \
  --raw_data_path raw_data/Real_application \
  --sample_name P1CRC \
  --st_file HD.h5ad \
  --sc_ref_file adata_sc_all_reanno.h5ad

Default published notebook output:

output/sp_SVC_case/<sample_name>/sp_SVC.h5ad

For iST / Xenium style sc-SVC reconstruction:

python application_sc_SVC_recon.py \
  --sample_name P2CRC \
  --data_type Xenium \
  --raw_data_path raw_data/Real_application \
  --sc_ref_file adata_sc_all_reanno.h5ad \
  --select_ct T

Default published notebook outputs:

output/sc_SVC_case/<sample_name>_<data_type>/<select_ct>/sc_SVC_expr.h5ad
output/sc_SVC_case/<sample_name>_<data_type>/<select_ct>/sc_SVC_spatial.h5ad

Python API

from revise.framework import REVISEPipeline

pipeline = REVISEPipeline(config_path="revise/revise.yaml")
svc = pipeline.run(
    profile="application_sc",
    runtime_overrides={"platform": "iST", "confounding": "segmentation"},
    io_overrides={
        "data_root": "raw_data/Real_application",
        "output_root": "output/sc_SVC_case",
        "sample_name": "P2CRC",
        "st_file": "Xenium.h5ad",
        "sc_ref_file": "adata_sc_all_reanno.h5ad",
        "patient_key": "Patient",
    },
    set_overrides=["sc.select_ct=T"],
)

Notebooks

Area Files Purpose
Benchmark reproduce/benchmark/seg_benchmark.ipynb, spot_benchmark.ipynb, batch_benchmark.ipynb, imputation_benchmark.ipynb Inspect Sim2Real-ST benchmark outputs and PCC/SSIM/MSE trends
Application reconstruction reproduce/case/*_recon.ipynb, reproduce/case/sp_SVC_case.ipynb Rebuild paper application cases from raw inputs
Application analysis reproduce/case/*_analysis.ipynb, application_sc_SVC_analysis_case.ipynb Analyze SVC outputs, cell states, pathways, spatial patterns, and downstream figures
SMI case SMI/CosMx-SMI-REVISE_spSVC.ipynb CosMx SMI sp-SVC application example

ReadTheDocs links the maintained benchmark and case notebooks through docs/benchmark/ and docs/case/.

Repository Layout

  • revise/framework.py: public REVISEPipeline entry point.
  • revise/revise.yaml: routing profiles and default configuration.
  • revise/recon/: unified pipeline context and lifecycle orchestration.
  • revise/backend/: strategies, platform adapters, plugin registries, kernels, and lower-level operations.
  • revise/config/: config loader and internal runner configuration contracts.
  • revise/analysis/: benchmark metric and downstream analysis helpers.
  • reproduce/benchmark/: benchmark launchers and analysis notebooks.
  • reproduce/case/: real application reconstruction and analysis notebooks.
  • docs/: ReadTheDocs / Sphinx source.

License

REVISE is released under the MIT License.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

revise_svc-0.0.31.tar.gz (93.8 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

revise_svc-0.0.31-py3-none-any.whl (117.2 kB view details)

Uploaded Python 3

File details

Details for the file revise_svc-0.0.31.tar.gz.

File metadata

  • Download URL: revise_svc-0.0.31.tar.gz
  • Upload date:
  • Size: 93.8 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.10.15

File hashes

Hashes for revise_svc-0.0.31.tar.gz
Algorithm Hash digest
SHA256 9b409a3a8feadcfa4404f647a602c4dec5c84644c143cecaa4ffd61a0a659f07
MD5 948834105409021568a95dce2c6ab981
BLAKE2b-256 1be133c2128b399c2dc50fbf4c3fe8d34448a31bc863a413428d421d640a3e13

See more details on using hashes here.

File details

Details for the file revise_svc-0.0.31-py3-none-any.whl.

File metadata

  • Download URL: revise_svc-0.0.31-py3-none-any.whl
  • Upload date:
  • Size: 117.2 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.10.15

File hashes

Hashes for revise_svc-0.0.31-py3-none-any.whl
Algorithm Hash digest
SHA256 30dff6c09c5b22de3642c0a12003627d949ff0ad7b28258cb567e838a110b69e
MD5 2a09a476d0f4cdf2d510e121d15aa150
BLAKE2b-256 88425fadcb50fa8b81247e7fb9a8b86875b435e64675f0352e0509ff77a8a0ac

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page