BioMed Research International
Volume 2019 (2019), Article ID 6425963, 14 pages
https://doi.org/10.1155/2019/6425963

Research Article

A Novel Computer-Aided Diagnosis Scheme on Small Annotated Set: G2C-CAD

Guangyuan Zheng,1,2 Guanghui Han,³ Nouman Q. Soomro,⁴ Linjuan Ma,¹ Fuquan Zhang,5,6 Yanfeng Zhao,⁷ Xinming Zhao,⁷ and Chunwu Zhou⁷

¹Beijing Key Laboratory of Intelligent Information Technology, School of Computer Science and Technology, Beijing Institute of Technology, Beijing 100081, China
²School of Information and Technology, Shangqiu Normal University, No. 55 Pingyuan Road, Shangqiu, Henan Province, China
³School of Biomedical Engineering, Sun Yat-sen University, Guangzhou 510006, China
⁴Department of Software Engineering, Mehran University of Engineering and Technology, SZAB Campus, Khairpur Mirs 76062, Pakistan
⁵Fujian Provincial Key Laboratory of Information Processing and Intelligent Control, Minjiang University, Fuzhou 350121, China
⁶Digital Performance and Simulation Technology Lab., School of Computer Science and Technology, Beijing Institute of Technology, 100081 Beijing, China
⁷Department of Imaging Diagnosis, Cancer Institute and Hospital, Chinese Academy of Medical Sciences, Beijing 100021, China

Correspondence should be addressed to Nouman Q. Soomro; noumansoomro@gmail.com

Received 29 November 2018; Accepted 5 March 2019; Published 15 April 2019

Academic Editor: Giandomenico Roviello

Copyright © 2019 Guangyuan Zheng et al. This is an open access article distributed under the Creative Commons Attribution License, which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.

Abstract.

Purpose. Computer-aided diagnosis (CAD) can aid in improving diagnostic level; however, the main problem currently faced by CAD is that it cannot obtain sufficient labeled samples. To solve this problem, in this study, we adopt a generative adversarial network (GAN) approach and design a semisupervised learning algorithm, named G2C-CAD. Methods. From the National Cancer Institute (NCI) Lung Image Database Consortium (LIDC) dataset, we extracted four types of pulmonary nodule sign images closely related to lung cancer: noncentral calcification, lobulation, spiculation, and nonsolid/ground-glass opacity (GGO) texture, obtaining a total of 3,196 samples. In addition, we randomly selected 2,000 non-lesion image blocks as negative samples. We split the data 90% for training and 10% for testing. We designed a DCGAN generative adversarial framework and trained it on the small sample set. We also trained our designed CNN-based fuzzy Co-forest on the labeled small sample set and obtained a preliminary classifier. Then, coupled with the simulated unlabeled samples generated by the trained DCGAN, we conducted iterative semisupervised learning, which continually improved the classification performance of the fuzzy Co-forest until the termination condition was reached. Finally, we tested the fuzzy Co-forest and compared its performance with that of a C4.5 random decision forest and the G2C-CAD system without the fuzzy scheme, using ROC and confusion matrix for evaluation. Results. Four different types of lung cancer-related signs were used in the classification experiment: noncentral calcification, lobulation, spiculation, and nonsolid/ground-glass opacity (GGO) texture, along with negative image samples. For these five classes, the G2C-CAD system obtained AUCs of 0.946, 0.912, 0.908, 0.887, and 0.939, respectively. The average accuracy of G2C-CAD exceeded that of the C4.5 random decision tree by 14%. G2C-CAD also obtained promising test results on the LISS signs dataset; its AUCs for GGO, lobulation, spiculation, pleural indentation, and negative image samples were 0.972, 0.964, 0.941, 0.967, and 0.953, respectively. Conclusion. The experimental results show that G2C-CAD is an appropriate method for addressing the problem of insufficient labeled samples in the medical image analysis field. Moreover, our system can be used to establish a training sample library for CAD classification diagnosis, which is important for future medical image analysis.

1. Introduction

Pulmonary carcinomas are the most lethal disease in the world. Approximately 1.5 million people die due to pulmonary cancer every year—far higher than the mortality rate of other diseases [1]. In the United States alone, the lung cancer deaths in 2018 will reach 154,050 [2]. However, most pulmonary tumors cause no symptoms. Therefore, the disease is usually diagnosed at advanced stages, resulting in a low overall 5-year survival rate of approximately 14% [3]. In contrast, the 5-year survival rate of patients with stage IA non-small cell lung cancer that has been pathologically confirmed and resected or precision treated [4–6] can reach 83% [7–9]. Thus, early lung cancer detection can sharply decrease the lung cancer mortality rate [10, 11]. Pulmonary primary cancers manifest as nodules in the early stage. Compared to chest X rays, computed tomography (CT) has shown higher sensitivity in detecting small lung nodules [3, 12]. Currently, CT screening is the most recommended method for finding nodules [13, 14]. The associated increase in spiral CT screening has led to a growing burden on radiologists [15]. Despite the higher resolution available today, it is still difficult for radiologists to distinguish malignant nodules from benign ones in low-dose CT (LDCT) images. The rates of resected benign pulmonary nodules can reach 50% during surgery [16–19]. These unnecessary surgeries cause physical and mental pain and impose additional financial burdens on patients.

A “sign” in a CT lung scans refers to a radiologic finding that suggests a specific disease process. Understanding the meaning of a sign implies an understanding of the findings on the CT scan [20]. Signs are also called “CT features,” “CT manifestation,” “CT patterns,” or sometimes “CT findings” [21]. Lobulation signs [22, 23], spiculation signs [15, 24–28], and some texture signs [29, 30] play crucial roles in radiologists’ ability to differentiate benign from malignant nodules [31, 32]. Noncentral calcification, such as punctate sign or eccentric sign calcification, usually indicates that a nodule is malignant [33–35]. Therefore, it is highly important to study methods for identifying the signs of pulmonary nodules automatically to assist radiologists in diagnosing pulmonary malignant nodules.

