 Methodology
 Open Access
 Open Peer Review
 Published:
Cluster ensemble based on Random Forests for genetic data
BioData Miningvolume 10, Article number: 37 (2017)
Abstract
Background
Clustering plays a crucial role in several application domains, such as bioinformatics. In bioinformatics, clustering has been extensively used as an approach for detecting interesting patterns in genetic data. One application is population structure analysis, which aims to group individuals into subpopulations based on shared genetic variations, such as single nucleotide polymorphisms. Advances in DNA sequencing technology have facilitated the obtainment of genetic datasets with exceptional sizes. Genetic data usually contain hundreds of thousands of genetic markers genotyped for thousands of individuals, making an efficient means for handling such data desirable.
Results
Random Forests (RFs) has emerged as an efficient algorithm capable of handling highdimensional data. RFs provides a proximity measure that can capture different levels of cooccurring relationships between variables. RFs has been widely considered a supervised learning method, although it can be converted into an unsupervised learning method. Therefore, RFderived proximity measure combined with a clustering technique may be well suited for determining the underlying structure of unlabeled data. This paper proposes, RFcluE, a cluster ensemble approach for determining the underlying structure of genetic data based on RFs. The approach comprises a cluster ensemble framework to combine multiple runs of RF clustering. Experiments were conducted on highdimensional, real genetic dataset to evaluate the proposed approach. The experiments included an examination of the impact of parameter changes, comparing RFcluE performance against other clustering methods, and an assessment of the relationship between the diversity and quality of the ensemble and its effect on RFcluE performance.
Conclusions
This paper proposes, RFcluE, a cluster ensemble approach based on RF clustering to address the problem of population structure analysis and demonstrate the effectiveness of the approach. The paper also illustrates that applying a cluster ensemble approach, combining multiple RF clusterings, produces more robust and higherquality results as a consequence of feeding the ensemble with diverse views of highdimensional genetic data obtained through bagging and random subspace, the two key features of the RF algorithm.
Background
Clustering is an unsupervised learning technique aimed at uncovering the underlying natural structure of data. In data analysis, clustering is the process of partitioning objects into groups based on their similarities, where objects in the same group are more similar to one another than to objects in different groups. Clustering plays an essential role in several application domains, such as text mining, image segmentation, and bioinformatics. In bioinformatics, clustering has been extensively used as an approach for detecting interesting patterns in genetic data. Such an approach is formally used to find the underlying population substructure from genetic data without considering prior information. The analysis of population structures is a crucial prerequisite for any further analysis of genetic data, such as genomewide association mapping [1] for reducing false positive rates, and forensics [2] for developing reference panels to provide information on an individual’s ancestry. This kind of analysis aims to group individuals into subpopulations based on shared genetic variations. Single nucleotide polymorphisms (SNPs) are the most common type of genetic variation used to infer population structure. SNPs occur when a single nucleotide from a DNA sequence differs at the same position between individuals. An SNP has three categories: homozygous with the common allele (genotype AA), heterozygous (genotype AB), and homozygous with the rare allele (genotype BB). Advances in DNA sequencing technology have facilitated the attainment of genetic datasets with exceptional sizes. Genetic data usually contain hundreds of thousands of genetic markers genotyped for thousands of individuals. Thus, an efficient means for handling such highdimensional data is desirable.
Two major clustering approaches have been developed to infer the structure of populations from genetic data: distancebased and dimension reductionbased approaches. AWclust [3] is a distancebased approach that consists of constructing an allelesharing distance (ASD) matrix between all pairs of individuals in the genetic data. It then applies hierarchical clustering to infer clusters of individuals from the ASD matrix using Ward’s algorithm. PCAclust [4] is a dimension reductionbased clustering approach that involves applying principal component analysis (PCA) to reduce the dimensions of the genetic data. It then applies a modelbased clustering algorithm (i.e., a Gaussian mixture model clustering) to the set of relevant principal components.
Inferring population structures from genetic data can be defined as a problem of determining how to assign N individuals using l genetic markers to K subpopulations. This paper proposes a new approach for inferring population structures from genetic data. The proposed approach is based on Random Forests (RFs). Our motivation for using RFs is twofold: First, its capability of handling highdimensional data of thousands of individuals and hundreds of thousands of markers, which makes it a suitable solution for the problem of population structure analysis. Second, RFs provides a natural method for measuring proximities between individuals; this measure weighs the cooccurrence between markers such that the more correlated a marker is with other markers, the more it will affect the proximity between individuals. Therefore, it can handle the linkage nature among genetic markers. In genetics, linkage refers to a correlation between types of an allele that appear at different loci, especially when a genome is densely genotyped due to linkage disequilibrium (LD) [5]. LD refers to the nonrandom association of particular alleles, which plays a major role in discovering population structures from genetic data. RF clustering, in which RFderived proximity is combined with a clustering technique, is well suited for discovering the underlying structure of unlabeled data [6, 7]. However, the main concern underlying the RF algorithm is that, for each run, a different proximity matrix is generated due to its random nature, therefore producing a different clustering result each time. Thus, this paper proposes a Random Forest cluster Ensemble (RFcluE) approach to discover the underlying structure of genetic data. Within this approach, a cluster ensemble framework is utilized to combine the results of multiple runs of RF clustering toward obtaining a more reliable and robust clustering result than a single run of RF clustering.
Methods
Random Forests
Random Forests (RFs) has emerged as an efficient algorithm capable of handling highdimensional data [8]. RFs was formally developed by Leo Breiman [8] as a classification and regression ensemble learning method. This method is based on a combination of bagging [9] and random subspace [10]. Bagging is the process of aggregating the results of multiple trees, where each tree is grown on a bootstrap sample of the objects. A bootstrap sample of a specified size is drawn with replacement from the original data. Random subspace refers to the selection of a random subset of variables as candidates for splitting at each node. Rather than considering all variables as candidates for splitting, RFs considers only a subset of variables, thus reducing the correlation between trees.
In the context of population structure analysis, individuals are the objects, while SNP markers are the variables. Thus, in RFs, a forest is constructed by building multiple decision trees. To build a tree, the algorithm first creates a root node containing a bootstrap sample of the individuals. Then, at each node, the algorithm selects a random subset of the markers to search over, and subsequently determines the best split markers based on a splitting criterion. A splitting criterion usually maximizes some measure of node purity, which means the degree to which individuals of a node belong to one class. In RFs, the Gini index [11] is used as a splitting criterion to select the best split at each node. The Gini index measures how well a potential split of a node is in separating the individuals into two known classes. Consequently, the Gini index at node n is defined as:
where \( {\overset{\hat{\mkern6mu} }{p}}_c^n=\frac{n_c}{n} \) is the proportion of individuals that are of class c at node n. The Gini index is minimized when all individuals in the node are of the same class, increasing as the individuals in the node are spread more evenly among different classes. The gain for splitting node n based on marker x _{ i }, Gain (x _{ i }, n), is defined as the difference between the impurity at node n and the weighted average of impurities at each child node of n. That is,
where n ^{L} and n ^{R} are the left and right child nodes of the parent node n, respectively, and w _{ L } and w _{ R } are the proportions of individuals assigned to the left and right child nodes. Based on the gain value, the marker x _{ i } with the lowest impurity is selected to split individuals at node n.
This process of splitting is repeated until an unpruned tree is formed. The generated forest contains a significant amount of information about the relationship between the markers and the individuals that can be used for prediction, variable importance, proximity calculation, missing data imputation, and outlier detection. RFderived proximity, a byproduct of a random forest, is defined based on similar individuals ending up in the same leaf node more often than dissimilar individuals. This proximity can capture different levels of cooccurring relationships between markers.
RFs is widely considered a supervised learning method, although it can be adapted as an unsupervised learning method to derive proximity matrix from unlabeled data [6]. Recently, unsupervised RFs has been successfully applied in a wide variety of domains, including bioinformatics [6, 12], image and document analysis [13,14,15], networking [16], cloud computing [17, 18], manufacturing [19], remote sensing [20], and chemometrics [21].
To use RFs for unsupervised learning, the RF algorithm must first randomly generate synthetic data based on the original dataset, in which a random forest is built to distinguish the original data from the synthetic data. One approach for generating the synthetic data is to randomly draw synthetic individuals from marginal distributions of each observed marker in the original data [4]. Hence, the synthetic class has a distribution of independent random markers, where each marker follows the same distribution as the corresponding marker in the original data.
Cluster ensemble
A cluster ensemble is an effective approach for combining different clusterings of the same dataset into a more robust and higherquality clustering than any individual clustering. A cluster ensemble typically consists of two components: an ensemble constructor and a consensus function. An ensemble constructor generates a set of different partitions of the dataset, which is referred to as “base clusterings” or “ensemble members.” On the other hand, a consensus function combines the base clusterings of the ensemble and produces a single clustering as the ultimate output of the cluster ensemble.
Regarding the ensemble constructor, several methods have been proposed to obtain ensemble members, including applying different clustering algorithms [22, 23], applying the same clustering algorithm with random parameter initializations [24,25,26], projecting data onto different subspaces [26,27,28], and data subsampling [25, 29, 30].
The consensus function is critical in the cluster ensemble for performing the combination task. Different approaches have been proposed, including featurebased, graphbased, and pairwisebased approaches. The featurebased approach deals with the problem of cluster ensemble as the clustering of categorical data [31, 32]. Specifically, each ensemble member provides a cluster label as a new feature describing each object. Thus, any categorical clustering can be exploited to find the consensus clustering. The graphbased approach represents the cluster ensemble as a graph, which is then divided into a number of clusters using graph partition technique [26, 28]. Lastly, the pairwisebased approach represents the information from multiple base clusterings as a coassociation matrix that contains cooccurrence relationships between all pairs of objects, which can be used as an input to any similaritybased clustering to derive the final partition [23, 25, 27, 33, 34].
Cluster ensemble based on Random Forests
The proposed approach, the Random Forest cluster Ensemble (RFcluE), is based on the concept of a cluster ensemble, where RF clustering is used as a base clustering method. The general framework for the RFcluE approach is shown in Fig. 1. The RFcluE approach has two stages: The first stage is ensemble construction, followed by the consensus function stage. The first stage takes a genetic dataset as an input and then outputs a set of partitions. The second stage takes the set of partitions as an input and produces a final clustering result as an output.
Let G = {g _{1}, g _{2}, …, g _{ N }} represent a set of N individuals, where g _{ i } is a genotype profile of individual i that consists of l genetic markers. A cluster ensemble first constructs a set of partitions (i.e., ensemble members), P = {P _{1}, P _{2}, …, P _{ M }}, by applying the base clustering method M times. Each run of the base clustering method returns a set of clusters,\( {P}_i=\left\{{C}_i^1,{C}_i^2,\dots .,{C}_i^{k_i}\right\}, \)such that \( {\bigcup}_{j=1}^{k_i}{C}_i^j=G \), where k _{ i } is the number of clusters in the i ^{th} clustering and \( {C}_i^j \) is the j ^{th} cluster of the i ^{th} partition, for i = 1, 2, …, M. Then, the consensus function is applied to the set of generated partitions, P, in order to find a new partition, P ^{∗}, that better represents the properties of each partition in P of the cluster ensemble.
Ensemble construction
The ensemble construction based on RFs is used to create the base clusterings. Clustering using RFs is generally composed of three steps:

