SIMS is a pipeline for building interpretable and accurate classifiers for identifying any target on single-cell rna-seq data. The SIMS model is based on a sequential transformer, a transformer model specifically built for large-scale tabular datasets.
SIMS takes in a list of arbitrarily many expression matrices along with their corresponding target variables. We assume the matrix form cell x gene
, and NOT gene x cell
, since our training samples are the transcriptomes of individual cells.
The code is run with python
. To use the package, we recommend using a virtual environment such as miniconda which will allow you to install packages without harming your system python
.
If using conda, run
- Create a new virtual environment with
conda create --name=<NAME> python=3.9
- Enter into your virtual environment with
conda activate NAME
Otherwise, enter your virtual environment of choice and
- Install the SIMS package with
pip install --use-pep517 git+https://github.com/braingeneers/SIMS.git
- Set up the model training code in a
MYFILE.py
file, and run it withpython MYFILE.py
. A tutorial on how to set up training code is shown below.
To train a model, we can set up a SIMS class in the following way:
from scsims import SIMS
from pytorch_lightning.loggers import WandbLogger
logger = WandbLogger(offline=True)
sims = SIMS(data=['my/data/file.h5ad'], class_label='class_label')
sims.setup_trainer(accelerator="gpu", devices=1, logger=logger)
sims.train()
This will automatically load in your .h5ad
file, where the class_label
is assumed to be a valid column in the .obs
attribute. Alternatively, if your labels are stored in a separate csv, you may also initialize the class like
sims = SIMS(data=['my/data/file.h5ad'], labelfiles=['my/label/file.csv'], class_label='class_label')
sims.train()
This will set up the underlying dataloaders, model, model checkpointing, and everything else we need. Model checkpoints will be saved every training epoch.
To load in a model to infer new cell types on an unlabeled dataset, we load in the model checkpoint, point to the label file that we originally trained on, and run the predict
method on new data.
sims = SIMS(weights_path='myawesomemodel.ckpt', labelfiles=['my/label/file.csv'], class_label='class_label')
cell_predictions = sims.predict('my/new/unlabeled.h5ad')
Finally, to look at the explainability of the model, we similarly run
explainability_matrix = sims.explain('my/new/unlabeled.h5ad') # this can also be labeled data, of course
To customize the underlying pl.Trainer
and SIMS model params, we can initialize the SIMS model like
from pytorch_lightning.loggers import WandbLogger
from pytorch_lightning.callbacks import EarlyStopping, LearningRateMonitor
from scsims import SIMS
wandb_logger = WandbLogger(project=f"My Project", name=f"SIMS Model Training") # set up the logger to log data to Weights and Biases
sims = SIMS(data=adata, class_label='class_label')
sims.setup_model(n_a=64, n_d=64, weights=sims.weights) # weighting loss inversely proportional by label freq, helps learn rare cell types (recommended)
sims.setup_trainer(
logger=wandb_logger,
callbacks=[
EarlyStopping(
monitor="val_loss",
patience=50,
),
LearningRateMonitor(logging_interval="epoch"),
],
num_epochs=100,
)
sims.train()
This will train the SIMS model on the given expression matrices with target variable given by the class_label
column in each label file.
If you are using SIMS only for predictions using an already trained model, github codespaces is the recommended way to use this tool. You can also use this pipeline to train it in smaller datasets as the computing services offered in codespaces are modest. To use this tool in github codespaces start by forking the repo in your github account. Then create a new codespace with the SIMS repo as the Repository of choice. Once inside the newly created environment pull the latest SIMS image:
docker pull jmlehrer/sims:latest
Run the docker container mounting the file folder containing datasets and model checkpoints to the filesystem:
docker run -it -v /path/to/local/folder:/path/in/container [image_name] /bin/bash
Run main.py to check if the installation has been completed. You can alter this file as shown above to perform the different tasks.
python main.py