Parameter-based transfer learning for severity classification of atopic dermatitis using hyperspectral imaging

Abstract

Background/purpose: Because atopic dermatitis (AD) is a chronic inflammatory skin condition that causes structural changes, there is a growing need for noninvasive research methods to evaluate this condition. Hyperspectral imaging (HSI) captures skin structure features by exploiting light wavelength variations in penetration depth. In this study, parameter-based transfer learning was deployed to classify the severity of AD using HSI. Therefore, we aimed to obtain an optimal combination of classification results from the four models after constructing different source- and target-domain datasets.

Methods: We designated psoriasis, skin cancer, eczema, and AD datasets as the source datasets, and the set of images acquired via hyperspectral camera as the target dataset for wavelength-specific AD classification. We compared the severity classification performances of 96 combinations of sources, models, and targets.

Results: The highest classification performance of 83% was achieved when ResNet50 was trained on the augmented psoriasis dataset as the source, with the resulting parameters used to train the model on the target Near-infrared radiation (NIR) dataset. The second highest classification accuracy of 81% was achieved when ResNet50 was trained on the unaugmented psoriasis dataset as the source, with the resulting parameters used to train the model on the target R dataset. ResNet50 demonstrated potential as a generalized model for both the source and target data, also confirming that the psoriasis dataset is an effective training resource.

Conclusion: The present study not only demonstrates the feasibility of the severity classification of AD based on hyperspectral images, but also showcases combinations and research scalability for domain exploration.

Keywords: atopic dermatitis; domain selection; hyperspectral imaging; severity classification; transfer learning.