Skip to content

求助:无法复现出论文的结果 #13

@ggoodd-any

Description

@ggoodd-any

高老师您好,我以The Changes You Want To See(CYWS)的代码框架为基础尝试复现MetaUAS的结果。训练数据集只使用InpatinedCocoDataset,数据增强方式大部分保持CYWS的数据增强模式,同时遵循DRAEM论文中的方法,增加了使用Perlin噪声合成的异常,以实现local region changes。
模型方面完全遵循MetaUAS的配置,使用了代码提供的模型。学习率设置为1e-4,Weight Decay 为,BatchSize设置为128,训练30epochs。然后直接使用这个模型在MVTecAD数据集上进行测试。模型测试代码与MetaUAS的配置完全相同,使用oneprompt_seed1.json配置。
训练结果为:

Name I-AUROC I-AP I-F1max P-AUROC P-AP P-F1max P-AUPRO
bottle 0.974 0.992 0.961 0.901 0.537 0.486 0.630
cable 0.650 0.790 0.763 0.927 0.338 0.374 0.710
capsule 0.649 0.885 0.905 0.804 0.088 0.135 0.366
carpet 0.878 0.968 0.909 0.914 0.514 0.512 0.555
grid 0.815 0.932 0.862 0.785 0.160 0.268 0.419
hazelnut 0.951 0.971 0.923 0.982 0.754 0.705 0.909
leather 0.999 1.000 0.995 0.979 0.406 0.419 0.930
metal_nut 0.708 0.907 0.912 0.540 0.224 0.257 0.494
pill 0.601 0.905 0.916 0.606 0.142 0.251 0.459
screw 0.611 0.827 0.858 0.779 0.025 0.059 0.487
tile 0.846 0.941 0.865 0.820 0.507 0.480 0.620
toothbrush 0.744 0.904 0.845 0.820 0.296 0.382 0.493
transistor 0.628 0.579 0.608 0.798 0.207 0.279 0.527
wood 0.988 0.996 0.968 0.894 0.541 0.543 0.622
zipper 0.686 0.906 0.881 0.692 0.090 0.119 0.277
Avg 0.782 0.900 0.878 0.816 0.322 0.351 0.567

与论文的结果相差比较大。
是不是我在复现的过程中忽略了某些重要的细节?例如需要进行梯度裁剪,或者需要在训练数据集中加入一部分MVTec-AD的训练数据集一起训练。

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions