Performance evaluation of mask R-CNN for lung segmentation using computed tomographic images

다국어 초록 (Multilingual Abstract)

Image segmentation techniques based on machine learning are able to improve diagnostic and therapeutic accuracy by localizing target areas. The accuracy and efciency of these techniques are dependent on network architecture and loss minimization method because the performance of a machine learning model is determined by training strategies. In this study, the lung segmentation based on computed tomographic images was performed by using mask regional convolutional neural networks (R-CNNs) with various feature extraction networks and optimizers. The efects of the feature extraction networks and optimizers on the trained mask R-CNNs were evaluated in terms of total training loss, segmentation accuracy and training time. The results showed that the convergence of total loss values during network training was afected by the architectures of the feature extraction networks as well as the optimizers. The lung segmentation accuracy and training time of the mask R-CNN were mainly dependent on the optimizer and network architecture, respectively. Among the various optimizers, the ASGD optimizer maximized lung segmentation accuracy, and the training time was reduced by the feature extraction network including general convolution layers and feature pyramid network (FPN). In conclusion, it is important to apply the optimal network architecture and optimizer to the mask R-CNN for maximizing its performance, and the optimized mask R-CNN can be potentially used for improving diagnostic and therapeutic accuracy.

참고문헌 (Reference)

1 T. Peng, 7 : 137794-, 2019

2 M. Everingham, 111 : 98-, 2015

3 K. Harrison, 34 : 74-, 2022

4 J. Sousa, 12 : 1959-, 2022

5 F. Sforazzini, 4 : 2-, 2022

6 B. H. Cho, 10 : 611-, 2020

7 B. T. Polyak, 30 : 838-, 1992

8 J. Duchi, 12 : 2121-, 2011

9 L. Geng, 24 : 27-, 2019

10 P. Tang, 352-, 2018

11 T.-Y. Lin, 2117-, 2017

12 B. Romera-Paredes, 312-, 2016

13 M. H. Hesamian, 32 : 582-, 2019

14 J. Wang, 176-, 2016

15 A. N. J. Raj, 7 : 349-, 2021

16 N. Saeedizadeh, 1 : 100007-, 2021

17 K. He, 2961-, 2017

18 A. A. Novikov, 37 : 1865-, 2018

19 J. Bergstra, 13 : 281-305, 2012

20 S. Ren, 28 : 91-, 2015

21 B. C. Russell, 77 : 157-, 2008

22 K. He, 770-, 2016

23 I. Loshchilov, "arXiv: 1711. 05101"

24 A. Garcia-Garcia, "arXiv: 1704. 06857"

25 D. P. Kingma, "arXiv: 1412. 6980"

26 M. D. Zeiler, "arXiv: 1212. 5701"

27 L. Bottou, "Neural networks: tricks of the trade" Springer 421-436, 2012

28 J. S. Kirby, "LUNGX challenge for computerized lung nodule classification" 3 : 044506-, 2016