atlas

Research map / Large language models / Instruction and fine-tuning

Cell2Sentence: Teaching Large Language Models the Language of Biology

ICML 2024 · Large language models › Instruction and fine-tuning · approach: Fine-tuning for reasoning data · 62 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.

Introduces Cell2Sentence, which converts single-cell gene expression profiles into rank-ordered gene-name "cell sentences" so that GPT-2 models can be fine-tuned to generate cells, predict cell types, and convert back to expression.

Problem

Applying pretrained large language models to single-cell transcriptomics, which is normally handled by specialized neural networks that do not leverage language pretraining.

Approach

Represent each cell as a text sequence of gene names ordered by expression rank (a "cell sentence"), which is approximately reversible to expression via a log-linear rank-expression fit, so standard causal LMs can be fine-tuned on it with natural-language prompts and metadata.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add natural-language pretraining before Cell2Sentence training (ablation)GPT-2 Small · Immune tissue, 17 cell typesR2: 0.873 -> 0.949Helped
Add natural-language pretraining before Cell2Sentence training (ablation)GPT-2 Medium · Unseen immune tissue test dataAccuracy: 29.17 -> 74.26Helped
Increase model size from GPT-2 Small to GPT-2 MediumNL + C2S · Unseen immune tissue test dataAccuracy: 69.95 -> 74.26Helped
Generate cells with Cell2Sentence instead of scVICell generation · Immune tissue, generated cells, K=5KNN accuracy (Expr.): 43.06 -> 54.67Helped
Use LoRA instead of full fine-tuning (ablation)GPT-2 fine-tuning · Gene generationUnique genes %: uniqueness of genes in generated sentences as low as 70%Hurt
Use FP32 instead of FP16 (ablation)GPT-2 training · Cell sentence trainingTraining speed: 60% slowdownHurt
Add natural-language pretraining before Cell2Sentence training (ablation)GPT-2 Small · 35 cell types seen during trainingValid genes %: 97.84 -> 99.60Helped

Surprising findings

Limitations

Experimental setup

Datasets: human immune tissue single-cell dataset (Domínguez Conde et al.) GEO/HCA-derived scRNA-seq datasets for reconstruction analysis (127 datasets)

Benchmarks: immune tissue held-out test set

Metrics: Pearson R Spearman R R2 KNN accuracy accuracy F1 perplexity gene validity gene uniqueness MMD

Baselines: scVI GPT-2 Small (C2S) GPT-2 Medium (C2S)

Builds on: GPT-2 scVI Scanpy LoRA

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