Seminar

Probabilistic topic models for single-cell genomics

To join this seminar virtually: please request Zoom connection details from ea@stat.ubc.ca

Abstract: Building a comprehensive topic model has become an important research tool in single-cell genomics. With a topic model, we can decompose and ascertain distinctive cell topics shared across multiple cells, and the gene programs implicated by each topic can later serve as a predictive model in translational studies. In this talk, I will present a few topic modeling tools for single-cell RNA-sequencing (RNA-seq) data analysis. The first topic model builds on the Embedded Topic Model (ETM) and incorporates sparse-inducing priors to make the model more interpretable. I will showcase that it can be used to unravel cell types or cell states. The second topic model uncovers short-term RNA velocity patterns from a plethora of spliced and unspliced single-cell RNA-sequencing (RNA-seq) counts. I will show that modeling both types of RNA counts can improve robustness in statistical estimation and can reveal new aspects of dynamic changes that can be missed in static analysis. I will showcase that our modeling framework can be used to identify statistically significant dynamic gene programs in pancreatic cancer data. Our results discovered that seven dynamic gene programs (topics) are highly correlated with cancer prognosis and generally enrich immune cell types and pathways.

Margin-closed multivariate autoregressive time series models

To join this seminar virtually: please request Zoom connection details from ea@stat.ubc.ca

Abstract: In literature on multivariate modeling using copulas, it is typical to first model univariate margins and then the multivariate dependence structure between the marginal components. This is useful because there are many diagnostics that can help in selecting univariate models. Following the same idea, we derive the conditions for when a multivariate stationary Gaussian vector autoregressive (VAR) time series is closed under margins, i.e., it has univariate autoregressive (AR) margins or lower-dimensional vector autoregressive (VAR) margins. It leads to a copula model and it can be extended to a regime-switching setting. The constraint of the closure under margins can reduce the number of parameters in VAR and Markov switching vector autoregressive (MSVAR) models. Moreover, after transforming the stationary univariate margins into standard Gaussian, the property of closure under margins enables a new framework of modeling high-dimensional time series by modeling its low-dimensional sub-processes first and then modeling their dependence structure. The framework makes it more flexible in analyzing marginal behavior of the multivariate time series and also enables a multi-stage estimation procedure.

Speeding up Metropolis using Theorems

To join this seminar virtually: Please register here.

Title: Speeding up Metropolis using Theorems

Abstract: Markov chain Monte Carlo (MCMC) algorithms, such as the Metropolis algorithm, are designed to converge to complicated high-dimensional target distributions, to facilitate sampling. The speed of this convergence is essential for practical use. In this talk, we will present several theoretical probability results which can help improve the Metropolis algorithm's convergence speed. Specific topics will include: diffusion limits, optimal scaling, optimal proposal shape, tempering, adaptive MCMC, the Containment property, and the notion of adversarial Markov chains. The ideas will be illustrated using the simple graphical example available at probability.ca/met. No particular background knowledge will be assumed.

Uncertainty Quantification for Structure Learning and Interpretable Machine Learning

Date/Time: Thursday, February 8, 2024, 10:25am to 11:25am

To join this seminar virtually: Please request Zoom connection details from ea@stat.ubc.ca

Abstract: The reliability of machine learning in scientific discoveries and decision-making hinges on the critical role of uncertainty quantification (UQ). In this talk, I will discuss UQ in two challenging scenarios motivated by scientific and societal applications: selective inference for large-scale graph learning and UQ for model-agnostic machine learning interpretations. Specifically, the first part concerns graphical model inference when only irregular, patchwise observations are available, a common setting in neuroscience, healthcare, genomics, and econometrics. To filter out low-confidence edges due to the irregular measurements, I will present a novel inference method that quantifies the uneven edgewise uncertainty levels over the graph as well as an FDR control procedure; this is achieved by carefully disentangling the dependencies across the graph and consequently yields more reliable graph selection. In the second part, I will discuss the computational and statistical challenges associated with UQ for feature importance of any machine learning model. I will take inspiration from recent advances in conformal inference and utilize an ensemble framework to address these challenges. This leads to an almost computationally free, assumption-light, and statistically powerful inference approach for occlusion-based feature importance. For both parts of the talk, I will highlight the potential applications of my research in science and society as well as how it contributes to more reliable and trustworthy data science.