(i)
Constructing a forest in an unsupervised fashion.

(ii)
Parsing the constructed forest to compute the proximities between individuals.

(iii)
Applying a clustering technique on the resulting proximity matrix.
The input of the ensemble constructor is a genetic dataset, G ЄR ^{N × l}, where N is the number of individuals and l is the number of genetic markers; and four parameters, specifically the number of trees (ntrees), the tree size controlled by specifying the maximum number of leaf nodes (MN), the number of clusters in each partition (k), and the ensemble size (M).
Since the base clustering method of the ensemble is RF clustering, the ensemble constructor first computes the RFderived proximity matrix. The algorithm that builds a random forest, RF, of size ntrees trees, where each tree has a maximum of MN leaf nodes in the unsupervised mode, is described in Algorithm 1. Based on the constructed forest, the RFderived proximity matrix, which denotes the similarity between each pair of individuals of size N × N, is calculated. Then, the proximity matrix S is converted to a dissimilarity matrix, D, by using \( D=\sqrt{\left(1s\right)} \). Lastly, the method applies Kmeans on this dissimilarity, after transforming it to Euclidean space using multidimensional scaling (MDS) [35], to partition the individuals into k clusters. The MDS technique used is classical scaling, where a N × N distance matrix is converted into a N × p configuration matrix. The configuration matrix contains the coordinates of N individuals in pdimensional space, where p < N; p is determined such that the dimension of the smallest space in which N individuals can be embedded, given D that contains the interdistances between individuals.
The output of the base clustering, RF clustering, is a single partition of the data. To construct a cluster ensemble of size M partitions, the base clustering method is repeated M times and, for each run, a different partition of data is generated such that the cluster ensemble is P = {P _{1}, P _{2}, …, P _{ M }}. The pseudocode of the ensemble construction of RFcluE is outlined in Algorithm 2.
Consensus function
Given a cluster ensemble P, P contains a set of M partitions,P = {P _{1}, P _{2}, …, P _{ M }}, produced by the ensemble construction. Each partition P _{ i } returns a set of clusters such that \( {P}_i=\left\{{C}_i^1,{C}_i^2,\dots .,{C}_i^{k_i}\right\} \), where k _{ i } is the number of clusters in P _{ i.} Each partition P _{ i } contains the cluster labels of N individuals, such that c(n) denotes the cluster label to which the individual n belongs. The goal of the consensus function is to find a new partition, P ^{∗},that combines the information from the cluster ensemble P. The pseudocode of the consensus function of RFcluE is outlined in Algorithm 3. It works as follows. First, the consensus function calculates the coassociation matrix (CO). CO summarizes the information in the ensemble P as the N × N matrix. This matrix denotes the similarity between any pair of N individuals as a proportion of M partitions in the ensemble P, in which they are assigned to the same cluster. Then, the consensus function applies agglomerative hierarchical clustering based on Ward’s minimum variance algorithm [36, 37] on the CO matrix to obtain the final partition, P ^{∗}. Ward’s algorithm is utilized because the inference of population structure needs an algorithm that minimizes the increase of withincluster variance each time an individual is added to a cluster.
Datasets
The performance of the RFcluE approach was empirically evaluated on three wellknown real datasets, namely a human genotype dataset from the PanAsian database [38], worldwide human genotype data from the HapMap project [39], and the worldwide human SNP dataset provided by Prof. Mark D. Shriver and described in [40, 41]. The population (i.e., truth label) that an individual belongs to is known for all individuals in all datasets. Table 1 describes the used datasets in terms of the number of individuals, the number of SNPs, and the number of populations.
Evaluation metrics
Many experiments were conducted to investigate the performance of the RFcluE approach. The performance evaluation comprised an assessment of the quality of the final clustering result of the approach. Besides, an assessment of the quality and diversity of the base clusterings, which are generated by the ensemble constructor, was conducted in order to study their impact on performance. Both quality and diversity were evaluated based on normalized mutual information (NMI).
NMI is a measure of agreement between two partitions based on information theory [28]. It treats the two partitions as nominal random variables. The NMI score between two partitions, A and B, is computed as:
MI(A, B) is the mutual information between two partitions, A and B, calculated as follows:
H(A) and H(B) are the entropy of partition A and partition B, respectively, and are calculated as:
where k _{ A } is the number of clusters in partition A, k _{ B } is the number of clusters in partition B, N _{ i } is the number of individuals in cluster i (C _{ i }) of partition A, N _{ j } is the number of individuals in cluster j (C _{ j }) of partition B, and N _{ ij } is the number of shared individuals between cluster i of partition A and cluster j of partition B (C _{ i } ∈ A and C _{ j } ∈ B).
Therefore, the NMI score becomes:
Note that 0 ≤ NMI (A, B) ≤ 1 , so it takes its maximum value if partitions A and B are identical, and its minimum value if partitions A and B are independent.
Let P represent a cluster ensemble that contains a set of generated M base partitions P = {P _{1}, P _{2}, …, P _{ M }}, P ^{∗} is the final clustering result of the cluster ensemble approach, and L is the truth population labels of individuals.
Based on NMI, the quality of the final clustering result P ^{∗} of an ensemble P is calculated as follows:
The diversity between two partitions, P _{ i }, P _{ j },is denoted as (1 − NMI(P _{ i }, P _{ j } ) ). Therefore, the diversity of an ensemble P is the average of all pairwise diversities among all pairs of partitions—P _{ i }, P _{ j } ∈ P—and can be calculated as follows:
where the higher the DS(P) value, the more diverse the ensemble.
The quality of cluster ensemble P is the average quality of all partitions, P _{ i } ∈ P, and can be calculated as follows:
In the comparison study, the adjusted Rand index (ARI) and accuracy (AC) were used, in addition to NMI.
The ARI [42] is a variation of the Rand index [43] that measures how often similar individuals are assigned to the same cluster and dissimilar individuals to different clusters. Given two partitions, A and B, the ARI between A and B is calculated as follows:
where k _{ A } and k _{ B } are the number of clusters in A and B, repectively. N _{ ij } is the number of individuals in both cluster i of partition A and cluster j in partition B; N _{ i.} is the number of individuals in cluster i of partition A; and N _{.j } is the number of individuals in cluster j in partition B. Obtaining a higher value of ARI is better, while random partitions yield values close to zero.
AC is used to measure the purity of the resulting clusters. To compute AC, each cluster is first assigned to the population label that is most frequent in that cluster. Then, AC is computed by counting the number of correctly assigned individuals and dividing the sum by the total number of individuals, N, as follows:
where N is the number of individuals, k is the number of clusters, n _{ i } is the number of individuals in cluster i, and m _{ i } is the number of individuals with the majority population label in cluster i.
Since each run of ensemble clustering would generate different results, all the metrics are reported as an average value of 20 random runs.
Results and discussion
Many experiments were conducted on the real genetic datasets described previously to assess the RFcluE approach in clustering highdimensional genetic data to infer population structure, including parameter analysis, consensus function, comparison study, and diversity and quality analysis.
Parameter analysis
The objective of the parameter analysis was to study the impact of the change in the parameters on RFcluE performance. In this analysis, both the diversity and quality of the ensemble (i.e., base clusterings), in addition to the quality of the final clustering, were considered. Running RFcluE involves the choice of two RF parameters, the number of trees in the forest (ntrees), and the tree size by specifying the maximum number of leaf nodes (MN). In addition to RF parameters, there is the ensemble size M, which is the number of times the base clustering method is executed. The last parameter is the number of clusters, k, as an input to the base clustering method. For the consensus function, the only parameter to be specified is the number of clusters for the final clustering result. To eliminate its effect in evaluation, the consensus function is forced to divide the individuals into K clusters, where K is the number of the true populations for the examined datasets. Therefore, the final clustering result can be evaluated against the corresponding truth population labels for the dataset.
Figure 2 plots the values of the diversity and quality of the ensemble as well as the quality of the ensemble’s final clustering to show the impact of the change in RF parameters. For each dataset, we tested these values (ntrees = {1000, 4000, 7000, 10000}, \( MN=\left\{\sqrt{N,}\ \frac{3}{2}\sqrt{N},2\sqrt{N},100\ \right\} \), M = 40, and \( k=\sqrt{N} \)), where N is the number of individuals in the examined dataset. From the plots, we were able to observe the insignificant impact of tree size on the quality of the ensemble’s final clustering of PanAsian and HapMap datasets. For Shriver’s dataset, the MN parameter had a minor impact, with lower values performing better than higher values. Consequently, we can conclude that the smallest value of the maximum number of leaf nodes, \( MN=\sqrt{N} \), is empirically sufficient to control the tree size in the forest. This value is also more efficient as it takes less time to run the RF algorithm. Additionally, the plots show that an increase in the number of trees is associated with a decrease in the diversity and an increase in the quality of the base clusterings, as well as an increase in the quality of the ensemble’s final clustering. These trends varied for each dataset. For the PanAsian dataset, there was a positive correlation between the performance improvement of the ensemble clustering and the number of trees, where a significant improvement was seen when the number of trees increased from 1000 to 4000. For the HapMap dataset, we observed similar, albeit minor, improvements as the number of trees increased. For Shriver’s dataset, the performance gain was negligible as the number of trees increased. From these observations, we can conclude that the number of trees is datasetdependent and must be sufficient to uncover the structure of the examined dataset.
Figure 3 shows the impact of ensemble size on the performance of ensemble clustering, considering both ensemble size and the number of trees. In this figure, the plots report the values of the diversity and quality of the ensemble as well as the quality of the final clustering, where the parameters are: (M = {10, 20, 30, 40, 50}, ntrees = {1000, 4000, 7000, 10000},\( MN=\sqrt{N} \), and \( k=\sqrt{N} \)). In general, we can see that the diversity and quality of the ensemble are similar across the five different ensemble sizes for all datasets. However, the quality of the ensemble’s final clustering improves as the ensemble size increases. The improvement in overall performance is dependent on the examined dataset, with the PanAsian dataset demonstrating the most significant improvement. We can also see that the impact of the ensemble size parameter is diminished as the number of trees in the forest is increased. On the other hand, for Shriver’s dataset, we can see stable performance despite a change in the number of trees and only slight improvement when increasing the ensemble size.
The last parameter is the number of clusters, k, as an input to the base clustering method. In order to study the impact of this parameter, three schemes were defined to determine the number of clusters, namely TrueK, FixedK, and RandomK. Specifically, let K and N represent the number of true clusters and the number of individuals in the examined dataset, respectively. The number of clusters for TrueK is k = K; for FixedK, the number of clusters is \( k=\sqrt{N} \), while for RandomK the number is random, selected such that \( k\ \epsilon\ \left[2,\sqrt{N}\right] \) for each run of the base clustering method. To compare the performance of the three schemes, an experiment was conducted using these parameters (M ={10, 20, 30, 40, 50}, ntrees = 10000, \( MN=\sqrt{N\ } \)). Fig. 4 shows, for each dataset, a bar plot of the NMI values of the three schemes across five ensemble sizes. Regardless of ensemble size, the FixedK scheme had higher NMI than the other two schemes for the HapMap and Shriver datasets. As for the PanAsian dataset, no significant difference was observed between the three schemes. This observation thus confirms the performance gain of the ensemble’s final clustering when the number of clusters in base clusterings is overproduced. Likewise, this observation also supports the recommendation that the value of k be set to greater than the expected number of clusters [44,45,46].
Consensus function
The consensus function of RFcluE, as presented previously, is composed of calculating the coassociation matrix and then applying Ward’s agglomerative clustering. This consensus function performs effectively by exploiting the coassociation between individuals in the ensemble. However, the ensemble can be explored by considering the association between clusters within different partitions in addition to the association between individuals. Linkbased similarity measures were proposed in [47] to improve the performance of CO by considering the association between clusters. These measures include connected triplebased similarity (CTS), SimRankbased similarity (SRS) and, finally, the approximate SimRankbased similarity (ASRS), which was introduced as an efficient variation of the SRS. Therefore, an experiment was conducted to study the impact of these measures on RFcluE performance when utilizing those measures in the consensus function instead of CO. Fig. 5 shows the NMI of applying CO, CTS, SRS, and ASRS to measure the similarity between different partitions of data in the consensus function. The consensus function was applied to the same ensemble, which was generated using this parameter settings (ntrees = 10000,\( MN=\sqrt{N} \), and \( k=\sqrt{N} \)). For all datasets, CO, CTS, and SRS demonstrated comparable performance, while ASRS provided the worst performance compared with other measures for the PanAsian dataset. However, ASRS provided the best performance for Shriver’s dataset when the ensemble size was greater than 30. However, the difference in performance between these measures was not statistically significant, with a pvalue < 0.05.
CO used in the consensus function of RFcluE, represents a similarity matrix in which any similaritybased clustering can be applied to obtain the final clustering result. In RFcluE, we applied Ward’s agglomerative hierarchical clustering. However, different clustering techniques can be applied to the CO, such as Kmeans and spectral clustering. Therefore, another experiment was conducted wherein these clustering techniques were applied to the CO to examine their impact on the performance of RFcluE. Fig. 6 shows the NMI of the three clustering techniques—Ward’s, Kmeans, and spectral clustering—when applied to the same ensemble. The parameter settings used for the base clustering method were (M = {10, 20, 30, 40, 50}, ntrees = 10000,\( MN=\sqrt{N} \), and \( k=\sqrt{N} \)). We can see that Ward’s clustering, applied with RFcluE, has the best performance compared with Kmeans and spectral clustering across all the examined datasets, demonstrating statistically significant performance, with a pvalue < 0.05.
Comparison study
A comparison study was conducted to assess the performance of the proposed approach, RFcluE, against AWclust [48] and PCAclust [4], the two most popular methods for population structure analysis. Moreover, the performance of the RFcluE approach was compared against RFclust. Table 2 and Fig. 7 present the performance of PCAclust, AWclust, RFclust, and RFcluE on the real datasets evaluated using ARI, AC, and NMI. In RFcluE, the clustering result is based on combining multiple runs of RF clustering using a cluster ensemble framework, while RFclust is a clustering method that calculates the average of proximities derived from multiple runs of the RF algorithm and then applies Ward’s agglomerative hierarchical clustering. For RFcluE, the ensemble size M = 40 and FixedK scheme were used. For RFclust, the number of forests was equal to the ensemble size. For both RFcluE and RFclust, the RF parameters were set such that ntrees = 10000 and \( MN=\sqrt{N} \). All the compared methods were forced to divide the data into the real number of clusters in the examined dataset. Below, a discussion of the performance of RFcluE, AWclust, and PCAclust is presented, followed by a detailed comparison between RFcluE and RFclust under the same RF parameter settings.
RFcluE, AWclust, and PCAclust
The performance of the RFcluE, AWclust, and PCAclust approaches, based on ARI, AC, and NMI measures, on three real datasets is compared in Table 2. Fig. 7 shows that RFcluE generally outperforms PCAclust and AWclust over the three datasets. The bar plot for the PanAsian dataset indicates that RFcluE yields a superior clustering result when compared to the other approaches based on ARI, AC, and NMI. For the HapMap dataset, PCAclust had the worst performance, while RFcluE had the best performance. For Shriver’s dataset, all approaches had comparable performance, while RFcluE performed better than the other approaches considering all measures.
RFcluE versus RFclust
First, the effect of RF parameters was compared for both RFclust and RFcluE. As shown in Fig. 8, when RFclust is used, its performance is in most cases slightly changed as the tree size increases. An exception is HapMap, which shows a slight degradation in performance as the tree size increases. This confirms that, like RFcluE, building trees with \( MN=\sqrt{N} \) is always sufficient for any dataset. On the other hand, RFclust performance was not affected by changing the number of trees per forest nor the number of forests, as shown in Fig. 9. However, its performance was slightly improved with HapMap when increasing the number of trees in the forest from 1000 to 4000, and was slightly improved thereafter. Overall, RFclust exhibited stable performance across different values of the number of trees per forest and the number of forests. In addition, small tree sizes are always efficient to provide robust results.
The plots in Fig. 8 show that when comparing the performance of RFcluE with that of RFclust under the same RF parameters, RFcluE performs significantly better for all parameter settings, especially for PanAsian datasets. Another parameter is the number of forests to be constructed, which represents the ensemble size within the RFcluE approach. Unlike RFclust performance, RFcluE performance was improved when increasing the number of forests (M), especially when using a smaller number of trees. As shown in Fig. 9, the performance of RFcluE was improved over that of RFclust as the number of trees was increased over different numbers of forests (M). One exception was that RFcluE performance for HapMap became similar to that of RFclust when increasing the number of trees from 1000 to 4000, and subsequently stabilized for both approaches. However, the performance of RFcluE was much better than that of RFclust over different numbers of trees for the PanAsian dataset. A similar observation can also be made for Shriver’s dataset, except that the performance of both approaches did not change much across different numbers of trees. This observation indicates that multiple forests with 1000 trees each were enough to discover the structure of Shriver’s dataset.
Overall, we can conclude that it is both crucial and more efficient to use RF clustering as a base clustering method within a cluster ensemble framework instead of averaging the proximities of several forests and then applying clustering. RFcluE exhibited better performance than RFclust, especially for clustering in the PanAsian dataset. One important observation was that the clustering performance of RFcluE was significantly improved by increasing the number of forests and the number of trees per forest; unlike RFclust, where these parameters became irrelative. Moreover, the performance of RFcluE was more robust than that of RFclust with respect to tree size as long as sufficient trees per forest were constructed.
Diversity and quality analysis
The final experiment was conducted to assess the relationship between the diversity and quality of the generated ensemble and its influence on the quality of the ensemble’s final clustering. The diversity of base clusterings is a major factor that could affect the performance of the cluster ensemble approach. On the other hand, the evaluation of the quality of base clusterings is necessary to determine improvements in the quality of the final clustering of the cluster ensemble approach. To perform this experiment, the diversity and quality of base clusterings, as well as the quality of the ensemble’s final clustering, were calculated by applying Eq. (9), Eq. (10), and Eq. (8), respectively.
One source of diversity in base clustering is the number of clusters as an input to the base clustering method. TrueK, FixedK, and RandomK schemes, identified earlier, could generate different levels of diversity among base clusterings. Consequently, an experiment was conducted in order to study the diversity and quality of base clusterings generated by these different schemes with the following parameters: (\( M=40,\kern0.5em ntrees=10000,\kern0.5em MN=\sqrt{N} \)). Table 3 reports the diversity and quality of base clusterings, as well as the quality of the ensemble’s final clustering over the three datasets. Based on this table, we can observe that the TrueK scheme has the least diversity and the best quality of base clusterings; however, it produces the lowest quality of the ensemble’s final clustering. FixedK produces the highest quality of the final clustering among the three schemes for all datasets. This result confirms that selecting a greater number of clusters for base clustering methods than expected would introduce diversity within the ensemble. Thus, higher diversity could lead to more significant improvement in the quality of the ensemble’s final clustering. To this end, we can conclude that the quality of the base clusterings is not correlated with the quality of the cluster ensemble approach based on RFs, while combining base clusterings could produce a higherquality final clustering result due to their diversity.
The other sources of diversity are bagging, random subspace, and synthetic data generation applied within unsupervised RF algorithms. Therefore, AWcluE and PCAcluE were developed as an ensemble version of the two single clustering methods, PCAclust and AWclust, to demonstrate how the diversity and quality of the base clustering method influence the performance of the entire ensemble, especially RF clustering as a base clustering method of RFcluE. Accordingly, PCAcluE and AWcluE are defined as ensemblebased clustering methods that apply different base clustering methods but utilize the same consensus function of RFcluE. On the one hand, PCAcluE applies PCA and then Kmeans as a base clustering method. On the other hand, AWcluE calculates ASD and then applies Kmeans as a base clustering method. For both methods, Kmeans with a random initialization is considered as a source of diversity that can produce different partitions of the data with varying accuracy.
Table 4 reports the results for each ensemble method over the three datasets using the same parameters (M = 40,\( k=\sqrt{N} \)). By comparing the diversity and quality between the three ensemblebased clustering methods, we can see that RFcluE has the most diverse ensemble across the three datasets with moderate quality. However, it achieves the best performance and exhibits greater improvements in the quality of the ensemble’s final clustering over that of base clusterings. From this experimental result, we conjecture that the RF clustering method is most beneficial when applied within a cluster ensemble framework. Computing the RF proximity enables viewing highdimensional genetic data from different angles via bagging and random subspace, thus contributing to a more diverse ensemble than the two other ensembleclustering methods. Thus, combining multiple RF clustering results using an ensemble approach produces better clustering result than a single RF clustering.
Conclusions
This paper has presented RFcluE, a cluster ensemble approach based on an RF algorithm, to address the problem of population structure analysis. This approach is composed of two stages: ensemble construction, in which an RFbased clustering method is applied to generate a set of clusterings for the same dataset; and consensus function, which integrates all the clusterings to produce a final data clustering. Many experiments were conducted to empirically investigate the potential of the RFcluE approach on real genetic datasets in order to uncover the substructure of populations. In addition, a comparison study was carried out to compare RFcluE performance against existing, popular clustering methods for population structure analysis. The experimental results illustrated that the proposed approach, RFcluE, outperformed the other clustering approaches, providing more accurate results. Moreover, the experimental results indicated that combining multiple clusterings, generated based on RFs, within a cluster ensemble produces high quality and robust clustering results in comparison to a single run of RF clustering. This improvement in performance is a consequence of feeding the ensemble with diverse views of highdimensional genetic data obtained through bagging and random subspace, the two key features of the RF algorithm. To conclude, the major contributions of this paper are proposing and evaluating a cluster ensemble approach based on RFs and demonstrating its effectiveness for highdimensional, real genetic data. The paper also illustrated that applying a cluster ensemble approach to combine multiple RF clusterings produces more robust and highquality clustering results than clustering based on averaging the proximities derived from multiple forests. Future work should include the application of the RFcluE approach to other highdimensional biological data.
Abbreviations
 AC:

