Chronic obstructive pulmonary disease prediction method based on reconstructed airway tree imageTechnical Field
The invention relates to the technical field of medical image processing, in particular to a chronic obstructive pulmonary disease prediction method based on a reconstructed airway tree image.
Background
Chronic Obstructive Pulmonary Disease (COPD) is a common respiratory Disease, seriously harms the physical and mental health of patients, and has become an important problem which must be faced by global public health. Thus, effective diagnostic means are also of great importance.
In traditional clinical Tests, lung respiratory Function Tests (PFTs) are frequently used, the ratio of forced expiratory volume (FEV1) in the first second after bronchodilation to Forced Vital Capacity (FVC) is less than 70% as a standard for accurate diagnosis, and PFTs measure macroscopic functional parameters of the whole lung and cannot provide structural information of an airway tree of COPD lungs. In addition, the ratio of FEV1 to FVC of healthy population decreases with age, so the method is also prone to misdiagnosis and missed diagnosis.
Meanwhile, the pulmonary function examination process in hospitals is very complicated and long in steps, and generally comprises ventilation function, respiratory regulation and pulmonary circulation function. The detection criteria are often different, and clinical diagnosis of COPD requires comprehensive assessment in combination with symptoms, health status and the like of patients, and some patients with severe disease cannot tolerate the disease and are difficult to distinguish from airflow limitation caused by other diseases. Meanwhile, medical staff and workers have limited knowledge about relevant knowledge, and clinical application is greatly limited.
In summary, the current diagnostic methods for COPD have many defects, which increase the workload of doctors and the pain of patients, and the diagnostic results are not satisfactory.
Disclosure of Invention
Aiming at the defects of the prior art, the invention provides a chronic obstructive pulmonary disease prediction method based on a reconstructed airway tree image, which realizes the prediction of chronic obstructive pulmonary disease.
A chronic obstructive pulmonary disease prediction method based on reconstructed airway tree images comprises the following steps:
step 1, based on a medical image segmentation and three-dimensional visualization method, respectively extracting airway trees of COPD patients and healthy people, and storing the airway tree images, wherein the specific method comprises the following steps:
step 1.1, reading collected multilayer CT image files of COPD patients and healthy people from the same hospital by adopting software, and generating reconstructed airway tree images of the healthy people and the COPD patients;
step 1.2, setting the background of the generated reconstructed airway tree images of healthy people and COPD patients to be white;
step 1.3, converting different visual angles to respectively capture the generated reconstructed airway tree images of healthy people and COPD patients, simultaneously controlling the size and resolution of the captured images to be consistent, and storing the captured airway tree images;
step 2, removing redundant background information of the intercepted airway tree image with the same size and resolution;
step 3, training and classifying by using the intercepted airway tree color images with different visual angles based on a 3-channel convolutional neural network model, and predicting whether people in the test set have COPD or not, wherein the specific method comprises the following steps:
3.1, setting a training set, a verification set and a test set of the convolutional neural network model according to the ratio of 8: 1 based on the image intercepted by the reconstructed airway tree;
step 3.2, constructing a convolutional neural network structure comprising an input layer, a convolutional layer, a pooling layer and a full-connection layer; the first layer of the convolutional neural network is an input layer, the second layer is a convolutional layer C1, the convolutional layer C1 is next to a Batch Normalization layer and a ReLU activation function layer, the third layer is a convolutional layer C2, the next to the Batch Normalization layer and the ReLU activation function layer, the fourth layer is a maximum pooling layer P1, the fifth layer is a convolutional layer C3, the sixth layer is a convolutional layer C4, the seventh layer is a maximum pooling layer P2, the eighth layer is an average pooling layer P3, the next to the dropout layer, and the ninth layer is a full-connection layer F2;
3.3, training the convolutional neural network model, and determining the optimal parameters of the convolutional neural network model according to the training time and the accuracy of the verification set;
step 3.4, inputting the test set into the trained convolutional neural network model, and further predicting whether people in the test set have COPD;
and 4, performing cross validation on the image set intercepted by the reconstructed airway tree to ensure the accuracy of the prediction result.
According to the technical scheme, the invention has the beneficial effects that: the method for predicting the chronic obstructive pulmonary disease based on the reconstructed airway tree image is accurate and efficient, and misdiagnosis and missed diagnosis are effectively avoided; meanwhile, the method can be used as a technical means for primarily screening the chronic obstructive pulmonary disease before the examiner confirms the diagnosis, and the complex process of fussy process and long steps of the past hospital pulmonary function examination is avoided, so that the pain of the examiner and the workload of the hospital are greatly reduced; in addition, the invention also provides a new way different from the past lung function examination, and aims to predict the chronic obstructive pulmonary disease from the perspective of reconstructing an airway tree image by multi-layer CT directly, thereby overcoming the defect that most of primary hospitals have no lung function instruments.
Drawings
Fig. 1 is a flowchart of a method for predicting chronic obstructive pulmonary disease based on a reconstructed airway tree image according to an embodiment of the present invention;
fig. 2 is a reconstructed airway tree image of a COPD patient and an airway tree image of a healthy person in a multi-slice CT image provided by an embodiment of the present invention, wherein (a) is the reconstructed airway tree image of the healthy person, and (b) is the reconstructed airway tree image of the COPD patient;
fig. 3 is an airway tree image of a healthy person captured from 3 different viewing angles based on CT image reconstruction according to an embodiment of the present invention; wherein (a) is a front view of an airway tree reconstructed by a healthy person based on a CT image; (b) a back view of an airway tree reconstructed for a healthy person based on a CT image; (c) a top oblique view of an airway tree reconstructed for a healthy person based on a CT image;
fig. 4 is an airway tree image of a COPD patient at 3 different viewing angles, which is taken after being reconstructed based on a CT image, provided by an embodiment of the present invention, wherein (a) is a front view of an airway tree which is reconstructed by the COPD patient based on the CT image; (b) a back view of an airway tree reconstructed based on CT images for COPD patients; (c) a top-down oblique view of an airway tree reconstructed based on CT images for COPD patients;
FIG. 5 is a diagram of removing redundant background images of an airway tree image reconstructed based on a CT image according to an embodiment of the present invention;
FIG. 6 is a schematic diagram of a simplified convolutional neural network model according to an embodiment of the present invention.
Detailed Description
The following detailed description of embodiments of the present invention is provided in connection with the accompanying drawings and examples. The following examples are intended to illustrate the invention but are not intended to limit the scope of the invention.
A method for predicting chronic obstructive pulmonary disease based on reconstructed airway tree images, as shown in fig. 1, includes the following steps:
step 1, based on a medical image segmentation and three-dimensional visualization method, respectively extracting airway trees of a COPD patient and a healthy person, wherein the specific method comprises the following steps:
step 1.1, reading collected multilayer CT image files of COPD patients and healthy persons from the same hospital by adopting software, and generating reconstructed airway tree images of the healthy persons and the COPD patients as shown in figure 2;
the data used in this example are all from the same hospital, and data acquisition of 190 patients with COPD and data acquisition of 90 healthy people are performed, and the training set, the validation set and the test set are set according to the ratio of 8: 1.
Step 1.2, in order to facilitate processing of the reconstructed airway tree image, namely, saving a large amount of time cost required for processing a large amount of image backgrounds, setting the generated backgrounds of the reconstructed airway tree images of healthy people and COPD patients to be white;
step 1.3, converting different visual angles to respectively perform screenshot on reconstructed airway tree images generated by healthy people and COPD patients, simultaneously controlling the size and resolution of the intercepted images to be consistent, and storing the intercepted airway tree images;
in this embodiment, images captured by healthy people based on 3 different viewing angles of the airway tree reconstructed by the CT image are shown in fig. 3; an image of a COPD patient taken from 3 different perspectives of an airway tree reconstructed from a CT image is shown in figure 4.
In this embodiment, in order to ensure that the converted view angles of the reconstructed airway tree images of each COPD patient and healthy person are the same, and therefore the view angles cannot be manually converted, a direction button provided by the method needs to be selected for operation in a three-dimensional visualization method, for example, when a Front image is selected to be captured, a Select 3D view button needs to be clicked on a page of the reconstructed image, and Front needs to be selected. Similarly, for the Back view and the oblique view (rotated 45 ° downward based on the front view), Back and Isometric will be selected in the Select 3D view button, respectively.
In order to ensure that the size and resolution of the reconstructed airway tree image are the same for each COPD patient and healthy person, a default value should be selected for three-dimensional visualization when saving the reconstructed airway tree image, and the reconstructed 3D airway tree image should not be enlarged or reduced, and likewise, should not be manually truncated.
Step 2, removing redundant background information of the intercepted airway tree image with the same size and resolution;
in this embodiment, the airway tree image with the redundant background information removed is shown in fig. 5. Since the smaller the image size when training the convolutional neural network, the lower the time cost consumed for training. Therefore, here, removing the extra white background does not affect the loss of image information.
The method for removing the redundant background is specifically as follows: detecting images from top to bottom and from bottom to top according to lines respectively, stopping when useful information of the images is detected, and recording the number of lines; and detecting the images from left to right and from right to left according to the columns respectively, stopping when useful information of the images is detected, and recording the number of the columns respectively. And (3) performing difference on the numerical values recorded in the two rows and the numerical values recorded in the two columns, comparing the difference values, selecting a larger numerical value to perform square interception on the airway tree image, and storing the acquired square image to be used as a data set for later training.
Step 3, training and classifying by using the intercepted airway tree color images with different visual angles based on a convolutional neural network model with 3 channels, and predicting whether people in the test set have COPD or not, wherein the specific method comprises the following steps:
3.1, setting a training set, a verification set and a test set of the convolutional neural network model according to the ratio of 8: 1 based on the image intercepted by the reconstructed airway tree;
step 3.2, constructing a convolutional neural network structure comprising an input layer, a convolutional layer, a pooling layer and a full-link layer as shown in FIG. 6; the first layer of the convolutional neural network is an input layer, the second layer is a convolutional layer C1, the convolutional layer C1 is next to a Batch Normalization layer and a ReLU activation function layer, the third layer is a convolutional layer C2, the next to the Batch Normalization layer and the ReLU activation function layer, the fourth layer is a maximum pooling layer P1, the fifth layer is a convolutional layer C3, the sixth layer is a convolutional layer C4, the seventh layer is a maximum pooling layer P2, the eighth layer is an average pooling layer P3, the next to the dropout layer, and the ninth layer is a fully-connected layer F2;
where the input layer size is 224 x 224, the filter size of the convolutional layer is 3 x 3, the second layer has 64 convolution kernels, and the fifth and sixth layers have 128 convolution kernels.
3.3, training the convolutional neural network model, and determining the optimal parameters of the convolutional neural network model according to the training time and the accuracy of the verification set;
step 3.4, inputting the test set into the trained convolutional neural network model, and further predicting whether people in the test set have COPD;
and 4, performing cross validation on the image set intercepted by the reconstructed airway tree to ensure the accuracy of the prediction result, namely, recycling the training set, the validation set and the test set, and averaging the accuracy rates of all the prediction results.
In this embodiment, the average value of the accuracy of the prediction results is 90%.
Finally, it should be noted that: the above examples are only intended to illustrate the technical solution of the present invention, but not to limit it; although the present invention has been described in detail with reference to the foregoing embodiments, it will be understood by those of ordinary skill in the art that: the technical solutions described in the foregoing embodiments may still be modified, or some or all of the technical features may be equivalently replaced; such modifications and substitutions do not depart from the spirit of the corresponding technical solutions and scope of the present invention as defined in the appended claims.