celldiffusion 0.1.2

CellDiffusion(Single-Cell graph neural Diffusion) is a physics-informed graph generative model to do scRNA-seq analysis. CellDiffusion investigates cellular dynamics utilizing an attention-based neural network.

About:

CellDiffusion(Single-Cell graph neural Diffusion) is a deep diffusion model to leverage multi-scale patterns in single-cell graphs and enhance scRNA-seq analysis. Single-cell transcriptomics are typically analyzed based on gene expression within individual cells and hypothetic cell adjacencies. However, existing computational methods often suffer from a lack of leveraging and integrating multi-scale dependencies in feature space, undermining their effectiveness and robustness in downstream applications like handling of batch effects, cell type identification, and cell fate inference. To tackle this challenge, we introduce CellDiffusion to incorporate long-range information propagation among cells to uncover cellular biology from their transcriptomics. CellDiffusion integrates both local and global diffusion processes to comprehensively capture cell relationships, ranging from fine-grained structures to large-scale patterns. This approach exhibits great perception of inherent cell types and potential lineages and preserves cell identities in batch-imbalanced datasets. CellDiffusion enhances various downstream tasks, including data integration, reference-based cell type annotation, unsupervised clustering, and trajectory inference.

This repository contains the source code for the paper "CellDiffusion: graph-based deep diffusion model leverages multi-scale dependencies among single cells", Yu-Chen Liu, Lei Jiang, Simon Liang Lu, Anqi Zou, Zedong Lin, Nidhi Siddharam Loni, Heng Pan, Vijaya B. Kolachalama, Dong Xu*, Juexin Wang* & Chao Zhang*.

Installation:

CellDiffusion is available on PyPI. To install CellDiffusion, run the following command:

pip install celldiffusion

Or grab this source codes:

git clone https://github.com/CZCBLab/CellDiffusion.git
cd CellDiffusion

Python=3.9.9 is required. See other requirements in the file requirements.txt.

Run CellDiffusion in Docker:

git clone https://github.com/CZCBLab/CellDiffusion.git
cd CellDiffusion

# Build the Docker image
sudo docker build -t celldiffusion .

# Run Docker container with CPU
sudo docker run -it -p 8888:8888 --restart always celldiffusion bash

# Or run Docker container with GPU
sudo docker run -it -p 8888:8888 --restart always --gpus all celldiffusion bash

# Start Jupyter Notebook
jupyter notebook --ip="0.0.0.0" --allow-root

'celldiffusion' could be changed into your image name. Please refer to Docker and NVIDIA Container Toolkit for more details about Docker installation.

Tutorials:

For data integration, please check the notebook file "CellDiffusion_tutorial_Data_Integration.ipynb".

For reference-based cell type annotation, please check the notebook file "CellDiffusion_tutorial_Annotation_(Label_Transfer).ipynb".

For clustering tasks, please check the notebook file "CellDiffusion_tutorial_Clustering.ipynb".

For trajectory tasks, please check the notebook file "CellDiffusion_tutorial_Trajectory_Inference.ipynb".