COMPUTERISED TOMOGRAPHY IMAGE PROCESSING
20220284584 · 2022-09-08
Inventors
- Regent LEE (Oxford (Botley), GB)
- Anirudh CHANDRASHEKAR (Oxford (Botley), GB)
- Vicente GRAU (Oxford (Botley), GB)
- Ashok HANDA (Oxford (Botley), GB)
Cpc classification
A61B6/504
HUMAN NECESSITIES
G06V10/774
PHYSICS
G06T7/30
PHYSICS
G06T3/40
PHYSICS
International classification
A61B6/00
HUMAN NECESSITIES
G06T3/40
PHYSICS
G06T7/30
PHYSICS
G06V10/774
PHYSICS
Abstract
Methods for training an algorithm to identify structural anatomical features, for example of a blood vessel, in a non-contrast computed tomography (NCT) image are described herein. The algorithm may comprise an image segmentation algorithm, a random forest classifier, or a generative adversarial network in examples described herein. In one embodiment, a method comprises receiving a labelled training set for a machine learning image segmentation algorithm. The labelled training set comprising a plurality of NCT images, each NCT image of the plurality of NCT images showing a targeted region of a subject, the targeted region including at least one blood vessel. The labelled training set further comprises a corresponding plurality of segmentation masks, each segmentation mask labelling at least one structural feature of a blood vessel in a corresponding NCT image of the plurality of NCT images. The method further comprises training a machine learning image segmentation algorithm, using the plurality of NCT images and the corresponding plurality of segmentation masks, to learn features of the NCT images that correspond to structural features of the blood vessels labelled in the segmentation masks, and output a trained image segmentation model. The method further comprises outputting the trained image segmentation model usable for identifying structural features of a blood vessel in an NCT image. Further methods are described herein for identifying anatomical features from an NCT image, and for establishing training sets. Computing apparatuses and computer readable media are also described herein.
Claims
1. A method for training a machine learning image segmentation algorithm to identify structural features of a blood vessel in a non-contrast computed tomography (NCT) image, the method comprising: receiving a labelled training set for the machine learning image segmentation algorithm, the labelled training set comprising: a plurality of NCT images, each NCT image of the plurality of NCT images showing a targeted region of a subject, the targeted region including at least one blood vessel; and a corresponding plurality of segmentation masks, each segmentation mask labelling at least one structural feature of a blood vessel in a corresponding NCT image of the plurality of NCT images; training a machine learning image segmentation algorithm, using the plurality of NCT images and the corresponding plurality of segmentation masks, to learn features of the NCT images that correspond to structural features of the blood vessels labelled in the segmentation masks, and output a trained image segmentation model; outputting the trained image segmentation model usable for identifying structural features of a blood vessel in an NCT image.
2. A method for training a classification algorithm to identify structural features of a blood vessel in a non-contrast computed tomography (NCT) image, the method comprising: receiving a labelled training set for the classification algorithm, the labelled training set comprising: a plurality of NCT images, each NCT image of the plurality of NCT images showing a targeted region of a subject, the targeted region including at least one blood vessel; and a corresponding plurality of segmentation masks, each segmentation mask labelling at least one structural feature of a blood vessel in a corresponding NCT image of the plurality of NCT images; dividing each NCT image and corresponding segmentation mask into sub-volumes, each sub-volume of the corresponding segmentation mask labelling at least one feature in a corresponding sub-volume in the corresponding NCT image; extracting, from each sub-volume of the NCT image, radiomic feature values for a set of radiomic features; training a classification algorithm, using the extracted radiomic feature values for each sub-volume of each NCT image and the corresponding sub-volume of the corresponding segmentation masks, to learn features of the NCT images that correspond to structural features of the blood vessels labelled in the segmentation masks, and output a trained classification model; outputting the trained classification model usable for identifying structural features of a blood vessel in an NCT image.
3. A method according to claim 2, wherein the classification algorithm comprises a random forest classification algorithm.
4. A method according to any preceding claim, wherein each segmentation mask has been generated from a corresponding contrast computed tomography (CCT) image, each CCT image corresponding to an NCT image of the plurality of NCT images and showing the features of the blood vessel in the targeted region of the corresponding NCT image.
5. A method according to any preceding claim, wherein the method further comprises generating the labelled training set.
6. A method according to any preceding claim, wherein the labelled training set has been established according to any of claims 23 to 25.
7. A method according to claim 1, wherein the machine learning image segmentation algorithm comprises a neural network.
8. A method according to any preceding claim, wherein the at least one blood vessel of the targeted region of the non-contrast CT image includes an artery or a vein, or wherein the at least one blood vessel of the targeted region includes the aorta.
9. A method according to any preceding claim, wherein the targeted region of the NCT image includes an aortic aneurysm.
10. A method according to any preceding claim, wherein the structural features of the blood vessel comprise one or more of: inner lumen, outer wall, intima/media, false lumen, calcification, thrombus, ulceration, atherosclerotic plaques.
11. A method according to any preceding claim, wherein each segmentation mask of the plurality of segmentation masks comprises a binary segmentation mask.
12. A computer-readable medium having instructions stored thereon which, when executed by one or more processors, cause the one or more processors to implement a method for training a machine learning image segmentation algorithm or to implement a method for training a classification algorithm according to any preceding claim.
13. A computer-readable medium having stored thereon computer-readable code representative of a trained image segmentation model for identifying structural features of a blood vessel in a non-contrast computed tomography (NCT) image or having code stored thereon computer-readable code representative of a trained classification algorithm for identifying structural features of a blood vessel in a non-contrast computed tomography (NCT) image.
14. A computing apparatus for training a machine learning image segmentation algorithm or a classification algorithm to identify structural features of a blood vessel in a non-contrast computed tomography (NCT) image, the apparatus comprising: one or more memory units; and one or more processors configured to execute instructions stored in the one or more memory units to perform the method of any of claims 1 to 11.
15. A method for identifying structural features of a blood vessel in an unlabelled non-contrast computed tomography (NCT) image, the method comprising: providing the NCT image to a trained image segmentation model, the trained image segmentation model trained to learn features of NCT images that correspond to structural features of blood vessels; generating, using the trained image segmentation model, predicted segmentation data for the provided NCT image, the predicted segmentation data for identifying the features of the blood vessel in the provided NCT image.
16. A method according to claim 15, wherein the trained image segmentation model has been trained according to the method of claim 1.
17. A computer readable medium having stored thereon predicted segmentation data generated using a method according to claim 15 or claim 16.
18. A computing apparatus for identifying structural features of a blood vessel in an unlabelled non-contrast computed tomography (NCT) image, the apparatus comprising: one or more memory units; and one or more processors configured to execute instructions stored in the one or more memory units to perform the method of any of claims 15 to 16.
19. A method for identifying structural features of a blood vessel from a non-contrast computed tomography (NCT) image, the NCT image showing a targeted region of a subject, the targeted region including at least one blood vessel, the method comprising: for each of a plurality of sub-volumes of the NCT image: sampling radiomic feature values for a set of radiomic features from the sub-volume; classifying the sub-volume as belonging to a structural feature of a set of structural features based on the sampled radiomic feature values; and identifying, from the classifications of the plurality of sub-volumes of the NCT image, structural features of a blood vessel shown in the targeted region.
20. A method according to claim 19, wherein classifying the sub-volume as belonging to a structural feature of a set of structural features based on the sampled radiomic feature values comprises using a trained classification algorithm.
21. A method according to claim 19, wherein classifying the sub-volume as belonging to a structural feature of a set of structural features based on the sampled radiomic feature values comprises comparing radiomic features with corresponding threshold values.
22. A computer-readable medium having instructions stored thereon which, when executed by one or more processors, cause the one or more processors to implement a method according to any of claims 15-16 or according to any of claims 19-21 to identify structural features of a blood vessel in a non-contrast computed tomography image.
23. A method for establishing a labelled training set for training a machine learning image segmentation algorithm or a classification algorithm to identify structural features of a blood vessel in a non-contrast computed tomography (NCT) image, the method comprising: receiving a plurality of NCT images, each NCT image showing a targeted region of a subject, the targeted region including at least one blood vessel; receiving a plurality of contrast computed tomography (CCT) images, each CCT image corresponding to an NCT image of the plurality of NCT images and showing the corresponding targeted region of the subject; adapting the plurality of CCT images to substantially align with the corresponding NCT images; and segmenting the plurality of adapted CCT images to generate a corresponding plurality of segmentation masks, each segmentation mask labelling at least one structural feature of the at least one blood vessel in the corresponding adapted CCT image; wherein the labelled training set includes pairs of NCT images and segmentation masks, each pair comprising a segmentation mask and the NCT image to which the adapted CCT image substantially aligns.
24. A method according to claim 23, wherein adapting the plurality of CCT images comprises orienting the plurality of CCT images to substantially align with the corresponding NCT image.
25. A method according to claim 23 or claim 24, wherein adapting the plurality of CCT images comprises scaling the plurality of CCT images to substantially align with the corresponding NCT images.
26. A labelled training set established according to the method of any of claims 23 to 25.
27. A computer-readable medium having instructions stored thereon which, when executed by one or more processors, cause the one or more processors to implement a method according to any of claims 23 to 25 to establish a training set.
28. A computing apparatus for establishing a labelled training set for training a machine learning image segmentation algorithm or a classification algorithm to identify structural features of a blood vessel in a non-contrast computed tomography (NCT) image for identifying structural features of a blood vessel in an unlabelled non-contrast computed tomography (NCT) image, the apparatus comprising: one or more memory units; and one or more processors configured to execute instructions stored in the one or more memory units to perform the method of any of claims 23 to 27.
29. A method for training a generative adversarial network (GAN) to generate a pseudo-contrast computed tomography (PCT) image from a non-contrast computed tomography (NCT) image, the GAN comprising a generator network and a discriminator network, the method comprising: receiving a training set comprising: a plurality of NCT images, each NCT image of the plurality of NCT images showing at least one anatomical structure; and a plurality of contrast computed tomography (CCT) images, each CCT image showing at least one anatomical structure; training the GAN, wherein training the GAN comprises: training the generator network, using the plurality of NCT images and feedback from the discriminator network, to generate PCT images; training the discriminator network, using the generated PCT images and the plurality of CCT images, to classify received images as PCT images or CCT images and to provide feedback to the generator network; and outputting a trained generator model to translate an input NCT image to a PCT image showing at least one anatomical structure.
30. A method according to claim 29, wherein the GAN is a conditional GAN.
31. A method according to claim 29, wherein the GAN is a cycle-GAN.
32. A method according to any of claims 29-31, wherein the generator network comprises a U-NET architecture.
33. A method according to any of claims 29-32, wherein the anatomical structure comprises a blood vessel.
34. A method according to any of claims 29-32, wherein the anatomical structure comprises a kidney.
35. A method according to any of claims 29-32, wherein the anatomical structure comprises a bowel.
36. A computer-readable medium having instructions stored thereon which, when executed by one or more processors, cause the one or more processors to implement a method for training a generative adversarial network (GAN) according to any of claims 29-35.
37. A method for identifying anatomical structures in a non-contrast computed tomography (NCT) image, the method comprising: providing the NCT image to a trained generator model, the trained generator model, the generator model trained as part of a generative adversarial network, the generator model trained to translate an input NCT image to a pseudo-contrast computed tomography (PCT) image showing at least one anatomical structure; and generating, using the trained generator model, a PCT image corresponding to the provided NCT image; and identifying, from the PCT image, structural features of the at least one anatomical structure.
38. A computer-readable medium having stored thereon computer-readable code representative of a trained generator model to generate a PCT image from an input NCT image showing at least one anatomical structure.
39. A method comprising: sending an unlabelled non-contrast computed tomography (NCT) image to a server, the NCT image showing a targeted region of a subject including at least one anatomical structure; and receiving, from the server, predicted segmentation data or a pseudo-contrast computed tomography (PCT) image for the NCT image, the predicted segmentation data or PCT image labelling structural features of the at least one anatomical structure.
Description
BRIEF DESCRIPTION OF THE FIGURES
[0071] Embodiments of the invention will now be described by way of example only, with reference to the accompanying figures, in which:
[0072]
[0073]
[0074]
[0075]
[0076]
[0077]
[0078]
[0079]
[0080]
[0081]
[0082]
[0083]
[0084]
[0085]
[0086]
[0087]
[0088]
[0089]
[0090]
[0091]
[0092]
[0093]
[0094]
[0095]
[0096]
[0097]
[0098]
[0099]
[0100]
[0101]
[0102]
[0103]
[0104]
[0105]
[0106]
[0107]
[0108]
[0109]
[0110]
[0111]
[0112]
[0113]
[0114]
[0115]
[0116]
[0117]
[0118] Throughout the description and the drawings, like reference numerals refer to like parts.
DETAILED DESCRIPTION
[0119] The present disclosure provides ways of training a machine learning algorithm, for example an image segmentation or random forest algorithm or a generative adversarial network (GAN), to identify anatomical structures, for example structural features of a blood vessel, in a non-contrast computed tomography (NCT) image, and further discloses methods for establishing a training set used to train the machine learning algorithm to identify such anatomical structures in an NCT image. The present disclosure further provides other ways of identifying structural features in an NCT image. Whilst various embodiments are described below, the invention is not limited to these embodiments, and variations of these embodiments may well fall within the scope of the invention which is to be limited only by the appended claims.
[0120] CT angiograms (CTAs) are widely used in all fields of cardiovascular surgery/medicine. When treatment of an artery, for example the aorta, is being considered, a medical/surgical professional usually requires a detailed view of the artery to differentiate the morphology/anatomy of the arterial structures. In the case of abdominal aortic aneurysms (AAAs), there is usually thrombus lining within the aneurysm sac. Full visualisation of the thrombus morphology, and its relation to the artery wall, is important for planning surgical intervention, for example by stenting or open repair.
[0121]
[0122] The inventors have discovered that although the structural features 120 and 130 are difficult to distinguish using an NCT image, there is a difference in radiodensity between the subregions that is identifiable from information embedded in an NCT image. This information can be exploited by a machine learning algorithm to identify structural features such as 120 and 130 of a targeted region like 110. For example, blood, thrombus, and arterial walls are all made up of different substances and differ in their densities. Blood is predominantly fluid, whereas thrombus is predominantly fibrinous and collagenous, with red cells/platelets, and arterial walls are predominantly smooth muscle cells, with collagen. Although it is difficult for the human eye to detect the difference between these entities, the inventors have shown that due to the distinct nature of these entities, raw data obtained from an NCT image contains information which enables a machine learning algorithm to discern one from the other. This has the advantage of overcoming the problems described above associated with administering a patient with a contrast agent.
[0123] The inventors were able to determine that there is a difference in Hounsfield unit intensity between different structural features of a blood vessel embedded in an NCT image, by first manually segmenting structural features, such as the thrombus and the inner lumen, from a CCT image of an abdominal aortic region of a patient as discussed in relation to
[0124] The inventors were also able to show that due to the distinct nature of the structural features of a blood vessel, such as structural features 120 and 130, the radiomic signatures of these regions are distinct. The inventors were able to exploit the distinct radiomic signatures of the regions within a blood vessel and devised a method for training a machine learning algorithm, such as a random forest classification algorithm, to identify structural features of a blood vessel from a NCT image as discussed below. Even without the use of a machine learning algorithm, the radiomic features values may be used to identify structural features.
[0125]
[0126]
[0127] Once the boundary between the structural features was determined, the inventors were able to compare the Hounsfield unit intensity of the different structural features 320, 330, and 340 of the NCT image.
[0128] A method for establishing a labelled training set for training a machine learning image segmentation algorithm or a classification algorithm (such as a random forest) to identify structural features of a blood vessel in an NCT image will now be described in relation to the flowchart shown in
[0129] At 510, the method comprises receiving a plurality of NCT images, each NCT image showing a targeted region of a subject, such as the targeted region 110 shown in
[0130] At 520, the method comprises receiving a plurality of CCT images, where each CCT image corresponds to an NCT image of the plurality of NCT images and shows the corresponding targeted region of the subject.
[0131] At 530, the method comprises adapting the plurality of CCT images to substantially align with the corresponding NCT images, where adapting the plurality of CCT images comprises orienting and scaling the plurality of CCT images to substantially align with the corresponding NCT images.
[0132] At 540, the method comprises segmenting the plurality of adapted CCT images to generate a corresponding plurality of segmentation masks, where each segmentation mask labels at least one structural feature of the at least one blood vessel of the targeted region in the corresponding adapted CCT image.
[0133] At 550, a labelled training set is established, wherein the labelled training set includes pairs of NCT images and segmentation masks, where each pair comprises a segmentation mask and the NCT image to which the adapted CCT image substantially aligns.
[0134] The method for establishing a labelled training set used to train a machine learning image segmentation algorithm or a classification algorithm, as described in
[0135] Computing apparatus 600 may comprise a computing device, a server, a mobile or portable computer and so on. Computing apparatus 600 may be distributed across multiple connected devices. Other architectures to that shown in
[0136] Referring to
[0137] Memory 620 is for storing data within computing apparatus 600. The one or more memories 620 may include a volatile memory unit or units. The one or more memories may include a non-volatile memory unit or units. The one or more memories 620 may also be another form of computer-readable medium, such as a magnetic or optical disk. One or more memories 620 may provide mass storage for the computing apparatus 600. Instructions for performing a method as described herein may be stored within the one or more memories 620.
[0138] The apparatus 600 includes a number of user interfaces including visualising means such as a visual display 630 and a virtual or dedicated user input device such as keyboard 640.
[0139] The communications module 650 is suitable for sending and receiving communications between processor 610 and remote systems.
[0140] The port 660 is suitable for receiving, for example, a non-transitory computer readable medium containing one or more instructions to be processed by the processor 610.
[0141] The processor 610 is configured to receive data, access the memory 620, and to act upon instructions received either from said memory 620 or a computer-readable storage medium connected to port 660, from communications module 650 or from user input device 640.
[0142] The computing apparatus 600 may receive, via the communications module 650, data representative of a plurality of non-contrast CT scans of a targeted region of a subject and data representative of a plurality of contrast CT scans of a targeted region of a subject. The data received via the communications module 650 relating to a non-contrast CT scan may be received prior to or subsequent to a contrast CT scan, and may comprise information relating to the measured intensity of the x-rays impinging the targeted region of the subject. The processor 610 may be configured to follow instructions stored in one or more memories 620 to use the received data to reconstruct the corresponding non-contrast and contrast CT images using various CT reconstruction techniques. Each CCT image that is reconstructed corresponds to a reconstructed NCT image of the plurality of reconstructed NCT images.
[0143] Processer 610 may be configured to follow instructions stored in the memory 620 to adapt the plurality of CCT images by orienting and scaling the plurality of CCT images to substantially align with the corresponding NCT images.
[0144] The processor 610 may be configured to follow further instructions stored in the memory 620 to segment the plurality of adapted CCT images to generate a corresponding plurality of segmentation masks, each segmentation mask labelling at least one structural feature of the at least one blood vessel of the targeted region in the corresponding adapted CCT image. The reconstructed CCT image comprises voxels/pixels, and the generated plurality of segmentation masks may be binary segmentation masks, where the voxels/pixels comprising structural feature of the blood vessel of the targeted region may be labelled with a 1 and the voxels/pixels comprising features in the image which are not structural features of the blood vessel may be labelled with a 0 (for example).
[0145] The processor 610 may be configured to follow instructions stored in the memory 620 to pair a generated segmentation mask with a corresponding NCT image to which the adapted CCT image substantially aligns.
[0146] Based on the above description, computing apparatus 600 can be used to establish a labelled training set for training a machine learning image segmentation algorithm or a classification algorithm, where the established labelled training set includes information relating to pairs of NCT images and segmentation masks, where each pair comprises a segmentation mask and the NCT image to which the corresponding adapted CCT image substantially aligns. The skilled person would appreciate that other architectures to that shown in
[0147]
[0148] At step 710, the method comprises receiving a labelled training set, such as the labelled training set described above in relation to
[0149] At step 720, the method comprises training a machine learning segmentation algorithm using the plurality of NCT images and the corresponding plurality of segmentation masks, to learn features of the NCT images that correspond to structural features of the blood vessels labelled in the segmentation masks.
[0150] At step 730, the method comprises output of a trained image segmentation model usable for identifying structural features of a blood vessel in an NCT image.
[0151] The method for training a machine learning image segmentation algorithm, as described above in relation to
[0152] The processor 610 may be configured to train a machine learning image segmentation algorithm to learn the features of NCT images that correspond to structural features of blood vessels of the targeted region using the plurality of NCT images and the corresponding plurality of segmentation masks. For each NCT image and the corresponding segmentation mask, the processor 610 may follow instructions stored in one or more memories 620 to compare the segmentation mask with the corresponding NCT image and adjust the internal weights of the image segmentation algorithm via backpropagation. Several iterations of the comparison between the NCT image and the corresponding segmentation mask may be performed for each NCT image from the plurality of NCT images and the corresponding segmentation masks until a substantially optimized setting for the internal weights is achieved. The processor 610 may follow further instructions stored in one or more memories 620 to perform image transformations at each iteration for each NCT image of the plurality of NCT images to diversify the input data set and maximise learning.
[0153] The processor 610 may be configured to follow further instructions to output the trained image segmentation model and store the trained image segmentation model in one or more memories 620. The trained image segmentation model may comprise for example the weights and biases established during training, along with any selected hyperparameters such as minibatch size or learning rate.
[0154]
[0155] At step 810, the method comprises providing the NCT image to a trained image segmentation model which may be trained according to the method described above in relation to
[0156] At step 820, the method comprises generating, using the trained image segmentation model, predicted segmentation data for the provided NCT image. The predicted segmentation data identifies features of the blood vessel in the provided NCT image.
[0157] The method for identifying structural features of a blood vessel in an unlabelled NCT image, as described above in relation to
[0158] The computing apparatus 600 may receive, via the communications module 650, data from an NCT scan of a subject. The received data may comprise information relating to the measured intensity of the x-rays impinging the targeted region of the subject.
[0159] The computing apparatus 600 may store a trained image segmentation model in one or more memories 620 of the computing apparatus 600, where the trained image segmentation model is trained to learn features of NCT images that correspond to structural features of blood vessels of a targeted region. The processor 610 may be configured to input the received data from the NCT scan to the trained image segmentation model.
[0160] The processor 610 may follow further instructions stored in memory 620 of the computing apparatus 600 to generate, using the trained image segmentation model, a predicted segmentation mask for the provided NCT image. The predicted segmentation mask identifies the structural features of the at least one blood vessel in a targeted region of the provided NCT image.
[0161]
[0162] The general U-NET architecture used for the experiment comprises two components: a downsampling/contraction path (as shown on the left in
[0163] The inventors sought to use a modified version of this established U-NET to predict aneurysmal components from a non-contrast CT scan. The architecture used in this experiment is illustrated clearly in
[0164] By training the model on the ImageNet database, the inventors sought to first teach the algorithm to classify the aorta from surrounding structures, and then to extend its performance to aortic segmentation.
[0165] Each axial CT slice and their respective image masks/segmentation masks were augmented through image transformations (shear, divergence) to diversify the input data set and maximize learning. The initial learning rate and weight decay was set to 4.0×10.sup.−2 and 1.0×10.sup.−7, respectively. The first training loop included a batch size of 25 images and was trained for a total of 16 cycles. Subsequently, the earlier pre-trained layers, which have learned to identify features common to all images (ex. edges, corners, shapes, etc.) were fine-tuned using differential learning rates for an additional 10 cycles. During the training paradigm, the DICE score metric, which reflects the similarity of the inner mask prediction to the ground truth inner mask, increased from 8.6% to approximately 69%.
[0166]
[0167]
[0168]
[0169] An experiment was performed to further investigate whether deep learning segmentation methods can be used to define and extract the subtle differences between the various components of the soft tissue (inner lumen and intra-lumen thrombus (ILT)/WS regions) directly from NCT images. The methodology is illustrated in
[0170] The above discussion illustrates that a machine learning image segmentation algorithm can be used to identify structural features in an NCT image. There is accordingly enough information in an NCT image to be able to identify structural features of a blood vessel. With this in mind, the inventors hypothesised that, given the different biochemical properties of blood within the lumen and intra-lumen thrombus, the radiomic features of these regions are distinct. This hypothesis was tested, and the methodology is illustrated in
[0171] The inventors devised two experiments to show how manual classification of radiomic features can be used to differentiate visually indistinct regions on non-contrast CT images. The two experiments and their results referenced herein as experiments 1 and 2 will be described below.
[0172] In the first experiment (experiment 1), the methodology of
[0173] The table in
[0174]
[0175]
[0176] Next, in Experiment 2 the inventors increased the size of the testing data set and modified the sub-sampling parameters. Once again, the methodology of
[0177] A 3-fold cross validation approach was employed to train and assess the random forest classifier. That is, the datasets for the same 75 patients were used three times with different randomly selected training/validation data and test data. The tables shown in
[0178]
[0179] The above discussion illustrates that radiomic features can be used to train a random forest classification algorithm to identify structural features within an NCT image.
[0180] At step 1910, the method comprises receiving a labelled training set. The labelled training set comprises information relating to a plurality of NCT images, where each NCT image of the plurality of NCT images shows a targeted region of a subject which includes at least one blood vessel. The training set further comprises a corresponding plurality of segmentation masks, where the segmentation masks are generated from a CCT image corresponding to an NCT image of the plurality of NCT images and each segmentation mask labels at least one structural feature of a blood vessel in a corresponding NCT image of the plurality of NCT images. The labelled training set may be similar to that described above in relation to the image segmentation model. The labelled training set may be generated by following the method described in relation to
[0181] At step 1920, the method comprises dividing each NCT image and the corresponding segmentation mask into sub-volumes, where each sub-volume of the corresponding segmentation mask labels at least one feature in a corresponding sub-volume in a corresponding NCT image. In some examples, the sub-volumes have a volume of 10 mm×10 mm×10 mm. However, it will be appreciated that the sub-volumes may be greater or smaller than 10 mm×10 mm×10 mm.
[0182] At step 1930, the method comprises extracting, from each sub-volume of the NCT image, radiomic feature values for a set of radiomic features.
[0183] At step 1940, the method comprises training a random forest classification algorithm, using the extracted radiomic feature values for each sub-volume of each NCT image and the corresponding sub-volume of the corresponding segmentation masks, to learn features of the NCT images that correspond to structural features of the blood vessels labelled in the segmentation masks.
[0184] At step 1950, the method comprises output of a trained random forest classification model usable for identifying structural features of a blood vessel in an NCT image.
[0185] The method for training a random forest classification algorithm, as described above in relation to
[0186]
[0187] At step 2010, the method comprises sampling radiomic feature values for a set of radiomic features for each of a plurality of sub-volumes of a NCT image, wherein the NCT image shows a targeted region of a subject, the targeted region including at least one blood vessel.
[0188] At step 2020, the method comprises classifying the sub-volume as belonging to a structural feature of a set of structural features based on the sampled radiomic feature values. Classifying may be performed by a trained random forest classification algorithm such as that output by the process of
[0189] At step 2030, the method comprises identifying, from the classifications of the plurality of sub-volumes of the NCT image, structural features of a blood vessel shown in the targeted region.
[0190] The method for identifying structural features of a blood vessel in a NCT image, as described above in relation to
[0191] The discussion above has demonstrated that a trained machine learning image segmentation model can be used to identify structural features of blood vessels in a NCT image, and similarly a trained random forest classification algorithm can be used to identify structural features of blood vessels in a NCT image.
[0192] Generative adversarial networks (GANs) have also been investigated. GANS are an approach to generative modelling using deep learning methods, for example convolutional networks. GANS are a class of deep learning architectures whereby two networks train simultaneously, with one network focused on data generation (generator) and the other network focused on data discrimination (discriminator). The generator network and the discriminator network ‘compete’ against each other, learning the statistical distribution of the training data, which in turn enables the generator to generate new examples from the same distribution. A known dataset serves as the initial training data for the discriminator network. Training the discriminator involves presenting it with samples from the training dataset, until it achieves acceptable accuracy. The generator network trains based on whether it succeeds in fooling the discriminator.
[0193] The inventors have demonstrated that GANs can be used to generate/produce a pseudo-contrast computed tomography (PCT) image from an input non-contrast computed tomography (NCT) image.
[0194] A conditional GAN (CGAN) is an extension to the GAN idea. In a conditional GAN, the generative model can be trained to generate new examples from the input domain, where the random vector from the latent space is provided with/conditioned by some additional value, such as a class value, a digit or so on. The discriminator model is also trained by being provided with both an input image that is real or fake and the additional input.
[0195] A cycle-GAN is an extension to the GAN idea. Traditionally, training an image-to-image translation model requires a dataset comprising many paired examples of input images and corresponding expected output images. A cycle-GAN is an approach to training image-to-image translation using the GAN model architecture, in which the generator models and discriminator models can be trained without the need for paired examples in the training data. A cycle-GAN may comprise two generator networks and two discriminator networks. One generator may take images from the first domain as input and output images for the second domain, and the second generator may take images from the second domain and generate images for the first domain. A first discriminator may determine the plausibility of the output image from the first generator and the second discriminator may determine the plausibility of the output image from the second network. Additionally, the output images from the first generator may be input to the second generator and vice versa in order to encourage cycle consistency—if an original input image is input to the first generator and the generated output image is input to the second generator, then it is desirable that the output from the second generator substantially matches the original image. Accordingly, a cycle-GAN may be thought of as two inter-related CGANS each comprising a generator and a discriminator, whereby each CGAN is trained to synthesize an image given an input image. A loss function is further used to update each of the CGANs based on cycle consistency. Cycle consistency loss compares an image input to the cycle-GAN with the generated output and updates the generator models in each training iteration.
[0196] In a test, 75 patients for which paired NCT and CCT images were available were randomly selected from an approved clinical study. A 2-D cycle-GAN and a CGAN were trained and evaluated for the purposes of generating a pseudo-contrast computed tomography (PCT) image from an input NCT image, using a 3-fold cross-validation approach with a training/testing split of 50:25 patients (a total of 11,243 images) respectively. The accuracies of the generated outputs were assessed using four metrics of clinical importance (aneurysm diameter, area, lumen/thrombus volume, and thrombus morphology) against that obtained from CCT images (ground truth). It was found that the generated PCT images bear a strong concordance with the ground truth and enable the assessment of important clinical metrics.
[0197]
[0198] The Non-Contrast-to-Contrast-Cycle-GAN (NC2C-Cycle-GAN) architecture comprises of 4 networks, where 2 of the networks are generator networks (G.sub.AB, and G.sub.BA) and 2 discriminator networks (only one of which, D.sub.B is shown in the figure). The generator networks and discriminator networks comprised neural networks. The generator and discriminator components in the NC2C-Cycle-GAN model architecture were explicitly defined as least-squares GAN and a 70×70 PatchGAN, respectively. The NC2C-Cycle-GAN model incorporates an additional least-squares loss function for the discriminator, which in turn, improves the training of the generative model. On the other hand, the discriminator D.sub.B goes through the image pairs, in 70×70 patches, and is trained to classify whether the image under question is “real” or “fake”.
[0199] The CGAN comprised a pixel to pixel CGAN (Pix2Pix-CGAN). Unlike the cycle-GAN, CGANs require paired non-contrast and contrast images. The generator (G.sub.BA) and discriminator (D.sub.B) components in the NC2C-Conditional GAN model (NC2C-CGAN) architecture were identical to those used in the NC2C-Cycle GAN (least-squares GAN and a 70×70 PatchGAN discriminator). Once again, the discriminator and the generator comprised neural networks.
[0200] A 3-fold cross-validation training paradigm with a training to testing data split of 50:25 patients (˜7,500: ˜3,750 2D axial slices) was employed for both deep learning architectures. Both deep learning GAN architectures were trained with a learning rate of 2.0×10.sup.−5 for 200 epochs on 256×256 images centered around the aorta. For the NC2C-Cycle-GAN architecture, four networks (2 generators+2 discriminators) were trained simultaneously and various loss functions were evaluated at each iteration to document model training. In addition to the loss metrics inherent to the networks, both an identity mapping function and a cycle consistency loss function were included to ensure appropriate style transfer and regularization of the generator to allow for image translation, respectively. On the other hand, two networks (1 generator+1 discriminator) were trained for the NC2C-CGAN architecture. Model weights were saved every 10 epochs and intermediate model predictions were generated from the NCT images within the training cohort. The generated PCT images were independently evaluated against the ground truth to assess model training and generated image quality.
[0201] Each testing cohort contained a unique set of 25 patients. During model training, the root-mean-square error (RMSE) between the generated PCT image and its corresponding CTA image decreases with epoch duration to plateau at 8.0±1.2 and 8.5±1.4 for the CGAN and cycle-GAN respectively. Similarly, the DICE.sub.I coefficient (which quantifies the similarity of segmented regions) increases with epoch duration to plateau at 91.8±0.6% and 92±0.4% for the CGAN and cycle-GAN respectively. The latest model from each fold was used for subsequent analysis.
[0202]
[0203] Comparison of the image properties of the NCT scans below the 10.sup.th percentile and above the 90.sup.th percentile mark highlighted one prominent difference with regards to the x-ray tube current used during image acquisition. Scans obtained with lower tube current values tended to produce images with poorer transformation accuracy.
[0204]
[0205] From the generated 2D-axial slices, the 3D volumes of these 6 patients were reconstructed. The aneurysmal regions for each of these patients are illustrated in
[0206] Evaluation of aneurysm morphology was performed using 4 metrics including diameter, area and volume measurements as well as overall ILT spatial morphology. This information is useful in defining the biological behaviour of an AAA during the natural history of the disease. Measurements derived from the two GAN models' outputs were compared against those obtained from GT segmentations. The NC2C-Cycle-GAN model was better at approximating the maximum lumen diameter per axial slice when compared with that of the NC2C-CGAN model (panel A of
[0207] For both 2-D measurements, the NC2C-Cycle-GAN model had a lower bias and a narrower CI range when compared to that of the NC2C-CGAN model (panels C and D of
[0208] Classification of ILT morphology within the aneurysmal sac was assessed by first isolating the aneurysmal region in the generative model outputs and the GT segmentations. Overall, this was documented by compiling the axial regional ILT classifications and identifying the predominant classification type. Pseudo-contrast images within the aneurysmal region produced by NC2C-Cycle-GAN (
[0209]
[0210] At step 2910, the method comprises receiving a training set, where the training set comprises: a plurality of NCT images, each NCT image of the plurality of NCT images showing at least one blood vessel, and a plurality of contrast computed tomography (CCT) images, each CCT image showing at least one blood vessel.
[0211] At step 2920, the method comprises training the GAN by training the generator network and the discriminator network, where training the generator network comprises using the plurality of NCT images and feedback from the discriminator network, to generate PCT images and training the discriminator network comprises using the generated PCT images and the plurality of CCT images, to classify received images as PCT images or CCT images and to provide feedback to the generator network.
[0212] At step 2930, the method comprises outputting a trained generator model to translate an input NCT image to a PCT image showing at least one blood vessel.
[0213] The method for training a GAN as described above in relation to
[0214]
[0215] At step 3010, the method comprises providing the NCT image to a trained generator model, the trained generator model trained as part of a generative adversarial network, as described above in relation to
[0216] At step 3020, the method comprises generating, using the trained generator model, a PCT image corresponding to the provided NCT image.
[0217] At step 3030, the method comprises identifying, from the PCT image, structural features of the at least one blood vessel.
[0218] The method for identifying structural features of a blood vessel in a NCT image, as described above in relation to
[0219]
[0220] At step 3110, the method comprises sending an unlabelled NCT image to a server, where the NCT image comprises a targeted region of a subject including at least one blood vessel. The server may contain instructions for identifying structural features of a blood vessel in an unlabelled NCT image.
[0221] At step 3120, the method comprises receiving, from the server, information indicative of a predicted segmentation mask for the NCT image, where the predicted segmentation mask labels structural features of the at least one blood vessel of the targeted region.
[0222] The method for obtaining structural features for an unlabelled NCT image, as described above in relation to
[0223] The methods described above in relation to
[0224]
[0225] It will be appreciated that embodiments of the present invention can be realised in the form of hardware, software or a combination of hardware and software. Any such software may be stored in the form of volatile or non-volatile storage such as, for example, a storage device like a ROM, whether erasable or rewritable or not, or in the form of memory such as, for example, RAM, memory chips, device or integrated circuits or on an optically or magnetically readable medium such as, for example, a CD, DVD, magnetic disk or magnetic tape. It will be appreciated that the storage devices and storage media are embodiments of machine-readable storage that are suitable for storing a program or programs that, when executed, implement embodiments of the present invention. Accordingly, embodiments provide a program comprising code for implementing a system or method as claimed in any preceding claim and a machine-readable storage storing such a program. Still further, embodiments of the present invention may be conveyed electronically via any medium such as a communication signal carried over a wired or wireless connection and embodiments suitably encompass the same.
[0226] Many variations of the methods described herein will be apparent to the skilled person. For example, the methods described herein can be used to identify/segment features in other blood vessels besides the aorta (e.g. other arteries or veins). Furthermore, the methods described herein can be used to analyse the behaviour of other organs, for example in the liver, bowel, spleen, or kidney.
[0227] Each feature disclosed in this specification (including any accompanying claims, abstract and drawings), may be replaced by alternative features serving the same, equivalent or similar purpose, unless expressly stated otherwise. Thus, unless expressly stated otherwise, each feature disclosed is one example only of a generic series of equivalent or similar features. The invention is not restricted to the details of any foregoing embodiments. The invention extends to any novel one, or any novel combination, of the features disclosed in this specification (including any accompanying claims, abstract and drawings), or to any novel one, or any novel combination, of the steps of any method or process so disclosed. The claims should not be construed to cover merely the foregoing embodiments, but also any embodiments which fall within the scope of the claims.