REVIEW 5 major objections 5 minor 17 references
Global dense vector representations for words or items using shared parameter alternating Tweedie model
T0 review · 5 major / 5 minor · reviewed 2026-08-10 · deepseek-v4-flash
Pith's one-line read Modeling word co-occurrence counts with a Tweedie distribution yields embeddings that beat GloVe and rival feature-based BERT on named entity recognition.
desk verdict A real statistical proposal — shared-parameter alternating Tweedie with damped Fisher scoring — but the main application fits log counts while the model and simulation assume raw counts, so the headline claims are not actually tested. 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 load-bearing object is the shared-parameter alternating Tweedie regression: a two-part likelihood in which zero entries contribute their exact probability $\Pr[Y_{ij}=0]=\exp(-\lambda_{ij})$ with $\lambda_{ij}=\mu_{ij}^{2-p_{ij}}/(\phi_{ij}(2-p_{ij}))$, and positive entries contribute the Tweedie exponential-family density whose mean-variance relation is $\mathrm{Var}(Y_{ij})=\phi_{ij}\mu_{ij}^{p_{ij}}$. Information is shared across the whole matrix by factoring the power and dispersion parameters as $p_{ij}=p_i+\tilde{p}_j$ and $\phi_{ij}=\phi_i\tilde{\phi}_j$, so each row and each column carries its own latent scalar while the vectors $w_i$ and $\tilde{w}_j$ carry the geometry. Estimation alternates between two Fisher-scoring passes: with all column vectors held fixed, each row block $(w_i,b_i)$ is updated by adding $I_{\beta_i}^{-1}U_{\beta_i}$ scaled by $\text{lr}/t^{1/4}$, and then the symmetric pass updates columns with rows held fixed; the information matrices are block-diagonal, so the row updates separate cleanly and each row's data can be retrieved once. The normalizing term $c(y_{ij},\phi_{ij},p_{ij})$ in the Tweedie density is dropped from the loss because it does not depend on the regression parameters, leaving the deviance-like loss $-\sum_{ij}(y_{ij}\theta_{ij}-\kappa(\theta_{ij}))$. The power and dispersion values themselves are set before fitting by regressing the log sample variance on the log sample mean over the data matrix and reading off slope $\hat{p}$ and intercept $\hat{\delta}=\log\hat{\phi}$.
What would settle it
Fit a comparison embedding method — GloVe's weighted least squares, or the same Tweedie loss with the power fixed at $p=1$ — to the exact same distance-weighted Wikipedia WordPiece counts used for SA-Tweedie, and run both sets of vectors through the identical BiLSTM with the same seeds; if the test F1 gap against 91% collapses, the improvement comes from the counts and preprocessing rather than from the Compound Poisson Gamma likelihood. The distributional premise can be checked in the same pass: bin held-out rows of $\log(X_{ij}+1)$ by fitted mean and compare empirical variance to the assumed $\phi\mu^p$; the paper's own Table 1 already finds estimated $p$ values between 0 and 1 on several mean intervals, where the Tweedie family is undefined, so an explicit mean-variance check on the log scale would settle whether the likelihood is misspecified.
Extended reading notes
Core claim
The central claim is that a full generative model of the co-occurrence matrix — with $\log \mu_{ij} = w_i^\top \tilde{w}_j + b_i + \tilde{b}_j$, zero probability $\Pr[Y_{ij}=0]=\exp(-\lambda_{ij})$, and shared parameters $p_{ij}=p_i+\tilde{p}_j$, $\phi_{ij}=\phi_i\tilde{\phi}_j$ — can be estimated at Wikipedia scale and yields embeddings that outperform standard alternatives in a lightweight downstream model. Fitted to $\log(X_{ij}+1)$ of distance-weighted WordPiece co-occurrences from the January 2022 Wikipedia dump (about 4 billion tokens, 26,531 token types), 300-dimensional SA-Tweedie vectors give a simple BiLSTM about 95% weighted F1 on the CoNLL-2003 validation set and 91% on the test set, versus roughly 88% for GloVe under the same architecture and seeds; the paper reads this as more effective than BERT's feature-based embeddings, whose reported validation F1 is 91%. The paper further claims that the estimation scheme is decisive: Fisher scoring with a learning-rate factor $\text{lr}/t^{1/4}$ reaches a lower final training loss than unscaled Fisher scoring, which can stall and occasionally jump to a much worse solution, and than Adam, which reduces loss more slowly both within epochs and across iterations; nine simulated raw-Tweedie datasets support the comparison. Taken together, the claims position SA-Tweedie as a resource-efficient route to global word representations that rival much larger contextual models when used as features.
Load-bearing premise
The load-bearing premise is that the log-transformed co-occurrence responses actually follow the Tweedie law being fitted, a premise the paper never checks on its real data because its simulations validate the likelihood only on raw Tweedie counts, and if that premise fails the likelihood is misspecified and the NER result is not a clean test of the model.
Editorial extensions
If this is right
- Feature-based use of SA-Tweedie embeddings (91% test F1 with a one-layer BiLSTM) lands close to fine-tuned BERT (92.4–92.8 test F1) while using none of BERT's 110–345 million pretrained parameters.
- The objective is corpus-efficient: a single 4-billion-token Wikipedia dump outperforms GloVe's 42-billion-token vectors in the identical downstream architecture, implying more signal extracted per token.
- The estimation recipe transfers: Fisher scoring with a $t^{-1/4}$ decaying learning rate provides reliable convergence for alternating dot-product models, avoiding Adam's slow descent and unscaled Fisher scoring's late-iteration stalls.
- The same likelihood machinery applies to any sparse two-sided count matrix, such as user-item interaction counts for recommender systems, as long as the additive row/column sharing of $p$ and the multiplicative sharing of $\phi$ are appropriate.
- Pseudo-likelihood is a closed path for this problem: with no covariates observed, the surrogate conditional distributions push the alternating updates in wrong directions, so full-likelihood or alternating full-likelihood estimation is required.
Reading between the lines
- The NER gain may owe more to the variance weighting implicit in the deviance loss, which down-weights high-mean pairs, than to the zero-mass Tweedie form; retraining GloVe's weighted least squares on the identical Wikipedia counts would isolate how much of the improvement is the model rather than the counts and preprocessing.
- Because the paper validates the likelihood only on raw counts while fitting $\log(X+1)$, the NER number is the strongest evidence for the model; the paper's own Table 1, with fitted $p$ between 0 and 1 on several intervals, hints that a zero-inflated Tweedie or another log-scale family might fit the actual response better.
- The $t^{-1/4}$ schedule is a heuristic whose practical role is likely stabilization under a non-stationary alternating objective; comparing other damped step rules (line search, cosine decay, per-coordinate scaling) on the same simulated data would show how much the specific choice matters.
- The additive $p_{ij}=p_i+\tilde{p}_j$ factorization is motivated by symmetric item-item matrices; user-item matrices are asymmetric and far sparser, so applying the model to recommender data will likely require separate treatment of row and column power parameters.
Signed reviews
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper proposes a Shared-parameter Alternating Tweedie (SA-Tweedie) model for co-occurrence matrices, in which the log of each co-occurrence count is modeled as the inner product of a row embedding and a column embedding plus row/column biases, with Tweedie noise and a zero mass. The authors derive score functions and Fisher information for the Tweedie likelihood, introduce a learning-rate-adjusted Fisher scoring update, and compare it with plain Fisher scoring and Adam on simulated data. They then apply the method to a Wikipedia corpus with WordPiece tokens, building embeddings that are fed into a simple BiLSTM for CoNLL-2003 named entity recognition, where they report about 91% test weighted F1, above their GloVe and random baselines and above the feature-based BERT embeddings result cited from Devlin et al. (2019). The paper also describes a SQLite-based storage and CPU-GPU pipelining strategy to make the algorithm scalable.
Significance. If the statistical model were correctly specified for the response actually used, the paper would offer a potentially interesting probabilistic alternative to GloVe that can produce useful token embeddings with a comparatively light architecture. The manuscript also demonstrates a nontrivial engineering effort: storing the sparse matrix in SQLite, using rowid ranges, chunked GPU computation with einsum, and CPU-GPU queue-based parallelism. The central claim, however, is not adequately supported because the model fitted to the Wikipedia data is not the model developed in Section 2, and the simulation does not validate the modeling assumptions used in the application. These issues bear directly on the NER result and on the claimed advantage of the proposed optimizer, so the significance of the contribution as stated cannot be affirmed.
major comments (5)
- [Section 4.1, Eq. (2.6), Section 3] The model developed in Section 2 assumes the response Y_ij itself is Tweedie-distributed with mean exp(eta_ij) and zero mass exp(-lambda_ij). Section 4.1, however, states that the authors 'decided to model the log-counts using the alternating Tweedie regression model,' i.e., the response is Z_ij = log(X_ij + 1). No derivation is given for the claim that log-transformed weighted counts follow a Tweedie law; indeed a log transformation changes the zero atom (counts 0 and 1 both map to 0) and alters the support, so the likelihood in equations (2.6)-(2.7) is not the correct likelihood for the data actually fitted. The simulation in Section 3 generates raw Tweedie counts, not log-counts, so the simulation does not validate the application and the NER result does not test the proposed model.
- [Section 2.2, Table 1] The estimated power parameters in Table 1 include several intervals with p-hat between 0 and 1 (intervals 1, 2, 3, and 7). The paper itself states on page 20 that for these values 'Tweedie distribution is not well defined' and that a zero-inflated Tweedie distribution may be needed. Yet the subsequent application uses the Tweedie likelihood and score equations with these p values as if they were valid. This means that, for a substantial fraction of the data, the likelihood used for estimation is not the likelihood of any proper distribution, undermining both the estimation procedure and the interpretation of the resulting embeddings.
- [Section 2.1, 'Loss' definition, Eq. (2.8)] The loss used for the convergence criterion and for the loss-curve comparisons is defined as -sum_{i,j}(y_ij*theta_ij - kappa(theta_ij)). This is not the negative log-likelihood of the proposed Tweedie model: for y_ij = 0 the Tweedie log-likelihood contributes -lambda_ij = -mu_ij^(2-p)/(phi_ij(2-p)), whereas the loss contributes kappa(theta_ij) = mu_ij^(2-p)/(2-p), missing the 1/phi factor. The updates in (2.8), by contrast, use the score and Fisher information derived from the actual Tweedie likelihood. The algorithm therefore optimizes a different objective from the one whose score and information matrix it uses, and the reported loss curves do not measure likelihood maximization.
- [Section 3, Figures 4-8] The simulation study only reports training-loss trajectories and score-vector norms; it never checks whether the estimated embedding vectors recover the true vectors used to generate the data. Because the loss is the same objective being optimized, 'outperforms' in the simulation partly reflects the algorithm's progress on its own training criterion rather than statistical estimation accuracy. Moreover, the claimed advantage of the learning-rate-adjusted Fisher scoring over plain Fisher scoring is small (the final losses differ only after the second decimal place in most of the eight replicates), and no standard errors or tests are provided to show the difference is meaningful.
- [Section 4.2, Tables 3 and 4] The comparison with BERT is not an apples-to-apples evaluation. The BERT numbers reproduced in Table 4 come from Devlin et al. (2019) using cased WordPiece models and a different BiLSTM setup, while the SA-Tweedie experiments lowercase all tokens and use a 512-dimensional BiLSTM. The feature-based BERT row labeled 'Embeddings' reports a validation F1 of 91.0 and no test F1, whereas the paper's statement that SA-Tweedie is 'much more effective than BERT embeddings' is based on this non-comparable setting. The claim is therefore overstated and should be removed or replaced with a matched comparison.
minor comments (5)
- [Figure 1] Figure 1 describes a '0 vs positive: Bernoulli' component with a logit link, which suggests a two-stage hurdle model, but the equations in Section 2 do not include such a Bernoulli component and instead use the Tweedie zero mass exp(-lambda). Section 5 also refers to a 'two-stage Bernoulli-Tweedie model.' These descriptions are inconsistent and should be reconciled.
- [Equation (2.3)] The infinite sum in equation (2.3) contains an undefined variable x; it should likely read y^{ka_ij - 1}. The notation for the product and the indices elsewhere in the equation also needs cleaning.
- [Abstract and Section 1] The abstract and introduction mention that the Tweedie model handles 'positive probability mass for zero observations,' but the connection to the actual Compound Poisson Gamma model with p in (1,2) could be stated more precisely, especially since later intervals use p outside that range.
- [Section 4.2] The text says '4 Billon tokens' (typo for 'billion') and refers to the 'January 20th, 2022 version of the Wikipedia dump' without specifying the exact preprocessing (e.g., sentence splitting, window size k) used to build the co-occurrence matrix, making the experimental setup only partially reproducible.
- [Section 5] The convergence claim for the alternating procedure is asserted through an analogy with the generalized EM algorithm, but no formal theorem or proof is given; the two conditions stated are not verified for the SA-Tweedie model. This should either be made rigorous or explicitly labeled as a conjecture.
Circularity Check
No substantive circularity: the NER benchmark is external, and the optimizer comparison is an optimization-loss evaluation; at most a minor non-load-bearing self-citation.
full rationale
The paper's main external claim, the 91% weighted F1 on CoNLL-2003, is tested with a separate BiLSTM trained on annotated labels; the SA-Tweedie embeddings are learned unsupervised from Wikipedia co-occurrence data, so the downstream benchmark does not reduce to the training loss. The simulation comparison of Fisher scoring variants and Adam is measured on the same negative Tweedie deviance these updates optimize; that is a legitimate optimizer benchmark rather than a prediction of held-out data, so it does not make the central claims circular. The Section 4.1 decision to apply the Tweedie likelihood to log(X+1) after deriving it for raw counts (and after Section 3 simulated raw counts) is a distributional assumption gap that the paper itself flags when estimating p in intervals where the Tweedie law is not well defined (Section 2.2, Table 1); this is a correctness and misspecification risk, not a circular reduction, because the loss and score equations remain well-defined functions of the chosen response and the NER evaluation is independent. The only self-citation, Kim (2023), appears in Section 5 as a pointer for omitted pseudo-likelihood details; the pseudo-likelihood discussion is argued from non-identifiability and surrogate-distribution considerations, and it is not load-bearing for the SA-Tweedie or NER claims. No uniqueness theorem or ansatz is imported from prior work by the same authors; p and phi estimates come from the data's mean-variance relationship. Overall, the derivation chain is not circular; the low score reflects only a minor non-load-bearing self-citation.
Assumptions & free parameters
free parameters (8)
- Tweedie power p per interval =
0.485 to 1.230 in Table 1
- Dispersion intercept delta = log(phi) per interval =
-1.483 to 3.030 in Table 1
- Initial learning rate lr =
0.5
- Learning rate decay exponent =
1/4
- Convergence threshold epsilon and maxit =
1e-4 and 100
- Number of chunks per row =
30, 10, 4 for dimensions 300, 100, 50
- Embedding dimension d =
100 and 300
- Cooccurrence window size k =
not specified
assumptions (5)
- domain assumption The log-transformed cooccurrence count log(X_ij + 1) follows a Tweedie distribution with log link.
- ad hoc to paper The Tweedie power parameter decomposes as p_ij = p_i + p_j and the dispersion as phi_ij = phi_i * phi_j.
- domain assumption The dispersion and power parameters p and phi can be treated as known after a piecewise linear mean-variance regression on the same data.
- ad hoc to paper The per-block Fisher information matrices used in Algorithm 1 are invertible at every iteration.
- ad hoc to paper The alternating update converges because the log likelihood is bounded and sufficient statistics are linear in the data, as in the EM algorithm.
Cite this review
Pith. "Pith review of Global dense vector representations for words or items using shared parameter alternating Tweedie model." pith.science (2026). https://pith.science/paper/3ZNRQFJT
@misc{pith2026250100623,
author = {Pith},
title = {Pith review of: Global dense vector representations for words or items using shared parameter alternating Tweedie model},
year = {2026},
howpublished = {\url{https://pith.science/paper/3ZNRQFJT}},
note = {Machine review of arXiv:2501.00623}
}
read the original abstract
In this article, we present a model for analyzing the cooccurrence count data derived from practical fields such as user-item or item-item data from online shopping platform, cooccurring word-word pairs in sequences of texts. Such data contain important information for developing recommender systems or studying relevance of items or words from non-numerical sources. Different from traditional regression models, there are no observations for covariates. Additionally, the cooccurrence matrix is typically of so high dimension that it does not fit into a computer's memory for modeling. We extract numerical data by defining windows of cooccurrence using weighted count on the continuous scale. Positive probability mass is allowed for zero observations. We present Shared parameter Alternating Tweedie (SA-Tweedie) model and an algorithm to estimate the parameters. We introduce a learning rate adjustment used along with the Fisher scoring method in the inner loop to help the algorithm stay on track of optimizing direction. Gradient descent with Adam update was also considered as an alternative method for the estimation. Simulation studies and an application showed that our algorithm with Fisher scoring and learning rate adjustment outperforms the other two methods. Pseudo-likelihood approach with alternating parameter update was also studied. Numerical studies showed that the pseudo-likelihood approach is not suitable in our shared parameter alternating regression models with unobserved covariates.
Figures
Figures from the paper (8 more)
Reference graph
Works this paper leans on
-
[1]
Bonat, W. H. and Kokonendji, C. C. (2017). Flexible tweedie regression models for continuous data. Journal of Statistical Computation and Simulation, 87(11):2138–2152
work page 2017
-
[2]
Sigler, E., Litwin, M., Gray, S., Chess, B., Clark, J., Berner, C., McCandlish, S., Radford, A., Sutskever, I., and Amodei, D. (2020). Language models are few-shot learners. In
work page 2020
-
[3]
Larochelle, H., Ranzato, M., Hadsell, R., Balcan, M., and Lin, H., editors, Advances in Neural Information Processing Systems, volume 33, pages 1877–1901. Curran Associates, Inc
work page 1901
-
[4]
Dempster, A. P., Laird, N. M., and Rubin, D. B. (1977). Maximum likelihood from in- complete data via the em algorithm. Journal of the Royal Statistical Society: Series B (Methodological), 39(1):1–22
work page 1977
-
[5]
Devlin, J., Chang, M.-W., Lee, K., and Toutanova, K. (2019). BERT: Pre-training of deep bidirectional transformers for language understanding. In North American Chapter of the Association for Computational Linguistics
work page 2019
-
[6]
Freestone, M. and Karmaker, S. S. (2024). Word embeddings revisited: Do llms offer some- thing new? ArXiv, abs/2402.11094
arXiv 2024
-
[7]
Kim, T. (2023). Matrix factorization and prediction for high dimensional zero inflated co- occurrence count data via shared parameter alternating generalized linear regression with application in nlp. PhD Dissertation, Kansas State University
work page 2023
-
[8]
Snyder, W., Chen, K., Kakade, S., Jain, P., and Farhadi, A. (2024). Matryoshka represen- Kim & Wang /Matrix factorization and prediction using SA-Tweedie 43 tation learning. In Proceedings of the 36th International Conference on Neural Information Processing Systems, NIPS ’22, Red Hook, NY, USA. Curran Associates Inc
work page 2024
Show all 17 references
-
[9]
Pennington, J., Socher, R., and Manning, C. (2014). GloVe: Global vectors for word represen- tation. In Proceedings of the 2014 Conference on Empirical Methods in Natural Language Processing (EMNLP), pages 1532–1543, Doha, Qatar. Association for Computational Lin- guistics
2014
-
[10]
E., Neumann, M., Iyyer, M., Gardner, M., Clark, C., Lee, K., and Zettlemoyer, L
Peters, M. E., Neumann, M., Iyyer, M., Gardner, M., Clark, C., Lee, K., and Zettlemoyer, L. (2018). Deep contextualized word representations. ArXiv, abs/1802.05365
2018 arXiv
-
[11]
Raffel, C., Shazeer, N., Roberts, A., Lee, K., Narang, S., Matena, M., Zhou, Y., Li, W., and Liu, P. J. (2020). Exploring the limits of transfer learning with a unified text-to-text transformer. Journal of Machine Learning Research, 21(140):1–67
2020
-
[12]
Raza, S., Rahman, M., Kamawal, S., Toroghi, A., Raval, A., Navah, F., and Kazemeini, A. (2024). A comprehensive review of recommender systems: Transitioning from theory to practice
2024
-
[13]
M., Subramanian, R., Tan, X
Reizenstein, J., Rungta, R., Saladi, K., Schelten, A., Silva, R., Smith, E. M., Subramanian, R., Tan, X. E., Tang, B., Taylor, R., Williams, A., Kuan, J. X., Xu, P., Yan, Z., Zarov, I., Zhang, Y., Fan, A., Kambadur, M., Narang, S., Rodriguez, A., Stojnic, R., Edunov, S., and S...
2023
-
[14]
Polosukhin, I. (2017). Attention is all you need
2017
-
[15]
and Manning, C
Wang, M. and Manning, C. D. (2013). Effect of non-linear deep architecture in sequence labeling. In Proceedings of the Sixth International Joint Conference on Natural Language Processing, pages 1285–1291, Nagoya, Japan. Asian Federation of Natural Language Pro- cessing
2013
-
[16]
Gao, Q., Macherey, K., Klingner, J., Shah, A., Johnson, M., Liu, X., Kaiser, L., Gouws, S., Kato, Y., Kudo, T., Kazawa, H., Stevens, K., Kurian, G., Patil, N., Wang, W., Young, C., Smith, J., Riesa, J., Rudnick, A., Vinyals, O., Corrado, G., Hughes, M., and Dean, J. (2016). Go...
2016 arXiv
-
[17]
J., and Yalcın, S
Zanbouri, K., Darbandi, M., Nassr, M., Heidari, A., Navimipour, N. J., and Yalcın, S. (2024). A gso-based multi-objective technique for performance optimization of blockchain- based industrial internet of things. International Journal of Communication Systems, 37(15):e5886
2024
Reviewed August 10, 2026 · model on record in the stance chip above.
Discussion (0). Continue with ORCID to comment.