arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2210.01640cs.LG
收藏

测试时训练的混淆

Mixup for Test-Time Training

Hong Kong Baptist University(香港浸会大学);Nanyang Technological University(南洋理工大学)

Bochao Zhang, Rui Shao, Jingda Du, PC Yuen

摘要:

测试时训练为解决领域迁移问题提供了一条新途径。在该框架中,在训练阶段和测试阶段之间插入一个测试时训练阶段。在测试时训练阶段,通常用测试样本更新模型的部分。然后将更新后的模型用于测试阶段。然而,利用测试样本进行测试时训练具有一些局限性。首先,它会导致过拟合的测试时间过程,从而损害性能的主要任务。此外,在不改变其他部分的情况下更新模型的一部分将引起失配问题。因此,很难在主要任务上做得更好。针对上述问题,本文提出了在测试时训练中使用混合训练(MixTTT)来控制模型参数的变化,同时完成测试时的训练过程。我们从理论上证明了它的贡献,在缓解更新的部分和静态部分的主要任务作为一个特定的正则化效果测试时训练的不匹配问题。MixTTT可以用作一般基于测试时训练的方法中的附加模块,以进一步提高其性能。实验结果表明了该方法的有效性。

英文摘要:

Test-time training provides a new approach solving the problem of domain shift. In its framework, a test-time training phase is inserted between training phase and test phase. During test-time training phase, usually parts of the model are updated with test sample(s). Then the updated model will be used in the test phase. However, utilizing test samples for test-time training has some limitations. Firstly, it will lead to overfitting to the test-time procedure thus hurt the performance on the main task. Besides, updating part of the model without changing other parts will induce a mismatch problem. Thus it is hard to perform better on the main task. To relieve above problems, we propose to use mixup in test-time training (MixTTT) which controls the change of model's parameters as well as completing the test-time procedure. We theoretically show its contribution in alleviating the mismatch problem of updated part and static part for the main task as a specific regularization effect for test-time training. MixTTT can be used as an add-on module in general test-time training based methods to further improve their performance. Experimental results show the effectiveness of our method.

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

↑