One of the main methods is using Computer-Aided Detection/Diagnosis (CAD). The earliest conception of CAD appeared in the 1960s [36, 37]. Its early idea was attempting to “fully automate the chest exam.” Over the decades, this expectation has subsided (which seems to have happened to the early enthusiasm regarding the capabilities of artificial intelligence systems in general). Currently, the general agreement is that the focus should be on making useful computer-generated information available to physicians for decision support rather than trying to make a computer act like a diagnostician [38]. Various works on the CAD based on CT signs have been published. Han G et al. [22] designed a sliding-window-based framework to detect lobulation sign. Suzuki K et al. [39] developed a computer-aided diagnostic (CAD) scheme to distinguish benign from malignant nodules in LDCT scans using a massive training artificial neural network (MTANN) and concluded that spiculation sign is a highly differentiated feature for distinguishing benign from malignant nodules. CAD systems based on texture sign features have been investigated in several studies [40–42]. Since AlexNet won the ImageNet challenge by a large margin in 2012, deep learning techniques have flourished rapidly in the image detection field. Compared to traditional classification algorithms, deep learning can extract most distinctive features automatically and can implement end-to-end operations. However, the size of the dataset required to train a high capacity deep learning framework is quite large, while generating labeled training data in the medical image analysis field is very expensive [43].

To address the dilemma of having only a small annotation set available for lung nodule sign recognition, in this paper, we propose a semisupervised generative adversarial network (GAN) and a convolutional neural network (CNN)-based Co-forest CAD scheme. We apply the designed scheme to classify four types of nodule signs that are highly related to lung cancer. We call this proposed scheme G2C-CAD in abbreviation.

Overall workflow of G2C-CAD is illustrated in Figure 1. In the stage A, we train a GAN and a Co-forest on the available small sample set. Then, in stage B, we use the trained GAN to generate a synthetic nodule patch and transfer it to the trained CNN discriminator. From the discriminator, we gain the CNN-extracted features of the synthesized nodule patch. In stage C, the CNN features are provided to the fuzzy Co-forest pretrained on the original sample set to conduct semisupervised learning for the five types of ROI patches. Finally, the process iterates between stages B and C until a termination condition is met.

Figure 1: GAN- and CNN-based Co-forest computer aided diagnosis (G2C-CAD) system workflow.

The rest of this paper is organized as follows. In Section 2, we review the existing lung nodule classification algorithms. Section 3 presents our G2C-CAD algorithm. We introduce the experimental method in Section 4 and present an analysis in Section 5. Section 6 concludes this paper.

2. Related Work

Discriminating between benign or malignant nodules has attracted the interest of a large number of researchers. The early discrimination methods were based primarily on traditional machine learning algorithms such as k-nearest neighbors (k-NN), linear discriminant analysis (LDA), Bayes [44], rule based schemes, decision trees (DT), and the support vector machine (SVM). Krewer et al. [45] tested several classifiers, including DT, k-NN, and SVM, on extracted texture and shape features to discriminate malignant from benign nodules. By analyzing the experimental results, they found that partly solid and nonsolid nodules have a higher malignancy rate than do solid nodules. Colin Jacobs et al. [46] developed and evaluated a computer aided diagnostic system for classifying lung nodules into solid, partly solid, and nonsolid nodules. This CADx system performs statistical classification on the nodules’ intensity-, texture-, and segmentation-based features using the k-NN algorithm. Xiabi Liu et al. [47] proposed a feature selection method based on the FIsher criterion and genetic optimization (FIG) to address Common CT Imaging Signs of Lung (CISL) disease recognition problems. They applied the FIG feature selection algorithm to bag-of-visual-words features, wavelet transform-based features, the Local Binary Pattern, CT Value Histogram features, and others. Their results showed that FIG achieved high computational efficiency and was highly effective. Tao Sun et al. [48] investigated an SVM-based CADx system for lung cancer classification using a total of 488 input features that included textural features, patient characteristics, and morphological features to train the classifier. Hidetaka Arimura et al. [49] developed a computerized scheme to automatically detect lung nodules in LDCT images for lung cancer screening. They extracted possible nodule images using a ring average filter, identified a set of nodule candidates by applying a multiple-gray-level thresholding technique, and removed false positives by using two rule-based schemes on the localized image features related to morphology and gray levels. Tao Sun et al. [50] proposed a CADx system to predict the characteristics of solitary pulmonary nodules in lung CT to diagnose early stage lung cancer. In their CADx system an SVM model was constructed that exploited curvelet transform texture features, 3 patient demographic features, and 9 morphological features. Fangfang Han et al. [51] constructed a CADx system based on 50 categories of 3D textural features extracted from gray levels, a curvature cooccurrence matrix, and gradients as well as other nodule volume data derivatives.

The above CAD systems mainly utilized features obtained by traditional feature extraction algorithms. Traditional feature acquisition is based on manual design and selection, which requires experts with specialized heuristic knowledge. The features obtained in this way are low-level features near the pixel level, and the work scope is relatively small. In the image analysis workflow, the final performance of the system also depends on the quality of prior preprocessing or segmentation stages. Therefore, in the traditional CAD solutions, tuning the classification performance is both complicated and arduous [52].

The emergence of the convolutional neural network (CNN) [53] solved this dilemma. In 2012, the emergence of AlexNet sparked a revival in the image detection field through deep learning techniques based on CNN features, and CNNs have subsequently been used extensively in pulmonary imaging analysis. Compared to traditional algorithms, CNNs can extract more distinctive features automatically. Two studies [52, 54] demonstrated that CNNs are a promising technique in lung nodule identification. Deep learning techniques have the inherent superiority of being able to automatically extract features and adjust the performance seamlessly.

In traditional CAD algorithms, training only needs to seek an optimal discriminant surface from the manually designed feature space. In contrast, deep learning networks simultaneously attempt to find both the most significant discriminate features among large numbers of high-level features and an optimal classification surface. As a result, training a deep learning network requires a massive amount of labeled samples [55]—a requirement that cannot be met in the medical image analysis field because professional annotation is too expensive [56]. Data augmentation techniques produce only limited effects. Although transfer learning can alleviate the problem of the lack of training examples to a certain extent, the significant feature sequences vary between different classification tasks. Thus, transfer learning is not the most suitable method to cope with medical image analysis tasks.

Generative adversarial networks (GANs) [57] have demonstrated the promising ability to generate visually realistic images. A GAN trained with limited annotated samples can generate large numbers of realistic images. Chuquicusma MJ et al. [58] conducted visual Turing tests to evaluate the degree of realism in nodule images generated by a DCGAN and showed that the generated samples can be used to boost the diagnostic power by mining high-level discriminative image features and that the resulting features can be used to train both radiologists and deep networks.

