OCOO-T : A Simple and Scalable Virtual Cell Model for Transcriptional Perturbation Response Prediction

Danning Jiang; Lipeng Lai; Qirun Wang; Yalong Zhao; Zheming An; Zhiwen Yan

OCOO-T : A Simple and Scalable Virtual Cell Model for Transcriptional Perturbation Response Prediction

T0 review · 1 major / 0 minor · reviewed 2026-06-27 · grok-4.3

Challenge this review Re-run · record.json Download PDF Read on arXiv ↗ Review JSON

Pith's one-line read A vanilla Transformer with flow-matching and adaptive normalization predicts single-cell transcriptional responses to perturbations at state-of-the-art accuracy.

desk verdict OCOO-T is a minimalist vanilla Transformer with flow-matching for perturbation prediction that claims SOTA on three benchmarks, but the abstract supplies zero numbers or protocol details to check the claim. read the letter →

arxiv 2606.12838 v2 pith:MVTAGSH3 submitted 2026-06-11 q-bio.QM cs.AIcs.LGq-bio.GN

Danning Jiang , Zhiwen Yan , Qirun Wang , Zheming An , Yalong Zhao , Lipeng Lai This is my paper · ORCID

classification q-bio.QMcs.AIcs.LGq-bio.GN

keywords virtualcelltranscriptionalperturbationflowmatchingtransformersingle-cellomicsresponsedrugdiscoverygeneexpressionprediction

verification ladder T0 review T1 audit T2 compute T3 formal

The pith

A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.

The reading

The paper establishes that a minimalist flow-matching model built on a standard Transformer stack can predict how cells change their gene expression after genetic, chemical, or cytokine perturbations. It does so by treating the response as a continuous denoising process and injecting perturbation type, dosage, and cell identity only through adaptive layer normalization plus in-context tokens. This approach avoids the auxiliary encoders, hierarchical VAEs, or gene-interaction graphs used in prior work. A sympathetic reader would care because simpler architectures could make large-scale virtual-cell simulations practical for drug discovery and regulatory-network inference. Evaluations on Tahoe100M, Replogle, and PBMC data show the model matches or exceeds existing methods while scaling to long expression profiles via patching.

What carries the argument

Vanilla Transformer stack performing flow-matching denoising on continuous gene-expression profiles, conditioned by adaptive layer normalization and in-context tokens.

What would settle it

A new benchmark dataset containing strong, previously unseen gene-regulatory interactions where any method that explicitly encodes those interactions significantly outperforms OCOO-T on held-out perturbations.

Watch

Extended reading notes

Core claim

OCOO-T formulates transcriptional perturbation response prediction as a continuous-time flow-matching denoising task performed by a vanilla Transformer that operates directly on continuous gene-expression vectors; perturbation embeddings, dosage, and cell specificity are supplied solely through adaptive layer normalization and in-context tokens, enabling state-of-the-art accuracy across diverse perturbations and cell types on Tahoe100M, Replogle, and PBMC benchmarks together with linear scaling to long profiles through patching and depatching.

Load-bearing premise

That perturbation type, dosage, and cell identity supplied only through adaptive layer normalization and in-context tokens are sufficient to capture relevant biological response dynamics without gene-interaction priors or hierarchical encoders.

Editorial extensions

If this is right

The model scales linearly to full-length transcriptional profiles by patching and depatching cellular contexts.
Performance remains competitive across genetic, chemical, and cytokine perturbations as well as multiple cell types.
Architectural complexity can be reduced while preserving or improving accuracy on existing single-cell perturbation benchmarks.
In-silico cellular simulation becomes feasible at larger scale because the design avoids dedicated encoder-decoder modules.

Reading between the lines

Editorial extensions of the paper, not claims the author makes directly.

If the minimalist conditioning proves sufficient, explicit gene-interaction graphs may be unnecessary for many perturbation-prediction tasks.
The same patching strategy could be tested on other high-dimensional single-cell modalities such as chromatin accessibility or protein abundance.
Training cost and iteration speed for virtual-cell models would drop if the vanilla-Transformer baseline continues to match specialized architectures.

Share X Bluesky LinkedIn Reddit HN

Signed reviews

No signed human review yet.

Editorial analysis

A structured set of objections, weighed in public.

Desk editor's note, referee report, simulated authors' rebuttal, and a circularity audit.

Desk Editor's Note

read the letter

