arXiv Machine Learning

Response Magnitude as a Dominant Signal for Held-Out CRISPRi Perturbation Effect Prediction

arXiv:2608. 00152v1 Announce Type: new Abstract: Predicting the magnitude of a CRISPRi perturbation's transcriptomic effect on held-out target genes is an important open problem in single-cell biology.

arXiv Machine Learning
Aug 31

Locked Evaluation Surfaces: Transfer Failure and Sampling-Depth Entanglement in CRISPRi Perturbation-Effect Prediction

The study evaluates a frozen Geneformer representation for predicting CRISPRi perturbation effects under a tightly controlled, pre‑registered protocol. While the representation shows significant predictive power within the Virtual Cell Challenge dataset, it fails to transfer to external screens, with negative zero‑shot Spearman correlations. The analysis also reveals that the VCC endpoint is heavily influenced by sampling depth, as cell count alone explains most of the variance, indicating a sampling‑depth entanglement that could mask transfer failures in less controlled settings.

By Mehrdad Shoeibi, Niloofar Yousefi
arXiv AI
Sep 2

SCALE:Scalable Conditional Atlas-Level Endpoint transport for virtual cell perturbation prediction

SCALE is a conditional transport model that treats cells as unordered sets to predict treated cell populations without requiring cell-level matching. It uses a shared set-aware encoder and a conditional DiT backbone to learn latent transport, enabling endpoint supervision that is directly delta-aligned. Across diverse perturbation types—including genetic, chemical, developmental, and immune—SCALE accurately recovers gene‑expression changes, response directions, and population structure, outperforming competing methods on CRISPR data and successfully prioritizing cytokines that elicit distinct immune responses.

By Shuizhou Chen, Lang Yu, Xueqin Lin, Xinjie Mao, Songming Zhang, Xinyu Gu, Hao Wu, Sheng Xu, Kedu Jin, Lei Bai, Quan Qian, Qin Chen, Qiang Gao, Siqi Sun, Zhangyang Gao
arXiv AI
Oct 16

AI Scientists for Building Virtual-Cell Models

The paper introduces CellForge, a team of autonomous agents that can autonomously generate, train, and validate predictive models for single‑cell perturbation data across transcriptomic, epigenomic, and proteomic contexts. By surveying literature, parsing data, and engaging in critic‑mediated discussions among specialized agents, CellForge produces end‑to‑end model architectures and code without human specification, achieving performance competitive with state‑of‑the‑art hand‑crafted methods. Human evaluators found CellForge’s research plans to be more scientifically sound and innovative than those produced by single large language models or deep‑research systems.

By Xiangru Tang, Jiapeng Chen, Daniel Shao, Zhuoyun Yu, Yan Cui, Weixu Wang, Fang Wu, Yuchen Zhuang, Wenqi Shi, Zhi Huang, Arman Cohan, Xihong Lin, Fabian Theis, Smita Krishnaswamy, Mark Gerstein
arXiv Computation and Language
Sep 11

Biology-in-the-loop: Amortized Adaptive Hit Discovery in CRISPR Screens

The paper introduces AssayBench-Loop, a large benchmark of 1,389 CRISPR screens across five phenotype categories, and builds on it to develop AssayLoop, a sequential experimental design framework that combines a transformer-based acquisition policy (AssayFormer) trained on historical data with LLM-derived biological priors. AssayLoop achieves a 5.67‑fold enrichment over random selection, recovering 27.7% of hits after testing only about 5% of the library, and outperforms existing adaptive-design methods and standalone LLMs. The authors also present AssayLLM, extending the approach directly to an LLM via task‑specific post‑training, and show that performance improves with more historical training data and transfers to unseen phenotype categories.

By Carl Edwards, Edward De Brouwer, Xiner Li, Namkyeong Lee, Ehsan Hajiramezanali, Anne Biton, Sara Mostafavi, Gabriele Scalia
arXiv Machine Learning
Oct 9

Retrieval-Augmented Generation for Predicting Cellular Responses to Gene Perturbation

The paper introduces PT‑RAG, a retrieval‑augmented generation framework that enhances deep learning models for predicting cellular transcriptional responses to gene perturbations. PT‑RAG adds a two‑stage retrieval module—semantic retrieval followed by a Gumbel‑Softmax selector—to supply contextually relevant perturbation information, improving predictive accuracy across cell types and perturbations. Experiments on STATE‑style and scGPT backbones show consistent gains, such as a 5.9% reduction in 2‑Wasserstein distance for cross‑cell‑type predictions.

By Andrea Giuseppe Di Francesco, Andrea Rubbi, Rishabh Jain, Pietro Li\`o
arXiv Machine Learning
Oct 12

Differentiable Retrieval-Augmented Generation for Predicting Cellular Responses to Gene Perturbation

The paper introduces PT‑RAG, a retrieval‑augmented generation framework that enhances cellular perturbation response prediction by incorporating context from related perturbations. PT‑RAG uses a two‑stage retrieval process—semantic retrieval to shortlist candidate perturbations followed by a differentiable Gumbel‑Softmax selector that adapts the selection based on cell state and query perturbation. Experiments on two backbones, a frozen STATE‑style generator and a fully trained scGPT, show that PT‑RAG improves distributional similarity and predictive quality, achieving a 5.9% reduction in 2‑Wasserstein distance on cross‑cell‑type tasks.

By Andrea Giuseppe Di Francesco, Andrea Rubbi, Rishabh Jain, Pietro Li\`o
arXiv AI
Oct 1

CellMSA: Context Modeling for Single-Cell Representation Learning

CellMSA introduces a novel single‑cell representation learning framework that leverages a multiple‑sequence‑alignment‑inspired context model. For each target cell, it retrieves relevant cells across batches and related cell types, summarizing cross‑cell patterns into a context‑dependent gene‑pair representation that is fed into a pair‑aware encoder. Pretraining on a massive human single‑cell corpus (≈109 million cells) and subsequent benchmarks demonstrate consistent performance gains over existing methods.

By Suyuan Zhao, Minghao Liu, Yizhen Luo, Zaiqing Nie