InceptionV3 기반의 심장비대증 분류 정확도 향상 연구

• Published in: Journal of biomedical engineering research Vol. 43; no. 1; pp. 45 - 51
• Main Authors: 정우연, 김정훈, Jeong, Woo Yeon, Kim, Jung Hun
• Format: Journal Article
• Language: Korean
• Published: 대한의용생체공학회 01.02.2022
• Subjects: 의공학; Deep learning; Cardiomegaly; Inception; Artificial intelligence; Classification
• ISSN: 1229-0807; 2288-9396

The purpose of this study is to improve the classification accuracy compared to the existing InceptionV3 model by proposing a new model modified with the fully connected hierarchical structure of InceptionV3, which showed excellent performance in medical image classification. The data used for model training were trained after data augmentation on a total of 1026 chest X-ray images of patients diagnosed with normal heart and Cardiomegaly at Kyungpook National University Hospital. As a result of the experiment, the learning classification accuracy and loss of the InceptionV3 model were 99.57% and 1.42, and the accuracy and loss of the proposed model were 99.81% and 0.92. As a result of the classification performance evaluation for precision, recall, and F1 score of Inception V3, the precision of the normal heart was 78%, the recall rate was 100%, and the F1 score was 88. The classification accuracy for Cardiomegaly was 100%, the recall rate was 78%, and the F1 score was 88. On the other hand, in the case of the proposed model, the accuracy for a normal heart was 100%, the recall rate was 92%, and the F1 score was 96. The classification accuracy for Cardiomegaly was 95%, the recall rate was 100%, and the F1 score was 97. If the chest X-ray image for normal heart and Cardiomegaly can be classified using the model proposed based on the study results, better classification will be possible and the reliability of classification performance will gradually increase.