arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.02253cs.LGcs.CV
收藏

参数选择性持续测试时适应

Parameter-Selective Continual Test-Time Adaptation

College of Electric and Information, Northeast Agricultural University(东北农业大学电气与信息学院);Institute of Automation, Chinese Academy of Sciences(中国科学院自动化研究所)

Jiaxu Tian, Fan Lyu

AI总结:

针对持续测试时适应中Mean Teacher全参数更新导致的错误累积与灾难性遗忘,提出PSMT,通过学生选择性蒸馏和教师基于Fisher信息的掩码更新实现关键参数保护,并在多个基准上取得最优性能。

摘要:

持续测试时适应(CTTA)旨在在连续领域偏移下,于测试阶段使预训练模型适应不断变化的环境。大多数现有CTTA方法基于Mean Teacher(MT)结构,该结构包含学生模型和教师模型:学生模型利用教师模型生成的伪标签进行更新,教师模型随后通过指数移动平均策略更新。然而,这些方法不加区分地更新MT模型的所有参数;也就是说,一些涉及跨不同领域共享知识的关键参数可能被擦除,从而加剧错误累积和灾难性遗忘。本文提出Parameter-Selective Mean Teacher(PSMT,参数选择性平均教师)方法,能够在领域偏移下有效更新MT网络中的关键参数。首先,我们在学生模型中引入选择性蒸馏机制,利用过去知识正则化新知识,从而减轻错误累积的影响。其次,为避免灾难性遗忘,在教师模型中,我们通过Fisher信息创建掩码,利用指数移动平均选择性地更新参数,并对关键参数采取保护措施。大量实验结果证明,PSMT在多个基准数据集上优于最先进方法。代码可在https://github.com/JiaxuTian/PSMT获取。

英文摘要:

Continual Test-Time Adaptation (CTTA) aims to adapt a pretrained model to ever-changing environments during the test time under continuous domain shifts. Most existing CTTA approaches are based on the Mean Teacher (MT) structure, which contains a student and a teacher model, where the student is updated using the pseudo-labels from the teacher model, and the teacher is then updated by exponential moving average strategy. However, these methods update the MT model indiscriminately on all parameters of the model. That is, some critical parameters involving sharing knowledge across different domains may be erased, intensifying error accumulation and catastrophic forgetting. In this paper, we introduce Parameter-Selective Mean Teacher (PSMT) method, which is capable of effectively updating the critical parameters within the MT network under domain shifts. First, we introduce a selective distillation mechanism in the student model, which utilizes past knowledge to regularize novel knowledge, thereby mitigating the impact of error accumulation. Second, to avoid catastrophic forgetting, in the teacher model, we create a mask through Fisher information to selectively update parameters via exponential moving average, with preservation measures applied to crucial parameters. Extensive experimental results verify that PSMT outperforms state-of-the-art methods across multiple benchmark datasets. Our code is available at \url{https://github.com/JiaxuTian/PSMT}.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