Determination of sample size for a multi-class classifier based on single-nucleotide polymorphisms: a volume under the surface approach
© Liu et al.; licensee BioMed Central Ltd. 2014
Received: 25 June 2013
Accepted: 4 June 2014
Published: 14 June 2014
Data on single-nucleotide polymorphisms (SNPs) have been found to be useful in predicting phenotypes ranging from an individual’s class membership to his/her risk of developing a disease. In multi-class classification scenarios, clinical samples are often limited due to cost constraints, making it necessary to determine the sample size needed to build an accurate classifier based on SNPs. The performance of such classifiers can be assessed using the Area Under the Receiver Operating Characteristic (ROC) Curve (AUC) for two classes and the Volume Under the ROC hyper-Surface (VUS) for three or more classes. Sample size determination based on AUC or VUS would not only guarantee an overall correct classification rate, but also make studies more cost-effective.
For coded SNP data from D(≥2) classes, we derive an optimal Bayes classifier and a linear classifier, and obtain a normal approximation to the probability of correct classification for each classifier. These approximations are then used to evaluate the associated AUCs or VUSs, whose accuracies are validated using Monte Carlo simulations. We give a sample size determination method, which ensures that the difference between the two approximate AUCs (or VUSs) is below a pre-specified threshold. The performance of our sample size determination method is then illustrated via simulations. For the HapMap data with three and four populations, a linear classifier is built using 92 independent SNPs and the required total sample sizes are determined for a continuum of threshold values. In all, four different sample size determination studies are conducted with the HapMap data, covering cases involving well-separated populations to poorly-separated ones.
For multi-classes, we have developed a sample size determination methodology and illustrated its usefulness in obtaining a required sample size from the estimated learning curve. For classification scenarios, this methodology will help scientists determine whether a sample at hand is adequate or more samples are required to achieve a pre-specified accuracy. A PDF manual for R package “SampleSizeSNP” is given in Additional file 1, and a ZIP file of the R package “SampleSizeSNP” is given in Additional file 2.
KeywordsArea under the receiver operating characteristic curve Classification HapMap data Heterogeneous stock mice data Probability of correct classification Receiver operating characteristic Sample size determination
Data on single-nucleotide polymorphisms (SNPs) have been found to be useful in predicting an individual’s class membership or his/her response to a drug, susceptibility to environmental factors such as toxins, and the risk of developing a particular disease, among others [1–5]. The classification literature provides a variety of classifiers (e.g., Support Vector Machine, genetic programming, Neural Networks and Logistic Regression) and sample size determination methods [6–10], but most of these are only applicable to continuous data.
Recently Liu et al.  developed an optimal Bayes classifier and a linear classifier for coded SNP data from two classes, and obtained a normal approximation to the probability of correct classification (PCC) for each classifier. They also proposed a sample size determination methodology to determine an adequate sample size, which ensures that the difference between the two approximate PCCs is below a pre-specified threshold value. Using Monte Carlo simulations, Liu et al.  assessed the validity of their approximations. Furthermore, they illustrated the performance of their sample size determination method via simulations and a real data analysis using the HapMap data on two populations—Chinese and Japanese.
While Liu et al.  showed that their sample size determination method is competitive, they also pointed out that an additional maximization step is required in order to determine the discrimination values for each of their classifiers; see their REMARK1 in their article for more details. When there are three or more classes, however, determination of such discrimination values is not only more difficult, but also increases the overall computational burden. In a two-class scenario, a well known way to overcome this difficulty is to consider the Receiver Operating Characteristic (ROC) curve, which plots the True Positive Rates vs. False Positives Rates, at various discrimination values [12, 13]. Note that the ROC allows the discrimination value to be varied and it simultaneously explores all possible combinations of the correct classification rates . The Area Under the ROC curve (AUC) is commonly used as a scalar performance measure, which allows classifiers to be compared independent of the discrimination values. Unfortunately, the AUC measure is only applicable to a two-class scenario. A popular extension of the AUC measure, known as the Volume Under the ROC hyper-Surface (VUS) measure, is often used in a multi-class scenario (see e.g., Landgrebe and Duin  and Landgrebe and Paclik 2010 ).
This article revisits the problem of sample size determination in classification scenarios involving coded SNP data, but uses the AUC and the VUS as performance measures for two-class and multi-class scenarios, respectively. More specifically, for coded SNP data from D(≥2) classes, we derive an optimal Bayes classifier and obtain a normal approximation to its probability of correct classification, which is denoted by P C C(∞). We also derive a linear classifier and obtain a normal approximation to its probability of correct classification, which is denoted by . For an overall assessment of each of the classifiers, we define the scalar measures AUC (for two-class) and VUS (for multi-class), and correspondingly define the quantities for each classification scenario. For the two-class scenario, we propose to determine the sample size n for which , where γ∈(0,1) is a pre-specified threshold value. Whereas, for the multi-class scenario, we propose to determine the sample size n for which . A computational method to determine the total sample size for various values of γ is described. Monte Carlo simulations are carried out to corroborate our theoretical approximations, and the performance of our sample size determination method is assessed via simulations and analysis of the HapMap data consisting of 3 and 4 populations, respectively. In all, four different sample size determination studies are conducted with the HapMap data, covering cases involving well-separated populations to poorly-separated ones. Details are given in the data analysis section.
R software was used to carry out all the computations. A PDF manual for R package “SampleSizeSNP” is given in Additional file 1, and a ZIP file of the R package “SampleSizeSNP” is given in Additional file 2.
For an m such that , the data vector consists only of m SNPs, which are statistically independent. That is, the rest of the (p−m) correlated SNPs are not used for classification.
- 2.For each k=1,…,D and j=1,…,m, we postulate Hardy-Weinberg equilibrium, according to which the probability mass function of the coded SNP (X j ) belonging to class k is given by
where θk,j is the minor allele frequency at locus j in class k, and by definition θk,j∈(0.01,0.5). Here, θk,j<0.5 because it is the minor allele frequency, and θk,j>0.01 ensures that the polymorphism is not a mutation. For each k=1,…,D, let denote the parameter vector corresponding to the class C k .
There is a percentage ρ of the m SNPs with marginal effect on any two classes, and let l=⌊ρ m⌋ be the number of SNPs with marginal effects.
The optimal classifier and its PCC
In Additional file 3: Appendix 1, we derive a normal approximation for P C C(∞), as l→∞. That is, for large l, we show that
where ϕ is the (D−1)-dimensional multivariate normal density, is a multiple integral, and are (D−1)×1 vectors, and Σl,k is a (D−1)×(D−1) matrix. All these quantities are defined in Additional file 3: Appendix 1.
In Additional file 3: Appendix 4, we give an expression for (4) for the case D=3.
A linear classifier and its PCC
for all k′≠k, where , are the maximum likelihood estimators of , respectively. Also, the values of the weights wj,n(k,k′) in (5) are determined in the following way: For each j=1,…,m and k′≠k, suppose we test the hypothesis versus . Then wj,n(k,k′)=1 if is rejected; else wj,n(k,k′)=0. In Additional file 3: Appendix 2, we use the large sample theory to derive a Wald test of level α to test versus , and an expression for the power, , of this test, when .
AUC and VUSfor the optimal and linear classifiers
By replacing ξk.k by [see (8)] in the above definitions of R O C,A U C and the VUS, we obtain corresponding expressions for the linear classifier in (5). We denote the resulting ones as and . In Additional file 3: Appendix 4, we derive these expressions for the case D=3.
Computation of VUS
As is evident from (9), the computation of VUS involves high dimensional integration. Given below is a brief description of the steps involved in the computation of VUS. For ease of exposition, we will denote ξ k =ξk,k, k=1,…,D. First, we randomly generate the thresholds (see (9)) and compute the corresponding satisfying (7). Note that the contributes to the integration in VUS only if all the ξ k ’s are positive.
To find as many values that contribute to the integration as possible, we use the ant colony optimization algorithm, where only the values corresponding to the values that contribute to the integration are retained. However, these are perturbed by a small noise and the resulting values are used as seeds for the next iteration. Then, we use the genetic algorithm to obtain another value located in a different region within (0,1) k , which also contributes to the integration. We use the ant colony algorithm and the genetic algorithm alternatively to eventually generate a dense set of values that contribute to the integration. Note that the process is such that the newly generated values are appended to all the previously generated values.
Now, to compute the volume, V U S(∞), we use the convhulln function in the qhull R-package. Note that the convhulln function is designed to determine the convex hull of a set of D-dimensional points and thus compute the volume of the hull. In view of this, in order to compute the volume, V U S(∞), a base of (this is same as the vector, except that one of its components, e.g. the first component, is set to 0) is appended to the original . Since in each iteration the new values are appended to the old values from the previous iterations, and the VUS is concave, the computed VUS is supposed to increase in value with each iteration. We stop appending the new values when |V U S o l d −V U S n e w |<0.001. When this criterion is satisfied, we obtain the value of V U S(∞). Similarly, the values of are calculated.
Sample size determination using VUS or AUC
For the case D=2, we determine the sample size n satisfying the condition: . A simulation study for the case D=2 is carried out in Additional file 3: Appendix 5 to assess the performance of our sample size determination algorithm.
Monte Carlo simulations
Before we illustrate the performance of our sample size determination method based on AUC or VUS, we present results from an extensive Monte Carlo simulation study conducted to verify the accuracy of the approximations for and , respectively, and study their behavior as a function of n and other parameters. Here, we present the numerical assessments based on the VUS for the cases D=3 and 4, respectively. However, as mentioned above, the assessments based on the AUC for the case D=2 are given in Additional file 3: Appendix 5. Henceforth, we will set n k =n for all k=1,…,D, and we will use n instead of to simplify notations.
Performance of optimal and linear classifiers
D = 3
D = 4
Sample size determination: here, D = 3 and 4 , and n is the sample size for each class satisfying:
m = 30
m = 50
m = 100
m = 200
Application to the HapMap data
The aim of the International HapMap Project is to develop a haplotype map of the human genome, which will describe the common patterns of human DNA sequence variation.
The HapMap data (Phase III) consists of eleven populations with about p=1.2×106 SNPs. Here, we consider the following nine populations in order to illustrate our sample size determination algorithm: ASW—African ancestry in Southwest USA with 87 subjects; CEU—Utah residents with Northern and Western European ancestry from CEPH collection with 167 subjects; CHB—the Han Chinese individuals from Beijing with 137 subjects; CHD—Chinese in Metropolitan Denver, Colorado with 109 subjects; GIH—Gujarati Indians in Houston, Texas with 101 subjects; JPT—the Japanese individuals from Tokyo with 113 subjects; MEX—Mexican ancestry in Los Angeles, California with 86 subjects; TSI—Toscans in Italy (TSI) with 102 subjects; and YRI—Yoruba in Ibadan, Nigeria with 203 subjects. With these, we created four sample size determination studies, of which the first three involve three populations (D=3), and the last study involves four populations (D=4). More specifically, we conducted our sample size determination studies with the following population groupings: (I) (CEU, GIH, MEX); (II) (ASW, TSI, YRI); (III) (CHB, JPT, CHD); and (IV) (CHB, JPT, CHD, GIH).
Based on all the available subjects, we extracted pair-wise independent SNPs using the following steps. Suppose L is a set of SNPs, then: (I) form a set S with one SNP from L and update S after the next step; (II) from the remaining SNPs in L, choose one SNP that is independent of every SNP in S using Kendall’s τ coefficient as a test statistic to test pair-wise independence, and then add this new SNP to S. Here, we concluded independence if the Kendall’s τ-value <0.05; (III) Repeat (II) until each remaining SNP in L is correlated with at least one SNP in S. This procedure yielded a set S with m=92 pair-wise independent SNPs, and with these we built our linear classifier.
For example, if we set γ=0.10 (i.e., ), then in the three population (CEU,GIH, MEX) case, the and about 62 observations are required for each class with a total sample size of 186, whereas in the three population (ASW, TSI, YRI) case, the and about 150 observations are required for each class with a total sample size of 450. Note that, for γ=0.10, in study (I) the required sample sizes for each population is less than what is currently available, whereas in study (II), we would need 63 and 48 more observations for the populations ASW and TSI, respectively. For the three population (CHB, JPT and CHD) case, if we set γ=0.10 then the and about 244 observations are required for each class with a total sample size of 732. Clearly, for study (III) at least 100 more observations are needed for each population (CHB, JPT and CHD) when γ=0.10. Finally, for the four population (CHB, JPT, CHD, GIH) case, setting γ=0.10 yields that the and about 279 samples are required for each class with a total sample of 1,116. Once again, at least 150 more observations are needed for each of the four populations when γ=0.10.
The results from the four HapMap studies suggest that the value is large and the required total sample size is small when the populations are well-separated [as in study (I)]. Whereas, when the populations are moderately-separated [as in study (II), where the populations ASW and YRI may be similar], the value reduces and the required total sample size increases moderately. When the populations are poorly-separated [as in study (III), where all the three populations may be similar], the value reduces even further and there is a substantial increase in the required total sample size. Finally, in the four population study, where three of the populations are poorly-separated, once again we see a further reduction in the value and a corresponding increase in the required total sample size. Although not reported here, we also considered other well-/moderate-/poorly- separated cases with the HapMap data and observed similar results as the ones reported here.
It is well known in the classification literature that the performance of a classifier depends on how well separated the classes are. Similarly, the studies above involving the HapMap data show that the performance of our sample size determination methodology also depends on the extent of separation between populations. While our methodology provides a formal way of determining an approximate total sample size for each specified value of γ, it is clear from the HapMap data analysis that it is not possible to propose a universal γ value. Nevertheless, if the classes are well-separated or moderately-separated, then we believe that γ=0.10 may be a good choice for many frequently encountered data sets in classification problems.
We have built an optimal Bayes classifier and a linear classifier based on coded SNP data from two or more classes. For these classifiers, we have considered the two commonly used scalar performance measures, the Area Under the ROC curve (AUC) and the Volume Under the ROC hyper-Surface (VUS), which allow classifiers to be compared independent of discrimination values. We have illustrated the performance of a sample size determination methodology, which selects the smallest total sample size n such that the criterion is satisfied. While the approximations to the VUS (or AUC) obtained here provide the necessary theoretical justification, the simulations and the HapMap data analysis presented here illustrate the practical value of our sample size determination method.
The fact that the HapMap contains data on multiple populations belonging to similar or dissimilar geographical locations enabled us to test the performance of our sample size determination method on three different multi-class scenarios involving well-separated, moderately-separated, and poorly-separated populations. We have shown that the the extent of separation between the populations and the choice of threshold value affect the total sample size required to satisfy the criterion. With regard to the choice of the threshold value γ in other practical contexts, we recommend that the user take into consideration the cost of obtaining more samples and choose an appropriate value of γ that gives an acceptable precision. In other words, if the cost of sampling is affordable then the user may want to sample more to achieve a higher precision (lower γ value) using our classifier; otherwise, the user has to settle for a higher γ value that makes use of all the available samples. We also infer from our HapMap data analysis that a value of may indicate the extent of separation between the classes. Thus, the value of could also give some prior guidance on the choice of γ values, especially in instances where the cost of sampling is a serious concern.
In summary, for multiple classes, we have developed an asymptotic methodology based on AUC or VUS to estimate the learning curve of SNP classifiers. It is shown that the required total sample size can be obtained from the estimated learning curve for each pre-specified threshold value. In classification problems, sample size determination is important due to cost considerations. This methodology will help scientists determine if a sample at hand is adequate or more observations are necessary to achieve a pre-specified accuracy, and thus help users strike an optimal balance between precision and cost.
We would like to thank the Editor and the two reviewers for their careful reading and insightful suggestions, which greatly improved the content and the presentation of the article. T.N.S. was supported by a grant from the National Security Agency [H98230-11-1-0188] and the National Science Foundation [#1309665].
- Guzzetta G, Jurman G, Furlanello C: A machine learning pipeline for quantitative phenotype prediction from genotype data. BMC Bioinformatics. 2010, 11 (Suppl 8): S3-10.1186/1471-2105-11-S8-S3.View ArticlePubMed CentralPubMedGoogle Scholar
- Lee SH, van der Werf JHJ, Hayes BJ, Goddard ME, Visscher PM: Predicting unobserved phenotypes for complex traits from whole-genome SNP data. Plos Genet. 2008, 4: e1000231-10.1371/journal.pgen.1000231.View ArticlePubMed CentralPubMedGoogle Scholar
- Nunkesser R, Bernholt T, Schwender H, Ickstadt K, Wegener I: Detecting high-order interactions of single nucleotide polymorphisms using genetic programming. Bioinformatics. 2007, 23: 3280-3288. 10.1093/bioinformatics/btm522.View ArticlePubMedGoogle Scholar
- Wray NR, Goddard ME, Visscher PM: Prediction of individual genetic risk to disease from genome-wide association studies. Genome Res. 2007, 17: 1520-1528. 10.1101/gr.6665407.View ArticlePubMed CentralPubMedGoogle Scholar
- Zhou N, Wang L: Effective selection of informative SNPs and classification on the HapMap genotype data. BMC Bioinformatics. 2007, 8: 484-492. 10.1186/1471-2105-8-484.View ArticlePubMed CentralPubMedGoogle Scholar
- De Valpine P, Bitter HM, Brown MPS, Heller J: A simulation-approximation approach to sample size planning for high-dimensional classification studies. Biostatistics. 2009, 10: 424-435. 10.1093/biostatistics/kxp001.View ArticlePubMed CentralPubMedGoogle Scholar
- Dobbin KK, Simon RM: Sample size determination in microarray experiments for class comparison and prognostic classification. Biostatistics. 2005, 6: 27-38. 10.1093/biostatistics/kxh015.View ArticlePubMedGoogle Scholar
- Dobbin KK, Simon RM: Sample size planning for developing classifiers using high-dimensional DNA microarray data. Biostatistics. 2007, 8: 101-117. 10.1093/biostatistics/kxj036.View ArticlePubMedGoogle Scholar
- Dobbin KK, Zhao Y, Simon RM: How large a training set is needed to develop a classifier for microarray data. Clin Cancer Res. 2008, 14: 108-114. 10.1158/1078-0432.CCR-07-0443.View ArticlePubMedGoogle Scholar
- Mukherjee S, Tamayo P, Rogers S, Rifkin R, Engle A, Campbell C, Golub TR, Mesirov JP: Estimating dataset size requirements for classifying DNA microarray data. J Comput Biol. 2003, 10: 119-142. 10.1089/106652703321825928.View ArticlePubMedGoogle Scholar
- Liu X, Wang Y, Rekhaya R, Sriram TN: Sample size determination for classifiers based on single-nucleotide polymorphisms. Biostatistics. 2012, 13: 217-227. 10.1093/biostatistics/kxr053.View ArticlePubMedGoogle Scholar
- Metz C: Basic principles of ROC analysis. Seminars Nucl Med. 1978, 3: 283-298.View ArticleGoogle Scholar
- Fawcett T: An introduction to ROC analysis. Pattern Recogn Lett. 2005, 27: 861-874.View ArticleGoogle Scholar
- Landgrebe T, Duin RPW: Approximating the multiclass ROC by pairwise analysis. Pattern Recogn Lett. 2007, 28: 1747-1758. 10.1016/j.patrec.2007.05.001.View ArticleGoogle Scholar
- Landgrebe T, Paclik P: The ROC skeleton for multiclass ROC estimation. Pattern Recogn Lett. 2010, 31: 949-958. 10.1016/j.patrec.2009.12.037.View ArticleGoogle Scholar
This article is published under license to BioMed Central Ltd. This is an Open Access article distributed under the terms of the Creative Commons Attribution License (http://creativecommons.org/licenses/by/2.0), which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly credited.