MOFA2: training a model in R
2024-01-05
This vignette contains a detailed tutorial on how to train a MOFA model using R. A concise template script can be found here. Many more examples on application of MOFA to various multi-omics data sets can be found here.
1 Load libraries
library(data.table)
library(MOFA2)2 Is MOFA the right method for my data?
MOFA (and factor analysis models in general) are useful to uncover variation in complex data sets that contain multiple sources of heterogeneity. This requires a relatively large sample size (at least ~15 samples). In addition, MOFA needs the multi-modal measurements to be derived from the same samples. It is fine if you have samples that are missing some data modality, but there has to be a significant degree of matched measurements.
3 Preprocessing the data
3.1 Normalisation
Proper normalisation of the data is critical. The model can handle three types of data: continuous (modelled with a gaussian likelihood), small counts (modelled with a Poisson likelihood) and binary measurements (modelled with a bernoulli likelihood). Non-gaussian likelihoods give non-optimal results, we recommend the user to apply data transformations to obtain continuous measurements. For example, for count-based data such as RNA-seq or ATAC-seq we recommend size factor normalisation + variance stabilisation (i.e. a log transformation).
3.2 Feature selection
It is strongly recommended that you select highly variable features (HVGs) per assay before fitting the model. This ensures a faster training and a more robust inference procedure. Also, for data modalities that have very different dimensionalities we suggest a stronger feature selection fort he bigger views, with the aim of reducing the feature imbalance between data modalities.
4 Create the MOFA object
To create a MOFA object you need to specify three dimensions: samples, features and view(s). Optionally, a group can also be specified for each sample (no group structure by default). MOFA objects can be created from a wide range of input formats, including:
- a list of matrices: this is recommended for relatively simple data.
- a long data.frame: this is recommended for complex data sets with multiple views and/or groups.
- MultiAssayExperiment: to connect with Bioconductor objects.
- Seurat: for single-cell genomics users. See this vignette
4.1 List of matrices
A list of matrices, where each entry corresponds to one view. Samples are stored in columns and features in rows.
Let’s simulate some data to start with
data <- make_example_data(
n_views = 2,
n_samples = 200,
n_features = 1000,
n_factors = 10
)[[1]]
lapply(data,dim)## $view_1
## [1] 1000 200
##
## $view_2
## [1] 1000 200Create the MOFA object:
MOFAobject <- create_mofa(data)Plot the data overview
plot_data_overview(MOFAobject)
In case you are using the multi-group functionality, the groups can be specified
using the groups argument as a vector with the group ID for each sample.
Keep in mind that the multi-group functionality is a rather advanced option that we
discourage for beginners. For more details on how the multi-group inference works,
read the FAQ section and
check this vignette.
N = ncol(data[[1]])
groups = c(rep("A",N/2), rep("B",N/2))
MOFAobject <- create_mofa(data, groups=groups)Plot the data overview
plot_data_overview(MOFAobject)
4.2 Long data.frame
A long data.frame with columns sample, feature, view, group (optional), value
might be the best format for complex data sets with multiple omics and potentially multiple groups of data. Also, there is no need to add rows that correspond to missing data:
filepath <- system.file("extdata", "test_data.RData", package = "MOFA2")
load(filepath)
head(dt)## sample feature view value
## 1: sample_0_group_1 feature_0_view_0 view_0 2.08
## 2: sample_1_group_1 feature_0_view_0 view_0 0.01
## 3: sample_2_group_1 feature_0_view_0 view_0 -0.11
## 4: sample_3_group_1 feature_0_view_0 view_0 -0.82
## 5: sample_4_group_1 feature_0_view_0 view_0 -1.13
## 6: sample_5_group_1 feature_0_view_0 view_0 -0.25Create the MOFA object
MOFAobject <- create_mofa(dt)## Creating MOFA object from a data.frame...print(MOFAobject)## Untrained MOFA model with the following characteristics:
## Number of views: 2
## Views names: view_0 view_1
## Number of features (per view): 1000 1000
## Number of groups: 1
## Groups names: single_group
## Number of samples (per group): 100
## Plot data overview
plot_data_overview(MOFAobject)
5 Define options
5.1 Define data options
- scale_groups: if groups have different ranges/variances,
it is good practice to scale each group to unit variance. Default is
FALSE - scale_views: if views have different ranges/variances,
it is good practice to scale each view to unit variance. Default is
FALSE
data_opts <- get_default_data_options(MOFAobject)
head(data_opts)## $scale_views
## [1] FALSE
##
## $scale_groups
## [1] FALSE
##
## $center_groups
## [1] TRUE
##
## $use_float32
## [1] TRUE
##
## $views
## [1] "view_0" "view_1"
##
## $groups
## [1] "single_group"5.2 Define model options
- num_factors: number of factors
- likelihoods: likelihood per view (options are “gaussian”, “poisson”, “bernoulli”). Default is “gaussian”.
- spikeslab_factors: use spike-slab sparsity prior in the factors?
Default is
FALSE. - spikeslab_weights: use spike-slab sparsity prior in the weights?
Default is
TRUE. - ard_factors: use ARD prior in the factors?
Default is
TRUEif using multiple groups. - ard_weights: use ARD prior in the weights?
Default is
TRUEif using multiple views.
Only change the default model options if you are familiar with the underlying mathematical model.
model_opts <- get_default_model_options(MOFAobject)
model_opts$num_factors <- 10
head(model_opts)## $likelihoods
## view_0 view_1
## "gaussian" "gaussian"
##
## $num_factors
## [1] 10
##
## $spikeslab_factors
## [1] FALSE
##
## $spikeslab_weights
## [1] FALSE
##
## $ard_factors
## [1] FALSE
##
## $ard_weights
## [1] TRUE5.3 Define training options
- maxiter: number of iterations. Default is 1000.
- convergence_mode: “fast” (default), “medium”, “slow”. For exploration, the fast mode is sufficient. For a final model, consider using medium" or even “slow”, but hopefully results should not change much.
- gpu_mode: use GPU mode? (needs cupy installed and a functional GPU).
- verbose: verbose mode?
train_opts <- get_default_training_options(MOFAobject)
head(train_opts)## $maxiter
## [1] 1000
##
## $convergence_mode
## [1] "fast"
##
## $drop_factor_threshold
## [1] -1
##
## $verbose
## [1] FALSE
##
## $startELBO
## [1] 1
##
## $freqELBO
## [1] 56 Build and train the MOFA object
Prepare the MOFA object
MOFAobject <- prepare_mofa(
object = MOFAobject,
data_options = data_opts,
model_options = model_opts,
training_options = train_opts
)Train the MOFA model. Remember that in this step the MOFA2 R package connets with the
mofapy2 Python package using reticulate. This is the source of most problems when
running MOFA. See our FAQ section
if you have issues. The output is saved in the file specified as outfile. If none is specified, the output is saved in a temporary location.
outfile = file.path(tempdir(),"model.hdf5")
MOFAobject.trained <- run_mofa(MOFAobject, outfile, use_basilisk=TRUE)## Warning: Output file /tmp/Rtmp4Ki5AA/model.hdf5 already exists, it will be replaced## Connecting to the mofapy2 package using basilisk.
## Set 'use_basilisk' to FALSE if you prefer to manually set the python binary using 'reticulate'.If everything is successful, you should observe an output analogous to the following:
######################################
## Training the model with seed 1 ##
######################################
Iteration 1: time=0.03, ELBO=-52650.68, deltaELBO=837116.802 (94.082647669%), Factors=10
(...)
Iteration 9: time=0.04, ELBO=-50114.43, deltaELBO=23.907 (0.002686924%), Factors=10
#######################
## Training finished ##
#######################
Saving model in `/var/folders/.../model.hdf5.../tmp/Rtmp4Ki5AA/model.hdf5.7 Downstream analysis
This finishes the tutorial on how to train a MOFA object from R. To continue with the downstream analysis, follow this tutorial
Session Info
sessionInfo()## R version 4.3.2 Patched (2023-11-13 r85521)
## Platform: x86_64-pc-linux-gnu (64-bit)
## Running under: Ubuntu 22.04.3 LTS
##
## Matrix products: default
## BLAS: /home/biocbuild/bbs-3.18-bioc/R/lib/libRblas.so
## LAPACK: /usr/lib/x86_64-linux-gnu/lapack/liblapack.so.3.10.0
##
## locale:
## [1] LC_CTYPE=en_US.UTF-8 LC_NUMERIC=C
## [3] LC_TIME=en_GB LC_COLLATE=C
## [5] LC_MONETARY=en_US.UTF-8 LC_MESSAGES=en_US.UTF-8
## [7] LC_PAPER=en_US.UTF-8 LC_NAME=C
## [9] LC_ADDRESS=C LC_TELEPHONE=C
## [11] LC_MEASUREMENT=en_US.UTF-8 LC_IDENTIFICATION=C
##
## time zone: America/New_York
## tzcode source: system (glibc)
##
## attached base packages:
## [1] stats graphics grDevices utils datasets methods base
##
## other attached packages:
## [1] data.table_1.14.10 pheatmap_1.0.12 lubridate_1.9.3 forcats_1.0.0
## [5] stringr_1.5.1 dplyr_1.1.4 purrr_1.0.2 readr_2.1.4
## [9] tidyr_1.3.0 tibble_3.2.1 ggplot2_3.4.4 tidyverse_2.0.0
## [13] MOFA2_1.12.1 BiocStyle_2.30.0
##
## loaded via a namespace (and not attached):
## [1] tidyselect_1.2.0 farver_2.1.1 filelock_1.0.3
## [4] fastmap_1.1.1 GGally_2.2.0 digest_0.6.33
## [7] timechange_0.2.0 lifecycle_1.0.4 magrittr_2.0.3
## [10] compiler_4.3.2 rlang_1.1.2 sass_0.4.8
## [13] tools_4.3.2 utf8_1.2.4 yaml_2.3.8
## [16] corrplot_0.92 ggsignif_0.6.4 knitr_1.45
## [19] S4Arrays_1.2.0 labeling_0.4.3 reticulate_1.34.0
## [22] DelayedArray_0.28.0 plyr_1.8.9 RColorBrewer_1.1-3
## [25] abind_1.4-5 HDF5Array_1.30.0 Rtsne_0.17
## [28] withr_2.5.2 BiocGenerics_0.48.1 grid_4.3.2
## [31] stats4_4.3.2 fansi_1.0.6 ggpubr_0.6.0
## [34] colorspace_2.1-0 Rhdf5lib_1.24.1 scales_1.3.0
## [37] cli_3.6.2 mvtnorm_1.2-4 rmarkdown_2.25
## [40] crayon_1.5.2 generics_0.1.3 reshape2_1.4.4
## [43] tzdb_0.4.0 cachem_1.0.8 rhdf5_2.46.1
## [46] splines_4.3.2 zlibbioc_1.48.0 parallel_4.3.2
## [49] BiocManager_1.30.22 XVector_0.42.0 matrixStats_1.2.0
## [52] basilisk_1.14.1 vctrs_0.6.5 Matrix_1.6-4
## [55] carData_3.0-5 jsonlite_1.8.8 dir.expiry_1.10.0
## [58] car_3.1-2 bookdown_0.37 IRanges_2.36.0
## [61] hms_1.1.3 S4Vectors_0.40.2 rstatix_0.7.2
## [64] ggrepel_0.9.4 irlba_2.3.5.1 magick_2.8.2
## [67] jquerylib_0.1.4 glue_1.6.2 codetools_0.2-19
## [70] ggstats_0.5.1 cowplot_1.1.2 uwot_0.1.16
## [73] RcppAnnoy_0.0.21 stringi_1.8.3 gtable_0.3.4
## [76] munsell_0.5.0 pillar_1.9.0 basilisk.utils_1.14.1
## [79] htmltools_0.5.7 rhdf5filters_1.14.1 R6_2.5.1
## [82] evaluate_0.23 lattice_0.22-5 highr_0.10
## [85] backports_1.4.1 png_0.1-8 broom_1.0.5
## [88] bslib_0.6.1 Rcpp_1.0.11 nlme_3.1-164
## [91] SparseArray_1.2.3 mgcv_1.9-1 xfun_0.41
## [94] MatrixGenerics_1.14.0 pkgconfig_2.0.3