Accuracy
 ARI:

Adjusted Rand index
 ASD:

Allelesharing distance
 ASRS:

Approximate SimRankbased similarity
 CO:

Coassociation matrix
 CTS:

Connectedtriplebased similarity
 LD:

Linkage disequilibrium
 MDS:

Multidimensional scaling
 NMI:

Normalized mutual information
 PCA:

Principal component analysis
 RFcluE:

Random Forest cluster Ensemble
 RFclust:

Random Forest clustering
 SNPs:

Single nucleotide polymorphisms
 SRS:

SimRankbased similarity
References
 1.
Marchini J, Cardon LR, Phillips MS, Donnelly P. The effects of human population structure on large genetic association studies. Nat Genet. 2004;36:512–7.
 2.
Kidd KK, Pakstis AJ, Speed WC, Grigorenko EL, Kajuna SL, Karoma NJ, Kungulilo S, Kim JJ, Lu RB, Odunsi A. Developing a SNP panel for forensic identification of individuals. Forensic Sci Int. 2006;164:20–32.
 3.
Gao X, Starmer J. Human population structure detection via multilocus genotype clustering. BMC Genet. 2007;8:34.
 4.
Patterson N, Price AL, Reich D. Population structure and eigenanalysis. PLoS Genet. 2006;2:e190.
 5.
