REVIEW 4 major objections 4 minor 1 cited by
Clustering in hyperbolic balls
T0 review · 4 major / 4 minor · reviewed 2026-08-09 · deepseek-v4-flash
Pith's one-line read This paper establishes a mathematical foundation for k-means and EM clustering inside hyperbolic balls, using a Möbius-invariant barycenter and Möbius mixture distributions.
desk verdict A useful clustering proposal for Poincaré balls, but the k-means update doesn't optimize its stated objective and the EM stationarity equation has a sign error; still worth a referee's time because the flaws are identifiable and likely fixable. 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 conformal barycenter, defined as the unique minimizer of the geodesically convex energy $\widetilde H_n(a) = -\sum_i w_i \log\big((1-|a|^2)(1-|y_i|^2)/\rho(y_i,a)\big)$, computed by the gradient-flow system (9)-(10) that transports the weighted points to a configuration whose weighted Euclidean sum is zero and then maps the origin back through the inverse Möbius transformation. The second pillar is the Möbius family of distributions with densities $p(x;a,s)\propto\big((1-|x|^2)(1-|a|^2)/\rho(a,x)\big)^s$, which is conformally invariant and transitively acted upon by the group of Möbius isometries of the Poincaré ball.
What would settle it
Take three points in the Poincaré disc with strongly unequal weights (for example, weights 100, 1, 1), compute the weighted barycenter by the paper's gradient-flow routine, and compare it with the numerical minimizer of (11) found by direct optimization; if they disagree, the center update in both algorithms can compute the wrong centers.
Extended reading notes
Core claim
The paper's central claim is that a hyperbolic analogue of k-means and of Gaussian-mixture EM exists and is computable. For k-means the center of each cluster is the conformal barycenter, defined as the unique minimizer of a geodesically convex function; for mixtures the cluster model is the Möbius family of probability densities, and maximum likelihood estimates the location parameter as the weighted conformal barycenter. Both algorithmic steps — moving points by a hyperbolic gradient flow until their weighted sum vanishes and then pulling the origin back by the resulting Möbius transformation — are inherited from this geometry, so the cluster centers transform equivariantly under isometries of the ball. The paper further presents the EM equations, including a closed-form update for the concentration parameter in the disc and a digamma-function equation in higher dimensions, and reports experiments in $\mathbb{B}^2$ and $\mathbb{B}^3$.
Load-bearing premise
The weakest load-bearing step is the weighted barycenter computation: the paper assumes that running the gradient-flow system (9) with weighted points drives the weighted Euclidean sum to zero and that the Möbius preimage of zero is exactly the unique minimizer of the weighted objective (11), but it gives no proof or citation for the weighted case.
Editorial extensions
If this is right
- Data embedded in a Poincaré ball of any dimension can be clustered by k-means with cluster centers updated by the weighted gradient-flow routine instead of the Euclidean average.
- Mixtures of Möbius distributions can be learned by EM, with responsibilities computed from the Möbius density and parameters updated by weighted-barycenter and concentration-parameter estimation.
- Because both the barycenter and the Möbius family are conformally invariant, the clustering outcome is equivariant under Möbius transformations: applying an isometry of the ball to the data applies the same isometry to the inferred cluster centers.
- The reported experiments in the disc and the three-dimensional ball show that the algorithms recover mixture parameters close to ground truth when clusters are well separated; the same experiments show overlapping components are not recovered, so practical use requires diagnostics such as monitoring responsibilities and log-likelihood.
- The M-step is numerically tractable: the concentration parameter has a closed-form update in the Poincaré disc and a digamma-equation solution in higher dimensions.
Reading between the lines
- Beyond the paper, the same conformal-barycenter machinery could define other Euclidean algorithms on hyperbolic balls, such as k-medians or principal geodesic analysis, provided the corresponding objectives are geodesically convex; the paper does not address these.
- The unproved weighted-convergence step suggests a testable mathematical conjecture: for any positive weights and initial points in the ball, the flow (9) with $f=(K/N)\sum_j \omega_j x_j$ drives the weighted sum to zero, and the limiting Möbius transformation pulls the origin to the weighted barycenter.
- The reported failures on overlapping mixtures imply that, as with Euclidean EM, initialization and identifiability matter; a natural extension is an initialization strategy based on the unweighted barycenter or on k-means++ in hyperbolic distance, but the paper does not propose one.
Signed reviews
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper proposes a mathematical framework for unsupervised learning in Poincaré balls. It defines conformal barycenters and weighted barycenters, introduces a family of Möbius distributions, and then presents two algorithms: k-means clustering whose centers are conformal barycenters, and an EM algorithm for mixtures of Möbius distributions. Experiments are reported for two- and three-dimensional balls. The central claim is that these algorithms provide rigorous, well-founded clustering tools for hyperbolic representations.
Significance. If correct, the paper would address a genuine gap: hyperbolic machine learning lacks standard unsupervised tools, and the conformal-invariance properties of the proposed barycenter and distribution family are appealing. The exposition is clear, and the experiments demonstrate that the proposed updates terminate and recover parameters in some simple settings. However, the paper relies on two unreviewed preprints by the same first author for the two central mathematical objects, and, more importantly, the algorithms as printed do not optimize the objectives they state. The k-means update minimizes a log-ratio barycenter objective rather than squared hyperbolic distance, and the n-dimensional EM M-step for concentration parameters has a sign error in the printed stationarity equation. These are load-bearing inconsistencies, not presentation issues.
major comments (4)
- [Section 5.1, Eq. (8), and Section 6.1] The k-means objective is stated as J = sum_i sum_j kappa_ij d_hyp^2(xi_i, a_j), but step iv replaces each center by the conformal barycenter, i.e. the minimizer of H_2 or H_tilde in Eq. (8)/(11). No theorem in the paper shows that the minimizer of (8) minimizes the sum of squared hyperbolic distances. In the one-dimensional real slice, stationarity for J gives artanh(a) = N^{-1} sum artanh(y_i), while stationarity for (8) gives N a/(1-a^2) = sum y_i/(1 - a y_i); these are different for generic configurations, so a Lloyd iteration can increase J. Section 6.1 inherits the same mismatch in all dimensions. The paper therefore does not implement the k-means objective it defines.
- [Section 6.2, Remark 3, Eq. (16)] The stationarity equation printed in Remark 3 has the digamma difference reversed. Differentiating (16) gives ψ(1+s-n/2) - ψ(1+s-n) = H_tilde(hat a) / (N sum_i w_i). The printed equation has the opposite order. Since ψ is increasing, for s > n-1 the printed left-hand side is negative while the right-hand side is nonnegative (and positive except in degenerate cases), so no solution exists. Thus the M-step for concentration parameters in the n-dimensional EM algorithm is inconsistent with the objective, and the B1/B2 results are not produced by the stated EM update.
- [Section 3.2, steps ii-iv] The weighted-barycenter routine is asserted without proof or citation. It is not shown that solutions of (9) with f = (K/N) sum_j w_j x_j drive sum_j w_j x_j(T) to zero, nor that the inverse Möbius preimage of zero minimizes the weighted objective (11). The unweighted case is attributed to [18], but the weighted case is used in every cluster-center update in both k-means and EM. The text also assumes strictly positive weights, while EM responsibilities can vanish, so the routine's range of applicability to the E-step is not stated.
- [Section 7] The statement that convergence 'can be established in an analogous way' to Euclidean counterparts is not supported. Euclidean Lloyd convergence relies on the centroid update minimizing the cluster objective, which is exactly the property missing here, and EM monotonicity relies on an exact M-step, which the current n-dimensional s-update does not provide. The paper cannot claim an analogous convergence argument without addressing the objective/update mismatch and the stationarity error.
minor comments (4)
- [Section 3.2, Definition 4] Definition 4 refers to the unique minimum of 'the function (8)', but the weighted objective should be (11).
- [Section 3.2] The weights are denoted w_i in the text and omega_j in the algorithm; the notation should be unified.
- [Section 6.2, iii-1] The mixing-probability update writes sum_{i=1}^n gamma_im; the index should be N, not n.
- [Sections 4 and 5.3] The central objects in Definition 3 and Eq. (13) are imported from the preprints [17] and [18] without restating or proving the required convexity, uniqueness, normalization, and MLE properties; the paper should either make these results self-contained or clearly mark them as imported assumptions.
Circularity Check
EM location updates are the barycenter objective by construction, and the barycenter apparatus is imported from load-bearing self-citations; the k-means objective mismatch is a correctness risk rather than circularity.
-
uniqueness imported from authors
[Section 3.1, pp. 3-4; ODE system (9)-(10) and steps iii-iv]
"For a sufficiently large T, it holds that x1(T)+···+xN(T)=0. This means (see [18, Proposition 3]) that zero (center of the ball) is the conformal barycenter of points x1(T),...,xN(T). Furthermore, due to [18, Theorem 12], there exists a (unique up to an orthogonal transformation) Möbius transformation h ∈ Gn, such that x1(T)=h(y1),...,xN(T)=h(yN)."
The computation of every conformal/weighted barycenter—hence every k-means cluster center and every EM location update—is justified by a uniqueness/computation theorem imported from [18], a preprint by the paper's own first author. The text supplies no proof of the weighted version (Section 3.2) and no independent check of [18, Prop. 3 / Thm. 12]. The algorithm's output is forced by that prior self-authored theorem, invoked as an external mathematical fact rather than derived or independently verified here.
-
self definitional
[Section 4, eq. (13); Section 4.2; Section 5.2 EM step ii-2]
"p(x;a,s)=π^{n/2}Γ(1+s−n/2)/Γ(1+s−n)·((1−|x|²)(1−|a|²)/ρ(a,x))^s ... the maximum likelihood estimations â and ŝ are (see [18]): a) â is the conformal barycenter ... ii-2) Compute weighted barycenters a(j)m of points ξi ... with weights γim using the method explained in subsection 3.2."
The Möbius density's dependence on the location parameter a is exactly exp(s·log((1−|a|²)(1−|x|²)/ρ(a,x))), so maximizing the likelihood over a is, by construction, minimizing H_n in eq. (8)—the same objective used to define the conformal barycenter in Definition 3. The EM M-step then 'computes weighted barycenters', i.e. repeats this same minimization with weights γim (eq. 11). Thus the EM location estimates are not an independent derivation from data; the barycenter definition is baked into the density, and the paper presents this identity as an MLE result imported from [18].
full rationale
Most of the paper's mathematical machinery is not derived inside the paper: the conformal barycenter (Definition 3), its gradient-flow computation, and the Möbius family (eq. 13) are taken from [17] and [18], both involving the first author. Those citations are load-bearing because every cluster-center update in k-means and every EM M-step calls the weighted-barycenter routine. I do not treat citation of prior mathematical work as circular per se, but here the cited preprints are neither machine-checked nor independently benchmarked, and the text does not reprove the key uniqueness and convergence facts. More sharply, the Möbius density (13) is defined by the same conformal factor that defines the barycenter objective (8), so the statement 'MLE â is the conformal barycenter' (Section 4.2) is true by construction; the EM's weighted-barycenter M-step then re-executes that same minimization. That makes the EM location learning partially definitional rather than an independent prediction. The k-means section, by contrast, is best described as an objective mismatch—the center update minimizes H_n while the stated objective J is squared hyperbolic distance—which is a correctness risk, not a circularity. The synthetic experiments on data generated from the same Möbius mixtures provide a self-contained sanity check but no external falsification. Overall, there is real self-citation load-bearing and one by-construction identity, but the algorithms also contain independent ingredients (assignment rules, responsibilities, concentration-parameter updates), so the circularity score is moderate rather than severe.
Assumptions & free parameters
free parameters (4)
- Initial concentration parameter s* =
2 (suggested default)
- Concentration initializations s_i =
sampled from Exp(1)+1
- Gradient-flow constant K =
unspecified negative value
- Number of clusters k =
assumed known per experiment
assumptions (6)
- standard math H_n(a) in (8) is geodesically convex and has a unique minimizer
- standard math The ODE system (9)-(10) is the hyperbolic gradient flow for H_n and drives unweighted configurations to zero sum
- ad hoc to paper The weighted version (system (9) with f = K/N Σ ω_j x_j) drives Σ ω_i x_i(T) to 0
- domain assumption The density in (13) is a valid probability density on Bn
- domain assumption Moeb_n(a,s) is conformally invariant and G_n acts transitively
- domain assumption The MLE equations (15)-(16) are concave with unique solutions
invented entities (2)
-
Conformal barycenter (Definition 3)
-
Möbius distributions Moeb_n(a,s)
Cite this review
Pith. "Pith review of Clustering in hyperbolic balls." pith.science (2026). https://pith.science/paper/VK5O6WK7
@misc{pith2026250119247,
author = {Pith},
title = {Pith review of: Clustering in hyperbolic balls},
year = {2026},
howpublished = {\url{https://pith.science/paper/VK5O6WK7}},
note = {Machine review of arXiv:2501.19247}
}
abstract
The idea of representations of the data in negatively curved manifolds recently attracted a lot of attention and gave a rise to the new research direction named {\it hyperbolic machine learning} (ML). In order to unveil the full potential of this new paradigm, efficient techniques for data analysis and statistical modeling in hyperbolic spaces are necessary. In the present paper rigorous mathematical framework for clustering in hyperbolic spaces is established. First, we introduce the $k$-means clustering in hyperbolic balls, based on the novel definition of barycenter. Second, we present the expectation-maximization (EM) algorithm for learning mixtures of novel probability distributions in hyperbolic balls. In such a way we lay the foundation of unsupervised learning in hyperbolic spaces.
Figures
Figures from the paper (10 more)
Forward citations
Cited by 1 Pith paper
-
Cartan Networks: Group theoretical Hyperbolic Deep Learning
Cartan networks compose solvable-group homomorphisms with isometries to define hyperbolic layers, and the paper reports competitive benchmark performance.
Reference graph
Works this paper leans on
-
[17]
Conformal and holomorphic barycenters in hyperbolic balls
Ja ´cimovi´c, V . & Kalaj, D. Conformal and holomorphic barycenters in hyperbolic balls. arXiv:2410.02257. (2024)
work page Pith review arXiv 2024
-
[18]
A group-theoretic framework for machine learning in hyperbolic spaces
Ja ´cimovi´c, V . A group-theoretic framework for machine learning in hyperbolic spaces. arXiv:2501.06934. (2025)
arXiv 2025
-
[1]
Smith, A.L., Asta, D.M. & Calder, C.A. The geometry of con tinuous latent space models for network data. Statistical Science: A Review Journal of the Institute of Ma thematical Statistics. 34, 428 (2019)
work page 2019
-
[2]
Sala, F., De Sa, C., Gu, A. & Ré, C. Representation tradeof fs for hyperbolic embeddings. Proceedings of the 35th International Conference on Machine Learning . pp. 4460-4469 (2018)
work page 2018
-
[3]
Arvanitidis, G., Hansen, L.K. & Hauberg, S. Latent Space Oddity: on the Curvature of Deep Generative Models. International Conference on Learning Representations . (2018)
work page 2018
-
[4]
Nickel, M. & Kiela, D. Poincaré embeddings for learning h ierarchical representations. Proceedings of the 31st International Conference on Neural Information Processin g Systems. pp. 6341-6350 (2017)
work page 2017
-
[5]
Tifrea, A., Bécigneul, G. & Ganea, O.-E. Poincaré GloV e: Hyperbolic Word Embeddings. International Confer- ence on Learning Representations . (2019)
work page 2019
-
[6]
Muscoloni, A., Thomas, J.M., Ciucci, S., Bianconi, G. & C annistraci, C.V . Machine learning meets complex networks via coalescent embedding in the hyperbolic space. Nature Communications. 8, 1615 (2017)
work page 2017
Show all 19 references
-
[7]
Chami, I., Wolf, A., Juan, D.-C., Sala, F., Ravi, S. & Ré, C . Low-dimensional hyperbolic knowledge graph embeddings. arXiv:2005.00545. (2020)
2020 arXiv
-
[8]
& Lio, P
Corso, G., Ying, R., Pándy, M., V eli ˇckovi´c, P ., Leskovec, J. & Lio, P . Neural distance embeddings for biological sequences. Proceedings of the 35th International Conference on Neural Information Processing Systems . pp. 18539-18551 (2021)
2021
-
[9]
& Nickel, M
Klimovskaia, A., Lopez-Paz, D., Bottou, L. & Nickel, M. P oincaré maps for analyzing complex hierarchies in single-cell data. Nature Communications. 11, 2966 (2020)
2020
-
[10]
& Wang, Z
Li, A., Y ang, B., Huo, H., Chen, H., Xu, G. & Wang, Z. Hyper bolic neural collaborative recommender. IEEE Transactions on Knowledge and Data Engineering . 35, 9114-9127 (2022)
2022
-
[11]
& V argas, S
Chamberlain, B.P ., Hardwick, S.R., Wardrope, D.R., Dz ogang, F., Daolio, F. & V argas, S. Scalable hyperbolic recommender systems. arXiv:1902.08648. (2019)
2019 arXiv
-
[12]
& Pantazis, D
Baker, C., Suárez-Méndez, I., Smith, G., Marsh, E.B., F unke, M., Mosher, J.C., Maestú, F., Xu, M. & Pantazis, D. Hyperbolic graph embedding of MEG brain networks to study brain alterations in individuals with subjective cognitive decline. IEEE Journal of Biomedical and Health...
2024
-
[13]
& Serrano, M.A
Allard, A. & Serrano, M.A. Navigable maps of structural brain networks across species. PLoS Computational Biology. 16, e1007584 (2020)
2020
-
[14]
& Rubin, D.B
Dempster, A.P ., Laird, N.M. & Rubin, D.B. Maximum likel ihood from incomplete data via the EM algorithm. Journal of the Royal Statistical Society: Series B . 39, 1-22 (1977)
1977
-
[15]
& Sra, S
Banerjee, A., Dhillon, I.S., Ghosh, J. & Sra, S. Cluster ing on the unit hypersphere using von Mises-Fisher distributions. Journal Of Machine Learning Research . 6, 1345-1382 (2005) 15 Clustering in hyperbolic balls A P REPRINT
2005
-
[16]
& Hamelryck, T
Mardia, K.V ., Barber, S., Burdett, P .M., Kent, J.T. & Hamelryck, T. Mixture models for spherical data with appli- cations to protein bioinformatics. Directional Statistics F or Innovative Applications. F orum for Interdisciplinary Mathematics. pp. 15-32 (2022)
2022
-
[19]
& Stegun, I.A
Abramowitz, M. & Stegun, I.A. Handbook of mathematical functions with formulas, graphs, and mathematical tables. (U.S. Government Printing Office, 1972) 16
1972
Reviewed August 9, 2026 · model on record in the stance chip above.
Discussion (0). Continue with ORCID to comment.