Bayesian Tensor Filtering: Smooth, Locally-Adaptive Factorization of Functional Matrices

by   Wesley Tansey, et al.

We consider the problem of functional matrix factorization, finding low-dimensional structure in a matrix where every entry is a noisy function evaluated at a set of discrete points. Such problems arise frequently in drug discovery, where biological samples form the rows, candidate drugs form the columns, and entries contain the dose-response curve of a sample treated at different concentrations of a drug. We propose Bayesian Tensor Filtering (BTF), a hierarchical Bayesian model of matrices of functions. BTF captures the smoothness in each individual function while also being locally adaptive to sharp discontinuities. The BTF model is agnostic to the likelihood of the underlying observations, making it flexible enough to handle many different kinds of data. We derive efficient Gibbs samplers for three classes of likelihoods: (i) Gaussian, for which updates are fully conjugate; (ii) Binomial and related likelihoods, for which updates are conditionally conjugate through Pólya--Gamma augmentation; and (iii) Black-box likelihoods, for which updates are non-conjugate but admit an analytic truncated elliptical slice sampling routine. We compare BTF against a state-of-the-art method for dynamic Poisson matrix factorization, showing BTF better reconstructs held out data in synthetic experiments. Finally, we build a dose-response model around BTF and show on real data from a multi-sample, multi-drug cancer study that BTF outperforms the current standard approach in biology. Code for BTF is available at


page 1

page 2

page 3

page 4


Kernelized Bayesian Matrix Factorization

We extend kernelized matrix factorization with a fully Bayesian treatmen...

Drug response prediction by inferring pathway-response associations with Kernelized Bayesian Matrix Factorization

A key goal of computational personalized medicine is to systematically u...

Bayesian Poisson Tensor Factorization for Inferring Multilateral Relations from Sparse Dyadic Event Counts

We present a Bayesian tensor factorization model for inferring latent gr...

Hierarchical Compound Poisson Factorization

Non-negative matrix factorization models based on a hierarchical Gamma-P...

Bayesian Probabilistic Matrix Factorization: A User Frequency Analysis

Matrix factorization (MF) has become a common approach to collaborative ...

Robust and Practical Solution of Laplacian Equations by Approximate Elimination

We introduce a new algorithm and software for solving linear equations i...

Please sign up or login with your details

Forgot password? Click here to reset