Reich DE, Cargill M, Bolk S, Ireland J, Sabeti PC, Richter DJ, Lavery T, Kouyoumjian R, Farhadian SF, Ward R, Lander ES. Linkage disequilibrium in the human genome. Nature. 2001;411:199–204.
 6.
Shi T, Horvath S. Unsupervised learning with random Forest predictors. J Comput Graph Stat. 2006;15:118–38.
 7.
Breiman L, Cutler A. Random forests manual (version 4.0). In Technical Report of the University of California. Berkeley: Department of Statistics, University of California; 2003.
 8.
Breiman L. Random forests. Mach Learn. 2001;45:5–32.
 9.
Breiman L. Bagging predictors. Mach Learn. 1996;24:123–40.
 10.
Tin Kam H. The random subspace method for constructing decision forests. IEEE Trans Pattern Anal Mach Intell. 1998;20:832–44.
 11.
Breiman L, Friedman J, Stone CJ, Olshen RA. Classification and regression trees. Wadsworth, New York: Wadsworth Inc.; 1984.
 12.
Pouyan MB, Birjandtalab J, Nourani M. Distance metric learning using random forest for cytometry data. In: 2016 38th annual international conference of the IEEE engineering in medicine and biology society (EMBC); 1620 Aug. 2016; 2016. p. 2590.
 13.
