Deep Message Passing on Sets
| Autor: | Junier B. Oliva, Marc Niethammer, Yifeng Shi |
|---|---|
| Rok vydání: | 2020 |
| Předmět: |
FOS: Computer and information sciences
Computer Science - Machine Learning Theoretical computer science Computer science Connection (vector bundle) Message passing Statistical relational learning Machine Learning (stat.ML) General Medicine Graph Machine Learning (cs.LG) Set (abstract data type) Kernel (image processing) Statistics - Machine Learning Interpretability Block (data storage) |
| Zdroj: | AAAI |
| ISSN: | 2374-3468 2159-5399 |
| DOI: | 10.1609/aaai.v34i04.6031 |
| Popis: | Modern methods for learning over graph input data have shown the fruitfulness of accounting for relationships among elements in a collection. However, most methods that learn over set input data use only rudimentary approaches to exploit intra-collection relationships. In this work we introduce Deep Message Passing on Sets (DMPS), a novel method that incorporates relational learning for sets. DMPS not only connects learning on graphs with learning on sets via deep kernel learning, but it also bridges message passing on sets and traditional diffusion dynamics commonly used in denoising models. Based on these connections, we develop two new blocks for relational learning on sets: the set-denoising block and the set-residual block. The former is motivated by the connection between message passing on general graphs and diffusion-based denoising models, whereas the latter is inspired by the well-known residual network. In addition to demonstrating the interpretability of our model by learning the true underlying relational structure experimentally, we also show the effectiveness of our approach on both synthetic and real-world datasets by achieving results that are competitive with or outperform the state-of-the-art. Comment: 11 pages, 8 figures |
| Databáze: | OpenAIRE |
| Externí odkaz: |
|