Testing non-linear effects

Introduction

Typical analysis using regression models assumes a linear affect of the covariate on the response. Here we consider testing non-linear effects in the case of 1) continuous and 2) ordered categorical variables.

We demonstrate this feature on a lightly modified analysis of PBMCs from 8 individuals stimulated with interferon-β (Kang, et al, 2018, Nature Biotech).

Standard processing

Here is the code from the main vignette:

library(dreamlet)
library(muscat)
library(ExperimentHub)
library(scater)

# Download data, specifying EH2259 for the Kang, et al study
eh <- ExperimentHub()
sce <- eh[["EH2259"]]

# only keep singlet cells with sufficient reads
sce <- sce[rowSums(counts(sce) > 0) > 0, ]
sce <- sce[, colData(sce)$multiplets == "singlet"]

# compute QC metrics
qc <- perCellQCMetrics(sce)

# remove cells with few or many detected genes
ol <- isOutlier(metric = qc$detected, nmads = 2, log = TRUE)
sce <- sce[, !ol]

# set variable indicating stimulated (stim) or control (ctrl)
sce$StimStatus <- sce$stim

sce$id <- paste0(sce$StimStatus, sce$ind)

# Create pseudobulk
pb <- aggregateToPseudoBulk(sce,
  assay = "counts",
  cluster_id = "cell",
  sample_id = "id",
  verbose = FALSE
)

Continuous variable

Consider the continuous variable Age. Typical analysis only considers linear effects using a single regression coefficient, but we also want to consider the non-linear effects of age. We can peform a basis expansion using splines instead use 3 coefficients to model the age effect.

# Simulate age between 18 and 65
pb$Age <- runif(ncol(pb), 18, 65)

# formula included non-linear effects of Age
# by using a natural spline of degree 3
# This corresponds to using 3 coefficients instead of 1
form <- ~ splines::ns(Age, 3)

# Normalize and apply voom/voomWithDreamWeights
res.proc <- processAssays(pb, form, min.count = 5)

# Differential expression analysis within each assay
res.dl <- dreamlet(res.proc, form)

# The spline has degree 3, so there are 3 coefficients
# estimated for Age effects
coefNames(res.dl)
## [1] "(Intercept)"          "splines::ns(Age, 3)1" "splines::ns(Age, 3)2"
## [4] "splines::ns(Age, 3)3"
# Jointly test effects of the 3 spline components
# The test of the 3 coefficients is performed with an F-statistic
topTable(res.dl, coef = coefNames(res.dl)[2:4], number = 3)
## DataFrame with 3 rows and 9 columns
##         assay          ID splines..ns.Age..3.1 splines..ns.Age..3.2
##   <character> <character>            <numeric>            <numeric>
## 1 CD4 T cells     PIP5K1A             0.822322              6.00327
## 2 CD4 T cells      TATDN1             2.122103              4.86031
## 3 CD4 T cells       CASC4             0.975087              6.39073
##   splines..ns.Age..3.3   AveExpr         F     P.Value adj.P.Val
##              <numeric> <numeric> <numeric>   <numeric> <numeric>
## 1              1.14697   3.17359   23.4020 3.09271e-06 0.0448195
## 2              1.51265   3.96489   14.2325 7.16916e-05 0.4347344
## 3              1.04721   3.54874   13.6922 8.99947e-05 0.4347344

Ordered categorical

We can also test non-linear effects in the case of categorical variables with a natural ordering to the categories. Consider time course data with 4 time points. Each time point is a category and has a natural ordering from first to last.

We have multiple options to model the time course.

  • Continuous: Modeling time point as a continuous variable uses a single regression coefficient to model the linear effects of the time course. This is simple, models the order of the time points, but ignores non-linear effects

    Model using as.numeric(TimePoint)

  • Categorical: Including time point as a typical categorical variable uses estimated the mean response value for each category. So it estimates 4 coefficients. While this can be useful for comparing two categories, it ignores the order of the time points.

    Model using factor(TimePoint)

  • Ordered categorical: Here, the trend across ordered time points is modled using orthogonal polynomials. The trend is decomposed into independent linear, quadratic, etc., effects that can be tested either jointly or by themselves.

    Model using:

    ord <- c("time_1", "time_2", "time_3", "time_4")
    ordered(factor(TimePoint), ord)