Kumar J, Doermann D. Unsupervised classification of structurally similar document images. In: 2013 12th International Conference on Document Analysis and Recognition; 2528 Aug. 2013; 2013. p. 1225–9.
 14.
Pei Y, Kou L, Zha H. Anatomical structure similarity estimation by random forest. In: 2016 IEEE international conference on image processing (ICIP); 2528 Sept. 2016; 2016. p. 2941–5.
 15.
Du S, Chen S. Detecting cosalient objects in large image sets. IEEE Sig Process Lett. 2015;22:145–8.
 16.
Wang Y, Xiang Y, Zhang J. Network traffic clustering using random Forest proximities. In: 2013 IEEE international conference on communications (ICC); 913 June 2013; 2013. p. 2058–62.
 17.
Uriarte RB, Tsaftaris S, Tiezzi F. Service clustering for autonomic clouds using random Forest. In: 2015 15th IEEE/ACM international symposium on cluster, cloud and grid computing; 47 may 2015; 2015. p. 515–24.
 18.
Uriarte RB, Tiezzi F, Tsaftaris SA. Supporting autonomic Management of Clouds: service clustering with random Forest. IEEE Trans Netw Serv Manag. 2016;13:595–607.
 19.
Puggini L, Doyle J, McLoone S. Fault detection using random Forest similarity distance. IFACPapersOnLine. 2015;48:583–8.
 20.
