Joint Probability Estimation Using Tensor Decomposition and Dictionaries
This addresses the challenge of estimating joint distributions for hybrid data where traditional methods like Gaussian Mixture Models perform poorly, offering a novel approach for applications requiring accurate probability estimation.
The paper tackles the problem of non-parametric estimation of joint probabilities for mixed discrete and continuous variables from 2D marginals, proposing a method that combines dictionary representations with tensor decomposition to handle hybrid distributions and various distribution families, resulting in better classification rates and lower error rates compared to state-of-the-art estimators.
In this work, we study non-parametric estimation of joint probabilities of a given set of discrete and continuous random variables from their (empirically estimated) 2D marginals, under the assumption that the joint probability could be decomposed and approximated by a mixture of product densities/mass functions. The problem of estimating the joint probability density function (PDF) using semi-parametric techniques such as Gaussian Mixture Models (GMMs) is widely studied. However such techniques yield poor results when the underlying densities are mixtures of various other families of distributions such as Laplacian or generalized Gaussian, uniform, Cauchy, etc. Further, GMMs are not the best choice to estimate joint distributions which are hybrid in nature, i.e., some random variables are discrete while others are continuous. We present a novel approach for estimating the PDF using ideas from dictionary representations in signal processing coupled with low rank tensor decompositions. To the best our knowledge, this is the first work on estimating joint PDFs employing dictionaries alongside tensor decompositions. We create a dictionary of various families of distributions by inspecting the data, and use it to approximate each decomposed factor of the product in the mixture. Our approach can naturally handle hybrid $N$-dimensional distributions. We test our approach on a variety of synthetic and real datasets to demonstrate its effectiveness in terms of better classification rates and lower error rates, when compared to state of the art estimators.