arXiv:2509.25713cs.LGcs.CV2025-09被引 1

无标签情况下提升长尾数据生成质量,解决多数类主导问题

Reweighted Flow Matching via Unbalanced OT for Label-free Long-tailed Generation

  • 用不平衡最优传输构建条件向量场,无需类别标签
  • 通过密度比重加权,使少数类生成质量显著提升
  • 适合长尾分布生成任务,尤其在无标签场景下表现优异

流匹配近期成为连续时间生成建模的强大框架。但在处理长尾分布时,标准流匹配存在多数类偏差,导致少数类生成质量低且无法匹配真实类别比例。本文提出无标签不平衡最优传输重加权流匹配(UOT-RFM),一种在类别不平衡分布下进行生成建模的新方法,无需任何类别标签信息。该方法利用小批量不平衡最优传输(UOT)构建条件向量场,并通过基于逆重加权策略缓解多数类偏差。重加权依赖于一个无标签的多数度量,定义为目标分布与UOT边际的密度比,该度量基于数据几何结构量化多数程度,无需类别标签。将此度量引入训练目标后,UOT-RFM在理论上实现一阶修正(k=1),实证上通过高阶修正(k>1)显著改善尾部类别生成效果。模型在长尾基准测试中优于现有流匹配基线,同时在平衡数据集上保持竞争力。

原文摘要 · Abstract (English)

Flow matching has recently emerged as a powerful framework for continuous-time generative modeling. However, when applied to long-tailed distributions, standard flow matching suffers from majority bias, producing minority modes with low fidelity and failing to match the true class proportions. In this work, we propose Unbalanced Optimal Transport Reweighted Flow Matching (UOT-RFM), a novel framework for generative modeling under class-imbalanced (long-tailed) distributions that operates without any class label information. Our method constructs the conditional vector field using mini-batch Unbalanced Optimal Transport (UOT) and mitigates majority bias through a principled inverse reweighting strategy. The reweighting relies on a label-free majority score, defined as the density ratio between the target distribution and the UOT marginal. This score quantifies the degree of majority based on the geometric structure of the data, without requiring class labels. By incorporating this score into the training objective, UOT-RFM theoretically recovers the target distribution with first-order correction ($k=1$) and empirically improves tail-class generation through higher-order corrections ($k > 1$). Our model outperforms existing flow matching baselines on long-tailed benchmarks, while maintaining competitive performance on balanced datasets.

生成模型长尾分布无监督最优传输

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