arXiv:2605.01468cs.CVcs.AI2026-05中稿 · CVPR被引 1

通过生成边界附近样本,提升长尾数据中尾部类别的分类性能。

Decision Boundary-aware Generation for Long-tailed Learning

论文配图:Decision Boundary-aware Generation for Long-tailed Learning
图 1 · 摘自论文原文
  • 基于决策边界感知生成,聚焦尾部类别附近的样本。
  • 在多个基准上显著提升尾部类别准确率,减少类别间混淆。
  • 适合长尾学习、数据增强及分类器优化研究者使用。

长尾数据会使决策边界偏向头部类别,降低尾部类别的准确率。基于扩散的生成增强方法通过生成额外数据缓解此问题,而头到尾迁移进一步减轻了由长尾数据集继承的生成器偏差。然而,我们发现头到尾迁移虽有助于平衡分类器的决策空间,却也引发潜在的非局部特征混杂,导致类别间特征纠缠、决策边界重叠和尾部类别分布偏移。为此,我们首先识别出边界模糊性问题,并提出决策边界感知生成(DBG)框架,通过生成具有信息量的近边界样本,促进近边界表征学习。整体上,DBG在重平衡长尾数据的同时,使决策空间更具可分性。在多个标准长尾基准测试中,DBG持续提升尾部类别与整体准确率,且类别间重叠更少。DBG代码已公开于 https://github.com/keepdigitalabc-svg/DBG。

原文摘要 · Abstract (English)

Long-tailed data bias decision boundaries toward head classes and degrade tail class accuracy. Diffusion-based generative augmentation address this problem by generating additional data, while head-to-tail transfer further mitigate the generator bias inherit from long-tailed dataset. However, we show that while head-to-tail transfer helps balance the decision space of the classifier, it also induces latent non-local feature mixing that entangles inter-class features, causing decision boundary overlap and tail class distribution shift. To address this, we first identify the problem of boundary ambiguity and then propose Decision Boundary-aware Generation (DBG) framework, which promotes near-boundary representation learning by generating informative near-boundary samples. Overall, DBG rebalances the long-tailed dataset while yielding more separable decision space for long-tailed learning. Across standard long-tailed benchmarks, DBG consistently improves tail class and overall accuracy with less inter-class overlap. The code of DBG is available at https://github.com/keepdigitalabc-svg/DBG.

长尾学习生成增强决策边界扩散模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。