提出新方法实现高效多对多匹配,避免退化为一对一匹配。
Many-to-Many Matching via Sparsity Controlled Optimal Transport
- 通过行/列匹配预算约束显式控制多对多关系
- 采用变形q-熵正则化使匹配更贴近预算上限
- 适用于需要灵活匹配的场景,如数据挖掘与聚类
多对多匹配旨在将一个集合中的多个点与另一集合中的多个点匹配,是众多数据挖掘问题的基础。该问题可自然建模为最优传输(OT)框架。然而,现有OT方法要么无法实现真正的多对多匹配,要么需精细调节正则化参数才能获得满意结果。本文提出一种新型多对多匹配方法,显式编码多对多约束,防止退化为一对一匹配。方法包含两个组件:一是对运输计划每行每列施加匹配预算约束,规定每个点最多可匹配的点数;二是采用变形q-熵正则化,鼓励每个点尽可能达到其匹配预算。尽管变形q-熵最初用于稀疏化运输计划,本文将其用于避免退化为一对一匹配。通过惩罚算法优化目标函数,该方法计算高效且理论上保证收敛。在多种任务上的实验表明,所提方法能生成有意义的多对多匹配,性能优异。
原文摘要 · Abstract (English)
Many-to-many matching seeks to match multiple points in one set and multiple points in another set, which is a basis for a wide range of data mining problems. It can be naturally recast in the framework of Optimal Transport (OT). However, existing OT methods either lack the ability to accomplish many-to-many matching or necessitate careful tuning of a regularization parameter to achieve satisfactory results. This paper proposes a novel many-to-many matching method to explicitly encode many-to-many constraints while preventing the degeneration into one-to-one matching. The proposed method consists of the following two components. The first component is the matching budget constraints on each row and column of a transport plan, which specify how many points can be matched to a point at most. The second component is the deformed $q$-entropy regularization, which encourages a point to meet the matching budget maximally. While the deformed $q$-entropy was initially proposed to sparsify a transport plan, we employ it to avoid the degeneration into one-to-one matching. We optimize the objective via a penalty algorithm, which is efficient and theoretically guaranteed to converge. Experimental results on various tasks demonstrate that the proposed method achieves good performance by gleaning meaningful many-to-many matchings.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。