The paper's main move is to strip away the auxiliary encoders, hierarchical VAEs, and gene-interaction priors that most prior work uses, and instead run a standard Transformer directly on patched expression profiles under a flow-matching objective. Perturbation, dosage, and cell-type information are injected only via adaptive layer norm and in-context tokens. That is the actual novelty: a deliberately simple architecture that still claims to scale to long transcriptional profiles and beat existing methods on Tahoe100M, Replogle, and PBMC.

What it does cleanly is state the design choice up front and tie it to the scaling argument via patching and depatching. If the empirical results hold, the reduced complexity could matter for people who want something easier to train and extend than the current crop of AIVC models.

The soft spot is obvious and not minor: the abstract asserts state-of-the-art performance across diverse perturbations and cell types but gives no metrics, no baseline descriptions, no error bars, and no evaluation protocol. Without those, the central performance claim cannot be assessed, and the weakest assumption—that adaptive layer norm plus in-context tokens are enough without biological priors—remains untested in the supplied text. The stress-test note is right that there is no internal contradiction or missing derivation, but that only means the claim is purely empirical; the evidence for it is simply absent here.

This is the kind of paper a computational biology reading group might discuss if the full manuscript contains the missing numbers and a reproducible setup. It is worth a serious referee's time because the task is central to drug-discovery modeling and the minimalist framing is worth pressure-testing, even if heavy revision will be needed to make the results verifiable.

Referee Report

1 major / 0 minor

Summary. The paper introduces OCOO-T, a minimalist flow-matching-based virtual cell model that uses a vanilla Transformer operating directly on continuous gene expression profiles to predict single-cell transcriptional responses to genetic, chemical, and cytokine perturbations. Perturbation embeddings, dosage, and cell specificity are incorporated via adaptive layer normalization and in-context tokens, with patching/depatching for scalability to long profiles. The central claim is that this simple architecture achieves state-of-the-art performance on the Tahoe100M, Replogle, and PBMC benchmarks across diverse perturbations and cell types.

Significance. If the performance claims hold with proper validation, this would be significant for AIVC modeling by showing that standard flow-matching and Transformer components can suffice without auxiliary encoders, hierarchical VAEs, or gene-interaction priors, potentially improving scalability and reproducibility. The emphasis on a parameter-light design using established techniques is a strength for the field.

major comments (1)

[Abstract] Abstract: the assertion of state-of-the-art performance on Tahoe100M, Replogle, and PBMC benchmarks provides no quantitative metrics, baseline details, error analysis, or statistical comparisons, which is load-bearing for the central empirical claim and prevents verification of the reported improvements.

Simulated Author's Rebuttal

1 responses · 0 unresolved

We thank the referee for their review and constructive comment. We address the concern about the abstract below and will incorporate the suggested changes in the revised manuscript.

read point-by-point responses

Referee: [Abstract] Abstract: the assertion of state-of-the-art performance on Tahoe100M, Replogle, and PBMC benchmarks provides no quantitative metrics, baseline details, error analysis, or statistical comparisons, which is load-bearing for the central empirical claim and prevents verification of the reported improvements.

Authors: We agree that the abstract would be strengthened by including key quantitative metrics to support the SOTA claim. In the revised version, we will add concise performance highlights (e.g., primary metrics and baseline comparisons on each benchmark) drawn directly from the results tables, while preserving the abstract's length and readability. This addresses the verification concern without altering the manuscript's core claims. revision: yes

Circularity Check

0 steps flagged · score 0.0 of 10

No significant circularity detected

full rationale

The paper introduces OCOO-T as a minimalist flow-matching Transformer model for perturbation response prediction, relying on standard components (vanilla Transformer, adaptive layer norm, in-context tokens, patching) and reports empirical SOTA results on external benchmarks (Tahoe100M, Replogle, PBMC). No equations, derivations, or predictions are presented that reduce by construction to fitted parameters or self-defined quantities. No load-bearing self-citations, uniqueness theorems, or ansatzes imported from prior author work are invoked; the central claims rest on benchmark performance rather than internal definitional closure. This is a standard empirical modeling paper with no circular steps.

Assumptions & free parameters 0 free parameters · 0 assumptions · 0 invented entities

Abstract supplies no explicit free parameters, axioms, or invented entities; the model appears to rest on standard assumptions of flow-matching and Transformer architectures without additional ad-hoc constructs.

how reviews work

0 comments

Cite this review

Pith. "Pith review of OCOO-T : A Simple and Scalable Virtual Cell Model for Transcriptional Perturbation Response Prediction." pith.science (2026). https://pith.science/paper/MVTAGSH3

