---
description: Reduce high-dimensional gene expression data from
  individual cells into a lower-dimensional space for visualization.
  This lab explores PCA, tSNE and UMAP.
subtitle:  Seurat Toolkit
title:  Dimensionality Reduction
---

<div>

> **Note**
>
> Code chunks run R commands unless otherwise specified.

</div>

## Data preparation

First, let's load all necessary libraries and the QC-filtered dataset
from the previous step.

``` {r}
# Activate conda environment to get the correct python path
reticulate::use_condaenv("seurat", conda = "/opt/conda/bin/conda")
reticulate::py_discover_config()

suppressPackageStartupMessages({
    library(Seurat)
    library(ggplot2) # plotting
    library(patchwork) # combining figures
    library(scran)
})
```

``` {r}
# download pre-computed data if missing or long compute
fetch_data <- TRUE

# url for source and intermediate data
path_data <- "https://export.uppmax.uu.se/naiss2023-23-3/workshops/workshop-scrnaseq"
path_file <- "data/covid/results/seurat_covid_qc.rds"
if (!dir.exists(dirname(path_file))) dir.create(dirname(path_file), recursive = TRUE)
if (fetch_data && !file.exists(path_file)) download.file(url = file.path(path_data, "covid/results/seurat_covid_qc.rds"), destfile = path_file)
alldata <- readRDS(path_file)
```

## Feature selection

We first need to define which features/genes are important in our
dataset to distinguish cell types. For this purpose, we need to find
genes that are highly variable across cells, which in turn will also
provide a good separation of the cell clusters.

``` {r}
suppressWarnings(suppressMessages(alldata <- FindVariableFeatures(alldata, selection.method = "vst", nfeatures = 2000, verbose = FALSE, assay = "RNA")))
top20 <- head(VariableFeatures(alldata), 20)

LabelPoints(plot = VariableFeaturePlot(alldata), points = top20, repel = TRUE)
```

## Z-score transformation

Now that the genes have been selected, we now proceed with PCA. Since
each gene has a different expression level, it means that genes with
higher expression values will naturally have higher variation that will
be captured by PCA. This means that we need to somehow give each gene a
similar weight when performing PCA (see below). The common practice is
to center and scale each gene before performing PCA. This exact scaling
called Z-score normalization is very useful for PCA, clustering and
plotting heatmaps. Additionally, we can use regression to remove any
unwanted sources of variation from the dataset, such as `cell cycle`,
`sequencing depth`, `percent mitochondria` etc. This is achieved by
doing a generalized linear regression using these parameters as
co-variates in the model. Then the residuals of the model are taken as
the *regressed data*. Although perhaps not in the best way, batch effect
regression can also be done here. By default, variables are scaled in
the PCA step and is not done separately. But it could be achieved by
running the commands below:

``` {r}
alldata <- ScaleData(alldata, vars.to.regress = c("percent_mito", "nFeature_RNA"), assay = "RNA")
```

## PCA

Performing PCA has many useful applications and interpretations, which
much depends on the data used. In the case of single-cell data, we want
to segregate samples based on gene expression patterns in the data.

To run PCA, you can use the function `RunPCA()`.

``` {r}
alldata <- RunPCA(alldata, npcs = 50, verbose = F)
```

We then plot the first principal components.

``` {r}
#| fig-height: 4
#| fig-width: 12

wrap_plots(
    DimPlot(alldata, reduction = "pca", group.by = "orig.ident", dims = 1:2),
    DimPlot(alldata, reduction = "pca", group.by = "orig.ident", dims = 3:4),
    DimPlot(alldata, reduction = "pca", group.by = "orig.ident", dims = 5:6),
    ncol = 3
) + plot_layout(guides = "collect")
```

To identify which genes (Seurat) or metadata parameters (Scater/Scran)
contribute the most to each PC, one can retrieve the loading matrix
information. Unfortunately, this is not implemented in Scater/Scran, so
you will need to compute PCA using `logcounts`.

``` {r}
#| fig-height: 6
#| fig-width: 14

VizDimLoadings(alldata, dims = 1:5, reduction = "pca", ncol = 5, balanced = T)
```

