# Training non default regression models

#### Etienne Becht

#### June 2020

# Introduction

This vignette explains how to specify non-default machine learning frameworks and their hyperparameters when applying Infinity Flow. We will assume here that the basic usage of Infinity Flow has already been read, if you are not familiar with this material I suggest you first look at the basic usage vignette

This vignette will cover:

1. Loading the example data
2. Note on package design
3. The `regression_functions` argument
4. The `extra_args_regression_params` argument
5. Neural networks

# Loading the example data

Here is a single R code chunk that recapitulates all of the data preparation covered in the basic usage vignette.

    if(!require(devtools)){
        install.packages("devtools")
    }
    if(!require(infinityFlow)){
        library(devtools)
        install_github("ebecht/infinityFlow")
    }

    library(infinityFlow)

    data(steady_state_lung)
    data(steady_state_lung_annotation)
    data(steady_state_lung_backbone_specification)

    dir <- file.path(tempdir(), "infinity_flow_example")
    input_dir <- file.path(dir, "fcs")
    write.flowSet(steady_state_lung, outdir = input_dir)
    #> [1] "/tmp/RtmpePd0Os/infinity_flow_example/fcs"

    write.csv(steady_state_lung_backbone_specification, file = file.path(dir, "backbone_selection_file.csv"), row.names = FALSE)

    path_to_fcs <- file.path(dir, "fcs")
    path_to_output <- file.path(dir, "output")
    path_to_intermediary_results <- file.path(dir, "tmp")
    backbone_selection_file <- file.path(dir, "backbone_selection_file.csv")

    targets <- steady_state_lung_annotation$Infinity_target
    names(targets) <- rownames(steady_state_lung_annotation)
    isotypes <- steady_state_lung_annotation$Infinity_isotype
    names(isotypes) <- rownames(steady_state_lung_annotation)

    input_events_downsampling <- 1000
    prediction_events_downsampling <- 500
    cores = 1L

# Note on package design

The `infinity_flow()` function which encapsulates the complete Infinity Flow computational pipeline uses two arguments to respectively select regression models and their hyperparameters. These two arguments are both lists, and should have the same length. The idea is that the first list, `regression_functions` will be a list of model templates (XGBoost, Neural Networks, SVMs…) to train, while the second will be used to specify their hyperparameters. The list of templates is then fit to the data using parallel computing with socketing (using the `parallel` package through the `pbapply` package), which is more memory efficient.

# The `regression_functions` argument

This argument is a list of functions which specifies how many models to train per well and which ones. Each type of machine learning model is supported through a wrapper in the infinityFlow package, and has a name of the form `fitter_*`. See below for the complete list:

    print(grep("fitter_", ls("package:infinityFlow"), value = TRUE))
    #> [1] "fitter_glmnet"  "fitter_linear"  "fitter_nn"      "fitter_svm"
    #> [5] "fitter_xgboost"

fitter_ function | Backend | Model type
--- | --- | ---
fitter_xgboost | XGBoost | Gradient boosted trees
fitter_nn | Tensorflow/Keras | Neural networks
fitter_svm | e1071 | Support vector machines
fitter_glmnet | glmnet | Generalized linear and polynomial models
fitter_lm | stats | Linear and polynomial models

These functions rely on optional package dependencies (so that you do not need to install e.g. Keras if you are not planning to use it). We need to make sure that these dependencies are however met:

           optional_dependencies <- c("glmnetUtils", "e1071")
           unmet_dependencies <- setdiff(optional_dependencies, rownames(installed.packages()))
           if(length(unmet_dependencies) > 0){
               install.packages(unmet_dependencies)
           }
           for(pkg in optional_dependencies){
               library(pkg, character.only = TRUE)
           }