Semisupervised learning is a type of machine learning method that combines supervised and unsupervised learning. It can be applied when only a small number of labeled data exist, but a large number of unlabeled data are available. Semisupervised learning is important for reducing the cost of acquiring labeled data and improving classifier performance. Commonly used methods include EM with generative mixture models, self-training, cotraining, transductive support vector machines, and graph-based methods [59]. Co-forest is a cotraining based semisupervised learning algorithm that first learns an initial classifier from a small amount of labeled data and then refines the classifier by further exploiting a larger number of unlabeled data to boost the classifier’s performance. When applying micro calcification detection for breast cancer diagnosis, Ming Li et al. [60] showed that Co-forest can successfully enhance the performance of a model trained on only a small amount of diagnosed samples by utilizing the available undiagnosed samples.

The above works inspired us to exploit a GAN to enlarge and enrich a training set of pulmonary nodules. It also motivated us to implement the designed G2C-CAD.

3. Materials and Methods

3.1. Experimentation Materials

Insufficient labeled samples represents a barrier to CAD progress. The emergence of GAN [57] begun to change this situation. A GAN consists of two main parts: a generator G and a discriminator D. The G is used to learn the distribution of real images. Then it generates realistic images to attempt to fool the D. The D attempts to perform true and false discriminations concerning received images. Throughout the process, the G strives to make the generated image more realistic, while the D tries to identify the true and false images. This process is equivalent to a game with two opponents. Over time, the G and D eventually achieve a dynamic equilibrium in which images generated by the generator are highly similar to the real image distribution, and the discriminator cannot determine whether a sample is drawn from the true data or generated by the generator. DCGAN [61] is a GAN extension in which a CNN is introduced to conduct unsupervised training. The ability of the CNN to extract features is used to enhance the training of the generation network. Building on this idea, we constructed a 32 × 32 input scale DCGAN. The discriminator architecture in DCGAN has 4 layers, as shown in Figure 2.

Figure 2: The architecture of the generator and the discriminator in the DCGAN model.

We tested a GAN trained from 9 samples to generate unlabeled-nodule sign patches, and the result is gratifying, as shown in Figure 3.

(a) Original sign patches

(b) Patches generated by DCGAN

Figure 3: Samples of DCGAN-generated sign patches and real samples. The image on the left shows examples of the seed samples used for training. Most of the generated samples in (b) are realistic; although some of the generated patches can be recognized as being fake (the 5th one in row 3), they do help to boost the CAD system’s performance.

3.2. CNN Feature-Based Fuzzy Co-Forest Method

As discussed in Section 3.1, when we gain a trained GAN, we also get a trained CNN discriminator simultaneously. For each image patch passed though the discriminator, we obtain a 128-dimensional CNN feature vector from the last convolutional layer. Features of 4 × 4 is hard for eyes to discern; in Figure 4 we show an example of 32-dimensional 16 × 16 CNN features of a sign patch extracted from layer 1 of D.

(a) Input patch

(b) Corresponding features

Figure 4: (a) An original input sign patch; (b) the corresponding 32 CNN features abstracted from the layer 1 of the discriminator D.

Cotraining random forest (Co-forest) is an upgraded algorithm for the cotraining paradigm. The standard cotraining algorithm has two strong assumptions: the samples distribution is consistent with that of the target functions and the different features extracted from the same data should be conditionally independent. In most cases, however, these two strong assumptions are difficult to satisfy. Co-forest uses an ensemble consisting of multiple classifiers to avoid the constraints of standard cotraining. The specific structure of random forests enables the Co-forest to take advantage of semisupervised learning and ensemble learning to better learn the distribution of the training data. Existing Co-forest works are based on traditional manually designed features [60, 62]. Here, we try to extend the Co-forest approach to deep neural networks by utilizing the CNN features obtained from the GAN’s discriminator.

For each generated realistic sign from DCGAN, we can extract a 128-element 4 × 4 CNN feature vector. Each 4 × 4 CNN feature can be transformed to a 1 × 16 vector. If we assume that DCGAN runs N times, then we will obtain N CNN feature vectors from the discriminator. These CNN feature vectors for the N image patches can be expressed by a matrix:

where in (1) f is a 16-element vector transformed from a 4 × 4 feature patch. Every row in Original_F represents the 128-dimensional CNN features from one input image patch. The features in each column are produced from the same filter. The n in represents the th sign in N, and m represents the mth feature of a sign. We build a complete reference vector , . The cosine similarity of any feature f to r can be calculated as a:

where is an element of f. We use a distance matrix A of the relative cosine distances to r to replace the feature matrix Original_F:

From matrices (1) and (4), we can see that for any two elements and , a smaller difference between the values of and indicates that their corresponding features and are more similar. The elements in each column of A are a series of continuous-valued data. To build a Co-forest, the first step is to construct decision trees by utilizing the labeled samples [63]. Assume that the total number of samples is and that the classes are c₁, c₂, …, , k. To build a decision tree, we randomly select S samples. These S samples are sorted on by the values of , where x represents the xth column of A. We calculate the middle value between and as a split point . The class information entropy of S is

where represents the proportion of the category j samples relative to all samples, and is a subset of S constructed by all the elements of S belonging to class , .

When selecting feature as the splitting node of the decision tree, the information entropy of S is

Then, we compute the information gain:

The split information entropy is

and the information gain ratio is

In a column of attributes A[x], a suitable threshold to divide A[x] into two intervals is , . This split produces the maximal information gain ratio. This is then selected as the parent node to generate two children. This process is conducted recursively to build a decision tree until a termination criterion is matched.

In the reasoning process, if a sample’s attribute value falls into a small region around , after it is disturbed by noise, it can easily be misclassified. As shown in Figure 5, suppose we have a sample whose real attribute values are a1=0.977 and a2=0.827; however, due to noise during collection, a1 is changed to 0.973. In a traditional decision tree, this sample would be misclassified as .

(a)

(b)

Figure 5: Traditional classification process.

To avoid this fragility, when constructing a decision tree, we utilize a fuzzy scheme [63]. In a traditional decision tree, when a sample’s attribute a is greater than , the sample belongs to either or . We modify this crisp classifying method by selecting a neighborhood threshold ε around the split point , as shown in Figures 6(a) and 6(b), so that the classification function becomes

where C(s) is a classification function that maps sample s to one of two weighted classes.

(a)

(b)

(c)

