atlas

Research map / Probabilistic and scientific modeling / Molecules and proteins

CellPLM: Pre-training of Cell Language Model Beyond Single Cells

ICLR 2024 · Probabilistic and scientific modeling › Molecules and proteins · approach: Biological sequence transformers · 75 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes CellPLM, a pre-trained transformer that treats cells as tokens and tissues as sentences, pre-trained on scRNA-seq and spatial transcriptomics with a Gaussian mixture latent prior, and evaluated on denoising, imputation, annotation and perturbation prediction.

Problem

Existing single-cell pre-trained models treat genes as tokens and model only within-cell gene dependencies, ignoring cell-cell relations, and suffer from limited and noisy single-cell data.

Approach

Model a "cell language model" by treating cells as tokens within a tissue, masking gene expressions and predicting them from other unmasked cells; jointly pre-train on scRNA-seq and spatial transcriptomics with 2D sinusoidal positional encodings and a Gaussian mixture variational latent prior with a batch-aware decoder.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Fine-tune CellPLMCellPLM · PBMC 5KRMSE: 0.775 ± 0.002 -> 0.725 ± 0.001Helped
Fine-tune CellPLMCellPLM · PBMC 5KMAE: 0.565 ± 0.001 -> 0.551 ± 0.001Helped
Fine-tune CellPLMCellPLM · JurkatRMSE: 0.423 ± 0.001 -> 0.391 ± 0.001Helped
Fine-tune CellPLMCellPLM · JurkatMAE: 0.351 ± 0.001 -> 0.320 ± 0.001Helped
Fine-tune CellPLMCellPLM · Lung2Corr: 0.227 ± 0.011 -> 0.318 ± 0.015Helped
Fine-tune CellPLMCellPLM · Lung2Cosine: 0.360 ± 0.014 -> 0.481 ± 0.011Helped
Fine-tune CellPLMCellPLM · Liver2Corr: 0.253 ± 0.014 -> 0.328 ± 0.011Helped
Fine-tune CellPLMCellPLM · Liver2Cosine: 0.399 ± 0.012 -> 0.481 ± 0.010Helped
Fine-tune CellPLMCellPLM · MSF1: 0.703 -> 0.766 ± 0.007Helped
Fine-tune CellPLMCellPLM · hPancreasF1: 0.739 ± 0.006 -> 0.749 ± 0.010Helped
Fine-tune CellPLMCellPLM · hPancreasPrecision: 0.761 ± 0.004 -> 0.753 ± 0.010Hurt
Use pre-trained weights rather than training from scratch (ablation)CellPLM · Lung2 spatial transcriptomic imputationCorr: 0.058 ± 0.020 -> 0.318 ± 0.015Helped

Surprising findings

Limitations

Experimental setup

Datasets: HTCA scRNA-seq HCA scRNA-seq GEO scRNA-seq CosMx NSCLC SRT PBMC 5K Jurkat Lung2 Liver2 MS hPancreas Adamson Perturb-Seq Norman Perturb-Seq

Benchmarks: PBMC 5K denoising Jurkat denoising Lung2 imputation Liver2 imputation MS cell type annotation hPancreas cell type annotation Adamson perturbation Norman perturbation

Metrics: ARI NMI RMSE MAE Pearson correlation cosine similarity F1 precision

Baselines: scGPT Geneformer PCA DeepImpute scGNN 2.0 GraphSCI SAVER DCA MAGIC scImpute SpaGE stPlus

Builds on: scBERT scGPT Flowformer SpaFormer scVI

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy