Enhancing Unsupervised Graph Few-shot Learning via Set Functions and Optimal Transport

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Liu, Yonghao, Giunchiglia, Fausto, Li, Ximing, Huang, Lan, Feng, Xiaoyue, Guan, Renchu
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866913643145199616
author Liu, Yonghao
Giunchiglia, Fausto
Li, Ximing
Huang, Lan
Feng, Xiaoyue
Guan, Renchu
author_facet Liu, Yonghao
Giunchiglia, Fausto
Li, Ximing
Huang, Lan
Feng, Xiaoyue
Guan, Renchu
contents Graph few-shot learning has garnered significant attention for its ability to rapidly adapt to downstream tasks with limited labeled data, sparking considerable interest among researchers. Recent advancements in graph few-shot learning models have exhibited superior performance across diverse applications. Despite their successes, several limitations still exist. First, existing models in the meta-training phase predominantly focus on instance-level features within tasks, neglecting crucial set-level features essential for distinguishing between different categories. Second, these models often utilize query sets directly on classifiers trained with support sets containing only a few labeled examples, overlooking potential distribution shifts between these sets and leading to suboptimal performance. Finally, previous models typically require necessitate abundant labeled data from base classes to extract transferable knowledge, which is typically infeasible in real-world scenarios. To address these issues, we propose a novel model named STAR, which leverages Set funcTions and optimAl tRansport for enhancing unsupervised graph few-shot learning. Specifically, STAR utilizes expressive set functions to obtain set-level features in an unsupervised manner and employs optimal transport principles to align the distributions of support and query sets, thereby mitigating distribution shift effects. Theoretical analysis demonstrates that STAR can capture more task-relevant information and enhance generalization capabilities. Empirically, extensive experiments across multiple datasets validate the effectiveness of STAR. Our code can be found here.
format Preprint
id arxiv_https___arxiv_org_abs_2501_05635
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Enhancing Unsupervised Graph Few-shot Learning via Set Functions and Optimal Transport
Liu, Yonghao
Giunchiglia, Fausto
Li, Ximing
Huang, Lan
Feng, Xiaoyue
Guan, Renchu
Machine Learning
Graph few-shot learning has garnered significant attention for its ability to rapidly adapt to downstream tasks with limited labeled data, sparking considerable interest among researchers. Recent advancements in graph few-shot learning models have exhibited superior performance across diverse applications. Despite their successes, several limitations still exist. First, existing models in the meta-training phase predominantly focus on instance-level features within tasks, neglecting crucial set-level features essential for distinguishing between different categories. Second, these models often utilize query sets directly on classifiers trained with support sets containing only a few labeled examples, overlooking potential distribution shifts between these sets and leading to suboptimal performance. Finally, previous models typically require necessitate abundant labeled data from base classes to extract transferable knowledge, which is typically infeasible in real-world scenarios. To address these issues, we propose a novel model named STAR, which leverages Set funcTions and optimAl tRansport for enhancing unsupervised graph few-shot learning. Specifically, STAR utilizes expressive set functions to obtain set-level features in an unsupervised manner and employs optimal transport principles to align the distributions of support and query sets, thereby mitigating distribution shift effects. Theoretical analysis demonstrates that STAR can capture more task-relevant information and enhance generalization capabilities. Empirically, extensive experiments across multiple datasets validate the effectiveness of STAR. Our code can be found here.
title Enhancing Unsupervised Graph Few-shot Learning via Set Functions and Optimal Transport
topic Machine Learning
url https://arxiv.org/abs/2501.05635