Background

Microarray data analysis is a high throughput method used to gain information about gene functions inside cells. This information is in turn used to detect the presence or absence of disease [13], and gain a better understanding of a disease mechanism [4].

A particularly useful application of microarray technology uses microarray data to detect the presence of disease by combining gene expression levels from a number of genes, to provide information on whether disease is present (classification) or the risk for the occurrence of disease in the future (prediction). While very complex classifiers can be constructed, a number of authors have expressed concern with the “black box” nature of these approaches [5] preferring simpler more interpretable classifiers in clinical applications [6, 7]. It is noted that the preference for the latter kind of classifiers should not be at the expense of their performance.

Classification involves, at its most fundamental level, a comparison between expression levels in one or more genes between two or more conditions (e.g., disease versus no disease). This comparison can be based on a fairly heuristic criterion (e.g., fold-change in gene expression [8]), or by using parametric or non-parametric statistical methods [912]. There are several advantages and disadvantages with each of these methods. For example, it is biologically plausible that genes with large differential expression levels should be part of a classification criterion. However, the fold-change criterion does not take gene expression variability into account and determining a cutoff is an arbitrary exercise [13]. On the other hand, parametric statistical methods, which are based on some variant of the t-test, provide some sense of one’s confidence on the gene expression difference, but frequently lose the intuitive appeal of heuristic methods like fold-change (e.g., when even small differences are statistically significant). In addition, parametric methods make strong and frequently untenable assumptions regarding the distribution of gene expression levels [13]. Non-parametric methods, which are based on ranking gene expression levels, are expected to lose some information because of the use of ranks instead of actual gene-expression data. However, such methods are robust to deviations from parametric assumptions [13], and are less vulnerable to biases arising from data normalization and other pre-processing steps [14], which are plausibly assumed to be rank-preserving [6, 7].

The fact that the TSP provides classifiers based on only two genes is also an attractive compromise in the so-called “bias-variance” tradeoff [15]. As a classifier’s performance is a combination of variance (random error) and bias (systematic error), in many cases, high-dimensional classifiers with low bias (due to good performance in the current sample) have large variances (i.e., poor precision) in new samples. By contrast, simpler (and thus more rigid) classifiers, while possibly having higher levels of bias, are less influenced by a specific sample and may have better overall performance (smaller variance) in multiple samples.

The simple TSP classifiers, it was hoped, would perform sufficiently well both in the current sample as well as in new samples. The TSP is a rank-based classifier in the sense that it uses the rankings of gene expression levels within a gene profile rather than the levels themselves, an approach with significant advantages due to the nonparametric nature of the classification technique. The central idea behind the TSP classifier is that it identifies two genes whose gene expression ranking changes between the two conditions under consideration. This change lends itself to a simple biological interpretation as an inversion of mRNA abundance of the two genes in the two conditions under consideration. The pair of genes selected by the TSP [6], referred to as the top scoring pair (TSP), is found by the following approach: Consider G genes which have been profiled by microarray analysis. Let n1 be the number of experiments from the first class with expression levels \(Y_{i}=\{Y_{i,1}, Y_{i,2}, \cdots, Y_{i,n_{1}}\}\phantom {\dot {i}\!}\), and let n2 be the number of experiments from the second class with expression levels \(\phantom {\dot {i}\!}Y_{i}=\{Y_{i,n_{1}+1}, Y_{i,n_{1}+2}, \cdots, Y_{i,n}\}\), where n=n1+n2. Given a pair of genes (i,j), 1≤ijG, the reversal score of the pair was defined in [6] as

$$ \Delta_{ij} = \left| P(Y_{i} > Y_{j}|C=1) - P(Y_{i} > Y_{j}|C=2)\right| $$
(1)

where P(Y i >Y j |C=m) denotes the probability that the expression level of gene i is larger than the expression level of gene j in samples from class C, with C being equal to m=1,2. The score Δ ij can be empirically approximated by the expression [6]

$$ D_{ij} =\left|\frac{\sum_{\ell=1}^{n_{1}}{{I}_{1}(Y_{i,\ell}>Y_{j,\ell})}}{n_{1}} - \frac{\sum_{\ell=n_{1}+1}^{n}{{I}_{2}(Y_{i,\ell}>Y_{j,\ell})}}{n_{2}}\right| $$
(2)