Figure 6: Here, (a) and (b) are fuzzy classification processes, and (c) is the decision tree.

As shown in Figure 6(c), along one branch, the final classification is calculated as follows:

where is one class from .

When , under the fuzzy classifying scheme, the classification result of disturbed example s is . It is obvious that the maximal probability of the final decision result is . Based on this classification example, we can see that the fuzzy decision scheme is more robust to noise.

The classification result of the fuzzy Co-forest is a multiple-label probability distribution of the union of the decision trees’ output; we consider the class with the highest probability as the final output.

Let LB denote the labeled set. G(z) denotes the process of generating a new fake sign-image patch utilizing the trained DCGAN. There are N classifiers in the Co-forest ensemble . . We denote one of the classifiers in ensemble, , as the concomitant ensemble of and create a subensemble that includes all the classifiers except . For a newly generated sample from G, if the max fuzzy vote sum of the classifiers in concomitant ensemble exceeds a preset threshold , the sample will be copied to a newly labeled set with the new assigned label. Based on [64], the process iterates until is larger than Then, the set is used to refine . denotes the classification error rate, and , where is the predicted confidence of on , and is the size of set .

Based on the fuzzy decision tree scheme, we construct the fuzzy Co-forest as shown in Algorithm 1.

Input: the labeled set LB.

The threshold of the confidence, T is random trees’ number

Process:

Initialize a random forest containing T trees.

for i in

= 0.5

= 0

endfor

t = 0

Loop until all of the trees in Random Forest unchanged

t = t+1

z = a new random vector

for i in

= ClassificationErrorRate(,L)

=

if( <)

for in

if )

endfor

endfor

for i in

if()

= LearnRandomTree

endfor

endLoop

Output:

Algorithm 1: G2C-CAD.

4. Experiments

4.1. Datasets

We collected sample instances from both the LIDC-IDRI and LISS datasets. LIDC-IDRI [65] consists of pulmonary medical image files (such as CT scans and X-rays) with corresponding pathological annotations. The data were collected by the National Cancer Institute to study early cancer detection in high-risk populations. LISS [21] consists a set of CISLs collected by the Cancer Institute and Hospital at the Chinese Academy of Medical Sciences and the Beijing Institute of Technology intended for computer-aided detection and diagnosis research and medical education. LISS contains 271 CT scans and 677 abnormal regions, including nine categories of CISLs.

4.2. LIDC-IDRI Instances

In the LIDC-IDRI CT imaging slices, most of the annotated nodules have diameters less than 32 pixels, as shown in Figure 7. Therefore, we choose 32 × 32 as the input ROI size.

Figure 7: The diameter-size distribution of the nodules in the LIDC-IDRI dataset.

A higher degree of lobulation, speculation, and nonsolid texture signs indicates a greater probability of malignant nodules [66]. A calcification sign usually indicates a benign nodule [66–68] except when it has a noncentral appearance. Signs of subtlety [69], internal structure [27], sphericity, and margin [33] have not been clearly proven to have a strong relationship with malignancy. To simplify the comparisons in this experiment, we selected nodules from LIDC with calcification =4 (noncentral calcification), lobulation >=4, spiculation >=4, texture <=2, and malignancy >=3 as experimental instances based on the selection rules shown in Table 1.

Table 1: The scheme of sign selection.

Used as signs
Abandoned

Subtlety
-
-

Internal structure
-
-

Calcification
=4
<>4

Sphericity
-
-

Margin
-
-

Lobulation
>=4
<4

Spiculation
>=4
<4

Texture
<=2
>2

LIDC-IDRI contains 21,057 annotated nodules. We selected the 4 category nodules with signs such as noncentral calcification, lobulation, speculation, and nonsolid/GGO texture that have a high prevalence of malignancy as the experimental objects. For the probability of malignancy in these nodules, we adopted the average value of 4 radiologists’ scores. The noncentral calcification, lobulation, spiculation, and nonsolid/GGO texture signs are illustrated in Table 2. In addition, we randomly extracted 32 × 32-pixel image patches from slices not annotated by any radiologist as negative samples. In total, we used five types of image blocks in our experiment.

Table 2: Sign samples of calcification, lobulation, spiculation, and nonsolid/GGO texture.

Noncentral calcification
Lobulation
Spiculation
Nonsolid/GGO texture

Subtlety
4.67
5.0
5.0
4.0
4.0
5.0
3.67
3.0

Internal structure
1.0
1.0
1.0
1.0
1.0
1.0
1.0
1.0

Calcification
4.0
4.0
6.0
6.0
6.0
6.0
6.0
6.0

Sphericity
3.33
2.67
4.25
4.0
5.0
5.0
4.33
4.0

Margin
3.67
4.67
4.5
4.0
2.0
4.0
2.67
1.0

Lobulation
1.67
2.0
4.25
4.0
4.0
1.0
2.67
1.0

Spiculation
1.33
1.33
2.25
4.0
4.0
5.0
1.67
2.0

Texture
5.0
4.67
4.75
4.0
5.0
4.0
1.33
1.0

Based on the center point of the merged regions annotated by 4 experts, we extracted 32 × 32-pixel image blocks as the experimental input by following the selection criteria shown in Table 1. Among these samples, we ensured that a given category of ROI patches for a single patient does not appear in any two subsets simultaneously, which helped ensure that the specificity of the trained individual networks is as high as possible.

4.3. LISS Instances

From LISS, we selected signs of lobulation, spiculation, pleural indentation, and GGO associated with malignant lung cancer [70–74] as the experimental objects. The number of samples in each category is shown in Table 3.

Table 3: Number of selected instances of LISS.

CISL Category
# of Lesion Regions

GGO
45

Lobulation
41

Spiculation
29

Pleural Indentation
45

Negative
80

4.4. Evaluation Criteria

To evaluate the performance of the algorithm presented in this paper, we considered the following criteria.

(1)ROC: The Receiver Operator Characteristic curve (ROC) is a method that comprehensively and simultaneously reflects the sensitivity and specificity of the classification result. By comparing the classification results of different samples with the annotation labels, a series of sensitivity and specificity scores is calculated. Then, a curve is drawn using sensitivity as the ordinate and 1 − specificity as the abscissa. A larger area under the curve (AUC) indicates a higher diagnostic accuracy. On the ROC curve, the point closest to the top left of the coordinate diagram is the critical value that reflects the highest sensitivity and specificity.(2)Confusion matrix: A confusion matrix is also called an error matrix, and it is a visual representation of the classification effect. A confusion matrix can be used to describe the relationship between the real category attribute of the sample data and the recognition result. It is a method for evaluating classifier performance and is widely used in pattern recognition. A confusion matrix is also a performance evaluation method that scholars often use when solving practical application problems.