In this vignette we will train all of these models. Note that if you do it on your own data, it make take quite a bit of memory (remember that the output expression matrix will be a numeric matrix of size `(prediction_events_downsampling x number of wells) rows x (number of wells x number of models)`.

To train multiple models we create a list of these fitter_* functions and assign this to the `regression_functions` argument that will be fed to the `infinity_flow` function. The names of this list will be used to name your models.

    regression_functions <- list(
        XGBoost = fitter_xgboost, # XGBoost
        SVM = fitter_svm, # SVM
        LASSO2 = fitter_glmnet, # L1-penalized 2nd degree polynomial model
        LM = fitter_linear # Linear model
    )

# The `extra_args_regression_params` argument

This argument is a list of list (so of the form `list(list(...), list(...), etc.)`) of length `length(regression_functions)`. Each element of the extra_args_regression_params object is thus a list. This lower-level list will be used to pass named arguments to the machine learning fitting function. The list of `extra_args_regression_params` is matched with the list of machine learning models `regression_functions` using the order of the elements in these two lists (e.g. the first regression model is matched with the first element of the list of arguments, then the seconds elements are matched together, etc…).

    backbone_size <- table(read.csv(backbone_selection_file)[,"type"])["backbone"]
    extra_args_regression_params <- list(
         ## Passed to the first element of `regression_functions`, e.g. XGBoost. See ?xgboost for which parameters can be passed through this list
        list(nrounds = 500, eta = 0.05),

        # ## Passed to the second element of `regression_functions`, e.g. neural networks through keras::fit. See https://keras.rstudio.com/articles/tutorial_basic_regression.html
        # list(
        #         object = { ## Specifies the network's architecture, loss function and optimization method
        #             model = keras_model_sequential()
        #             model %>%
        #                 layer_dense(units = backbone_size, activation = "relu", input_shape = backbone_size) %>%
        #                 layer_dense(units = backbone_size, activation = "relu", input_shape = backbone_size) %>%
        #                 layer_dense(units = 1, activation = "linear")
        #             model %>%
        #                 compile(loss = "mean_squared_error", optimizer = optimizer_sgd(lr = 0.005))
        #             serialize_model(model)
        #         },
        #         epochs = 1000, ## Number of maximum training epochs. The training is however stopped early if the loss on the validation set does not improve for 20 epochs. This early stopping is hardcoded in fitter_nn.
        #         validation_split = 0.2, ## Fraction of the training data used to monitor validation loss
        #         verbose = 0,
        #         batch_size = 128 ## Size of the minibatches for training.
        # ),

        # Passed to the third element, SVMs. See help(svm, "e1071") for possible arguments
        list(type = "nu-regression", cost = 8, nu=0.5, kernel="radial"),

        # Passed to the fourth element, fitter_glmnet. This should contain a mandatory argument `degree` which specifies the degree of the polynomial model (1 for linear, 2 for quadratic etc...). Here we use degree = 2 corresponding to our LASSO2 model Other arguments are passed to getS3method("cv.glmnet", "formula"),
        list(alpha = 1, nfolds=10, degree = 2),

        # Passed to the fourth element, fitter_linear. This only accepts a degree argument specifying the degree of the polynomial model. Here we use degree = 1 corresponding to a linear model.
        list(degree = 1)
    )

We can now run the pipeline with these custom arguments to train all the models.

    if(length(regression_functions) != length(extra_args_regression_params)){
        stop("Number of models and number of lists of hyperparameters mismatch")
    }
    imputed_data <- infinity_flow(
        regression_functions = regression_functions,
        extra_args_regression_params = extra_args_regression_params,
        path_to_fcs = path_to_fcs,
        path_to_output = path_to_output,
        path_to_intermediary_results = path_to_intermediary_results,
        backbone_selection_file = backbone_selection_file,
        annotation = targets,
        isotype = isotypes,
        input_events_downsampling = input_events_downsampling,
        prediction_events_downsampling = prediction_events_downsampling,
        verbose = TRUE,
        cores = cores
    )
    #> Using directories...
    #>  input: /tmp/RtmpePd0Os/infinity_flow_example/fcs
    #>  intermediary: /tmp/RtmpePd0Os/infinity_flow_example/tmp
    #>  subset: /tmp/RtmpePd0Os/infinity_flow_example/tmp/subsetted_fcs
    #>  rds: /tmp/RtmpePd0Os/infinity_flow_example/tmp/rds
    #>  annotation: /tmp/RtmpePd0Os/infinity_flow_example/tmp/annotation.csv
    #>  output: /tmp/RtmpePd0Os/infinity_flow_example/output
    #> Parsing and subsampling input data
    #>  Downsampling to 1000 events per input file
    #>  Concatenating expression matrices
    #>  Writing to disk
    #> Logicle-transforming the data
    #>  Backbone data
    #>  Exploratory data
    #>  Writing to disk
    #>  Transforming expression matrix
    #>  Writing to disk
    #> Harmonizing backbone data
    #>  Scaling expression matrices
    #>  Writing to disk
    #> Fitting regression models
    #>  Randomly selecting 50% of the subsetted input files to fit models
    #>  Fitting...
    #>      XGBoost
    #>
    #>  22.13754 seconds
    #>      SVM
    #>
    #>  1.56838 seconds
    #>      LASSO2
    #>
    #>  12.95937 seconds
    #>      LM
    #>
    #>  0.121284 seconds
    #> Imputing missing measurements
    #>  Randomly drawing events to predict from the test set
    #>  Imputing...
    #>      XGBoost
    #>
    #>  1.82065 seconds
    #>      SVM
    #>
    #>  1.750707 seconds
    #>      LASSO2
    #>
    #>  1.844726 seconds
    #>      LM
    #>
    #>  0.08630109 seconds
    #>  Concatenating predictions
    #>  Writing to disk
    #> Performing dimensionality reduction
    #> 10:45:16 UMAP embedding parameters a = 1.262 b = 1.003
    #> 10:45:16 Read 5000 rows and found 17 numeric columns
    #> 10:45:16 Using Annoy for neighbor search, n_neighbors = 15
    #> 10:45:16 Building Annoy index with metric = euclidean, n_trees = 50
    #> 0%   10   20   30   40   50   60   70   80   90   100%
    #> [----|----|----|----|----|----|----|----|----|----|
    #> **************************************************|
    #> 10:45:17 Writing NN index file to temp file /tmp/RtmpePd0Os/file85ae5329f52d
    #> 10:45:17 Searching Annoy index using 1 thread, search_k = 1500
    #> 10:45:18 Annoy recall = 100%
    #> 10:45:19 Commencing smooth kNN distance calibration using 1 thread with target n_neighbors = 15
    #> 10:45:20 Initializing from normalized Laplacian + noise (using irlba)
    #> 10:45:20 Commencing optimization for 1000 epochs, with 101766 positive edges using 1 thread
    #> 10:45:41 Optimization finished
    #> Exporting results
    #>  Transforming predictions back to a linear scale
    #>  Exporting FCS files (1 per well)
    #> Plotting
    #>  Chopping off the top and bottom 0.005 quantiles
    #>  Shuffling the order of cells (rows)
    #>  Producing plot
    #> Background correcting
    #>  Transforming background-corrected predictions. (Use logarithm to visualize)
    #>  Exporting FCS files (1 per well)
    #> Plotting
    #>  Chopping off the top and bottom 0.005 quantiles
    #>  Shuffling the order of cells (rows)
    #>  Producing plot

Our model names are appended to the predicted markers in the output. For more discussion about the outputs (including output files written to disk and plots), see the basic usage vignette

       print(imputed_data$bgc[1:2, ])
    #>     FSC-A       FSC-H      FSC-W   SSC-A        SSC-H      SSC-W CD69-CD301b
    #> 1 52791.9 -0.03905850 -0.1031236 4331.57 -0.007255098 -0.1183022  0.70785284
    #> 2 49252.2 -0.07876111 -0.8108239 1780.66 -1.257845408 -0.9648148 -0.05681061
    #>      Zombie      MHCII        CD4       CD44        CD8       CD11c
    #> 1 539.40082 -0.8931286 1.78052493  0.6013719  0.3874686 -2.41721342
    #> 2 -13.60832 -0.6730070 0.01557133 -1.0178377 -0.6396222  0.03860166
    #>          CD11b       F480       Ly6C    Lineage  CD45a488 FJComp-PE(yg)-A
    #> 1  0.432357303 -1.2049765 -1.3564584 1.53710514 0.7212552       2.3103466
    #> 2 -0.001956604 -0.9236521  0.7308933 0.05430416 0.3671689       0.9698867
    #>        CD24      CD103     Time CD137.LASSO2_bgc CD137.LM_bgc CD137.SVM_bgc
    #> 1 -1.441091 -1.0354256 3494.209      -0.12268145   -0.6225719    -0.6213091
    #> 2 -1.766836  0.4153181 3417.409       0.02751279    0.1100712    -0.3520929
    #>   CD137.XGBoost_bgc CD28.LASSO2_bgc CD28.LM_bgc CD28.SVM_bgc CD28.XGBoost_bgc
    #> 1        -0.1613588       0.8537910   0.1912968    0.9443784        0.9803510
    #> 2         0.2037864      -0.2415615   0.2004835   -0.1195522        0.0485431
    #>   CD49b(pan-NK).LASSO2_bgc CD49b(pan-NK).LM_bgc CD49b(pan-NK).SVM_bgc
    #> 1              -0.09593482           -0.1807528            -0.5808465
    #> 2               0.43043289            0.2458367             0.7738870
    #>   CD49b(pan-NK).XGBoost_bgc KLRG1.LASSO2_bgc KLRG1.LM_bgc KLRG1.SVM_bgc
    #> 1                -0.4260228        0.3171454    0.3548617     0.6270122
    #> 2                 0.9223872        0.4890908    0.2834585     0.6974098
    #>   KLRG1.XGBoost_bgc Ly-49c/F/I/H.LASSO2_bgc Ly-49c/F/I/H.LM_bgc
    #> 1       -0.06216572               0.1684994          0.23734129
    #> 2        0.41351539              -0.1372351          0.03149387
    #>   Ly-49c/F/I/H.SVM_bgc Ly-49c/F/I/H.XGBoost_bgc Podoplanin.LASSO2_bgc
    #> 1           -1.4134629               -0.4056362            -0.9566363
    #> 2           -0.4964237                0.3703906            -0.2852313
    #>   Podoplanin.LM_bgc Podoplanin.SVM_bgc Podoplanin.XGBoost_bgc SHIgG.LASSO2_bgc
    #> 1       -0.59183777         -1.0906160             -0.6393116     2.117139e-17
    #> 2        0.03212325          0.1273272             -0.3055337     2.117139e-17
    #>    SHIgG.LM_bgc SHIgG.SVM_bgc SHIgG.XGBoost_bgc SSEA-3.LASSO2_bgc SSEA-3.LM_bgc
    #> 1 -4.306325e-16  4.063431e-16       1.86559e-16         0.2516162    -0.1554570
    #> 2  1.139460e-15 -2.216939e-16       1.86559e-16        -0.3957177    -0.1254614
    #>   SSEA-3.SVM_bgc SSEA-3.XGBoost_bgc TCR Vg3.LASSO2_bgc TCR Vg3.LM_bgc
    #> 1      0.4223681          0.1608146         -0.1513909     -0.2389296
    #> 2     -0.7481376         -0.3097776         -0.1604972     -0.0211420
    #>   TCR Vg3.SVM_bgc TCR Vg3.XGBoost_bgc rIgM.LASSO2_bgc   rIgM.LM_bgc
    #> 1      -0.3023001           -0.458102    2.892201e-16 -6.751540e-16
    #> 2      -0.4609283           -0.134714   -3.388169e-16  1.098922e-16
    #>    rIgM.SVM_bgc rIgM.XGBoost_bgc    UMAP1    UMAP2 PE_id
    #> 1 -5.075502e-16     3.482785e-16  50.5329 394.6491     1
    #> 2  1.204868e-16    -2.797585e-16 145.4289 530.9091     1

# Neural networks

Neural networks won’t build in knitr for me but here is an example of the syntax if you want to use them.

Note: there is an issue with serialization of the neural networks and socketing since I updated to R-4.0.1. If you want to use neural networks, please make sure to set

    cores = 1L

    optional_dependencies <- c("keras", "tensorflow")
    unmet_dependencies <- setdiff(optional_dependencies, rownames(installed.packages()))
    if(length(unmet_dependencies) > 0){
         install.packages(unmet_dependencies)
    }
    for(pkg in optional_dependencies){
        library(pkg, character.only = TRUE)
    }

    invisible(eval(try(keras_model_sequential()))) ## avoids conflicts with flowCore...

    if(!is_keras_available()){
         install_keras() ## Instal keras unsing the R interface - can take a while
    }

    if (!requireNamespace("BiocManager", quietly = TRUE)){
        install.packages("BiocManager")
    }
    BiocManager::install("infinityFlow")

    library(infinityFlow)

    data(steady_state_lung)
    data(steady_state_lung_annotation)
    data(steady_state_lung_backbone_specification)

    dir <- file.path(tempdir(), "infinity_flow_example")
    input_dir <- file.path(dir, "fcs")
    write.flowSet(steady_state_lung, outdir = input_dir)

    write.csv(steady_state_lung_backbone_specification, file = file.path(dir, "backbone_selection_file.csv"), row.names = FALSE)

    path_to_fcs <- file.path(dir, "fcs")
    path_to_output <- file.path(dir, "output")
    path_to_intermediary_results <- file.path(dir, "tmp")
    backbone_selection_file <- file.path(dir, "backbone_selection_file.csv")

    targets <- steady_state_lung_annotation$Infinity_target
    names(targets) <- rownames(steady_state_lung_annotation)
    isotypes <- steady_state_lung_annotation$Infinity_isotype
    names(isotypes) <- rownames(steady_state_lung_annotation)

    input_events_downsampling <- 1000
    prediction_events_downsampling <- 500

    ## Passed to fitter_nn, e.g. neural networks through keras::fit. See https://keras.rstudio.com/articles/tutorial_basic_regression.html
    regression_functions <- list(NN = fitter_nn)

    backbone_size <- table(read.csv(backbone_selection_file)[,"type"])["backbone"]
    extra_args_regression_params <- list(
            list(
            object = { ## Specifies the network's architecture, loss function and optimization method
            model = keras_model_sequential()
            model %>%
            layer_dense(units = backbone_size, activation = "relu", input_shape = backbone_size) %>%
            layer_dense(units = backbone_size, activation = "relu", input_shape = backbone_size) %>%
            layer_dense(units = 1, activation = "linear")
            model %>%
            compile(loss = "mean_squared_error", optimizer = optimizer_sgd(lr = 0.005))
            serialize_model(model)
            },
            epochs = 1000, ## Number of maximum training epochs. The training is however stopped early if the loss on the validation set does not improve for 20 epochs. This early stopping is hardcoded in fitter_nn.
            validation_split = 0.2, ## Fraction of the training data used to monitor validation loss
            verbose = 0,
            batch_size = 128 ## Size of the minibatches for training.
        )
    )

    imputed_data <- infinity_flow(
        regression_functions = regression_functions,
        extra_args_regression_params = extra_args_regression_params,
        path_to_fcs = path_to_fcs,
        path_to_output = path_to_output,
        path_to_intermediary_results = path_to_intermediary_results,
        backbone_selection_file = backbone_selection_file,
        annotation = targets,
        isotype = isotypes,
        input_events_downsampling = input_events_downsampling,
        prediction_events_downsampling = prediction_events_downsampling,
        verbose = TRUE,
        cores = 1L
    )

# Conclusion

Thank you for following this vignette, I hope you made it through the end without too much headache and that it was informative. General questions about proper usage of the package are best asked on the Bioconductor support site to maximize visibility for future users. If you encounter bugs, feel free to raise an issue on infinityFlow’s github.

# Information about the R session when this vignette was built

    sessionInfo()
    #> R Under development (unstable) (2022-12-12 r83436)
    #> Platform: aarch64-apple-darwin20 (64-bit)
    #> Running under: macOS Monterey 12.6.1
    #>
    #> Matrix products: default
    #> BLAS:   /Library/Frameworks/R.framework/Versions/4.3-arm64/Resources/lib/libRblas.0.dylib
    #> LAPACK: /Library/Frameworks/R.framework/Versions/4.3-arm64/Resources/lib/libRlapack.dylib;  LAPACK version 3.10.1
    #>
    #> Random number generation:
    #>  RNG:     L'Ecuyer-CMRG
    #>  Normal:  Inversion
    #>  Sample:  Rejection
    #>
    #> locale:
    #> [1] C/en_US.UTF-8/en_US.UTF-8/C/en_US.UTF-8/en_US.UTF-8
    #>
    #> time zone: America/New_York
    #> tzcode source: internal
    #>
    #> attached base packages:
    #> [1] stats     graphics  grDevices utils     datasets  methods   base
    #>
    #> other attached packages:
    #> [1] e1071_1.7-12       glmnetUtils_1.1.8  infinityFlow_1.9.0 flowCore_2.11.0
    #>
    #> loaded via a namespace (and not attached):
    #>  [1] sass_0.4.4          generics_0.1.3      class_7.3-20.1
    #>  [4] gtools_3.9.4        shape_1.4.6         stringi_1.7.8
    #>  [7] lattice_0.20-45     digest_0.6.31       magrittr_2.0.3
    #> [10] evaluate_0.18       grid_4.3.0          iterators_1.0.14
    #> [13] fastmap_1.1.0       xgboost_1.6.0.1     foreach_1.5.2
    #> [16] jsonlite_1.8.4      Matrix_1.5-3        glmnet_4.1-6
    #> [19] survival_3.4-0      pbapply_1.6-0       codetools_0.2-18
    #> [22] jquerylib_0.1.4     cli_3.4.1           rlang_1.0.6
    #> [25] RProtoBufLib_2.11.0 Biobase_2.59.0      RcppAnnoy_0.0.20
    #> [28] uwot_0.1.14         matlab_1.0.4        splines_4.3.0
    #> [31] cachem_1.0.6        yaml_2.3.6          cytolib_2.11.0
    #> [34] tools_4.3.0         raster_3.6-11       parallel_4.3.0
    #> [37] BiocGenerics_0.45.0 vctrs_0.5.1         R6_2.5.1
    #> [40] png_0.1-8           proxy_0.4-27        matrixStats_0.63.0
    #> [43] stats4_4.3.0        lifecycle_1.0.3     stringr_1.5.0
    #> [46] S4Vectors_0.37.3    irlba_2.3.5.1       terra_1.6-47
    #> [49] bslib_0.4.1         glue_1.6.2          data.table_1.14.6
    #> [52] Rcpp_1.0.9          xfun_0.35           knitr_1.41
    #> [55] htmltools_0.5.4     rmarkdown_2.18      compiler_4.3.0
    #> [58] sp_1.5-1