where index indicates the th subject, 1≤n and I m (Yi,>Yj,)=1 if Yi,>Yj, in class m=1,2, and 0 otherwise.

Obviously, the larger the Δ ij , the higher the probability that the expression levels of genes i and j have reverse relative rankings in the two groups, and it is exactly this property that is used for classification by the TSP. More specifically, let (α,β) be the pair of genes that yields the maximum score Δ α β =max{Δ ij } (referred to as the Top Scoring Pair (TSP) [6]). Then the classification is performed as follows:

Assume that

$$P(Y_{\alpha} > Y_{\beta}|C=1) > P(Y_{\alpha} > Y_{\beta}|C=2) $$
(3)

i.e.,

$$ \frac{\sum_{\ell=1}^{n_{1}}{{I}_{1}(Y_{\alpha,\ell}>Y_{\beta,\ell})}}{n_{1}} > \frac{\sum_{\ell=n_{1}+1}^{n}{{I}_{2}(Y_{\alpha,\ell}>Y_{\beta,\ell})}}{n_{2}} $$
(4)

Then a new subject s whose measured expression levels for genes a and b are Yα,s and Yβ,s respectively, will be classified as belonging to the first class if Yα,s>Yβ,s, and to the second class otherwise.

The genes in the top scoring pair as selected by the TSP method may have a problem, as Lin et al. [5] also point out: the selected genes may not be a pair of genuinely up-regulated and down-regulated genes, but one of the selected genes in the pair happens to serve only as a reference or “pivot” gene that may lead to a high TSP score but a rather non-informative gene pair. Most researchers have used more complicated methods or selected more features in order to overcome the mentioned problems. In the proposed method we employ a simple statistic associated with the Receiver Operating Characteristic (ROC) curve that is commonly known as the Area Under the ROC curve (AUROC) or the Area Under the Curve (AUC), for short. The ROC curve and the AUC in particular have been widely used as a measure for microarray classification and other medical diagnostic tests (see, e.g., [1623].

The proposed method, referred to as AUCTSP (AUC-based TSP), uses similar ideas as the TSP, thus benefiting from the simplicity of the TSP approach, but enhances TSP by making the resulting classifier less prone to overfitting, achieving higher classification accuracy and avoiding the selection of pivot genes as members of the top scoring pair of genes.

Methods

In this manuscript we propose the AUCTSP, a classifier that works according to the same principle as TSP but differs from the latter in that the probabilities that determine the top scoring pair are computed based on the relative rankings of the two marker genes across all subjects instead of within each individual subject. Although the classification is still done on an individual-subject basis, consideration of all subject data in the estimation of the ranking reversals results in a classifier with higher accuracy. This performance superiority of AUCTSP over TSP is demonstrated through simulations and case studies (see “Results” section) involving classification in ovarian, leukemia, colon, prostate and breast cancers and diffuse large b-cell lymphoma.

The proposed AUCTSP classifier

The score that TSP computes is based on the probability P(Y i >Y j |C=m) that the expression level of gene i is larger than the expression level of gene j in samples from the m-th class, m=1,2. This probability was approximated in [6] by the proportion of individuals of class m with higher expression level in gene i than in gene j out of all individuals in class m, i.e., by the probability

$$ P_{\text{TSP}}(Y_{i} > Y_{j}|C=m)=\frac{\sum_{\ell=1}^{n_{m}}{{I}_{m}(Y_{i,\ell}>Y_{j,\ell})}}{n_{m}} $$
(5)

We propose to approximate the original probability P(Y i >Y j |C=m) by the probability that a randomly chosen individual from class m has an expression level for gene i that is larger than that of a randomly chosen individual from class m (m=1,2) for gene j.

The estimate of the original probability P(Y i >Y j |C=m) in the proposed AUCTSP method is given by

$$P_{\text{AUCTSP}}(Y_{i}>Y_{j}|C=m)=$$
$$ \frac{\sum_{k=1}^{n_{m}}\sum_{\ell=1}^{n_{m}}{{I}(Y_{i,k}>Y_{j,\ell})}}{n_{m}^{2}} $$
(6)

The numerator in Eq. 6 denotes the sum over all samples k, 1≤kn m , of the number of times that the expression level of gene i in sample k is larger than the expression level of gene j in some other sample k,1≤n m , from the same class m (m=1 or 2).

The probability PAUCTSP can be calculated by the Area Under the ROC Curve (AUC) [23]. The AUC statistic has been used extensively in diagnostic test validation [1820, 22, 23] and gene feature selection [21] in two-group settings. In our case here, group 1 is taken to be the set of expression levels of gene i in class m, and group 2 is taken to be the set of expression levels of gene j in the same class m. It is well established that, for independent samples, the AUC statistic is the minimum-variance unbiased estimate of P(X>Y) [24]. In correlated samples (as we have here, since the gene expression levels are measured on the same individual i=1,2,⋯,n m for m=1,2), it is expected that PAUCTSP is still an unbiased estimate of P(X>Y) and should generate more precise estimates of the probability P(Y i >Y j |C=m) compared to PTSP, unless the correlation of gene expression levels between genes i and j in the same individual is too high (thus leading to an inflated variance of the AUC-based estimator). In addition, the AUCTSP classifier, which is based on a summary measure derived from all subjects (compared to the single-subject approach in the TSP), has the potential to yield a top scoring pair that is less susceptible to the specific training data, thus further avoiding overfitting compared to the TSP. The better performance of AUCTSP is corroborated by our experimental results.

We highlight the following two points about our use of the AUC statistic in the proposed method: (i) the AUC statistic is traditionally applied on two groups one of which is the “healthy” and the other one the “diseased,” whereas in our method we apply it on gene expression profiles from the same (“healthy” or “diseased”) group; (ii) although the PAUCTSP is obtained from all subjects, the classification rule that we obtain in the AUCTSP classifier is still applied on the expression levels of the marker genes from the same single subject, exactly as in the TSP classifier.

To elucidate the intuition behind the AUCTSP classifier, consider the following example. Assume that the expression levels of a gene A for 5 healthy subjects are as given in Table 1. The probability that the expression level of A is less than the level of B in the healthy subjects is 5/5 = 1 while the probability that the level of A is less than the level of B in the diseased subjects is 0, yielding an overall TSP score \(D^{\tiny \mathrm {TSP}}_{AB}= 1\). Contrast the above with the situation involving two other genes, C and D (Table 1). The probability that the expression level of C is less than the level of D in the healthy subjects is 4/5 = 0.8, while the probability that the expression level of C is less than the level of D in the diseased subjects is 1/5 = 0.2. This yields an overall TSP score \(D^{\tiny \text {TSP}}_{CD}=0.6\), which is less than the score of pair A and B, and consequently the pair C and D would be discarded by the TSP. However, the distributions of the expression levels of C and D in the healthy (and the diseased) subjects exhibit greater separation than those for A and B and thus, using genes C and D for classification is arguably preferable.

Table 1 Gene expression levels in two genes

The above intuitive preference for pair (C,D) is supported by the score derived for these two genes according to the proposed AUCTSP approach. The non-parametric estimate of the AUC for pair (C, D) on the healthy subjects is 24/25 = 0.96, and on the diseased subjects it is 1/25 = 0.04. This yields an overall AUCTSP score of \(D^{\tiny \text {AUCTSP}}_{CD}=0.92\), while the corresponding AUCTSP score for the (A,B) gene pair is \(D^{\tiny \text {AUCTSP}}_{AB}=15/25 - 10/25 = 0.2\) and, therefore, the (C,D) pair is preferred over(A,B) by the proposed approach. We note here that the claim about the greater separation of the gene expression distributions is not based in any way on the actual values of the data, only on their ranking. This in turn means that the proposed method will be robust in selecting the top scoring pair and will not be affected by outliers in the gene expression data and will also be invariable to any rank-preserving normalization technique.

Results

The AUCTSP classifier was implemented in the C programming language. The evaluation of the methodology was based on (i) simulations and (ii) case studies.

Simulations

We compared the estimations given by TSP (Eq. 5) and AUCTSP (Eq. 6) for the probability P(X>Y) involved in the computation of the TSP and AUCTSP scores. We generated random expression levels for “genes” X and Y from normal distributions with different combinations of mean μ and deviation σ for different sample sizes, where μ X is greater than or equal to μ Y in all of the simulated cases. In this case, the probability P(X>Y) is given by the detectability index A z defined by Metz et al. [22] as:

$$ A_{z} = P(X > Y) = \Phi\left (\frac{\frac{|\mu_{X} - \mu_{Y}|}{\sigma_{X}}}{\sqrt{1 +\left (\frac{\sigma_{Y}}{\sigma_{X}}\right)^{2}}}\right) $$
(7)

where Φ() denotes the cumulative distribution function (CDF) of the standard normal distribution and μ X , σ X , and μ Y , σ Y denote the mean and standard deviation of the assumed normal distributions for X and Y, respectively.

The cases chosen for comparison are two normal distributions with:

  1. (i)

    small means (μ X =1, μ Y =0) with small variances (σ X =1, σ Y =1);

  2. (ii)

    small means (μ X =1, μ Y =0) with large variances (σ X =3, σ Y =3);

  3. (iii)

    large means (μ X =5, μ Y =0) with small variances (σ X =1, σ Y =1);

  4. (iv)

    large means (μ X =5, μ Y =0) with large variances (σ X =3, σ Y =3);

  5. (v)

    equal small means (μ X =1, μ Y =1) with a small variance for one distribution (σ X =1) and a large variance for the other distribution (σ Y =3);

  6. (vi)

    equal large means (μ X =5, μ Y =5) with a small variance for one distribution (σ X =1) and a large variance for the other distribution (σ Y =3).

The results for different sample sizes N=10,20,30,40 are shown in Table 2. Columns 4 and 5 show the estimates of probability P(X>Y) obtained by TSP and AUCTSP over 1000 random trials. The theoretical probability A z =P(X>Y) (see Eq. 7) is shown in the last column. With bold, we show the value that is closer to the theoretical value A z . As can be seen, for the cases where both simulated gene expression distributions have equal variances (cases i-iv), the AUCTSP and TSP estimates are virtually identical and are very close to the theoretical probability even for small sample sizes. In the two cases where the variance in one of the genes is greater (cases v-vi), both estimators do poorly for small sample size N and improve with increasing N, but the AUCTSP is always closer to the target quantity A z .

Table 2 Simulation results on estimation of P(X>Y) by TSP and AUCTSP

Next, we compared the capability of TSP and AUCTSP to identify the single informative pair of genes in the midst of other non-informative genes. For this purpose, we generated random normal expression levels for N “genes” from n1 “healthy” individuals and n2 “diseased” individuals, for all combinations of N=100, 200 and n1=n2=20, 40. In all these simulations the genes numbered 1 and 2 carry the differentiating information between the healthy and diseased groups, represented by normal distributions (NH() for the “healthy” and ND() for the “diseased”) that are different from N(0,1), as shown in Table 3. All remaining genes other than 1 and 2 have expression levels obtained from the same “non-informative” distribution N(0,1). The efficacy of each classifier is measured by how many times it is able to identify the pair of genes (1,2) as the top scoring pair. The results (as averages over 1000 simulations) are shown in Table 3. The rows correspond to cases exploring the effect of increasing variance and increasing differences in the means of the expression level distributions. As can be observed, the AUCTSP consistently outperforms the TSP, in some cases dramatically, even for small sample sizes.

Table 3 Simulation results for the ability of AUCTSP and TSP to identify the most informative gene pair

Case studies

We evaluated the performance of the AUCTSP classifier over the TSP classifier in 8 publicly available datasets:

  1. (i)

    Ovarian Cancer (Pepe et al., 2003 [17]) dataset which consists of 1536 genes with expression levels from 23 healthy and 30 diseased subjects;

  2. (ii)

    Acute Leukemia (Golub et al., 1999 [25]) dataset which consists of 3571 human genes with expression levels from 25 cases of acute myeloid (aka myelogenous) leukemia (AML) and 47 cases from acute lymphoblastic (aka lymphocytic) leukemia;

  3. (iii)

    Breast Cancer - Estrogen Receptor (ER) status (West et al., 2001 [26]) dataset which consists of the expression levels of 7129 genes in 49 tissues separated into two groups of 25 positive and 24 negative tissues based on the estrogen receptor (ER) status;

  4. (iv)

    Breast Cancer - Lymph Node (LN) status (West et al., 2001 [26]) dataset which consists of the expression levels of 7129 genes in 49 tissues separated into two groups of 24 positive and 25 negative tissues based on the lymph node (LN) status;

  5. (v)

    Diffuse Large B-Cell Lymphoma (DLBCL) to predict patient outcome (Alizadeh et al., 2000 [27]) dataset which consists of the expression levels of 7129 genes in 32 cured samples and 26 fatal or refractory disease samples.

  6. (vi)

    DLBCL versus Follicular Lymphoma (FL) (Alizadeh et al., 2000 [27]) dataset which consists of the expression levels of 7129 genes in 58 DLBCL samples and 19 FL samples;

  7. (vii)

    Colon Cancer (Alon et al., 1999 [28]) dataset which consists of the expression levels of 2000 genes from 40 subjects diagnosed with colon cancer and 22 healthy subjects;

  8. (viii)

    Prostate cancer (Singh et al., 2002 [29]) dataset which consists of the expression levels of 12533 genes from 52 subjects diagnosed with prostate cancer and 50 healthy subjects.

Top scoring pairs selected by TSP and AUCTSP

For each of these datasets, we applied AUCTSP and TSP and identified the top-scoring pairs obtained by AUCTSP and TSP. The selected pairs of genes are shown in Table 4 and the gene legend is shown in Table 5.

Table 4 Top scoring pairs of genes under TSP and AUCTSP
Table 5 Gene legend

Table 4 reports also (for informational purposes) the score that the selected pair by TSP and AUCTSP receives under the opposite classifier (AUCTSP and TSP, respectively). For example, the pair selected by TSP for the ovarian cancer dataset has a TSP score of 0.9 but it receives a score of 0.675 under AUCTSP, whereas the AUCTSP score of the pair selected by AUCTSP is 0.826, while the score given to it by TSP is 0.833. This shows that pairs selected by TSP may have significantly lower scores under AUCTSP.

The biological relevance of the selected genes was found by consulting the GENECARDS database [30] and the VarElect NGS Phenotyper [31]. All of the genes identified by AUCTSP have been reported in the existing literature to be indeed related to the corresponding disease, whereas some of the genes identified by TSP such as DARS for colon cancer have not been reported to be related. A full description of the biological findings on the genes selected by AUCTSP and TSP is given in the Additional file 1. The histograms of the selected genes are also given in the Additional file 2.

We also note that for the datasets examined, AUCTSP resulted in no ties, whereas TSP frequently selected multiple pairs of genes having the same highest TSP score (3 such pairs in the Leukemia dataset and 5 pairs in the Breast-ER dataset). We have identified the gene pair ultimately chosen by the TSP after applying the tie-breaking rule proposed by Geman et al. [6] with an asterisk (“*”) in Table 4. (For the case of the Breast-LN dataset, both the AUCTSP and TSP resulted in selecting a genetically modified gene (“Bacteriophage P1 Cre recombinase”) [32, 33] as member the top-scoring pair. The pair of genes selected by the AUCTSP and TSP after eliminating this gene from the dataset are marked with (“**”) in Table 4).

Furthermore, in order to check how far the selected genes (by either method) are from being non-informative “pivot” genes, we computed for each gene g the probability P g =P(gC1>gC2) that the expression levels of g in class C1 are greater than the expression levels of g in class C2, where C1,C2 are the two classes in the corresponding dataset. A value of P g close to 0.5 means that the gene is strongly non-informative. A value of P g close to 1 or close to 0 means that the gene is strongly informative. For the case where the value of P g is close to 0, we can simply inverse the ROC curve to compute the probability P(gC1<gC2), so that all informative genes are indicated by values of P g close to 1. The computation of P g was done by computing the AUC of the ROC curve corresponding to the expression values of gene g in classes C1 and C2. The results are shown in Table 6. The P g values for each member of a selected pair are shown in column 4, whereas column 5 shows the corresponding values \(\hat {P}_{g}\) if the ROC curve has to be inverted so that values closer to 1 indicate more informative genes. As can be seen, the genes selected by AUCTSP have better deviation from the 0.5 value of a non-informative gene in almost every case.

Table 6 Deviation of the genes selected by TSP and AUCTSP from the non-informative “pivot” gene

Classifier performance of AUCTSP vs. TSP

We also compared the performance of the proposed AUCTSP classifier vs. the TSP classifier in terms of accuracy for predicting the correct status of subjects in a “testing” set after the classification rule (i.e., the top-scoring pair and its associated probabilities under AUCTSP and TSP, respectively) is obtained from a “training” set.

For each of the eight datasets in our case study, we generated several training sets and testing sets, by randomly picking a percentage p of subjects to form the training set and using the remaining q=1−p percentage of subjects as the testing set, for different values of q=1%,5%,10%,15%,20%,30%. The actual size of the testing set was set to ⌈N·q⌉, where N is the size of the dataset, and the set of the training set was set to N−⌈N·q⌉. Our intention was to see how AUCTSP and TSP behave as the training set decreases, i.e., how well AUCTSP and TSP can “generalize” their classification rule. Each test was repeated for 1000 trials and the average of the classifier accuracy (i.e., the ratio of the sum of the true positive and true negative test cases identified by the classification rule obtained from the training set over the total number of test cases) was calculated over these trials for each training set.

The results for increasing sizes of test sets (equivalently, decreasing sizes of training sets) as percentages of subjects left out from the original dataset are shown in Table 7. The plot representations of the results listed in Table 7 are given in Figs. 1, 2, 3, 4, 5, 6, 7 and 8. These results show that the AUCTSP method performs better in terms of classification accuracy than the TSP method. The results indicate that the AUCTSP classifier is able indeed to identify useful marker genes from small training sets, in accordance with the “generalization” capability of the AUC statistic.

Fig. 1
figure 1

Comparison of TSP vs. AUCTSP classification accuracy for different sizes of training sets: OVARIAN dataset

Fig. 2
figure 2

Comparison of TSP vs. AUCTSP classification accuracy for different sizes of training sets: COLON dataset

Fig. 3
figure 3

Comparison of TSP vs. AUCTSP classification accuracy for different sizes of training sets: LEUKEMIA dataset

Fig. 4
figure 4

Comparison of TSP vs. AUCTSP classification accuracy for different sizes of training sets: BREAST-LN dataset

Fig. 5
figure 5

Comparison of TSP vs. AUCTSP classification accuracy for different sizes of training sets: BREAST-ER dataset

Fig. 6
figure 6

Comparison of TSP vs. AUCTSP classification accuracy for different sizes of training sets: DLBCL-FL dataset

Fig. 7
figure 7

Comparison of TSP vs. AUCTSP classification accuracy for different sizes of training sets: DLBCL dataset

Fig. 8
figure 8

Comparison of TSP vs. AUCTSP classification accuracy for different sizes of training sets: PROSTATE dataset

Table 7 Comparison of classifier accuracy by TSP and AUCTSP for decreasing size of training set

Discussion

AUCTSP maintains the basic advantages of TSP namely the data-driven and parameter-free machine learning features that resolve the parameter tuning issue without making any assumptions about the data used, as well as the production of easily interpretable classification rules. AUCTSP, however, improves TSP by avoiding overfitting and suffering less from small sample sizes, due to the fact that every sample is compared to all other samples in the same class rather than on only a single sample by sample comparison as in TSP. In addition, AUCTSP tends to avoid selection of non-informative pivot genes, which are a known problem of TSP. Concerning selection of genes whose over-expression or under-expression is due to reasons unrelated to the disease in question, we note that this is less likely to create a problem since pairs of genes rather than single genes have to be affected in that way.

Finally, we note that AUCTSP can be extended to select a number of k>1 pairs of genes, with the classification being made according to a majority voting rule among those k pairs of genes, as was done in [7], or to find triplets instead of pairs of genes as was done in [5]. As a non-parametric based technique, AUCTSP can also have potential benefits in areas such as RNA sequence analysis (see, e.g. [34]), but this extension is left for future work.

Conclusion

In this paper, we have proposed the AUCTSP, a simple yet reliable and robust rank-based classifier for gene expression classification. AUCTSP works according to the same principle as TSP but differs from the latter in that the probabilities that determine the top scoring pair are computed based on the relative rankings of the two marker genes across all subjects as opposed to for each individual subject. Results of calculating and comparing the AUCTSP and TSP probabilities for synthetic data as well as 8 publicly available datasets demonstrate the better performance of AUCTSP over TSP.