PULSAR-aligned / README.md
KuanP's picture
Update README.md
b630af3 verified
---
library_name: transformers
license: mit
base_model:
- KuanP/PULSAR-pbmc
---
# Model Card for PULSAR-pbmc
<!-- Provide a quick summary of what the model is/does. -->
**PULSAR** (Patient Understanding Leveraging Single-cell universAl Representation) is a multi-scale, multi-cellular foundation model for human peripheral blood mononuclear cells (PBMCs). It transforms a set of single-cell transcriptomes into an interpretable **donor-level embedding** that preserves single-cell resolution while capturing multicellular composition and coordination.
This repo hosts the **aligned PBMC model** (`PULSAR-aligned`) used to produce donor embeddings aligned for disease classification. A base-model is also available (see **Model Sources**).
## Model Details
### Model Description
<!-- Provide a longer summary of what this model is. -->
PULSAR (Patient Understanding Leveraging Single-cell universAl Representation) is a hierarchical, multi-scale foundation model for PBMC scRNA-seq that converts unordered sets of single cells into a 512-d donor embedding while preserving single-cell resolution. It integrates molecular priors from ESM2 protein embeddings, cellular representations via Universal Cell Embeddings (UCE, 1,280-d), and a Multicellular Transformer encoder–decoder trained with a high-masking, Masked Cell Modeling objective. Pretraining proceeds in two stages: a pan-tissue CELLxGENE corpus (≈36.2M cells; 6,807 samples) followed by continual pretraining on blood (≈8.74M cells; 2,588 samples). The resulting donor embeddings support zero-shot and lightweight-head downstream tasks, including large-scale reference mapping for disease classification (state-of-the-art accuracy with strong external generalization), regression of plasma proteomics from transcriptomes, forecasting of future outcomes (e.g., RA conversion in ACPA+ individuals and influenza vaccine responsiveness), and individualized cytokine perturbation modeling across donor, cellular, and gene levels. A “virtual instrument” conditions on cytokine protein embeddings to transform baseline donor states and, with the decoder and an optional UCE→expression head, generates perturbed cell distributions and gene programs. Attention over cells provides mechanistic interpretability, highlighting disease- and severity-relevant subsets and enriching for antigen-specific clonotypes in viral infection. PULSAR thus operationalizes the AI Virtual Cell vision by linking molecular, cellular, and multicellular organization into a unified, transferable representation for precision immunology.
- **Developed by:** Kuan Pang (Stanford University, kuanpang@stanford.edu)
- **Model type:** Transformer
- **License:** MIT
### Model Sources [optional]
<!-- Provide the basic links for the model. -->
- **Repository:** https://github.com/snap-stanford/PULSAR
- **Paper:** https://www.biorxiv.org/content/10.1101/2025.11.24.685470v1
- **Aligned version:** https://huggingface.co/KuanP/PULSAR-pbmc
## Uses
<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
### Direct Use
- Generate 512-d **donor embeddings** from PBMC scRNA-seq to:
- Perform **reference mapping/retrieval** (kNN) for disease phenotypes
### Out-of-Scope Use
The model might not work for tissue types other than PBMC, which also includes cell sorting samples.
## How to Get Started with the Model
Use the code below to get started with the model.
## Training Details
### Training Data
Stage-1 pretraining corpus: CZ CELLxGENE Census (LTS 2023-07-25), 36.2M cells, 6,807 samples across 53 tissues and 69 conditions.
Stage-2 continual pretraining (blood focus): 8.736M cells, 2,588 blood/PBMC samples (balanced sexes; broad ages).
More details can be found in the Paper and GitHub.
## Citation
<!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
**BibTeX:**
```
@article{pang2025pulsar,
title={PULSAR: a Foundation Model for Multi-scale and Multicellular Biology},
author={Pang, Kuan and Rosen, Yanay and Kedzierska, Kasia and He, Ziyuan and Rajagopal, Abhe and Gustafson, Claire E and Huynh, Grace and Leskovec, Jure},
journal={bioRxiv},
pages={2025--11},
year={2025},
publisher={Cold Spring Harbor Laboratory}
}
```