Skin Lesion Intelligent Diagnosis in Edge Computing Networks: A Federated Contrastive Learning Approach

Yanhang Shi,Xue Li,Siguang Chen
DOI: https://doi.org/10.1145/3595186
IF: 5
2023-05-01
ACM Transactions on Intelligent Systems and Technology
Abstract:In recent years, automatic skin lesion diagnosis methods based on artificial intelligence (AI) have achieved great success. However, the lack of labeled data, visual similarity between skin diseases and restriction on private data sharing remain the major challenges in skin lesion diagnosis. In this paper, first, we propose a federated contrastive learning framework (FCL) to to break down data silos and enhance the generalizability of diagnostic model to unseen data. Subsequently, by combining data features from different participated nodes, the proposed framework can improve the performance of contrastive training. To extract discriminative features during on-device training, we propose a contrastive learning based intelligent skin lesion diagnosis (CL-ISLD) scheme in edge computing networks. Specifically, a contrastive learning based dual encoder network is designed to overcome training sample scarcity by fully leveraging unlabeled samples for performance improvement. Meanwhile, we devise a maximum mean discrepancy (MMD) based supervised contrastive loss function, it can efficiently explore complex intra-class and inter-class variances of samples. Finally, the diagnosis simulations demonstrate that compared with existing methods, our proposed scheme can achieve superior accuracy in both on-device training and distributed training scenarios.
computer science, information systems, artificial intelligence
What problem does this paper attempt to address?