General Latent Feature Modeling for Data Exploration Tasks
This paper introduces a general Bayesian nonparametric latent feature model suitable to perform automatic exploratory analysis of heterogeneous datasets, where the attributes describing each object can be either discrete, continuous or mixed variables. The proposed model presents several important properties. First, it accounts for heterogeneous data while can be inferred in linear time with respect to the number of objects and attributes. Second, its Bayesian nonparametric nature allows us to automatically infer the model complexity from the data, i.e., the number of features necessary to capture the latent structure in the data. Third, the latent features in the model are binary-valued variables, easing the interpretability of the obtained latent features in data exploration tasks.
Latent feature models allow us to compact in a few features the immense redundant information present in the observed data, by capturing the statistical dependencies among the different objects and attributes. As a consequence, they appear as suitable tools to perform data exploratory analysis, i.e, they may help us to better understand the data .
There is an extensive literature in latent feature modeling of homogeneous data, where all the attributes describing each object in the database present the same (continuous or discrete) nature. In particular, these works assume that databases contain only either continuous data, usually modeled as Gaussian variables , or discrete, that can be either modeled by discrete likelihoods  or simply treated as Gaussian variables . However, there still exists a lack of works dealing with heterogeneous databases, which in fact are common in real applications. As motivating examples, Electronic Health Records from hospitals might contain lab measurements (often positive real-valued or real-valued data), diagnoses (categorical data) and genomic information (ordinal, count data and categorical data); also, a survey often contains diverse information about the participants such as age (count data), gender (categorical data), salary (positive real data), etc. Despite this diversity of data types, the standard approach when dealing with heterogeneous datasets is to treat all the attributes, either continuous or discrete, as Gaussian variables.
This paper presents a general latent feature model (GLFM) suitable for heterogeneous datasets, where the attributes describing each object can be either discrete, continuous or mixed variables. Specifically, we account for real-valued and positive real-valued as examples of continuous variables, and categorical, ordinal and count data as examples of discrete variables. The proposed model extends the essential building block of Bayesian nonparametric latent feature models, the Indian Buffet Process (IBP) by , to account for heterogeneous data while maintaining the model complexity of conjugate models. Among all the available latent feature models in the literature, we opt for the IBP due to two main reasons. First, the nonparametric nature of the IBP allows us to automatically infer the appropriate model complexity, i.e., the number of necessary features, from the data. Second, the IBP considers binary-valued latent features which has been shown to provide more interpretable results in data exploration than standard real-valued latent feature models . The standard IBP assumes real-valued observations combined with conjugate likelihood models, allowing for fast inference algorithms . However, we here aim at dealing with heterogeneous databases, such that conjugacy might not be straightforwardly available.
In order to propose a general observation model for the IBP that accounts for heterogeneous data while keeping the properties of conjugate models, we exploit two key ideas. First, we introduce an auxiliary real-valued variable (also called pseudo-observation), such that, conditioned on it, the model behaves as the standard linear-Gaussian IBP in . Second, we assume that there exists a function that transforms the pseudo-observation into the actual observation, mapping the real line into the (discrete or continuous) observation space of each attribute in the data. These two key ideas allow us to derive an efficient inference algorithm based on collapsed Gibbs sampling, which presents linear complexity with the number of objects and attributes in the data.
Our experiments provide examples of how to use the proposed model for data exploration in real-world datasets. Additionally, a software library implementing the GLFM, as well as the necessary scripts to perform automatic data exploration, is publicly available at https://github.com/ivaleraM/GLFM .
2General Latent Feature Model
We introduce the GLFM, which is a general Bayesian nonparametric latent feature model suitable for data exploration of heterogeneous datasets, where the attributes describing each object can be either discrete, continuous or mixed variables. Specifically, the GLFM accounts for the following data types:
The GLFM builds on the IBP , and therefore, it assumes that each observation can be explained by a potentially infinite-length binary vector whose elements indicate whether a latent feature is active or not for the -th object; and a (real-valued) weighting vector , whose elements weight the influence of each latent feature in the -th attribute
Figure 1 illustrates the GLFM by showing the corresponding graphical model together with an example of the generative model for an ordinal attribute taking values in the ordered set low, medium, high. The inference of the GLFM is performed using collapsed Gibbs sampling, which presents linear complexity with respect to the number of objects and the number of attributes . Additional details on the model, as well as on the inference algorithm can be found in .
The main goal of this section is to provide showcase examples about how to include the specific domain knowledge into the proposed GLFM to find and analyze the latent structure underlying data in different application domains, i.e., to perform data exploratory analysis. In particular, we here show examples of how to select the input data for the GLFM, as well as how to enter these data into the model, in order to obtain interpretable results that can be used to get a better understanding of the data.
|Attribute description||Type of variable|
|Stage of the cancer||Categorical with 2 categories|
|DES treatment level||Ordinal with 3 categories|
|Tumor size in cm||Count data|
|Serum Prostatic Acid Phosphatase (PAP)||Positive real-valued|
|Prognosis Status (outcome of the disease)||Categorical with 4 categories|
3.1Drug effect in a clinical trial for prostate cancer
Clinical trials are conducted to collect data regarding the safety and efficacy of a new drug before it can be sold in the consumer market, if ever. Concretely, the main goal of clinical trials is to prove the efficacy of a new treatment for a disease while ensuring its safety, i.e., check whether its adverse effects remain low enough for any dosage level of the drug. As an example, the publicly available Prostate Cancer dataset
In this section, we apply the proposed GLFM to the Prostate Cancer dataset to show that the proposed model can be efficiently used to discover the statistical dependencies in the data, which in this example corresponds to the effect of the different levels of treatment with DES in the suffering of prostate cancer and cardiovascular diseases. The prostate cancer dataset consists of 502 patients and 16 attributes, from which we make use of the five attributes listed in Table 1. The selection of these five attributes allows us not only to reduce the number of local minima in the posterior distribution of the proposed model due to the small sample size of the dataset, but also to focus on capturing the statistical dependencies between the target attributes, i.e, the relationship between the different levels of treatment with DES and the suffering of prostate cancer and cardiovascular diseases.
After running our model, we obtain four latent features. Figure 2 shows the effect of the inferred latent features, as well as the bias term, on each dimension/attribute of the data, where we can distinguish two groups of features. The first group accounts for patients in stage 3 and includes the bias term and the 2 first latent features. Within this group, the bias term – or equivalently pattern (0000) – and the first feature – or equivalently pattern (1000) – account for patients in stage 3 with a low average level of treatment with DES (refer to Figure ?). However, while the bias term models patients with low probability () of prostate cancer death, the first feature accounts for patients with higher probability () of prostate cancer death, which can be explained by a larger tumor size (refer to Figure ?). The second feature – or equivalently pattern (0100) –captures patients who exclusively received a high dosage (5 mg) of the drug (refer to Figure ?). These patients present a small tumor size and the lowest probability of prostatic cancer death, suggesting a positive effect of the drug as treatment for the cancer. However, they also present a significant increase in the probability of dying from a vascular disease (), indicating a potential adverse-effect of the drug that increases the risk of suffering from cardio-vascular diseases. Such observation is in agreement with previous studies .
The second group of features corresponds to the activation patterns (0010) and (0001), and accounts for patients in stage 4 with, respectively, mild and severe conditions. In particular, the third feature corresponds to patients with small tumor size, but intermediate values for the PAP biomarker, suggesting a certain spread degree of the tumor compared to the features in the first group, but not as severe as for patients with pattern (0001). Indeed, pattern (0001) models those patients in stage 4 with relatively high tumor size and the highest PAP values–it is thus not surprising that those patients present in turn the highest probability (above 50%) of prostatic death.
3.2Impact of Social Background on Mental Disorders
Several studies have analyzed the impact of social background in the development of mental disorders . Other studies have focused on finding and analyzing the co-occurring (comorbidity) pattern among the 20 most common psychiatric disorders . These studies found that the 20 most common disorders can be divided into three groups: i) externalizing disorders, which include substance use disorders (alcohol abuse and dependence, drug abuse and dependence and nicotine dependence); ii) internalizing disorders, which include mood and anxiety disorders (major depressive disorder (MDD), bipolar disorder and dysthymia, panic disorder, social anxiety disorder (SAD), specific phobia and generalized anxiety disorder (GAD), and pathological gambling (PG)); and iii) personality disorders (avoidant, dependent, obsessive-compulsive (OC), paranoid, schizoid, histrionic and antisocial personality disorders (PDs)). However, up to our knowledge, there is a lack of work on the impact of social background in the suffering of comorbid disorders.
In this section, we aim at extending the analysis in  to account for the influence of the social background of subjects (such as age, gender, etc.) in the probability of a subject suffering from comorbid disorders. To this end, in addition to the diagnoses of the above 20 psychiatric disorders, we also make use of the information provided by the NESARC, which includes a set of questions on the social background of participants. Specifically, in addition to the diagnoses of the most common 20 psychiatric disorders described above, we include the sex of the participants as input data to the proposed model. We model the gender information of the participants in the NESARC as a categorical variable with two categories: ‘male’, ‘female’. The percentage of males in the NESARC is approximately . Note also that the diagnoses of the 20 psychiatric disorders correspond to categorical variables with two possible categories, e.g., a patient suffering or not from a disorder.
After running our inference algorithm with the diagnoses of the disorders and the gender of subjects as input data, we obtain three latent features. Figure ? shows the probability of meeting each diagnostic criteria for the latent feature vectors listed in the legend and in the database (baseline). Note that the obtained latent features are similar to the ones in , i.e., feature 1 (pattern ) mainly models the seven personality disorders (PDs), feature 2 (pattern ) models alcohol and drug abuse disorders and the antisocial PD, while feature 3 (pattern ) models anxiety and mood disorders. Additionally, Figure ? shows the probability of being male and female for the latent feature vectors depicted in the legend and the empirical probability of being male and female in the database (baseline).
In Figure ?, we observe that having no active features (pattern ), which captures people that do not suffer from any disorder, increases the probability of being male with respect to the baseline probability, and therefore, it indicates that females tend to suffer in a higher extent from psychiatric disorders. Additionally, we observe that feature 1 (pattern ) increases the probability of being male, while feature 3 (pattern ) increases the probability of being female. Hence, from the analysis of Figure ?, we can conclude that, while women suffer more frequently from mood and anxiety disorders than men, PDs are more common in men.
In this paper, we have introduced the first available general latent feature model and its code implementation, which will ease researchers from diverse fields to analyze a wide range of heterogeneous, incomplete and noisy datasets in an automatic manner. We have showed the flexibility and applicability of the proposed GLFM by performing data exploratory analysis of diverse real-world datasets. Further results including higher dimensional spaces can be found in .
- For convenience, we here capitalize the vector .
- The stage of a cancer describes the size of a cancer and how far it has grown. Stage 3 means that the cancer is already quite large and may have started to spread into surrounding tissues or local lymph nodes. Stage 4 is more severe, and refers to a cancer that has already spread from where it started to another body organ. This is also called secondary or metastatic cancer. Find more details in http://www.cancerresearchuk.org/about-cancer/what-is-cancer/stages-of-cancer
- Mapping common psychiatric disorders: Structure and predictive validity in the National Epidemiologic Survey on Alcohol and Related Conditions.
Blanco, C., Krueger, R. F., Hasin, D. S., Liu, S. M., Wang, S., Kerridge, B. T., Saha, T., and Olfson, M. Journal of the American Medical Association Psychiatry
- The choice of treatment for cancer patients based on covariate information: application to prostate cancer.
Byar, D. P. and Green, S. B. Bulletin du Cancer
- Accelerated sampling for the indian buffet process.
Doshi-Velez, F. and Ghahramani, Z. In Proceedings of the 26th Annual International Conference on Machine Learning, ICML ’09, pp. 273–280, New York, NY, USA, 2009. ACM.
- Bayesian Nonparametric Poisson Factorization for Recommendation Systems.
Gopalan, P., Ruiz, F. J. R., Ranganath, R., and Blei, D. M. International Conference on Artificial Intelligence and Statistics (AISTATS)
- The Indian buffet process: an introduction and review.
Griffiths, T. L. and Ghahramani, Z. Journal of Machine Learning Research
- A Bayesian approach for estimating and replacing missing categorical data.
Li, X.-B. J. Data and Information Quality
- Applying cox regression to competing risks.
Lunn, M. and McNeil, D. Biometrics
- Bayesian nonparametric modeling of suicide attempts.
Ruiz, F. J. R., Valera, I., Blanco, C., and Perez-Cruz, F. Advances in Neural Information Processing Systems
- Bayesian nonparametric comorbidity analysis of psychiatric disorders.
Ruiz, F. J. R., Valera, I., Blanco, C., and Perez-Cruz, F. Journal of Machine Learning Research.
- Probabilistic low-rank matrix completion with adaptive spectral regularization algorithms.
Todeschini, A., Caron, F., and Chavent, M. In Burges, C.J.C., Bottou, L., Welling, M., Ghahramani, Z., and Weinberger, K.Q. (eds.), Advances in Neural Information Processing Systems 26, pp. 845–853. Curran Associates, Inc., Dec. 2013.
- General Latent Feature Models for Heterogeneous Datasets.
Valera, I., Pradier, M. F., and Ghahramani, Z. ArXiv: https://arxiv.org/abs/1706.03779
- Poverty, unemployment, and common mental disorders: population based cohort study.
Weich, S. and Lewis, G. BMJ
- Sex differences in rates of depression: cross-national perspectives.
Weissman, M. M., R., Bland, Joyce, P. R., Newman, S., Wells, J.E., and Wittchen, H.-U. Journal of Affective Disorders