repository
Strategy for Improving Generalizability of Medical Imaging Artificial Intelligence Model
- Author(s)
- 신서정
- Advisor
- 박래웅
- Department
- 일반대학원 의생명과학과
- Publisher
- The Graduate School, Ajou University
- Publication Year
- 2022-02
- Language
- eng
- Keyword
- Deep Learning; Medical Imaging; Style Transfer
- Alternative Abstract
- Recent developments in the field of artificial intelligence have helped develop high-performance artificial intelligence models in the medical imaging field. However, when applying these models in the actual medical site, performance declines are frequently observed. This is because the heterogeneity of medical imaging data is greater than that of the data obtained from other fields; this heterogeneity is a major obstacle to the growth of the artificial intelligence software market for medical imaging. Until now, to cope with domain shifts associated with the use of artificial intelligence in medical imaging, scientists have applied the color normalization and data-augmentation techniques to the training data, or the developed model has been finely adjusted to the test data. However, these techniques had limitations, such as the artificial adjustment of learning data diversity and the lack of data labels to fine-tune the developed model. In this paper, we use a generative model-based style-transfer technique to prevent any performance decline in the actual application stage of the artificial intelligence model for medical imaging analysis. When applying the artificial intelligence model to the actual medical sites, we use the generative model along with the artificial intelligence model developed by learning the medical image. The generative model adopted in this paper is CycleGAN, which converts landscape photographs into paintings of the artist Gogh and conversely. Using this model, the style of the actual medical field images that the artificial intelligence model should predict is changed to the style of the images learned by the artificial intelligence model. The analysis revealed that the artificial intelligence model showed significantly lower performance when predicting images of the actual field with styles different from the image that they learned. However, in images that passed through the generative model and were changed to a style familiar to the artificial intelligence model, the performance was restored. We confirmed that performance improvement of the proposed model was greater than that of the color-normalization methods used in the pathology field. Besides pathology images, brain magnetic resonance imaging (MRI) data in the field of radiology confirmed that changing the style improved the generalizability of the model. In addition, the mechanism of performance improvement was inferred by analyzing the image characteristics before and after the application of CycleGAN and the changes in the feature values extracted from the images. The results of this paper prove that the use of the style-transfer technique based on the generative model can be an effective strategy to prevent performance degradation because of data heterogeneity when the medical artificial intelligence model is applied.
- Fulltext
- Appears in Collections:
- Graduate School of Ajou University > Department of Biomedical Sciences > 4. Theses(Ph.D)
- Files in This Item:
- There are no files associated with this item.
Items in DSpace are protected by copyright, with all rights reserved, unless otherwise indicated.
-
- License
- STATISTICS
- Total Visit :4,071,009
- Total Download :1,967
- Today View :1,439