Here we simulated 4 time points, and perform differential expression analysis.

# Consider data generated across 4 time points
# While there are no time points in the real data
# we can add some for demonstration purposes
pb$TimePoint <- ordered(paste0("time_", rep(1:4, 4)))

# examine the ordering
pb$TimePoint
##  [1] time_1 time_2 time_3 time_4 time_1 time_2 time_3 time_4 time_1 time_2
## [11] time_3 time_4 time_1 time_2 time_3 time_4
## Levels: time_1 < time_2 < time_3 < time_4
# Use formula including time point
form <- ~TimePoint

# Normalize and apply voom/voomWithDreamWeights
res.proc <- processAssays(pb, form, min.count = 5)

# Differential expression analysis within each assay
res.dl <- dreamlet(res.proc, form)

# Examine the coefficient estimated
# for TimePoint it estimates
# linear (i.e. L)
# quadratic (i.e. Q)
# and cubic (i.e. C) effects
coefNames(res.dl)
## [1] "(Intercept)" "TimePoint.L" "TimePoint.Q" "TimePoint.C"
# Test only linear effect
topTable(res.dl, coef = "TimePoint.L", number = 3)
## DataFrame with 3 rows and 9 columns
##         assay          ID     logFC   AveExpr         t     P.Value adj.P.Val
##   <character> <character> <numeric> <numeric> <numeric>   <numeric> <numeric>
## 1 CD4 T cells        DCXR -0.671645   6.52867  -5.42880 4.78666e-05  0.393058
## 2 CD4 T cells        GGA2 -0.890112   4.98987  -5.26372 6.69370e-05  0.393058
## 3 CD8 T cells        FTH1 -0.759875  14.55380  -4.99199 8.13672e-05  0.393058
##           B     z.std
##   <numeric> <numeric>
## 1  1.686147  -5.42880
## 2  0.936906  -5.26372
## 3  1.627753  -4.99199
# Test linear, quadratic and cubic effcts
coefs <- c("TimePoint.L", "TimePoint.Q", "TimePoint.C")
topTable(res.dl, coef = coefs, number = 3)
## DataFrame with 3 rows and 9 columns
##         assay                   ID TimePoint.L TimePoint.Q TimePoint.C
##   <character>          <character>   <numeric>   <numeric>   <numeric>
## 1 CD8 T cells                 CD52    0.903032   -1.528456   -0.984065
## 2 CD8 T cells CCL5_ENSG00000161570    0.626201   -0.986502   -0.645468
## 3 CD8 T cells                  CD2    0.887578   -1.152415   -0.195616
##     AveExpr         F     P.Value adj.P.Val
##   <numeric> <numeric>   <numeric> <numeric>
## 1   8.69458   16.0946 1.90868e-05 0.0853298
## 2  11.88811   16.0433 1.94982e-05 0.0853298
## 3   9.67181   15.7793 2.17777e-05 0.0853298

Sample filtering

Due to variation in cell and read count for each sample, processAssays() filters out some sample. This filtering is summarized here:

details(res.dl)
##               assay n_retain    formula formDropsTerms n_genes n_errors
## 1           B cells       16 ~TimePoint          FALSE    1961        0
## 2   CD14+ Monocytes       16 ~TimePoint          FALSE    3087        0
## 3       CD4 T cells       16 ~TimePoint          FALSE    5262        0
## 4       CD8 T cells       16 ~TimePoint          FALSE    1030        0
## 5   Dendritic cells       13 ~TimePoint          FALSE     164        0
## 6 FCGR3A+ Monocytes       16 ~TimePoint          FALSE    1160        0
## 7    Megakaryocytes       13 ~TimePoint          FALSE     172        0
## 8          NK cells       16 ~TimePoint          FALSE    1656        0
##   error_initial
## 1         FALSE
## 2         FALSE
## 3         FALSE
## 4         FALSE
## 5         FALSE
## 6         FALSE
## 7         FALSE
## 8         FALSE