Probabilistic methods for designing functional protein structures

To join this seminar virtually: Please request Zoom connection details from ea@stat.ubc.ca

Abstract: The biochemical functions of proteins, such as catalyzing a chemical reaction or binding to a virus, are typically conferred by the geometry of only a handful of atoms.  This arrangement of atoms, known as a motif, is structurally supported by the rest of the protein, referred to as a scaffold.  A central task in protein design is to identify a diverse set of stabilizing scaffolds to support a motif known or theorized to confer function. This long-standing challenge is known as the motif-scaffolding problem.

In this talk, I describe a statistical approach I have developed to address the motif-scaffolding problem.  My approach involves (1) estimating a distribution supported on realizable protein structures and (2) sampling scaffolds from this distribution conditioned on a motif.  For step (1) I adapt diffusion generative models to fit example protein structures from nature.  For step (2) I develop sequential Monte Carlo algorithms to sample from the conditional distributions of these models.  I finally describe how, with experimental and computational collaborators, I have generalized and scaled this approach to generate and experimentally validate hundreds of proteins with various functional specifications.

Toward large-scale Bayesian inference with structured generative models

To join this seminar virtually: Please request Zoom connection details from ea@stat.ubc.ca

Abstract: Conditional generative models are an increasingly popular machine learning approach for performing Bayesian Inference in large-scale scientific applications. Despite their success, the sample-complexity of these methods often scale poorly with the growing dimensions of model parameters and observations. These models require large training datasets, which are typically unavailable with computationally expensive forward models as in numerical weather prediction. In these settings, it becomes crucial to identify the low-dimensional structure in posterior distributions and encode it in generative models for accurate inference. In this presentation, I will introduce an information-theoretic perspective for reducing the dimensions of parameters and observations in inverse problems with guarantees on the posterior approximation error. I will show how to identify relevant subspaces for these variables from either gradient evaluations of the log-likelihood function or by using score-based generative models when gradients are not available. The benefit of the dimension reduction approach will be showcased on a turbulent flow estimation problem from aerodynamics where traditional methods are unstable in small sample regimes.

Advancing Biomedical Data Science: Leveraging ChatGPT for genomics data & Testing data-driven hypothesis post-clustering

Date/Time: Thursday, January 25, 2024, 10:25am to 11:25am

To join this seminar virtually: Please request Zoom connection details from ea@stat.ubc.ca

Abstract: My research centers around bringing statistical insights and understanding to the practice of modern data science, and I will cover two projects related to this research vision in this talk.

In the first part of the talk, I will consider how to leverage large language models (LLMs) such as ChatGPT for biomedical discovery. While significant progress has been made in customizing large language models for biomedical data, these models often require extensive data curation and resource-intensive training. In the context of single-cell RNA-sequencing data, I will show that we can achieve surprisingly competitive results on many downstream tasks via a much simpler alternative: I input textual descriptions of genes into an off-the-shelf LLM, such as ChatGPT, to obtain low-dimensional representations of the genes, or “embeddings.” I then use these embeddings as features in downstream tasks. A similar approach enables LLM-derived embeddings of cells. This work highlights the potential of LLMs to provide meaningful and concise representations for biomedical data, and also raises a number of challenging statistical questions. Addressing these questions requires bringing principled statistical thinking to the practice of modern data science.

