中文

TdAttenMix:基于自上而下注意力引导的 Mixup 方法

计算机视觉与模式识别 2025-01-28 v1 人工智能

摘要

CutMix 是一种数据增强策略,通过裁剪和粘贴图像块来实现 mixup。现有方法选择随机区域或显著区域,往往与标签不一致,从而误导训练模型。据我们所知,我们首次将人类注视 Integrate 到 CutMix 中引导。由于人类注意力由高层识别和低层线索驱动,我们提出一种可控的自上而下注意力引导模块,以获取平衡自上而下与自下而上的注意力的通用人工注意力。所提出的TdAttenMix选择图像块并调整标签混合比例,聚焦于与当前标签相关的区域。实验结果表明,TdAttenMix 在八个不同基准测试中超越现有SOTA mixup方法。此外,我们引入一种基于人类注视的新指标,用于调查图像-标签不一致问题。项目页面:\url{https://github.com/morning12138/TdAttenMix}

关键词

引用

@article{arxiv.2501.15409,
  title  = {TdAttenMix: Top-Down Attention Guided Mixup},
  author = {Zhiming Wang and Lin Gu and Feng Lu},
  journal= {arXiv preprint arXiv:2501.15409},
  year   = {2025}
}