卷积神经网络
人工智能
计算机科学
元数据
模式识别(心理学)
图形
皮肤损伤
深度学习
人工神经网络
特征(语言学)
特征提取
上下文图像分类
病变
黑色素瘤诊断
边距(机器学习)
深层神经网络
机器学习
特征学习
作者
Thi Trang Nguyen,Vu Tien Sinh,Van Hieu Vu,Nguyễn Việt Anh
标识
DOI:10.21203/rs.3.rs-7109516/v1
摘要
Abstract Accurate classification of skin lesions is crucial for early detection of melanoma and other skin cancers. However, challenges such as inter-class similarity, intra-class variation, and limited labeled data persist. We propose a novel multi-modal graph neural network (GNN) that integrates three complementary modalities: deep visual features from pre-trained convolutional neural networks (CNNs), handcrafted descriptors (e.g., HSV histograms, fractals), and structured clinical metadata (e.g., age, sex, lesion location).Each lesion sample is represented as a node in a \(k\)-nearest neighbor graph constructed in the fused multi-modal feature space. A two-layer graph convolutional network is used to perform relational learning and classification. The model is trained with class-weighted cross-entropy loss and evaluated using repeated stratified 5-fold cross-validation.Experiments on the HAM10000 and ISIC2020 datasets show that our method consistently outperforms strong baselines. On HAM10000, the proposed GNN achieves an accuracy of 0.960 \(\pm\) 0.003, F1 score of 0.959 $\pm$ 0.004, and AUC of 0.999 $\pm$ 0.000. On ISIC2020, it reaches 0.999 $\pm$ 0.001 in both F1 and accuracy, with an AUC of 1.000. These results validate the effectiveness of multi-modal fusion and graph-based reasoning in skin lesion diagnosis and suggest the method's potential for real-world clinical deployment.
科研通智能强力驱动
Strongly Powered by AbleSci AI