4.5. Experimentation

From the LIDC-IDRI database, we acquired 590 noncentral calcification, 565 lobulation, 576 spiculation, 545 nonsolid/GGO texture sign patches, and 2,500 negative image patches. The experiment was conducted according to the following steps:

(1)Set the initial values of T, , and to 6, 0.5, and 0.6, respectively.(2)Train the GAN until the discrepancy cost reaches a balance, as illustrated in Figure 8.(3)Train a primary fuzzy Co-forest based on the original samples, utilizing the features exported from the trained DCGAN discriminator in Step .(4)Input a random vector to DCGAN and transmit the generated features to the concomitant random fuzzy decision forest in the primary Co-forest until for all classes. In this process, if the maximal weight of the fuzzy label from exceeds the threshold , store both features with the matched label; otherwise, discard the generated image and generate a new image.(5)Retrain the corresponding tree of .(6)Test the performance of the system.

Figure 8: Training discrepancy cost of DCGAN.

As a comparison, we trained a C4.5 random forest according to the same scheme which has been used by G2C-CAD, as the baseline method.

We also conducted experiments on samples obtained from LISS.

5. Results

We divided the dataset into two parts, 90% for training and 10% for validation. In this way, the nodule distribution in the validation subsets is consistent with the nodule distribution of the original dataset according to the radiologists’ consensus of their evaluations at the nodule level. The number of trees in the Co-forest classification model was 6, primary training was conducted on the training data, and finally, the system was further validated on the remaining 10%. The sensitivity and specificity of each class instance was calculated and compared using the ROC curves shown in Figure 9.

Figure 9: ROC curves for the 5 categories of sign ROI recognition results.

In Figure 9, the AUCs of noncentral calcification, negative image samples, lobulation, spiculation and nonsolid/GGO texture are 0.946, 0.939, 0.912, 0.908, and 0.887, respectively. From the curves in Figure 9, we can see that the G2C-CAD system achieves the highest overall classification accuracy on the calcification sign. Compared to noncentral calcification and negative image samples, the AUCs of lobulation, spiculation, and nonsolid/GGO texture are relatively lower.

To show the underlying classification error distribution, a confusion matrix of the 5 classification results is presented in Figure 10.

Figure 10: Confusion matrix of the predicted results.

The diagonal numbers in the confusion matrix represent the recognition accuracy rates of the corresponding category, and the nondiagonal elements are misclassification rates, i.e., the ratio of other category test samples that were misclassified to this class. From Figure 10 we can see that the noncentral calcification sign has the highest classification accuracy, while the nonsolid/GGO texture sign has the highest misclassification rate. Misclassified spiculation signs are mostly recognized as lobulation. By comparing the test samples, we find that many of these two types of samples have very similar textures. From the confusion matrix, we can also see that most misclassifications occur between lobulation, spiculation, and nonsolid/GGO texture signs.

6. Discussion

We performed a performance comparison on each category between G2C-CAD and the C4.5 random forest model using the confusion matrix. First, we constructed confusion matrices reflecting the accuracies of G2C-CAD and C4.5. Then, we calculated the difference matrix for those two confusion matrices as shown in Figure 11. The float numbers on the diagonal represent the differences between G2C-CAD and C4.5 regarding their classification accuracy for the corresponding category, while the float numbers on the nondiagonal represent their differences in the classification error rate, showing the numbers for each class that were misclassified into the corresponding category.

Figure 11: Performance difference matrix of G2C-CAD and C4.5.

Consequently, positive numbers on the diagonal and larger values indicate that G2C-CAD achieved a better performance than that of C4.5. For the elements that are not on the diagonal, the situation is the opposite. As shown in Figure 11, no positive values occur in the nondiagonal elements, which means that G2C-CAD possesses greater discrimination ability between each category. The element values on the diagonal are all positive, and the average value of the diagonal numbers in the difference confusion matrix is 0.144, demonstrating that our method has a better overall performance.

To verify the effectiveness of the fuzzy algorithm, we also conducted multiclassification experiments using G2C-CAD without the fuzzy algorithm and compared the performances of the two algorithms using a difference confusion matrix as shown in Figure 12.

Figure 12: Performance difference matrix of fuzzy G2C-CAD and nonfuzzy G2C-CAD.

From Figure 12, we can see that the discrimination performance of the CAD system employing the fuzzy algorithm is obviously better than that of the nonfuzzy one. Furthermore, as shown in Figure 12, the CAD performance of the fuzzy algorithm better distinguishes the difficult lobulation sign from the spiculation sign.

The experimental result on LISS shows higher performance; the areas under the ROC curve of GGO, lobulation, spiculation, pleural indentation, and negative image samples are 0.972, 0.964, 0.941, 0.967, and 0.953, respectively. By comparing the training dataset and test samples, we found that the main reason may be that the samples in different categories are more separable visually.

7. Conclusion

In this paper, by coupling a GAN with a semisupervised learning approach, we proposed a G2C-CAD method to detect signs that are highly correlated with malignant pulmonary nodules. We first trained a DCGAN on a small sample set. Then, we extracted the features from the CNN discriminator trained with the training samples and used them to train a primary fuzzy Co-forest classifier. Then, we use the trained DCGAN to generate large amounts of realistic fake samples. Based on these fake samples, we conducted semisupervised learning with the fuzzy Co-forest and finally obtained a classifier with excellent performance. By validating on the LIDC dataset, the area under the ROC curve for five sign types, noncentral calcification, negative image samples, lobulation, spiculation, and nonsolid/GGO texture, reached 0.946, 0.939, 0.912, 0.908, and 0.887, respectively. On the LISS dataset, the proposed system showed comprehensively higher classification performances than those of a trained C4.5 classifier. The experimental results show that the proposed G2C-CAD is an appropriate method for solving the problem of insufficient samples in the medical image analysis field. Moreover, our system can also be used to establish a training sample library for CAD classification diagnosis, which holds great significance for future medical image analysis.

In future work, we plan to combine this method with multiple-instance learning to perform weak supervised learning directly on CT slice images or to extend the algorithm, making it suitable for use in the 3D medical image classification field.