Peerbhay KY, Mutanga O, Ismail R. Random forests unsupervised classification: the detection and mapping of <italic>Solanum Mauritianum</italic> infestations in plantation forestry using Hyperspectral data. IEEE J Sel Top Appl Earth Obs Remote Sens. 2015;8:3107–22.
 21.
Afanador NL, Smolinska A, Tran TN, Blanchet L. Unsupervised random forest: a tutorial with case studies. J Chemom. 2016;30:232–41.
 22.
Swift S, Tucker A, Vinciotti V, Martin N, Orengo C, Liu X, Kellam P. Consensus clustering and functional interpretation of geneexpression data. Genome Biol. 2004;5:R94.
 23.
Ayad H, Kamel M. Finding natural clusters using multiclusterer combiner based on shared nearest neighbors. In Proceedings of the 4th international conference on Multiple classifier systems. Guildford, UK: SpringerVerlag; 2003. p. 166175.
 24.
Kim EY, Kim SY, Ashlock D, Nam D. MULTIK: accurate classification of microarray subtypes using ensemble kmeans clustering. BMC Bioinformatics. 2009;10:260.
 25.
Monti S, Tamayo P, Mesirov J, Golub T. Consensus clustering: a resamplingbased method for class discovery and visualization of gene expression microarray data. Machine Learning. 2003, 52:91118.
 26.