Whle all 16 samples are detained in B cells, only 9 are retained for megakaryocytes. This can result in a time point being dropped, and so the polynomial expansion for some cell types can have a lower degree. The combined results will then have NA values for these coefficients. For example, for TIMP1 in Megakaryocytes above there is not enought data to fit the cubic term, so TimePoint.C is NA.

Session Info

## R version 4.4.2 (2024-10-31)
## Platform: x86_64-pc-linux-gnu
## Running under: Ubuntu 24.04.1 LTS
## 
## Matrix products: default
## BLAS:   /usr/lib/x86_64-linux-gnu/openblas-pthread/libblas.so.3 
## LAPACK: /usr/lib/x86_64-linux-gnu/openblas-pthread/libopenblasp-r0.3.26.so;  LAPACK version 3.12.0
## 
## locale:
##  [1] LC_CTYPE=en_US.UTF-8       LC_NUMERIC=C              
##  [3] LC_TIME=en_US.UTF-8        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: Etc/UTC
## tzcode source: system (glibc)
## 
## attached base packages:
## [1] stats4    stats     graphics  grDevices utils     datasets  methods  
## [8] base     
## 
## other attached packages:
##  [1] GO.db_3.20.0                org.Hs.eg.db_3.20.0        
##  [3] AnnotationDbi_1.69.0        zenith_1.9.0               
##  [5] muscData_1.20.0             scater_1.35.0              
##  [7] scuttle_1.17.0              ExperimentHub_2.15.0       
##  [9] AnnotationHub_3.15.0        BiocFileCache_2.15.0       
## [11] dbplyr_2.5.0                muscat_1.21.0              
## [13] dreamlet_1.5.0              SingleCellExperiment_1.29.1
## [15] SummarizedExperiment_1.37.0 Biobase_2.67.0             
## [17] GenomicRanges_1.59.1        GenomeInfoDb_1.43.2        
## [19] IRanges_2.41.2              S4Vectors_0.45.2           
## [21] BiocGenerics_0.53.3         generics_0.1.3             
## [23] MatrixGenerics_1.19.0       matrixStats_1.4.1          
## [25] variancePartition_1.37.1    BiocParallel_1.41.0        
## [27] limma_3.63.2                ggplot2_3.5.1              
## [29] BiocStyle_2.35.0           
## 
## loaded via a namespace (and not attached):
##   [1] bitops_1.0-9              httr_1.4.7               
##   [3] RColorBrewer_1.1-3        doParallel_1.0.17        
##   [5] Rgraphviz_2.51.0          numDeriv_2016.8-1.1      
##   [7] tools_4.4.2               sctransform_0.4.1        
##   [9] backports_1.5.0           utf8_1.2.4               
##  [11] R6_2.5.1                  metafor_4.6-0            
##  [13] mgcv_1.9-1                GetoptLong_1.0.5         
##  [15] withr_3.0.2               prettyunits_1.2.0        
##  [17] gridExtra_2.3             cli_3.6.3                
##  [19] labeling_0.4.3            sass_0.4.9               
##  [21] KEGGgraph_1.67.0          SQUAREM_2021.1           
##  [23] mvtnorm_1.3-2             blme_1.0-6               
##  [25] mixsqp_0.3-54             parallelly_1.40.1        
##  [27] invgamma_1.1              RSQLite_2.3.9            
##  [29] shape_1.4.6.1             gtools_3.9.5             
##  [31] dplyr_1.1.4               Matrix_1.7-1             
##  [33] metadat_1.2-0             ggbeeswarm_0.7.2         
##  [35] abind_1.4-8               lifecycle_1.0.4          
##  [37] yaml_2.3.10               edgeR_4.5.1              
##  [39] mathjaxr_1.6-0            gplots_3.2.0             
##  [41] SparseArray_1.7.2         grid_4.4.2               
##  [43] blob_1.2.4                crayon_1.5.3             
##  [45] lattice_0.22-6            beachmat_2.23.5          
##  [47] msigdbr_7.5.1             annotate_1.85.0          
##  [49] KEGGREST_1.47.0           sys_3.4.3                
##  [51] maketools_1.3.1           pillar_1.10.0            
##  [53] knitr_1.49                ComplexHeatmap_2.23.0    
##  [55] rjson_0.2.23              boot_1.3-31              
##  [57] corpcor_1.6.10            future.apply_1.11.3      
##  [59] codetools_0.2-20          glue_1.8.0               
##  [61] data.table_1.16.4         vctrs_0.6.5              
##  [63] png_0.1-8                 Rdpack_2.6.2             
##  [65] gtable_0.3.6              assertthat_0.2.1         
##  [67] cachem_1.1.0              xfun_0.49                
##  [69] mime_0.12                 rbibutils_2.3            
##  [71] S4Arrays_1.7.1            Rfast_2.1.0              
##  [73] reformulas_0.4.0          iterators_1.0.14         
##  [75] statmod_1.5.0             nlme_3.1-166             
##  [77] pbkrtest_0.5.3            bit64_4.5.2              
##  [79] filelock_1.0.3            progress_1.2.3           
##  [81] EnvStats_3.0.0            bslib_0.8.0              
##  [83] TMB_1.9.15                irlba_2.3.5.1            
##  [85] vipor_0.4.7               KernSmooth_2.23-24       
##  [87] colorspace_2.1-1          rmeta_3.0                
##  [89] DBI_1.2.3                 DESeq2_1.47.1            
##  [91] tidyselect_1.2.1          curl_6.0.1               
##  [93] bit_4.5.0.1               compiler_4.4.2           
##  [95] graph_1.85.0              BiocNeighbors_2.1.2      
##  [97] DelayedArray_0.33.3       scales_1.3.0             
##  [99] caTools_1.18.3            remaCor_0.0.18           
## [101] rappdirs_0.3.3            stringr_1.5.1            
## [103] digest_0.6.37             minqa_1.2.8              
## [105] rmarkdown_2.29            aod_1.3.3                
## [107] XVector_0.47.0            RhpcBLASctl_0.23-42      
## [109] htmltools_0.5.8.1         pkgconfig_2.0.3          
## [111] lme4_1.1-35.5             sparseMatrixStats_1.19.0 
## [113] mashr_0.2.79              fastmap_1.2.0            
## [115] rlang_1.1.4               GlobalOptions_0.1.2      
## [117] UCSC.utils_1.3.0          DelayedMatrixStats_1.29.0
## [119] farver_2.1.2              jquerylib_0.1.4          
## [121] jsonlite_1.8.9            BiocSingular_1.23.0      
## [123] RCurl_1.98-1.16           magrittr_2.0.3           
## [125] GenomeInfoDbData_1.2.13   munsell_0.5.1            
## [127] Rcpp_1.0.13-1             babelgene_22.9           
## [129] viridis_0.6.5             EnrichmentBrowser_2.37.0 
## [131] RcppZiggurat_0.1.6        stringi_1.8.4            
## [133] zlibbioc_1.52.0           MASS_7.3-61              
## [135] plyr_1.8.9                parallel_4.4.2           
## [137] listenv_0.9.1             ggrepel_0.9.6            
## [139] Biostrings_2.75.3         splines_4.4.2            
## [141] hms_1.1.3                 circlize_0.4.16          
## [143] locfit_1.5-9.10           buildtools_1.0.0         
## [145] reshape2_1.4.4            ScaledMatrix_1.15.0      
## [147] BiocVersion_3.21.1        XML_3.99-0.17            
## [149] evaluate_1.0.1            RcppParallel_5.1.9       
## [151] BiocManager_1.30.25       nloptr_2.1.1             
## [153] foreach_1.5.2             tidyr_1.3.1              
## [155] purrr_1.0.2               future_1.34.0            
## [157] clue_0.3-66               scattermore_1.2          
## [159] ashr_2.2-63               rsvd_1.0.5               
## [161] broom_1.0.7               xtable_1.8-4             
## [163] fANCOVA_0.6-1             viridisLite_0.4.2        
## [165] truncnorm_1.0-9           tibble_3.2.1             
## [167] lmerTest_3.1-3            glmmTMB_1.1.10           
## [169] memoise_2.0.1             beeswarm_0.4.0           
## [171] cluster_2.1.8             globals_0.16.3           
## [173] GSEABase_1.69.0