Data Availability

The data used to support the findings of this study are available from the corresponding author upon request.

Disclosure

The publicly accessible medical images database of LIDC/IDRI is used.

Conflicts of Interest

The authors declare that there are no conflicts of interest regarding the publication of this paper.

Acknowledgments

This work was supported in part by National Natural Science Foundation of China (Grant nos. 60973059 and 81171407) and Program for New Century Excellent Talents in University of China (Grant no. NCET-10-0044).

References

“Cancers,” http://www.who.int/en/news-room/fact-sheets/detail/cancer.

“Cancer stat facts: common cancer sites,” https://seer.cancer.gov/statfacts/html/common.html.

S. Diederich, D. Wormanns, M. Semik et al., “Screening for early lung cancer with low-dose spiral CT: prevalence in 817 asymptomatic smokers,” Radiology, vol. 222, no. 3, pp. 773–781, 2002.

J. F. Williamson, “Brachytherapy technology and physics practice since 1950: A half-century of progress,” Physics in Medicine and Biology, vol. 51, no. 13, article no. R18, pp. R303–R325, 2006.

J. F. Williamson, S. K. Das, M. S. Goodsitt, and J. O. Deasy, “Introducing the medical physics dataset article,” Medical Physics, vol. 44, no. 2, pp. 349-350, 2017.

S. Balik, E. Weiss, N. Jan et al., “Evaluation of 4-dimensional computed tomography to 4-dimensional cone-beam computed tomography deformable image registration for lung cancer adaptive radiation therapy,” International Journal of Radiation Oncology • Biology • Physics, vol. 86, no. 2, pp. 372–379, 2013.

C. F. Mountain, “Revisions in the international system for staging lung cancer,” Chest, vol. 111, no. 6, pp. 1710–1717, 1997.

M. T. van Rens, J. M. van den Bosch, A. Brutel de la Rivière, and H. R. Elbers, “Prognostic assessment of 2,361 patients who underwent pulmonary resection for non-small cell lung cancer, stage I, II, and IIIA,” CHEST, vol. 117, no. 2, pp. 374–379, 2000.

E. F. Patz, S. Rossi, D. H. Harpole, J. E. Herndon, and P. C. Goodman, “Correlation of tumor size and survival in patients with stage IA non-small cell lung cancer,” Chest, vol. 117, no. 6, pp. 1568–1571, 2000.

C. I. Henschke, P. Boffetta, O. Gorlova, R. Yip, J. O. DeLancey, and M. Foy, “Assessment of lung-cancer mortality reduction from CT screening,” Lung Cancer, vol. 71, no. 3, pp. 328–332, 2011.

D. Kumar, A. Wong, and D. A. Clausi, “Lung nodule classification using deep features in CT images,” in Proceedings of the 12th Conference on Computer and Robot Vision, CRV 2015, pp. 133–138, Canada, June 2015.

M. Kaneko, K. Eguchi, and H. Ohmatsu, “Peripheral lung cancer: screening and detection with low-dose spiral CT versus radiography,” Radiology, vol. 201, no. 3, pp. 798–802, 1996.

S. Swensen, J. Jett, J. Sloan et al., “Screening for lung cancer with low-dose spiral computed tomography,” American Journal of Respiratory and Critical Care Medicine, vol. 165, no. 4, pp. 508–513, 2002.

E. Paci, D. Puliti, A. Lopes Pegna et al., “Mortality, survival and incidence rates in the ITALUNG randomised lung cancer screening trial,” Thorax, vol. 72, no. 9, pp. 825–831, 2017.

S. Iwano, T. Nakamura, Y. Kamioka, and T. Ishigaki, “Computer-aided diagnosis: A shape classification of pulmonary nodules imaged by high-resolution CT,” Computerized Medical Imaging and Graphics, vol. 29, no. 7, pp. 565–570, 2005.

A. Bernard and H. Shennib, “Resection of pulmonary nodules using video-assisted thoracic surgery,” The Annals of Thoracic Surgery, vol. 61, no. 1, pp. 202–205, 1996.

M. J. Mack, S. R. Hazelrigg, R. J. Landreneau, and T. E. Acuff, “Thoracoscopy for the diagnosis of the indeterminate solitary pulmonary nodule,” The Annals of Thoracic Surgery, vol. 56, no. 4, pp. 825–832, 1993.

S. Siegelman, E. Zerhouni, F. Leo, N. Khouri, and F. Stitik, “CT of the solitary pulmonary nodule,” American Journal of Roentgenology, vol. 135, no. 1, pp. 1–13, 1980.

B. A. Keagy, P. J. K. Starek, G. F. Murray, J. W. Battaglini, M. E. Lores, and B. R. Wilcox, “Major pulmonary resection for suspected but unconfirmed malignancy,” The Annals of Thoracic Surgery, vol. 38, no. 4, pp. 314–316, 1984.

J. Collins, “CT signs and patterns of lung disease,” Radiologic Clinics of North America, vol. 39, no. 6, pp. 1115–1135, 2001.

G. Han, X. Liu, F. Han et al., “The LISS—a public database of common imaging signs of lung diseases for computer-aided detection and diagnosis research and medical education,” IEEE Transactions on Biomedical Engineering, vol. 62, no. 2, pp. 648–656, 2015.

G. Han, X. Liu, N. Q. Soomro et al., “Empirical driven automatic detection of lobulation imaging signs in lung CT,” BioMed Research International, vol. 2017, 2017.

C.-Z. Shi, Q. Zhao, L.-P. Luo, and J.-X. He, “Size of solitary pulmonary nodule was the risk factor of malignancy,” Journal of Thoracic Disease, vol. 6, no. 6, pp. 668–676, 2014.

J. J. Erasmus, J. E. Connolly, H. P. McAdams, and V. L. Roggli, “Solitary pulmonary nodules: part I. Morphologic evaluation for differentiation of benign and malignant lesions,” RadioGraphics, vol. 20, no. 1, pp. 43–58, 2000.

K. Furuya, S. Murayama, H. Soeda et al., “New classification of small pulmonary nodules by margin characteristics on high-resolution CT,” Acta Radiologica, vol. 40, no. 5, pp. 496–504, 2016.

Z. Yang, S. Sone, S. Takashima et al., “High-resolution CT analysis of small peripheral lung adenocarcinomas revealed on screening helical CT,” American Journal of Roentgenology, vol. 176, no. 6, pp. 1399–1407, 2001.

