Saved in:
Bibliographic Details
Main Authors: Ma, Shaozhen, Huang, Wei, Wang, Hanchen, Wen, Dong, Zhang, Wenjie
Format: Preprint
Published: 2026
Subjects:
Online Access:https://arxiv.org/abs/2605.05689
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911654920323072
author Ma, Shaozhen
Huang, Wei
Wang, Hanchen
Wen, Dong
Zhang, Wenjie
author_facet Ma, Shaozhen
Huang, Wei
Wang, Hanchen
Wen, Dong
Zhang, Wenjie
contents Conditional generative models, particularly diffusion-based methods, have recently been applied to graph prediction by modeling the target as a conditional distribution given the input graph, yielding competitive results compared to deterministic predictor. However, existing diffusion-based prediction methods typically require expensive iterative denoising at inference and often suffer from unstable sampling, which motivates recent efforts to reduce inference denoising steps and enable stable sampling via techniques such as consistency training. Despite this progress, we find that existing consistency training methods for graph prediction could potentially fall into a shortcut solution: the model may attempt to satisfy the self-consistency constraint by ignoring the noisy target (i.e., assigning it negligible weight), ultimately collapsing into a purely deterministic predictor. To mitigate such shortcut solution, we propose GCCM, a graph contrastive consistency model that goes beyond isolated pairwise matching between the same target at different noise levels by introducing negative pairs into a contrastive consistency objective. This adds an additional separation requirement, making the shortcut solution no longer trivially sufficient to satisfy the proposed objective. Moreover, we apply feature perturbation to the input node/edge features to break identical conditioning on the input graph, so that the shortcut no longer yields the same predictions across noise levels and becomes less attractive. Extensive experiments on benchmark datasets demonstrate that GCCM mitigates the shortcut solution and yields consistent performance improvements in graph prediction compared to deterministic predictors.
format Preprint
id arxiv_https___arxiv_org_abs_2605_05689
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle GCCM: Enhancing Generative Graph Prediction via Contrastive Consistency Model
Ma, Shaozhen
Huang, Wei
Wang, Hanchen
Wen, Dong
Zhang, Wenjie
Artificial Intelligence
Conditional generative models, particularly diffusion-based methods, have recently been applied to graph prediction by modeling the target as a conditional distribution given the input graph, yielding competitive results compared to deterministic predictor. However, existing diffusion-based prediction methods typically require expensive iterative denoising at inference and often suffer from unstable sampling, which motivates recent efforts to reduce inference denoising steps and enable stable sampling via techniques such as consistency training. Despite this progress, we find that existing consistency training methods for graph prediction could potentially fall into a shortcut solution: the model may attempt to satisfy the self-consistency constraint by ignoring the noisy target (i.e., assigning it negligible weight), ultimately collapsing into a purely deterministic predictor. To mitigate such shortcut solution, we propose GCCM, a graph contrastive consistency model that goes beyond isolated pairwise matching between the same target at different noise levels by introducing negative pairs into a contrastive consistency objective. This adds an additional separation requirement, making the shortcut solution no longer trivially sufficient to satisfy the proposed objective. Moreover, we apply feature perturbation to the input node/edge features to break identical conditioning on the input graph, so that the shortcut no longer yields the same predictions across noise levels and becomes less attractive. Extensive experiments on benchmark datasets demonstrate that GCCM mitigates the shortcut solution and yields consistent performance improvements in graph prediction compared to deterministic predictors.
title GCCM: Enhancing Generative Graph Prediction via Contrastive Consistency Model
topic Artificial Intelligence
url https://arxiv.org/abs/2605.05689