CPA is a collaborative research project from Facebook AI Research (FAIR) and a computatiobal biology group of Prof. Fabian Theis (https://github.com/theislab) from Helmholtz Zentrum München.

CPA is a framework to learn effects of perturbations at the single-cell level. CPA encodes and learns phenotypic drug response across different cell types, doses and drug combinations. CPA allows:

• Out-of-distribution predicitons of unseen drug combinations at various doses and among different cell types.
• Learn interpretable drug and cell type latent spaces.
• Estimate dose response curve for each perturbation and their combinations.
• Access the uncertainty of the estimations of the model.

The repository is centered around the cpa module:

• cpa.train contains scripts to train the model.
• cpa.api contains user friendly scripts to interact with the model via scanpy.
• cpa.plotting contains scripts to plotting functions.
• cpa.model contains modules of cpa model.
• cpa.data contains data loader, which transforms anndata structure to a class compatible with cpa model.

Additional files and folders:

• datasets contains both versions of the data: raw and pre-processed.
• preprocessing contains notebooks to reproduce the datasets pre-processing from raw data.

• As a first step, download the contents of datasets/ and pretrained_models/ from this tarball.

To learn how to use this repository, check ./notebooks/demo.ipynb, and the following scripts:

• Note that hyperparameters in the demo.ipynb are set as default but might not work work for new datasets.

All the examples and the reproducbility notebooks for the plots in the paper could be found in the reproducibility repo.

• To prepare your data to train CPA, you need to add specific fields to adata object and perfrom data split. Examples on how to add necessary fields for multiple datasets used in the paper can be found in preprocessing/ folder.

There are two ways to train a cpa model:

• Using the command line, e.g.: python -m cpa.train --data datasets/GSM_new.h5ad --save_dir /tmp --max_epochs 1 --doser_type sigm
• From jupyter notebook: example in ./notebooks/demo.ipynb

Currently you can access the documentation via help function in IPython. For example:

from cpa.api import API

help(API)

from cpa.plotting import CPAVisuals

help(CPAVisuals)

A separate page with the documentation is coming soon.

If you have a question or noticed a problem, you can post an issue.

Please cite the following preprint if you find CPA useful in your research.

@article {Lotfollahi2021.04.14.439903,
	author = {Lotfollahi, Mohammad and Susmelj, Anna Klimovskaia and De Donno, Carlo and Ji, Yuge and Ibarra, Ignacio L. and Wolf, F. Alexander and Yakubova, Nafissa and Theis, Fabian J. and Lopez-Paz, David},
	title = {Learning interpretable cellular responses to complex perturbations in high-throughput screens},
	elocation-id = {2021.04.14.439903},
	year = {2021},
	doi = {10.1101/2021.04.14.439903},
	publisher = {Cold Spring Harbor Laboratory},

	URL = {https://www.biorxiv.org/content/early/2021/05/18/2021.04.14.439903},
	eprint = {https://www.biorxiv.org/content/early/2021/05/18/2021.04.14.439903.full.pdf},
	journal = {bioRxiv}
}

The preprint titled Learning interpretable cellular responses to complex perturbations in high-throughput screens can be found here.

This source code is released under the MIT license, included here.