Patch-wise Graph Contrastive Learning for Image Translation

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Jung, Chanyong, Kwon, Gihyun, Ye, Jong Chul
Natura: Preprint
Pubblicazione: 2023
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866910336344391680
author Jung, Chanyong
Kwon, Gihyun
Ye, Jong Chul
author_facet Jung, Chanyong
Kwon, Gihyun
Ye, Jong Chul
contents Recently, patch-wise contrastive learning is drawing attention for the image translation by exploring the semantic correspondence between the input and output images. To further explore the patch-wise topology for high-level semantic understanding, here we exploit the graph neural network to capture the topology-aware features. Specifically, we construct the graph based on the patch-wise similarity from a pretrained encoder, whose adjacency matrix is shared to enhance the consistency of patch-wise relation between the input and the output. Then, we obtain the node feature from the graph neural network, and enhance the correspondence between the nodes by increasing mutual information using the contrastive loss. In order to capture the hierarchical semantic structure, we further propose the graph pooling. Experimental results demonstrate the state-of-art results for the image translation thanks to the semantic encoding by the constructed graphs.
format Preprint
id arxiv_https___arxiv_org_abs_2312_08223
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Patch-wise Graph Contrastive Learning for Image Translation
Jung, Chanyong
Kwon, Gihyun
Ye, Jong Chul
Computer Vision and Pattern Recognition
Recently, patch-wise contrastive learning is drawing attention for the image translation by exploring the semantic correspondence between the input and output images. To further explore the patch-wise topology for high-level semantic understanding, here we exploit the graph neural network to capture the topology-aware features. Specifically, we construct the graph based on the patch-wise similarity from a pretrained encoder, whose adjacency matrix is shared to enhance the consistency of patch-wise relation between the input and the output. Then, we obtain the node feature from the graph neural network, and enhance the correspondence between the nodes by increasing mutual information using the contrastive loss. In order to capture the hierarchical semantic structure, we further propose the graph pooling. Experimental results demonstrate the state-of-art results for the image translation thanks to the semantic encoding by the constructed graphs.
title Patch-wise Graph Contrastive Learning for Image Translation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2312.08223