@misc{pith2026260612838,
  author       = {Pith},
  title        = {Pith review of: OCOO-T : A Simple and Scalable Virtual Cell Model for Transcriptional Perturbation Response Prediction},
  year         = {2026},
  howpublished = {\url{https://pith.science/paper/MVTAGSH3}},
  note         = {Machine review of arXiv:2606.12838}
}

read the original abstract

Predicting single-cell transcriptional responses to genetic, chemical and cytokine perturbations is a fundamental challenge in computational biology and AI Virtual Cell (AIVC) modeling, with direct implications for drug discovery and the elucidation of gene regulatory networks. Existing approaches often rely on auxiliary cell-state encoders, hierarchical variational autoencoders, dedicated Transformer encoder-decoder modules, or gene-interaction priors to compress high-dimensional expression profiles into latent representations. While effective, these designs increase architectural complexity and may limit scalability and generalizability. This paper introduces OCOO-T, a minimalist flow-matching-based AIVC model for transcriptional perturbation response prediction. OCOO-T utilizes a vanilla Transformer stack that operates directly on continuous gene expression profiles and formulates perturbation response prediction as a continuous-time denoising process. Perturbation embeddings, dosage information, and cell-line/cell-type specificity are integrated through adaptive layer normalization and in-context tokens. Comprehensive evaluations on Tahoe100M, Replogle, and PBMC benchmarks demonstrate that OCOO-T achieves state-of-the-art performance across diverse perturbations and cell types while effectively scaling to long transcriptional profiles through patching and depatching of cellular contexts. By leveraging the simplicity of Transformer-based denoising for single-cell omics, OCOO-T provides an effective and scalable framework for in-silico cellular simulation.

Figures

Figures reproduced from arXiv: 2606.12838 by the authors.

**Figure 1.** Overview of OCOO-T . A continuous expression profile is denoised by Transformer blocks conditioned on perturbation identity and cellular context. • SwiGLU FFN. OCOO-T replaces the standard ReLU feed-forward network with a SwiGLU feed-forward layer. The gated activation SiLU(W1x) ⊙ W2x improves the expressivity of the Transformer block while preserving a simple and scalable architecture. • RMSNorm. OCOO-T uses RMSNor… view at source ↗

**Figure 2.** Control cells (xc) are injected along with the input for perturbation prediction. • Genetic perturbations in the Replogle-Nadig benchmark share the same embedding space as gene tokens; in our experiments, these gene embeddings are initialized from ESM2 representations. • Cytokine stimulations in the PBMC benchmark are represented using ESM2 protein embeddings. For protein complexes, the final representation is obtai… view at source ↗

**Figure 3.** Patching enables the modeling of long-panel genes. Directly applying self-attention to full transcriptomic profiles is computationally expensive, because the cost of self-attention grows quadratically with sequence length, and the sequence length itself scales with the number of modeled genes. To make OCOO-T applicable to long gene panels, we adopt a simple patching and depatching strategy along the gene dimension (… view at source ↗

Figures from the paper (3 more)

**Figure 4.** Figure 4: Visualization of multi-dimensional performance across benchmarks. We compare OCOO-T with the following systems: • PerturbDiff is a conditional diffusion model that generates the distribution of perturbed single-cell transcriptomes by denoising from a reference control …

**Figure 5.** Figure 5: Comparison of the benchmark results between v-prediction and x-prediction under different patch sizes. vpred/xpred, v/x-prediction; p8/16/32, patch size 8/16/32. 4.4 Cellular Context Conditioning: Covariate Embeddings vs. Mean Control-Cell Profiles Cellular context is …

**Figure 6.** Figure 6: Comparisons of different cellular context injection methods on Raplogle-Nadig benchmark. S1: cell-line embeddings; S2–S7: mean control-cell profiles with set sizes 1, 4, 8, 16, 32, and 64, respectively [PITH_FULL_IMAGE:figures/full_fig_p016_6.png]

Discussion (0). Continue with ORCID to comment.

Forward citations

Cited by 1 Pith paper

Reviewed papers in the Pith corpus that reference this work. Sorted by Pith novelty score. Full citation record

PerturbMap: Cross-Context Transfer of Single-Cell Perturbation Responses
cs.AI 2026-07 conditional novelty 5.0 of 10

Reliability-weighted ridge transport of measured source perturbation responses improves recipient-context mean-effect MSE by 4.1% over a recipient-local low-rank base on Frangieh Perturb-CITE-seq.

Pith tools

Reviewed June 27, 2026 · model on record in the stance chip above.