arrow
Return

Positional diffusion: Graph-based diffusion models for set ordering

delete2024-10-01
delete0
delete
OA
AI
F
Francesco Giuliari
G
Gianluca Scarpellini
S
Stefano Fiorini
S
Stuart James
P
Pietro Morerio *
Y
Yiming Wang
A
Alessio Del Bue
DOI:10.1016/j.patrec.2024.10.010delete
deleteOriginal
deleteShare
deleteSave
View PDF
Abstract

Abstract

En 中文
Positional reasoning is the process of ordering an unsorted set of parts into a consistent structure. To address this problem, we present Positional Diffusion, a plug-and-play graph formulation with Diffusion Probabilistic Models. Using a diffusion process, we add Gaussian noise to the set elements' position and map them to a random position in a continuous space. Positional Diffusion learns to reverse the noising process and recover the original positions through an Attention-based Graph Neural Network. To evaluate our method, we conduct extensive experiments on three different tasks and seven datasets, comparing our approach against the stateof-the-art methods for visual puzzle-solving, sentence ordering, and room arrangement, demonstrating that our method outperforms long-lasting research on puzzle solving with up to +17% compared to the secondbest deep learning method, and performs on par against the state-of-the-art methods on sentence ordering and room rearrangement. Our work highlights the suitability of diffusion models for ordering problems and proposes a novel formulation and method for solving various ordering tasks. We release our code at https://github.com/IIT-PAVIS/Positional_Diffusion.
Keywords:
Graph neural networks
Diffusion probabilistic models
Positional reasoning
AI Summary

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

Pattern Recognition Letters cover
Pattern Recognition Letters
IF:
3.3
Papers:
7.8K
Citations:
1.6W

Organization

D
Durham University
Scholars:
1.3W
Papers: 1.5W
Citations: 2.1W
I
istituto italiano di tecnologia - iit
Scholars:
9.1K
Papers: 6.9K
Citations: 8