Yu Z, Wong HS, Wang H. Graphbased consensus clustering for class discovery from gene expression data. Bioinformatics. 2007;23:2888–96.
 27.
Fern XZ, Brodley CE. Random projection for high dimensional data clustering: A cluster ensemble approach. In Proceedings of the 20th International Conference on Machine Learning (ICML03), Washington, DC, USA. 2003: 186193.
 28.
Strehl A, Ghosh J. Cluster ensembles  a knowledge reuse framework for combining multiple partitions. Journal of Machine Learning Research. 2002;3:583617.
 29.
Dudoit S, Fridlyand J. Bagging to improve the accuracy of a clustering procedure. Bioinformatics. 2003;19:1090–9.
 30.
MinaeiBidgoli B, Topchy AP, Punch WF. A Comparison of Resampling Methods for Clustering Ensembles. In Proceedings of the International Conference on Artificial Intelligence; Las Vegas, Nevada, USA. 2004. p. 939945.
 31.
Topchy A, Jain AK, Punch W. A mixture model for clustering ensembles. In Proceedings of the 2004 SIAM International Conference on Data Mining. Lake Buena Vista, Florida: Society for Industrial and Applied Mathematics (SIAM); 2004. p. 379390.
 32.
Gionis A, Mannila H, Tsaparas P. Clustering aggregation. ACM Transactions on Knowledge Discovery from Data (TKDD). 2007;1:4.
 33.