M. F. McNitt-Gray, E. M. Hart, N. Wyckoff, J. W. Sayre, J. G. Goldin, and D. R. Aberle, “A pattern classification approach to characterizing solitary pulmonary nodules imaged on high resolution CT: Preliminary results,” Medical Physics, vol. 26, no. 6, pp. 880–888, 1999.

V. Ambrosini, S. Nicolini, P. Caroli et al., “PET/CT imaging in different types of lung cancer: an overview,” European Journal of Radiology, vol. 81, no. 5, pp. 988–1001, 2012.

F. Li, S. Sone, H. Abe, H. MacMahon, and K. Doi, “Malignant versus benign nodules at CT screening for lung cancer: comparison of thin-section CT findings,” Radiology, vol. 233, no. 3, pp. 793–798, 2004.

B. Ganeshan, E. Panayiotou, K. Burnand, S. Dizdarevic, and K. Miles, “Tumour heterogeneity in non-small cell lung carcinoma assessed by CT texture analysis: a potential marker of survival,” European Radiology, vol. 22, no. 4, pp. 796–802, 2012.

F. Li, S. Sone, H. Abe, H. MacMahon, S. G. Armato, and K. Doi, “Lung cancers missed at low-dose helical CT screening in a general population: comparison of clinical, histopathologic, and imaging findings,” Radiology, vol. 225, no. 3, pp. 673–683, 2002.

H. Chen, Y. Xu, Y. Ma, and B. Ma, “Neural network ensemble-based computer-aided diagnosis for differentiation of lung nodules on CT images clinical evaluation,” Academic Radiology, vol. 17, no. 5, pp. 595–602, 2010.

C. V. Zwirewich, S. Vedal, R. R. Miller, and N. L. Müller, “Solitary pulmonary nodule: high-resolution CT and radiologic-pathologic correlation,” Radiology, vol. 179, no. 2, pp. 469–476, 1991.

Y. J. Jeong, C. A. Yi, and K. S. Lee, “Solitary pulmonary nodules: detection, characterization, and guidance for further diagnostic workup and treatment,” American Journal of Roentgenology, vol. 188, no. 1, pp. 57–68, 2007.

A. N. Khan, H. H. Al-Jahdali, C. M. Allen, K. L. Irion, S. Al Ghanem, and S. S. Koteyar, “The calcified lung nodule: what does it mean?” Annals of Thoracic Medicine, vol. 5, no. 2, pp. 67–79, 2010.

H. C. Becker, W. J. Nettleton, P. H. Meyers, J. W. Sweeney, and C. M. Nice, “Digital computer determination of a medical diagnostic index directly from chest X-ray images,” IEEE Transactions on Biomedical Engineering, vol. BME-11, no. 3, pp. 67–72, 1964.

P. H. Meyers, C. M. Nice, H. C. Becker, W. J. Nettleton, J. W. Sweeney, and G. R. Meckstroth, “Automated computer analysis of radiographic images,” Radiology, vol. 83, no. 6, pp. 1029–1034, 1964.

B. Van Ginneken, B. Ter Haar Romeny, and M. Viergever, “Computer-aided diagnosis in chest radiography: a survey,” IEEE Transactions on Medical Imaging, vol. 20, no. 12, pp. 1228–1241, 2001.

K. Suzuki, F. Li, S. Sone, and K. Doi, “Computer-aided diagnostic scheme for distinction between benign and malignant nodules in thoracic low-dose CT by use of massive training artificial neural network,” IEEE Transactions on Medical Imaging, vol. 24, no. 9, pp. 1138–1150, 2005.

K. Kanazawa, Y. Kawata, N. Niki et al., “Computer-aided diagnosis for pulmonary nodules based on helical CT images,” Computerized Medical Imaging and Graphics, vol. 22, no. 2, pp. 157–167, 1998.

F. Han, H. Wang, G. Zhang et al., “Texture feature analysis for computer-aided diagnosis on pulmonary nodules,” Journal of Digital Imaging, vol. 28, no. 1, pp. 99–115, 2014.

H. Hu, Q. Wang, H. Tang, L. Xiong, and Q. Lin, “Multi-slice computed tomography characteristics of solitary pulmonary ground-glass nodules: Differences between malignant and benign,” Thoracic Cancer, vol. 7, no. 1, pp. 80–87, 2016.

G.-Y. Zheng, X.-B. Liu, and G.-H. Han, “Survey on medical image computer aided detection and diagnosis systems,” Ruan Jian Xue Bao/Journal of Software, vol. 29, no. 5, pp. 1471–1514, 2018.

F. H. Edwards, P. S. Schaefer, S. Callahan, G. M. Graeber, and R. A. Albus, “Bayesian statistical theory in the preoperative diagnosis of pulmonary lesions,” CHEST, vol. 92, no. 5, pp. 888–891, 1987.

