用混沌变换增强医学图像自监督学习,提升细微病灶分类效果。
Chaos-SSL: An Attention-Based Self-Supervised Learning Framework with Chaotic Transformation for Medical Image Classification

- 用一维混沌映射生成复杂非线性数据增强,替代传统简单变换。
- 在ISIC和APTOS数据集上分别达到92.61%和87.26%准确率,超越现有自监督方法。
- 通过注意力融合模型,结合专用特征与通用预训练特征,提升泛化能力。
自监督学习(SSL)已成为缓解医学图像分析中对大量标注数据依赖的重要方法。然而,依赖简单几何和色彩变换的标准SSL方法难以捕捉细微病灶所需的精细纹理特征。本文提出Chaos-SSL,一种用于医学图像分类的两阶段框架。第一阶段引入基于一维混沌映射(Logistic、Tent、Sine)的新型自监督预训练策略,作为对比学习中的复杂非线性增强。我们假设这些混沌变换能生成更具挑战性和语义丰富性的视图,迫使网络学习鲁棒的细粒度医学纹理表征。第二阶段设计了一种基于注意力的融合模型,动态结合Chaos-SSL模型的专有特征与更大的ImageNet预训练模型的通用特征。我们在两个公开数据集上验证该方法:ISIC 2018(皮肤病变)和APTO 2019(糖尿病视网膜病变)。结果表明,使用Tent映射预训练30轮后进行注意力融合的Chaos-SSL模型,在性能上可媲美当前最优水平,分别在ISIC 2018和APTO 2019上达到0.9261和0.8726的准确率,显著优于多个近期自监督方法。
原文摘要 · Abstract (English)
Self-Supervised Learning (SSL) has emerged as a powerful paradigm to mitigate the reliance on large, annotated datasets, a common bottleneck in medical image analysis. However, standard SSL methods, which rely on simple geometric and color augmentations, may fail to capture the fine-grained, complex textural details necessary for classifying subtle pathologies. This paper introduces Chaos-SSL, a novel two-stage framework for medical image classification. In the first stage, we propose a new self-supervised pre-training strategy that leverages 1D chaotic maps (Logistic, Tent, and Sine) as a complex, non-linear augmentation for contrastive learning. We hypothesize that these chaotic transformations create ``harder'' and more semantically-rich views, forcing a network to learn robust representations of fine-grained medical textures. In the second stage, we introduce an attention-based fusion model that dynamically combines the specialized features from our Chaos-SSL model with the general-purpose features of a larger, ImageNet-pre-trained model. We validate our method on two public datasets: ISIC 2018 (skin lesions) and APTOS 2019 (diabetic retinopathy). Our results demonstrate that the Chaos-SSL model pre-trained with a Tent map for 30 epochs, followed by attention fusion, achieves performance fully competitive with the state-of-the-art, yielding an accuracy of 0.9261 on ISIC 2018 and 0.8726 on APTOS 2019. This significantly outperforms existing SSL methods, including several recent approaches.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。