The second part of my talk is motivated by the practice of testing data-driven hypotheses. In biomedical sciences, it has become increasingly common to collect massive datasets without a pre-specified research question. In this setting, a data analyst might use the data both to generate a research question, and to test the associated null hypothesis. For example, in single-cell RNA-sequencing analyses, researchers often first cluster the cells, and then test for differences in the expected gene expression levels between the clusters to quantify up- or down-regulation of genes, annotate known cell types, and identify new cell types. However, this popular practice is invalid from a statistical perspective: once we have used the data to generate hypotheses, standard statistical inference tools are no longer valid. To tackle this problem, I developed a conditional selective approach to test for a difference in means between pairs of clusters obtained via k-means clustering. The proposed approach has appropriate statistical guarantees (e.g., selective Type 1 error control).

This talk features joint work with Lucy Gao (University of British Columbia), Daniela Witten (University of Washington), and James Zou (Stanford University).

Identifiable and interpretable nonparametric factor analysis

Date/Time: Thursday, January 18, 2024, 10:25am to 11:25am

To join this seminar virtually: Please request Zoom connection details from ea@stat.ubc.ca

Abstract: Factor models have been widely used to summarize the variability of high-dimensional data through a set of factors with much lower dimensionality. Gaussian linear factor models have been particularly popular due to their interpretability and ease of computation. However, in practice, data often violate the multivariate Gaussian assumption. To characterize higher-order dependence and nonlinearity, models that include factors as predictors in flexible multivariate regression are popular, with GP-LVMs using Gaussian process (GP) priors for the regression function and VAEs using deep neural networks. Unfortunately, such approaches lack identifiability and interpretability and tend to produce brittle and non-reproducible results. To address these problems by simplifying the nonparametric factor model while maintaining flexibility, we propose the NIFTY framework, which parsimoniously transforms uniform latent variables using one-dimensional nonlinear mappings and then applies a linear generative model. The induced multivariate distribution falls into a flexible class while maintaining simple computation and interpretation. We prove that this model is identifiable and empirically study NIFTY using simulated data, observing good performance in density estimation and data visualization. We then apply NIFTY to bird song data in an environmental monitoring application.

Learning-Rate-Free Methods for Scalable Bayesian Inference

To join this seminar virtually: Please request Zoom connection details from ea@stat.ubc.ca

Abstract: In recent years, particle-based variational inference (ParVI) methods such as Stein variational gradient descent have grown in popularity as scalable methods for Bayesian inference. Unfortunately, the properties of such methods invariably depend on hyperparameters such as the learning rate, which must be carefully tuned by practitioners in order to ensure convergence to the target measure at a suitable rate. In this work, we introduce a suite of new particle-based methods for scalable Bayesian inference based on coin betting, which are entirely learning-rate free. We illustrate the performance of our approach on a range of numerical examples, including several high-dimensional models and datasets, demonstrating comparable or superior performance to other ParVI algorithms with no need to tune a learning rate.

The design and analysis of annealing algorithms for sampling and generative modelling

To join this seminar virtually: Please request Zoom connection details from ea@stat.ubc.ca

Abstract: Generating samples from complex probability distributions is a fundamental challenge in statistical modelling. This problem is called "sampling" when we can access only an un-normalised density and "generative modelling" when we can only access a dataset of existing samples. In practice, this is generally impossible, and we must introduce a simpler reference distribution, such as a Gaussian, and manipulate its density and samples to approximate the target. In general, direct inference is reliable when the reference is close to the target and fragile when it is not. Annealing is a popular technique motivated by this principle and introduces a sequence of distributions that interpolates between the reference and target, ensuring the neighbouring distributions are close. An annealing algorithm specifies how to traverse this bridge of distributions to incrementally transform samples from the reference into samples approximating the target.

In this talk, I will discuss how annealing can be applied to sampling and generative modelling. As a case study, I will introduce parallel tempering (PT) and denoising diffusion models (DDM), two annealing algorithms recently gaining popularity in the sampling and generative modelling literature. I will show how our analysis of PT and DDM can provide insights into understanding the design choices behind their success and motivate the design choices for other annealing algorithms in sampling and generative modelling. Finally, I will discuss the applications of this body of work to Bayesian inference and PDE surrogate modelling.