Fred AL, Jain AK. Combining multiple clusterings using evidence accumulation. IEEE Transactions on Pattern Analysis and Machine Intelligence. 2005;27:835850.
 34.
IamOn N, Boongoen T, Garrett S, Price C. New cluster ensemble approach to integrative biological data analysis. International journal of data mining and bioinformatics. 2013;8:150168.
 35.
Pekalska E, Duin RPW. The Dissimilarity Representation for Pattern Recognition: Foundations And Applications. Singapore: World Scientific Publishing Co., Inc.; 2005.
 36.
Ward Jr JH. Hierarchical grouping to optimize an objective function. Journal of the American statistical association 1963;58:236244.
 37.
Ward Jr JH, Hook ME. Application of an hierarchial grouping procedure to a problem of grouping profiles. Educational and Psychological Measurement 1963.
 38.
The International HapMap C. A haplotype map of the human genome. Nature. 2005;437:1299–320.
 39.
Ngamphiw C, Assawamakin A, Xu S, Shaw PJ, Yang JO, Ghang H, Bhak J, Liu E, Tongsima S, Consortium HPAS. PanSNPdb: the panAsian SNP genotyping database. PLoS One. 2011;6:e21451.
 40.
Shriver MD, Kennedy GC, Parra EJ. The genomic distribution of human population substructure in four populations using 8525 SNPs. Human Genomics 2004, 1.
 41.
Shriver MD, Mei R, Parra EJ, Sonpar V, Halder I, Tishkoff SA, Schurr TG, Zhadanov SI, Osipova LP, Brutsaert TD, et al. Largescale SNP analysis reveals clustered and continuous patterns of human genetic variation. Human Genomics. 2005;2:81.
 42.
Hubert L, Arabie P. Comparing partitions. Journal of classification. 1985;2:193218.
 43.
Rand WM. Objective criteria for the evaluation of clustering methods. Journal of the American Statistical association. 1971;66:846850.
 44.
Ana L, Jain AK. Robust data clustering. In IEEE Computer Society Conference on Computer Vision and Pattern Recognition. Madison: IEEE Press; 2003: p. 128–133.
 45.
Hadjitodorov ST, Kuncheva LI, Todorova LP. Moderate diversity for better cluster ensembles. Information Fusion. 2006;7:264–75.
 46.
Kuncheva LI, Hadjitodorov ST. Using diversity in cluster ensembles. In Proceedings of the 2004 IEEE International Conference on Systems, Man, and Cybernetics (ICSMC). The Hague, Netherlands: IEEE; 2004. p. 12141219.
 47.
Iamon N, Garrett S. LinkCluE: A MATLAB Package for LinkBased Cluster Ensembles. Journal of Statistical Software. 2010.36:9
 48.
Gao X, Starmer JD. AWclust: pointandclick software for nonparametric population structure analysis. BMC Bioinformatics. 2008;9:77.
Acknowledgements
Not Applicable.
Funding
This research project was supported by a grant from the “Research Center of the Female Scientific and Medical Colleges”, Deanship of Scientific Research, King Saud University.
Availability of data and materials
PanAsian dataset is available in the PanAsian SNP Consortium website, http://www4a.biotec.or.th/PASNP/, HapMap dataset is available at ftp://ftp.ncbi.nlm.nih.gov/hapmap/phase_3/, and Shriver’s dataset was provided by Prof. Mark D. Shriver.
Author information
Affiliations
Contributions
LA designed and implemented the approach, designed and performed the experiments, analyze the results and wrote the manuscript; AH assisted with designing the approach and reviewed the manuscript. All authors read and approved the final manuscript.
Corresponding author
Correspondence to Luluah Alhusain.
Ethics declarations
Ethics approval and consent to participate
Not applicable.
Consent for publication
Not applicable.
Competing interests
The authors declare that they have no competing interests.
Publisher’s Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
Open Access This article is distributed under the terms of the Creative Commons Attribution 4.0 International License (http://creativecommons.org/licenses/by/4.0/), which permits unrestricted use, distribution, and reproduction in any medium, provided you give appropriate credit to the original author(s) and the source, provide a link to the Creative Commons license, and indicate if changes were made. The Creative Commons Public Domain Dedication waiver (http://creativecommons.org/publicdomain/zero/1.0/) applies to the data made available in this article, unless otherwise stated.
About this article
Received
Accepted
Published
DOI
Keywords
 Cluster ensemble
 Random Forests
 Genetic population
 Population structure analysis
 Random Forest proximity
 Highdimensional data
 Ensemble diversity
 Single nucleotide polymorphism
 Normalized mutual information