REVIEW 1 major objections 4 minor 31 references
On the (Non) Injectivity of Piecewise Linear Janossy Pooling
T0 review · 1 major / 4 minor · reviewed 2026-08-07 · deepseek-v4-flash
Pith's one-line read No piecewise-linear Janossy pooling is injective
desk verdict Settles the k>=2 Janossy pooling expressivity question with a genuinely new polytope-partition lemma; a solid paper with only minor presentation gaps. read the letter →
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
The reading
What carries the argument
The central object is the polytope-covering lemma (Theorem 3.2) that drives the negative result: for any finite partition of $\mathbb{R}^k$ into convex polytopes, there exists a strictly decreasing vector $w$ of length $n$ whose every $k$-dimensional order-preserving subvector falls inside a single polytope $P_0$. This reduces the non-injectivity of Janossy pooling to a counting argument: around such a $w$ the pooling map is affine, and the system of $k$ equations expressing the sum of perturbations over all $k$-subtuples has a nonzero solution in $\mathbb{R}^n$ because $n > k$. For the positive result, the key mechanism is a hypercube tessellation of $\mathbb{R}^d$ with side length $s = R(D)/2$, where $R(D)$ is the minimal pairwise distance among points of any multiset in the compact domain $D$; each hypercube carries a CPwL indicator of membership plus a CPwL coordinate feature that equals the point's location inside the cube and interpolates to zero at the margin boundary, so the pooled sum reveals each point uniquely.
What would settle it
Try to construct a CPwL function $f: \mathbb{R}^2 \to \mathbb{R}$ on a partition of the unit square into four quadrants and show that its 2-ary Janossy pooling on triples is injective; the theorem predicts such injectivity is impossible, so any explicit injective example would refute it. More directly, a computational search over polytope partitions of $\mathbb{R}^2$ for one that defeats Theorem 3.2—no strictly decreasing triple $w$ whose three ordered pairs share a polytope—would test the proof's engine.
Extended reading notes
Core claim
The paper's central claim is Theorem 3.1: let $C$ be any subset of $\mathbb{R}^d$ containing a line segment, let $f:(\mathbb{R}^d)^k \to \mathbb{R}^m$ be continuous piecewise linear, and let $n > k$. Then the $k$-ary Janossy pooling of $f$, defined by averaging $f$ over all ordered $k$-tuples of the input, is not injective on multisets of size $n$ from $C$. The proof hinges on a new geometric lemma (Theorem 3.2): for every finite partition of $\mathbb{R}^k$ into polytopes, there is a strictly decreasing vector $w=(w_1,\ldots,w_n)$ in $(0,1)^n$ and a single polytope $P_0$ such that every order-preserving $k$-subvector of $w$ lies in the interior of $P_0$. This makes the pooled output affine on a neighborhood of $w$, and since summing over $k$-element subsets leaves $k$ linear equations in $n > k$ unknowns, a nonzero perturbation preserves the pooled vector while changing the multiset. The companion positive result, Theorem 4.3, constructs, on a compact domain of multisets with $n$ distinct points, a CPwL function $f$ such that 1-ary pooling is injective and bi-Lipschitz; the construction tessellates $\mathbb{R}^d$ into hypercubes of side length $R(D)/2$ and attaches to each hypercube an indicator feature and a coordinate feature, so that the pooled output reveals every point of the multiset.
Load-bearing premise
The non-injectivity theorem depends on $f$ being continuous piecewise linear, so that its domain splits into finitely many polytopes on each of which $f$ is affine; if $f$ were smooth, injective Janossy pooling would be possible, and the theorem would no longer hold.
Editorial extensions
If this is right
- Set transformers and other $k$-ary pooling models with ReLU activations cannot faithfully represent multisets of size $n > k$ in general: for every such model there exist distinct multisets that its output cannot tell apart.
- The negative result provides a theoretical justification for the higher cost of sorting-based and quantile-based bi-Lipschitz encoders: no simple piecewise-linear pooling can match their injectivity guarantee.
- For datasets with well-separated points, the positive result shows that ordinary deep sets are sufficient, both injective and bi-Lipschitz, with the embedding dimension determined by the minimal separation $R(D)$.
- The construction's dimension scales as $(1/R(D))^d$, so the mathematical guarantee degrades as point clouds approach each other; near-duplicate points are exactly where specialized bi-Lipschitz layers are needed.
- The paper's conjecture that $k$-ary pooling is injective on multisets with multiplicity at most $k$, if true, would establish a precise expressivity hierarchy across pooling orders.
Reading between the lines
- A concrete, testable consequence of Theorem 3.1: for any ReLU set transformer with $k=2$, one can run an adversarial search over pairs of size-$n$ point sets ($n>2$) and expect to find identical embeddings; such an experiment would empirically demonstrate the collision guaranteed by the theorem.
- The geometric lemma may generalize beyond polytopal partitions: the argument uses only finiteness and convexity, so a similar non-injectivity could hold for any function that is affine on finitely many convex regions, e.g., hinge-type or other CPwL-like activations.
- The QM9 analysis in the paper (minimal within-molecule distances above 0.1 of the diameter for all sampled molecules) suggests a rule of thumb for practitioners: small-molecule point clouds are likely in the 'well-separated' regime where deep sets suffice, while continuous surfaces or dense clouds are not.
- One could implement the positive construction explicitly and measure, on real point clouds, whether the guaranteed injectivity translates into useful Lipschitz constants for downstream tasks such as learning Wasserstein distances.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper studies the injectivity and bi-Lipschitzness of k-ary Janossy pooling when the inner function f is continuous piecewise linear (CPwL). The main negative result (Theorem 3.1) states that if C contains a line segment and n > k, no CPwL k-ary Janossy pooling is injective on multisets of size n in M_n(C); the proof reduces to a combinatorial statement (Theorem 3.2) that every polytope partition of R^k contains a polytope P0 and a strictly ordered vector w in (0,1)^n all of whose ordered k-subvectors lie in the interior of P0, after which a homogeneous linear system in n > k unknowns yields a nontrivial perturbation with equal pooling value. The positive result (Theorem 4.3) constructs, for compact domains D of multisets with distinct elements, an explicit CPwL 1-ary pooling that is injective and bi-Lipschitz with respect to the Wasserstein distance, using a hypercube tessellation of scale R(D)/2. The paper also discusses the degenerate k=n case, the dependence of the embedding dimension on R(D), and a conjecture for bounded-multiplicity domains.
Significance. If correct, the negative result resolves the open question for k >= 2 and shows that injective CPwL multiset representations require either sorting-based maps or the degenerate k = n construction; this strengthens the case for bi-Lipschitz models built on sorting rather than standard Janossy pooling. The proofs are self-contained and rely on no fitted parameters: Theorem 3.1 is proven from the explicit combinatorial lemma in the appendix, and the construction in Theorem 4.3 is explicit. The compact-domain separation assumption is stated and justified in Proposition 4.4, and the limitations and conjecture are clearly identified. The main theorems give falsifiable predictions about expressivity of standard architectures, which is a useful contribution to the theory of permutation-invariant networks.
major comments (1)
- [Section 4, proof of Theorem 4.3, final paragraph] The bi-Lipschitz conclusion is justified by an unspecified auxiliary set D-hat, said to be a finite union of polytopes containing D and containing no multisets with repeated elements. This claim is not established. A finite union of grid hypercubes in the point space R^d is not a set of multisets, and the set of all multisets over such a union contains repeated points; if D-hat is intended as a set of multisets, its structure as a finite union of polytopes requires proof. Please give an explicit construction (for example, in sorted-coordinate space, fix an assignment of the n points to distinct grid cells and verify that each such region is a polytope) and prove that the constructed F is injective on that D-hat before applying [Sverdlov et al., 2024, Lemma 3.4]. As written, the bi-Lipschitz claim in Theorem 4.3 is not fully supported.
minor comments (4)
- [Appendix A.1, Proposition A.3] The sentence concluding that v_k lies in int(P0) because it lies in the interior of [0,1]^k is not a valid inference. The conclusion is nonetheless correct: since |POLY(v_k)| = 1 and the finitely many other closed polytopes do not contain v_k, there is a positive distance from v_k to their union, so a small ball around v_k is contained in P0. Please replace the argument.
- [Section 3, proof of Theorem 3.1] The proof refers to a finite polytope covering of [0,1]^k, while Theorem 3.2 is stated for a partition. Please clarify that the collection is the partition into linear regions of the symmetrized function (or take a common refinement) so that Theorem 3.2 applies directly.
- [Section 4.1] The sentence 'the ratio was not larger than 1/10' contradicts the histogram and the preceding paragraph; it should read 'not smaller than 1/10'.
- [Throughout] There are numerous typos and misspellings, including 'Currenlty', 'inejctivity', 'Lipshitz', 'partion', 'elments', 'Wasserstien', 'montonely', 'represetative', and 'a a finite union'. A careful proofreading pass is needed.
Circularity Check
No significant circularity; Theorems 3.1 and 4.3 are proven constructively from first principles.
full rationale
The derivation chain is self-contained. Theorem 3.1 is reduced to the new combinatorial Theorem 3.2, whose appendix proof uses only polytope-convexity arguments and Rockafellar's Accessibility Lemma; the final perturbation solves k homogeneous linear equations in n > k unknowns, with no fitted parameter. The d > 1 case reduces explicitly to d = 1 via an affine segment, and the compatibility of sorted and unsorted vectors is argued directly. Theorem 4.3 is a constructive construction: f is built from a cube tesselation of side s = R(D)/2 and a triangulated delta-margin, with injectivity demonstrated by the explicit recovery algorithm in Lemmas 4.5 and 4.6; the compactness/separation Proposition 4.4 is proven, not assumed. The only self-citation is [Sverdlov et al. 2024, Lemma 3.4], used solely to upgrade the constructively injective map to bi-Lipschitz; it is a general CPwL lemma whose assumptions do not include the target function, so it does not make the argument circular. There is a presentation gap in Appendix A.3 (membership in int(P0) needs the positive-distance argument from the finite family of other closed polytopes), and a harmless factor-n! slip in the k=n remark of Section 3.1, but these are correctness issues, not circularities. No 'prediction' in the paper is a fitted parameter renamed, and the QM9 separation histogram is illustrative motivation, not a fitted input to the theorems.
Assumptions & free parameters
free parameters (2)
- hypercube side length s =
R(D)/2
- margin width δ =
0 < δ < R(D)/4
assumptions (4)
- standard math Rockafellar's Theorem 6.1 (Accessibility Lemma): if v is in the interior of a convex set P and α>0, then a convex combination with positive weight on v is in the interior of P.
- standard math Goodman-Pach cell decomposition theorem: the δ-margin of a hypercube can be triangulated without new vertices so each simplex has vertices on both the inner boundary and the outer border.
- domain assumption Sverdlov et al. Lemma 3.4: a CPwL injective function on a finite union of compact polytopes is bi-Lipschitz with respect to a CPwL distance with the same zero set.
- standard math Two convex polytopes with empty-interior intersection lie in a common hyperplane (Boyd & Vandenberghe, Convex Optimization, §2.5.2).
Cite this review
Pith. "Pith review of On the (Non) Injectivity of Piecewise Linear Janossy Pooling." pith.science (2026). https://pith.science/paper/LMSITJW6
@misc{pith2026250520150,
author = {Pith},
title = {Pith review of: On the (Non) Injectivity of Piecewise Linear Janossy Pooling},
year = {2026},
howpublished = {\url{https://pith.science/paper/LMSITJW6}},
note = {Machine review of arXiv:2505.20150}
}
read the original abstract
Multiset functions, which are functions that map multisets to vectors, are a fundamental tool in the construction of neural networks for multisets and graphs. To guarantee that the vector representation of the multiset is faithful, it is often desirable to have multiset mappings that are both injective and bi-Lipschitz. Currently, there are several constructions of multiset functions achieving both these guarantees, leading to improved performance in some tasks but often also to higher compute time than standard constructions. Accordingly, it is natural to inquire whether simpler multiset functions achieving the same guarantees are available. In this paper, we make a large step towards giving a negative answer to this question. We consider the family of k-ary Janossy pooling, which includes many of the most popular multiset models, and prove that no piecewise linear Janossy pooling function can be injective. On the positive side, we show that when restricted to multisets without multiplicities, even simple deep-sets models suffice for injectivity and bi-Lipschitzness.
Figures
Reference graph
Works this paper leans on
-
[1]
Fourier sliced-wasserstein embedding for multisets and measures
Tal Amir and Nadav Dym. Fourier sliced-wasserstein embedding for multisets and measures. In The Thirteenth International Conference on Learning Representations, 2025. URL https://openreview.net/forum?id=BcYt84rcKq
work page 2025
-
[2]
Neural injective functions for multisets, measures and graphs via a finite witness theorem
Tal Amir, Steven Gortler, Ilai Avni, Ravina Ravina, and Nadav Dym. Neural injective functions for multisets, measures and graphs via a finite witness theorem. Advances in Neural Information Processing Systems, 36: 0 42516--42551, 2023
work page 2023
-
[3]
G-invariant representations using coorbits: Bi- L ipschitz properties, 2023
Radu Balan and Efstratios Tsoukanis. G-invariant representations using coorbits: Bi- L ipschitz properties, 2023
work page 2023
-
[4]
Permutation invariant representations with applications to graph deep learning
Radu Balan, Naveed Haghani, and Maneesh Singh. Permutation invariant representations with applications to graph deep learning. arXiv preprint arXiv:2203.07546, 2022
arXiv 2022
-
[5]
Stephen Boyd and Lieven Vandenberghe. Convex Optimization. Cambridge University Press, Cambridge, UK, 2004. ISBN 9780521833783
work page 2004
-
[6]
On dimensionality of feature vectors in MPNN s
C\' e sar Bravo, Alexander Kozachinskiy, and Cristobal Rojas. On dimensionality of feature vectors in MPNN s. In Ruslan Salakhutdinov, Zico Kolter, Katherine Heller, Adrian Weller, Nuria Oliver, Jonathan Scarlett, and Felix Berkenkamp, editors, Proceedings of the 41st International Conference on Machine Learning, volume 235 of Proceedings of Machine Learn...
work page 2024
-
[7]
Susanne C. Brenner and L. Ridgway Scott. The Mathematical Theory of Finite Element Methods, volume 15 of Texts in Applied Mathematics. Springer, New York, NY, 3 edition, 2008. ISBN 978-0-387-75933-3. doi:10.1007/978-0-387-75934-0. URL https://doi.org/10.1007/978-0-387-75934-0. Published: 22 December 2007, 3rd edition
-
[8]
Jameson Cahill, Joseph W Iverson, Dustin G Mixon, and Daniel Packer. Group-invariant max filtering. arXiv preprint arXiv:2205.14039, 2022
work page Pith review arXiv 2022
Show all 31 references
-
[9]
Iverson, and Dustin G
Jameson Cahill, Joseph W. Iverson, and Dustin G. Mixon. Towards a bilipschitz invariant theory, 2024
2024
-
[10]
On the h\"older stability of multiset and graph neural networks
Yair Davidson and Nadav Dym. On the h\"older stability of multiset and graph neural networks. In The Thirteenth International Conference on Learning Representations, 2025. URL https://openreview.net/forum?id=P7KIGdgW8S
2025
-
[11]
Low-dimensional invariant embeddings for universal geometric learning
Nadav Dym and Steven J Gortler. Low-dimensional invariant embeddings for universal geometric learning. Foundations of Computational Mathematics, pages 1--41, 2024
2024
-
[12]
Goodman and J \'a nos Pach
Jacob E. Goodman and J \'a nos Pach. Cell decomposition of polytopes by bending. Israel Journal of Mathematics, 64 0 (2): 0 129--138, June 1988. ISSN 1565-8511. doi:10.1007/BF02787218. URL https://doi.org/10.1007/BF02787218
1988 doi
-
[13]
Weisfeiler leman for euclidean equivariant machine learning
Snir Hordan, Tal Amir, and Nadav Dym. Weisfeiler leman for euclidean equivariant machine learning. In Proceedings of the 41st International Conference on Machine Learning, pages 18749--18784, 2024 a
2024
-
[14]
Complete neural networks for complete euclidean graphs
Snir Hordan, Tal Amir, Steven J Gortler, and Nadav Dym. Complete neural networks for complete euclidean graphs. In Proceedings of the AAAI Conference on Artificial Intelligence, volume 38, pages 12482--12490, 2024 b
2024
-
[15]
Set transformer: A framework for attention-based permutation-invariant neural networks
Juho Lee, Yoonho Lee, Jungtaek Kim, Adam Kosiorek, Seungjin Choi, and Yee Whye Teh. Set transformer: A framework for attention-based permutation-invariant neural networks. In International conference on machine learning, pages 3744--3753. PMLR, 2019
2019
-
[16]
Provably powerful graph networks
Haggai Maron, Heli Ben-Hamu, Hadar Serviansky, and Yaron Lipman. Provably powerful graph networks. Advances in neural information processing systems, 32, 2019
2019
-
[17]
Weisfeiler and leman go neural: Higher-order graph neural networks
Christopher Morris, Martin Ritzert, Matthias Fey, William L Hamilton, Jan Eric Lenssen, Gaurav Rattan, and Martin Grohe. Weisfeiler and leman go neural: Higher-order graph neural networks. In Proceedings of the AAAI conference on artificial intelligence, volume 33, pages 4602-...
2019
-
[18]
Murphy, Balasubramaniam Srinivasan, Vinayak Rao, and Bruno Ribeiro
Ryan L. Murphy, Balasubramaniam Srinivasan, Vinayak Rao, and Bruno Ribeiro. Janossy pooling: Learning deep permutation-invariant functions for variable-size inputs. In International Conference on Learning Representations, 2019. URL https://openreview.net/forum?id=BJluy2RcFm
2019
-
[19]
Quantum chemistry structures and properties of 134 kilo molecules
Raghunathan Ramakrishnan, Pavlo O Dral, Matthias Rupp, and O Anatole von Lilienfeld. Quantum chemistry structures and properties of 134 kilo molecules. Scientific Data, 1, 2014
2014
-
[20]
Tyrrell Rockafellar
R. Tyrrell Rockafellar. Convex Analysis. Princeton University Press, Princeton, NJ, 1970. ISBN 978-0691015866
1970
-
[21]
Blum, and Jean-Louis Reymond
Lars Ruddigkeit, Ruud van Deursen, Lorenz C. Blum, and Jean-Louis Reymond. Enumeration of 166 billion organic small molecules in the chemical universe database gdb-17. Journal of Chemical Information and Modeling, 52 0 (11): 0 2864--2875, 2012. doi:10.1021/ci300415d. PMID: 23088335
2012 doi
-
[22]
Barrett, Mateusz Malinowski, Razvan Pascanu, Peter Battaglia, and Timothy Lillicrap
Adam Santoro, David Raposo, David G.T. Barrett, Mateusz Malinowski, Razvan Pascanu, Peter Battaglia, and Timothy Lillicrap. A simple neural network module for relational reasoning. In Proceedings of the 31st International Conference on Neural Information Processing Systems, NI...
2017
-
[23]
On the expressive power of sparse geometric MPNN s
Yonatan Sverdlov and Nadav Dym. On the expressive power of sparse geometric MPNN s. In The Thirteenth International Conference on Learning Representations, 2025. URL https://openreview.net/forum?id=NY7aEek0mi
2025
-
[24]
FSW-GNN : A bi- L ipschitz WL -equivalent graph neural network, 2024
Yonatan Sverdlov, Yair Davidson, Nadav Dym, and Tal Amir. FSW-GNN : A bi- L ipschitz WL -equivalent graph neural network, 2024. URL https://arxiv.org/abs/2410.09118
2024
-
[25]
Universal representation of permutation-invariant functions on vectors and tensors
Puoya Tabaghi and Yusu Wang. Universal representation of permutation-invariant functions on vectors and tensors. In International Conference on Algorithmic Learning Theory, pages 1134--1187. PMLR, 2024
2024
-
[26]
Universal approximation of functions on sets
Edward Wagstaff, Fabian B Fuchs, Martin Engelcke, Michael A Osborne, and Ingmar Posner. Universal approximation of functions on sets. Journal of Machine Learning Research, 23 0 (151): 0 1--56, 2022
2022
-
[27]
Polynomial width is sufficient for set representation with high-dimensional features
Peihao Wang, Shenghao Yang, Shu Li, Zhangyang Wang, and Pan Li. Polynomial width is sufficient for set representation with high-dimensional features. In The Twelfth International Conference on Learning Representations, 2024. URL https://openreview.net/forum?id=34STseLBrQ
2024
-
[28]
How powerful are graph neural networks? In International Conference on Learning Representations, 2018
Keyulu Xu, Weihua Hu, Jure Leskovec, and Stefanie Jegelka. How powerful are graph neural networks? In International Conference on Learning Representations, 2018
2018
-
[29]
Deep sets
Manzil Zaheer, Satwik Kottur, Siamak Ravanbhakhsh, Barnab\' a s P\' o czos, Ruslan Salakhutdinov, and Alexander J Smola. Deep sets. In Proceedings of the 31st International Conference on Neural Information Processing Systems, NIPS'17, page 3394–3404, Red Hook, NY, USA, 2017. C...
2017
-
[30]
On the expressive power of spectral invariant graph neural networks
Bohang Zhang, Lingxiao Zhao, and Haggai Maron. On the expressive power of spectral invariant graph neural networks. In Proceedings of the 41st International Conference on Machine Learning, pages 60496--60526, 2024
2024
-
[31]
Exponential separations in symmetric neural networks
Aaron Zweig and Joan Bruna. Exponential separations in symmetric neural networks. In Alice H. Oh, Alekh Agarwal, Danielle Belgrave, and Kyunghyun Cho, editors, Advances in Neural Information Processing Systems, 2022. URL https://openreview.net/forum?id=jjlQkcHxkp0
2022
Reviewed August 7, 2026 · model on record in the stance chip above.
Discussion (0). Continue with ORCID to comment.