| Literature DB >> 34982041 |
Hoon Ko1, Jimi Huh2, Kyung Won Kim3,4, Heewon Chung1, Yousun Ko5, Jai Keun Kim2, Jei Hee Lee2, Jinseok Lee1.
Abstract
BACKGROUND: Detection and quantification of intra-abdominal free fluid (ie, ascites) on computed tomography (CT) images are essential processes for finding emergent or urgent conditions in patients. In an emergency department, automatic detection and quantification of ascites will be beneficial.Entities:
Keywords: artificial intelligence; ascites; computed tomography; deep residual U-Net
Mesh:
Year: 2022 PMID: 34982041 PMCID: PMC8764611 DOI: 10.2196/34415
Source DB: PubMed Journal: J Med Internet Res ISSN: 1438-8871 Impact factor: 5.428
Demographic and clinical data of participants in the control group and ascites group.
| Variables | Control group (n=200) | Ascites group (n=200) | ||
|
| ||||
|
|
| |||
|
|
| Female | 92 (46.0) | 101 (50.5) |
|
|
| Male | 108 (54.0) | 99 (49.5) |
|
| Age in years, mean (SD) | 59.7 (13.8) | 60.2 (15.3) | |
|
| ||||
|
| Large | 0 (0) | 92 (46.0) | |
|
| Moderate | 0 (0) | 47 (23.5) | |
|
| Small | 0 (0) | 61 (30.5) | |
|
| ||||
|
| Cancer | 14 (7.0) | 42 (21.0) | |
|
| Congestive heart failure | 0 (0) | 3 (1.5) | |
|
| Liver cirrhosis | 1 (0.5) | 51 (25.5) | |
|
| Acute liver failure | 0 (0) | 3 (1.5) | |
|
| Infection | 7 (3.5) | 28 (14.0) | |
|
| Blunt trauma | 5 (2.5) | 37 (18.5) | |
|
| Postoperative status | 32 (16.0) | 5 (2.5) | |
|
| Intestinal obstruction | 1 (0.5) | 10 (5.0) | |
|
| Renal failure | 0 (0) | 10 (5.0) | |
|
| Unknown cause of abdominal pain | 140 (70.0) | 11 (5.5) | |
Summary of training and testing data sets.
| Group | Training data, n (%) | Testing data, n (%) | Total, n (%) | |||||
|
| Subjects (n=160) | Images (n=6337) | Subjects (n=40) | Images (n=1635) | Subjects (n=200) | Images (n=7972) | ||
| Ascites | 80 (50.0) | 1969 (31.1) | 20 (50.0) | 492 (30.1) | 100 (50.0) | 2461 (30.9) | ||
| Control | 80 (50.0) | 4368 (68.9) | 20 (50.0) | 1143 (69.9) | 100 (50.0) | 5511 (69.1) | ||
Figure 1The architecture of our proposed model for ascites region segmentation based on a single abdomen computed tomography (CT) image. ReLU: rectified linear unit.
Hyperparameters of convolutional layers according to each layer and unit level.
| Model part, unit level, and layer | Kernel | Strides, n | Output size, pixels | |||||||
|
| Filter size, pixels | Filters, n |
|
| ||||||
|
| ||||||||||
|
| N/Aa | N/A | N/A | N/A | 256 × 256 × 3 | |||||
|
| ||||||||||
|
|
| |||||||||
|
|
| Convolutional layer 1 | 3 × 3 | 32 | 1 | 256 × 256 × 32 | ||||
|
|
| Convolutional layer 2 | 3 × 3 | 32 | 1 | 256 × 256 × 32 | ||||
|
|
| |||||||||
|
|
| Convolutional layer 3 | 3 × 3 | 64 | 2 | 128 × 128 × 64 | ||||
|
|
| Convolutional layer 4 | 3 × 3 | 64 | 1 | 128 × 128 × 64 | ||||
|
|
| |||||||||
|
|
| Convolutional layer 5 | 3 × 3 | 128 | 2 | 64 × 64 × 128 | ||||
|
|
| Convolutional layer 6 | 3 × 3 | 128 | 1 | 64 × 64 × 128 | ||||
|
|
| |||||||||
|
|
| Convolutional layer 7 | 3 × 3 | 256 | 2 | 32 × 32 × 256 | ||||
|
|
| Convolutional layer 8 | 3 × 3 | 256 | 1 | 32 × 32 × 256 | ||||
|
| ||||||||||
|
|
| |||||||||
|
|
| Convolutional layer 9 | 3 × 3 | 512 | 2 | 16 × 16 × 512 | ||||
|
|
| Convolutional layer 10 | 3 × 3 | 512 | 1 | 16 × 16 × 512 | ||||
|
| ||||||||||
|
|
| |||||||||
|
|
| Convolutional layer 11 | 3 × 3 | 256 | 1 | 32 × 32 × 256 | ||||
|
|
| Convolutional layer 12 | 3 × 3 | 256 | 1 | 32 × 32 × 256 | ||||
|
|
| |||||||||
|
|
| Convolutional layer 13 | 3 × 3 | 128 | 1 | 64 × 64 × 128 | ||||
|
|
| Convolutional layer 14 | 3 × 3 | 128 | 1 | 64 × 64 × 128 | ||||
|
|
| |||||||||
|
|
| Convolutional layer 15 | 3 × 3 | 64 | 1 | 128 × 128 × 64 | ||||
|
|
| Convolutional layer 16 | 3 × 3 | 64 | 1 | 128 × 128 × 64 | ||||
|
|
| |||||||||
|
|
| Convolutional layer 17 | 3 × 3 | 32 | 1 | 256 × 256 × 32 | ||||
|
|
| Convolutional layer 18 | 3 × 3 | 32 | 1 | 256 × 256 × 32 | ||||
|
| ||||||||||
|
|
| |||||||||
|
|
| Convolutional layer 19 | 1 × 1 | 1 | 1 | 256 × 256 × 1 | ||||
|
|
| |||||||||
|
|
| Sigmoid layer | N/A | N/A | N/A | 256 × 256 × 1 | ||||
aN/A: not applicable; this model part did not include this parameter.
Cross-validation results for the training data set comparing the mIoU for segmentation performance and AUROC for detection across models.
| Model | mIoUa (SD) | AUROCb (SD) |
| Deep residual U-Net (two residual blocks) | 0.86 (0.03) | 0.97 (0.02) |
| Deep residual U-Net (three residual blocks) | 0.86 (0.02) | 0.98 (0.01) |
| Deep residual U-Net (four residual blocks) | 0.87 (0.02) | 0.99 (0.01) |
| Deep residual U-Net (five residual blocks) | 0.69 (0.46) | 0.69 (0.01) |
| U-Net [ | 0.84 (0.02) | 0.96 (0.01) |
| Bidirectional U-Net [ | 0.82 (0.01) | 0.91 (0.01) |
| Recurrent residual U-Net [ | 0.74 (0.02) | 0.90 (0.01) |
amIoU: mean intersection over union; this is an index of the segmentation performance.
bAUROC: area under the receiver operating characteristic curve; this is an index of detection accuracy.
Effect of the number of convolutional layers in each residual block on cross-validation results with the training data set.
| Model | mIoUa (SD) | AUROCb (SD) |
| Deep residual U-Net with two convolutional layers in each residual block | 0.87 (0.02) | 0.99 (0.01) |
| Deep residual U-Net with three convolutional layers in each residual block | 0.83 (0.03) | 0.98 (0.02) |
| Deep residual U-Net with four convolutional layers in each residual block | 0.69 (0.02) | 0.69 (0.01) |
amIoU: mean intersection over union; this is an index of the segmentation performance.
bAUROC: area under the receiver operating characteristic curve; this is an index of the detection accuracy.
Effect of the number of convolutional layers in each residual block on the testing data set results for the deep residual U-Net model with four residual blocks.
| Model | mIoUa (SD) | AUROCb |
| Deep residual U-Net with two convolutional layers in each residual block | 0.87 (0.26) | 0.96 |
| Deep residual U-Net with three convolutional layers in each residual block | 0.84 (0.27) | 0.94 |
| Deep residual U-Net with four convolutional layers in each residual block | 0.74 (0.31) | 0.72 |
amIoU: mean intersection over union; this is an index of the segmentation performance.
bAUROC: area under the receiver operating characteristic curve; this is an index of the detection accuracy.
Segmentation performance and detection accuracy of artificial intelligence models with the testing data set.
| Model | mIoUa (SD) | AUROCb |
| Deep residual U-Net (two residual blocks) | 0.81 (0.33) | 0.87 |
| Deep residual U-Net (three residual blocks) | 0.86 (0.28) | 0.93 |
| Deep residual U-Net (four residual blocks) | 0.87 (0.26) | 0.96 |
| Deep residual U-Net (five residual blocks) | 0.70 (0.46) | 0.70 |
| U-Net [ | 0.80 (0.33) | 0.90 |
| Bidirectional U-Net [ | 0.77 (0.35) | 0.86 |
| Recurrent residual U-Net [ | 0.67 (0.41) | 0.81 |
amIoU: mean intersection over union; this is an index of the segmentation performance.
bAUROC: area under the receiver operating characteristic curve; this is an index of the detection accuracy.
Figure 2Five examples of ascites segmentation results using each model. A. The original computed tomography (CT) images and the ground-truth masking images. B. Our proposed model. C. The U-Net model. D. The bidirectional U-Net model. E. The recurrent residual U-Net model. Each row represents a different example of CT images. Blue represents the ground-truth masking images, and red represents the resultant segmented images.
Detection performance metrics of artificial intelligence models with the testing data set.
| Model | Sensitivity | Specificity | Accuracy | Balanced accuracy | Precision | F1 score |
| U-Net [ | 0.92 | 0.90 | 0.90 | 0.91 | 0.79 | 0.85 |
| Bidirectional U-Net [ | 0.94 | 0.86 | 0.88 | 0.90 | 0.74 | 0.83 |
| Recurrent residual U-Net [ | 0.85 | 0.81 | 0.82 | 0.83 | 0.66 | 0.74 |
| Deep residual U-Net | 0.96 | 0.96 | 0.96 | 0.96 | 0.91 | 0.93 |
Figure 3Examples of incorrect segmentation results. The left-hand column includes the original computed tomography (CT) images, the middle column includes the ground-truth masking images, and the right-hand column includes the segmented results by our deep residual U-Net algorithm. A. In a patient with a left ovarian cyst, our artificial intelligence (AI) algorithm detected fluid within the ovarian cyst as ascites. B. In a patient with a fully distended bladder, our AI algorithm detected fluid in the bladder as ascites. Red represents the resultant segmented images.
Comparison of the number of parameters for each U-Net model.
| Model | Trainable parameters, n | Nontrainable parameters, n | Total parameters, n |
| Our proposed model | 18,840,545 | 14,592 | 18,855,137 |
| U-Net [ | 34,600,353 | 14,016 | 34,614,369 |
| Bidirectional U-Net [ | 55,398,798 | 1408 | 55,400,197 |
| Recurrent residual U-net [ | 24,133,013 | 0 | 24,133,013 |