May, 2024
基于扩散模型的测试时间图像适应性用于鲁棒皮肤病变分类
Diffusion Model Driven Test-Time Image Adaptation for Robust Skin Lesion
Classification
TL;DR我们提出了一种测试时间图像适应方法,通过同时更新和预测测试图像来提高模型在测试数据上的准确性,通过扩散模型将目标测试图像反投影到源域,设计结构指导模块通过低通滤波添加细化操作,用于正则化扩散以保留结构信息,并引入自整合方案自动调整对适应和未适应输入的依赖关系,增强适应鲁棒性。在我们构建的ISIC2019-C和Dermnet-C损坏鲁棒性评估基准上进行的大量实验表明,我们的方法在各种损坏、体系结构和数据区域上使分类器更加鲁棒。我们的数据集和代码将在https://github.com/minghu0830/Skin-TTA_Diffusion上提供。