Papers › Rep the Set: Neural Networks for Learning Set Representations

Rep the Set: Neural Networks for Learning Set Representations

3 Apr 2019arXiv:1904.01962archive 2025-07-28

Konstantinos Skianis, Giannis Nikolentzos, Stratis Limnios, Michalis Vazirgiannis

In several domains, data objects can be decomposed into sets of simpler objects. It is then natural to represent each object as the set of its components or parts. Many conventional machine learning algorithms are unable to process this kind of representations, since sets may vary in cardinality and elements lack a meaningful ordering. In this paper, we present a new neural network architecture, called RepSet, that can handle examples that are represented as sets of vectors. The proposed model computes the correspondences between an input set and some hidden sets by solving a series of network flow problems. This representation is then fed to a standard neural network architecture to produce the output. The architecture allows end-to-end gradient-based learning. We demonstrate RepSet on classification tasks, including text categorization, and graph classification, and we show that the proposed neural network achieves performance better or comparable to state-of-the-art algorithms.

PaperPDFCode

In Syntology Open this paper in Syntology's Atlas, the map of the papers in Syntology's graph and their citations.

Code

giannisnik/repset officialmentioned in papermentioned on GitHubpytorch report

Repository list and official/mentioned flags are the archive's, frozen 2025-07-28. Reachability, where shown, is from one Syntology probe window (2026-09-16 to 2026-09-18); repositories not probed show nothing. GitHub stars are not tracked.

Code Syntology ran Syntology

Not run by Syntology. Nothing on this page verifies that the listed code works.

Tasks

General ClassificationGraph ClassificationText Categorization

Results from the paper archive 2025-07-28

TaskDatasetModelMetricValueRank at snapshotLeaderboardReport
Document Classification Amazon ApproxRepSet Accuracy 94.31 #1 of 3 Archive leaderboard report
Document Classification BBCSport ApproxRepSet Accuracy 95.73 #3 of 4 Archive leaderboard report
Document Classification Classic ApproxRepSet Accuracy 96.24 #2 of 2 Archive leaderboard report
Document Classification Recipe ApproxRepSet Accuracy 59.06 #1 of 2 Archive leaderboard report
Document Classification Reuters-21578 ApproxRepSet Accuracy 97.17 #1 of 8 Archive leaderboard report
Document Classification Twitter ApproxRepSet Accuracy 72.6 #1 of 3 Archive leaderboard report
Graph Classification IMDb-B ApproxRepSet Accuracy 71.46% #43 of 51 Archive leaderboard report
Graph Classification IMDb-M ApproxRepSet Accuracy 48.92% #30 of 36 Archive leaderboard report
Graph Classification MUTAG ApproxRepSet Accuracy 86.33% #56 of 74 Archive leaderboard report
Graph Classification PROTEINS ApproxRepSet Accuracy 70.74% #99 of 103 Archive leaderboard report
Graph Classification REDDIT-B ApproxRepSet Accuracy 80.3 #11 of 12 Archive leaderboard report
Text Classification 20NEWS ApproxRepSet Accuracy 76.18 #14 of 16 Archive leaderboard report
Text Classification Ohsumed ApproxRepSet Accuracy 64.06 #8 of 10 Archive leaderboard report

Ranks are positions in the archive's leaderboards as they stood at the 2025-07-28 snapshot. Results published since then are not among these rows, so a rank here is not a current standing.

Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections