Mar, 2024

通过内存网络在连续检测中防止灾难性遗忘

TL;DR现代预训练架构在持续对新任务进行微调时难以保留先前的信息。本文引入了一种基于内存的检测变换器架构,以适应预训练的 DETR 式检测器的新任务,并保留以前任务中的知识。我们提出一种用于从内存单元中高效检索信息的新颖的局部查询函数,旨在最小化遗忘。此外,我们在持续检测中确定了一项称为背景剥夺的根本挑战。当先前任务的对象类别在未来任务中重新出现时,可能没有标签,可能会被隐含地视为背景。这是持续检测或分割中不可避免的问题。引入的持续优化技术有效地解决了这一挑战。最后,我们评估了我们提出的系统在持续检测基准上的性能,并证明我们的方法在 MS-COCO 和 PASCAL-VOC 的持续检测任务上超过了现有技术水平,实现了 5-7% 的改进。