H. Krewer, B. Geiger, L. O. Hall et al., “Effect of texture features in computer aided diagnosis of pulmonary nodules in low-dose computed tomography,” in Proceedings of the IEEE International Conference on Systems, Man, and Cybernetics (SMC '13), pp. 3887–3891, IEEE, Manchester, UK, October 2013.

C. Jacobs, E. M. van Rikxoort, E. T. Scholten et al., “Solid, part-solid, or non-solid?: classification of pulmonary nodules in low-dose chest computed tomography by a computer-aided diagnosis system,” Investigative Radiology, vol. 50, no. 3, pp. 168–173, 2015.

X. Liu, L. Ma, L. Song, Y. Zhao, X. Zhao, and C. Zhou, “Recognizing common CT imaging signs of lung diseases through a new feature selection method based on fisher criterion and genetic optimization,” IEEE Journal of Biomedical and Health Informatics, vol. 19, no. 2, pp. 635–647, 2015.

T. Sun, J. Wang, X. Li et al., “Comparative evaluation of support vector machines for computer aided diagnosis of lung cancer in CT based on a multi-dimensional data set,” Computer Methods and Programs in Biomedicine, vol. 111, no. 2, pp. 519–524, 2013.

H. Arimura, S. Katsuragawa, K. Suzuki et al., “Computerized scheme for automated detection of lung nodules in low-dose computed tomography images for lung cancer screening,” Academic Radiology, vol. 11, no. 6, pp. 617–629, 2004.

T. Sun, R. Zhang, J. Wang, X. Li, and X. Guo, “Computer-aided diagnosis for early-stage lung cancer based on longitudinal and balanced data,” PLoS ONE, vol. 8, no. 5, Article ID e63559, 2013.

F. Han, C. L. Novak, S. Aylward et al., “A new 3D texture feature based computer-aided diagnosis approach to differentiate pulmonary nodules,” Proceedings of the SPIE, Article ID 86702Z, 2013.

K.-L. Hua, C.-H. Hsu, S. C. Hidayati, W.-H. Cheng, and Y.-J. Chen, “Computer-aided classification of lung nodules on computed tomography images via deep learning technique,” OncoTargets and Therapy, 2015.

Y. LeCun, L. Bottou, Y. Bengio, and P. Haffner, “Gradient-based learning applied to document recognition,” Proceedings of the IEEE, vol. 86, no. 11, pp. 2278–2323, 1998.

W. Shen, M. Zhou, F. Yang, C. Yang, and J. Tian, “Multi-scale convolutional neural networks for lung nodule classification,” in Proceedings of the International Conference on Information Processing in Medical Imaging, vol. 24, pp. 588–599, Springer, 2015.

N. Tajbakhsh and K. Suzuki, “Comparing two classes of end-to-end machine-learning models in lung nodule detection and classification: MTANNs vs. CNNs,” Pattern Recognition, vol. 63, pp. 476–486, 2017.

H. Greenspan, B. Van Ginneken, and R. M. Summers, “Guest editorial deep learning in medical imaging: overview and future promise of an exciting new technique,” IEEE Transactions on Medical Imaging, vol. 35, no. 5, pp. 1153–1159, 2016.

I. J. Goodfellow, J. Pouget-Abadie, M. Mirza et al., “Generative adversarial nets,” in Proceedings of the 28th Annual Conference on Neural Information Processing Systems 2014, NIPS 2014, pp. 2672–2680, Canada, December 2014.

M. J. Chuquicusma, S. Hussein, J. Burt, and U. Bagci, “How to fool radiologists with generative adversarial networks? A visual turing test for lung cancer diagnosis,” in Proceedings of the 2018 IEEE 15th International Symposium Biomedical Imaging, pp. 240–244, IEEE, 2018.

X. Zhu, “Semi-supervised learning literature survey,” Computer Science, University of Wisconsin-Madison, vol. 2, no. 3, p. 4, 2006.

M. Li and Z.-H. Zhou, “Improve computer-aided diagnosis with machine learning techniques using undiagnosed samples,” IEEE Transactions on Systems, Man, and Cybernetics: Systems, vol. 37, no. 6, pp. 1088–1098, 2007.

A. Radford, L. Metz, and S. Chintala, “Unsupervised representation learning with deep convolutional generative adversarial networks,” 2015 https://arxiv.org/abs/0712.3011.

Y. Liu, Z. Xing, C. Deng, P. Li, and M. Guo, “Automatically detecting lung nodules based on shape descriptor and semi-supervised learning,” in Proceedings of the 2010 International Conference on Computer Application and System Modeling, ICCASM 2010, pp. V1647–V1650, China, October 2010.

Y. Peng and P. Flach, “Soft discretization to enhance the continuous decision tree induction,” Integrating Aspects of Data Mining, Decision Support and Meta-Learning, vol. 1, no. 34, pp. 109–118, 2001.

N. Settouti, M. E. Habib Daho, M. E. Amine Lazouni, and M. A. Chikh, “Random forest in semi-supervised learning (Co-Forest),” in Proceedings of the 2013 8th International Workshop on Systems, Signal Processing and Their Applications, WoSSPA 2013, pp. 326–329, Algeria, May 2013.

S. G. Armato III, G. McLennan, L. Bidaut et al., “The lung image database consortium (lidc) and image database resource initiative (idri): a completed reference database of lung nodules on ct scans,” Medical Physics, vol. 38, no. 2, pp. 915–931, 2011.

D. E. Ost and M. K. Gould, “Decision making in patients with pulmonary nodules,” American Journal of Respiratory and Critical Care Medicine, vol. 185, no. 4, pp. 363–372, 2012.

M. M. Goodsitt, H. Chan, T. W. Way, S. C. Larson, E. G. Christodoulou, and J. Kim, “Accuracy of the CT numbers of simulated lung nodules imaged with multi-detector CT scanners,” Medical Physics, vol. 33, no. 8, pp. 3006–3017, 2006.

M. M. Goodsitt, H. Chan, T. W. Way, M. J. Schipper, S. C. Larson, and E. G. Christodoulou, “Quantitative CT of lung nodules: dependence of calibration on patient body size, anatomic region, and calibration nodule size for single- and dual-energy techniques,” Medical Physics, vol. 36, no. 7, pp. 3107–3121, 2009.

P. Opulencia, D. S. Channin, D. S. Raicu, and J. D. Furst, “RadLex™, and lung nodule image features,” Journal of Digital Imaging, vol. 24, no. 2, pp. 256–270, 2011.

J.-C. Wang, S. Sone, L. Feng et al., “Rapidly growing small peripheral lung cancers detected by screening CT: Correlation between radiological appearance and pathological features,” British Journal of Radiology, vol. 73, no. 873, pp. 930–937, 2000.

S. Takashima, Y. Maruyama, M. Hasegawa, A. Saito, M. Haniuda, and M. Kadoya, “High-resolution CT features: prognostic significance in peripheral lung adenocarcinoma with bronchioloalveolar carcinoma components,” Respiration, vol. 70, no. 1, pp. 36–42, 2003.

R. M. Lindell, T. E. Hartman, S. J. Swensen et al., “Five-year lung cancer screening experience: CT appearance, growth rate, location, and histologic features of 61 lung cancers,” Radiology, vol. 242, no. 2, pp. 555–562, 2007.

T. J. Kim, D. H. Han, K. N. Jin, and K. Won Lee, “Lung cancer detected at cardiac CT: prevalence, clinicoradiologic features, and importance of full–field-of-view images,” Radiology, vol. 255, no. 2, pp. 369–376, 2010.

M. Vazquez, D. Carter, E. Brambilla et al., “Solitary and multiple resected adenocarcinomas after CT screening for lung cancer: histopathologic features and their prognostic implications,” Lung Cancer, vol. 64, no. 2, pp. 148–154, 2009.