Dynamic graph message passing networks

Modelling long-range dependencies is critical for scene understanding tasks in computer vision. Although CNNs have excelled in many vision tasks, they are still limited in capturing long-range structured relationships as they typically consist of layers of local kernels. A fully-connected graph is b...

Full description

Bibliographic Details
Main Authors: Zhang, L, Xu, D, Arnab, A, Torr, PHS
Format: Conference item
Language:English
Published: IEEE 2020
_version_ 1826282152976711680
author Zhang, L
Xu, D
Arnab, A
Torr, PHS
author_facet Zhang, L
Xu, D
Arnab, A
Torr, PHS
author_sort Zhang, L
collection OXFORD
description Modelling long-range dependencies is critical for scene understanding tasks in computer vision. Although CNNs have excelled in many vision tasks, they are still limited in capturing long-range structured relationships as they typically consist of layers of local kernels. A fully-connected graph is beneficial for such modelling, however, its computational overhead is prohibitive. We propose a dynamic graph message passing network, that significantly reduces the computational complexity compared to related works modelling a fully-connected graph. This is achieved by adaptively sampling nodes in the graph, conditioned on the input, for message passing. Based on the sampled nodes, we dynamically predict node-dependent filter weights and the affinity matrix for propagating information between them. Using this model, we show significant improvements with respect to strong, state-of-the-art baselines on three different tasks and backbone architectures. Our approach also outperforms fully-connected graphs while using substantially fewer floating-point operations and parameters.
first_indexed 2024-03-07T00:39:32Z
format Conference item
id oxford-uuid:828ecf0f-d1cf-46b5-bf31-3c7fe38b17db
institution University of Oxford
language English
last_indexed 2024-03-07T00:39:32Z
publishDate 2020
publisher IEEE
record_format dspace
spelling oxford-uuid:828ecf0f-d1cf-46b5-bf31-3c7fe38b17db2022-03-26T21:38:24ZDynamic graph message passing networksConference itemhttp://purl.org/coar/resource_type/c_5794uuid:828ecf0f-d1cf-46b5-bf31-3c7fe38b17dbEnglishSymplectic ElementsIEEE2020Zhang, LXu, DArnab, ATorr, PHSModelling long-range dependencies is critical for scene understanding tasks in computer vision. Although CNNs have excelled in many vision tasks, they are still limited in capturing long-range structured relationships as they typically consist of layers of local kernels. A fully-connected graph is beneficial for such modelling, however, its computational overhead is prohibitive. We propose a dynamic graph message passing network, that significantly reduces the computational complexity compared to related works modelling a fully-connected graph. This is achieved by adaptively sampling nodes in the graph, conditioned on the input, for message passing. Based on the sampled nodes, we dynamically predict node-dependent filter weights and the affinity matrix for propagating information between them. Using this model, we show significant improvements with respect to strong, state-of-the-art baselines on three different tasks and backbone architectures. Our approach also outperforms fully-connected graphs while using substantially fewer floating-point operations and parameters.
spellingShingle Zhang, L
Xu, D
Arnab, A
Torr, PHS
Dynamic graph message passing networks
title Dynamic graph message passing networks
title_full Dynamic graph message passing networks
title_fullStr Dynamic graph message passing networks
title_full_unstemmed Dynamic graph message passing networks
title_short Dynamic graph message passing networks
title_sort dynamic graph message passing networks
work_keys_str_mv AT zhangl dynamicgraphmessagepassingnetworks
AT xud dynamicgraphmessagepassingnetworks
AT arnaba dynamicgraphmessagepassingnetworks
AT torrphs dynamicgraphmessagepassingnetworks