用自监督学习在少量标注数据下实现湿疹严重程度自动评估
Automated Measurement of Eczema Severity with Self-Supervised Learning
- 先用SegGPT做少样本分割,再用DINO特征分类
- 在真实场景图像上达到0.67的加权F1,优于现有方法
- 适合标注数据稀缺的皮肤疾病自动化诊断场景
利用数码相机拍摄的图像实现湿疹自动诊断,可帮助个体自我监测恢复情况。该过程包括先从图像中分割出湿疹区域,再测量分割区域的严重程度。当前最先进的自动化湿疹诊断方法依赖卷积神经网络(CNN)等深度神经网络,在准确测量湿疹严重程度方面表现优异,但这些方法需要大量标注数据训练,而获取此类数据较为困难。本文提出一种自监督学习框架,用于在有限训练数据条件下实现自动化湿疹诊断。该框架分为两个阶段:(i) 分割阶段,采用基于上下文学习的SegGPT算法,实现湿疹区域的少样本分割;(ii) 特征提取与分类阶段,从分割区域提取DINO特征,并输入多层感知机(MLP)进行四分类(湿疹严重程度)。在包含标注“真实场景”湿疹图像的数据集上评估,结果表明本方法优于现有深度学习方法,如微调后的ResNet-18(加权F1:0.44 ± 0.16)和视觉变压器(加权F1:0.40 ± 0.22),达到0.67 ± 0.01的加权F1。结果表明,自监督学习可在标注数据稀缺的自动化皮肤诊断中提供可行解决方案。
原文摘要 · Abstract (English)
Automated diagnosis of eczema using images acquired from digital camera can enable individuals to self-monitor their recovery. The process entails first segmenting out the eczema region from the image and then measuring the severity of eczema in the segmented region. The state-of-the-art methods for automated eczema diagnosis rely on deep neural networks such as convolutional neural network (CNN) and have shown impressive performance in accurately measuring the severity of eczema. However, these methods require massive volume of annotated data to train which can be hard to obtain. In this paper, we propose a self-supervised learning framework for automated eczema diagnosis under limited training data regime. Our framework consists of two stages: i) Segmentation, where we use an in-context learning based algorithm called SegGPT for few-shot segmentation of eczema region from the image; ii) Feature extraction and classification, where we extract DINO features from the segmented regions and feed it to a multi-layered perceptron (MLP) for 4-class classification of eczema severity. When evaluated on a dataset of annotated "in-the-wild" eczema images, we show that our method outperforms (Weighted F1: 0.67 $\pm$ 0.01) the state-of-the-art deep learning methods such as finetuned Resnet-18 (Weighted F1: 0.44 $\pm$ 0.16) and Vision Transformer (Weighted F1: 0.40 $\pm$ 0.22). Our results show that self-supervised learning can be a viable solution for automated skin diagnosis where labeled data is scarce.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。