Fused Partial Gromov-Wasserstein for Structured Objects

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Bai, Yikun, Wang, Shuang, Tran, Huy, Du, Hengrong, Wang, Juexin, Kolouri, Soheil
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909807569534976
author Bai, Yikun
Wang, Shuang
Tran, Huy
Du, Hengrong
Wang, Juexin
Kolouri, Soheil
author_facet Bai, Yikun
Wang, Shuang
Tran, Huy
Du, Hengrong
Wang, Juexin
Kolouri, Soheil
contents Structured data, such as graphs, is vital in machine learning due to its capacity to capture complex relationships and interactions. In recent years, the Fused Gromov-Wasserstein (FGW) distance has attracted growing interest because it enables the comparison of structured data by jointly accounting for feature similarity and geometric structure. However, as a variant of optimal transport (OT), classical FGW assumes an equal mass constraint on the compared data. In this work, we relax this mass constraint and propose the Fused Partial Gromov-Wasserstein (FPGW) framework, which extends FGW to accommodate unbalanced data. Theoretically, we establish the relationship between FPGW and FGW and prove the metric properties of FPGW. Numerically, we introduce Frank-Wolfe solvers and Sinkhorn solvers for the proposed FPGW framework. Finally, we evaluate the FPGW distance through graph matching, graph classification and graph clustering experiments, demonstrating its robust performance.
format Preprint
id arxiv_https___arxiv_org_abs_2502_09934
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Fused Partial Gromov-Wasserstein for Structured Objects
Bai, Yikun
Wang, Shuang
Tran, Huy
Du, Hengrong
Wang, Juexin
Kolouri, Soheil
Machine Learning
Structured data, such as graphs, is vital in machine learning due to its capacity to capture complex relationships and interactions. In recent years, the Fused Gromov-Wasserstein (FGW) distance has attracted growing interest because it enables the comparison of structured data by jointly accounting for feature similarity and geometric structure. However, as a variant of optimal transport (OT), classical FGW assumes an equal mass constraint on the compared data. In this work, we relax this mass constraint and propose the Fused Partial Gromov-Wasserstein (FPGW) framework, which extends FGW to accommodate unbalanced data. Theoretically, we establish the relationship between FPGW and FGW and prove the metric properties of FPGW. Numerically, we introduce Frank-Wolfe solvers and Sinkhorn solvers for the proposed FPGW framework. Finally, we evaluate the FPGW distance through graph matching, graph classification and graph clustering experiments, demonstrating its robust performance.
title Fused Partial Gromov-Wasserstein for Structured Objects
topic Machine Learning
url https://arxiv.org/abs/2502.09934