We can also plot the amount of variance explained by each PC.

``` {r}
#| fig-height: 4
#| fig-width: 5

ElbowPlot(alldata, reduction = "pca", ndims = 50)
```

Based on this plot, we can see that the top 8 PCs retain a lot of
information, while other PCs contain progressively less. However, it is
still advisable to use more PCs since they might contain information
about rare cell types (such as platelets and DCs in this dataset)

## tSNE

We will now run [BH-tSNE](https://arxiv.org/abs/1301.3342).

``` {r}
alldata <- RunTSNE(
    alldata,
    reduction = "pca", dims = 1:30,
    perplexity = 30,
    max_iter = 1000,
    theta = 0.5,
    eta = 200,
    num_threads = 0
)
# see ?Rtsne and ?RunTSNE for more info
```

We plot the tSNE scatterplot colored by dataset. We can clearly see the
effect of batches present in the dataset.

``` {r}
#| fig-height: 5
#| fig-width: 6

DimPlot(alldata, reduction = "tsne", group.by = "orig.ident")
```

## UMAP

We can now run [UMAP](https://arxiv.org/abs/1802.03426) for cell
embeddings.

``` {r}
alldata <- RunUMAP(
    alldata,
    reduction = "pca",
    dims = 1:30,
    n.components = 2,
    n.neighbors = 30,
    n.epochs = 200,
    min.dist = 0.3,
    learning.rate = 1,
    spread = 1
)
# see ?RunUMAP for more info
```

A feature of UMAP is that it is not limited by the number of dimensions
the data cen be reduced into (unlike tSNE). We can simply reduce the
dimentions altering the `n.components` parameter. So here we will create
a UMAP with 10 dimensions.

In Seurat, we can add in additional reductions, by default they are
named "pca", "umap", "tsne" etc. depending on the function you run. Here
we will specify an alternative name for the umap with the
`reduction.name` parameter.

``` {r}
alldata <- RunUMAP(
    alldata,
    reduction.name = "UMAP10_on_PCA",
    reduction = "pca",
    dims = 1:30,
    n.components = 10,
    n.neighbors = 30,
    n.epochs = 200,
    min.dist = 0.3,
    learning.rate = 1,
    spread = 1
)
# see ?RunUMAP for more info
```

UMAP is plotted colored per dataset. Although less distinct as in the
tSNE, we still see quite an effect of the different batches in the data.

``` {r}
#| fig-height: 4
#| fig-width: 12

wrap_plots(
    DimPlot(alldata, reduction = "umap", group.by = "orig.ident") + ggplot2::ggtitle(label = "UMAP_on_PCA"),
    DimPlot(alldata, reduction = "UMAP10_on_PCA", group.by = "orig.ident", dims = 1:2) + ggplot2::ggtitle(label = "UMAP10_on_PCA"),
    DimPlot(alldata, reduction = "UMAP10_on_PCA", group.by = "orig.ident", dims = 3:4) + ggplot2::ggtitle(label = "UMAP10_on_PCA"),
    ncol = 3
) + plot_layout(guides = "collect")
```

We can now plot PCA, UMAP and tSNE side by side for comparison. Have a
look at the UMAP and tSNE. What similarities/differences do you see? Can
you explain the differences based on what you learned during the
lecture? Also, we can conclude from the dimensionality reductions that
our dataset contains a batch effect that needs to be corrected before
proceeding to clustering and differential gene expression analysis.

``` {r}
#| fig-height: 4
#| fig-width: 12

wrap_plots(
    DimPlot(alldata, reduction = "pca", group.by = "orig.ident"),
    DimPlot(alldata, reduction = "tsne", group.by = "orig.ident"),
    DimPlot(alldata, reduction = "umap", group.by = "orig.ident"),
    ncol = 3
) + plot_layout(guides = "collect")
```

<div>

> **Discuss**
>
> We have now done Variable gene selection, PCA and UMAP with the
> settings we selected for you. Test a few different ways of selecting
> variable genes, number of PCs for UMAP and check how it influences
> your embedding.

</div>

## Z-scores & DR graphs

Although running a second dimensionality reduction (i.e tSNE or UMAP) on
PCA would be a standard approach (because it allows higher computation
efficiency), the options are actually limitless. Below we will show a
couple of other common options such as running directly on the scaled
data (z-scores) (which was used for PCA) or on a graph built from scaled
data. We will only work with UMAPs, but the same applies for tSNE.

### UMAP from z-scores

To run tSNE or UMAP on the scaled data, one first needs to select the
number of variables to use. This is because including dimensions that do
contribute to the separation of your cell types will in the end mask
those differences. Another reason for it is because running with all
genes/features also will take longer or might be computationally
unfeasible. Therefore we will use the scaled data of the highly variable
genes.

``` {r}
alldata <- RunUMAP(
    alldata,
    reduction.name = "UMAP_on_ScaleData",
    features = alldata@assays$RNA@var.features,
    assay = "RNA",
    n.components = 2,
    n.neighbors = 30,
    n.epochs = 200,
    min.dist = 0.3,
    learning.rate = 1,
    spread = 1
)
```

### UMAP from graph

To run tSNE or UMAP on the a graph, we first need to build a graph from
the data. In fact, both tSNE and UMAP first build a graph from the data
using a specified distance matrix and then optimize the embedding. Since
a graph is just a matrix containing distances from cell to cell and as
such, you can run either UMAP or tSNE using any other distance metric
desired. Euclidean and Correlation are usually the most commonly used.

``` {r}
# Build Graph
alldata <- FindNeighbors(alldata,
    reduction = "pca",
    assay = "RNA",
    k.param = 20,
    features = alldata@assays$RNA@var.features
)

alldata <- RunUMAP(alldata,
    reduction.name = "UMAP_on_Graph",
    umap.method = "umap-learn",
    graph = "RNA_snn",
    n.epochs = 200,
    assay = "RNA"
)
```

We can now plot the UMAP comparing both on PCA vs ScaledSata vs Graph.

``` {r}
#| fig-height: 4
#| fig-width: 12

p1 <- DimPlot(alldata, reduction = "umap", group.by = "orig.ident") + ggplot2::ggtitle(label = "UMAP_on_PCA")
p2 <- DimPlot(alldata, reduction = "UMAP_on_ScaleData", group.by = "orig.ident") + ggplot2::ggtitle(label = "UMAP_on_ScaleData")
p3 <- DimPlot(alldata, reduction = "UMAP_on_Graph", group.by = "orig.ident") + ggplot2::ggtitle(label = "UMAP_on_Graph")
wrap_plots(p1, p2, p3, ncol = 3) + plot_layout(guides = "collect")
```

## Genes of interest

Let's plot some marker genes for different cell types onto the
embedding.

  Markers                    Cell Type
  -------------------------- -------------------
  CD3E                       T cells
  CD3E CD4                   CD4+ T cells
  CD3E CD8A                  CD8+ T cells
  GNLY, NKG7                 NK cells
  MS4A1                      B cells
  CD14, LYZ, CST3, MS4A7     CD14+ Monocytes
  FCGR3A, LYZ, CST3, MS4A7   FCGR3A+ Monocytes
  FCER1A, CST3               DCs

``` {r}
#| fig-height: 9
#| fig-width: 12

myfeatures <- c("CD3E", "CD4", "CD8A", "NKG7", "GNLY", "MS4A1", "CD14", "LYZ", "MS4A7", "FCGR3A", "CST3", "FCER1A")
FeaturePlot(alldata, reduction = "umap", dims = 1:2, features = myfeatures, ncol = 4, order = T) +
    NoLegend() + NoAxes() + NoGrid()
```

<div>

> **Discuss**
>
> Select some of your dimensionality reductions and plot some of the QC
> stats that were calculated in the previous lab. Can you see if some of
> the separation in your data is driven by quality of the cells?

</div>

## Save data

We can finally save the object for use in future steps.

``` {r}
saveRDS(alldata, "data/covid/results/seurat_covid_qc_dr.rds")
```

## Session info

```{=html}
<details>
```
```{=html}
<summary>
```
Click here
```{=html}
</summary>
```
``` {r}
sessionInfo()
```

```{=html}
</details>
```