Disentangling Similarity and Relatedness in Topic Models
Organizations: Manchester Institute of Biotechnology, The University of Manchester, UK · Department of Computer Science, The University of Manchester, UK · Department of Chemistry, The University of Manchester, UK
Abstract
The recent success of large pre-trained language models (PLMs) has motivated their integration into topic modeling. However, PLM-augmented topic models differ from classical co-occurrence models such as Latent Dirichlet Allocation (LDA) not only in performance, but also in the type of semantic structure they capture. We formalize this distinction along two psycholinguistic axes: thematic relatedness (dog/bone) and taxonomic similarity (dog/wolf). To measure both axes over topic words, we construct a large synthetic benchmark of word pairs using LLM-based annotation and train a neural scorer on it. Across multiple corpora and model families, the scorer places different topic-model families at distinct positions within the joint similarity-relatedness space. The two scores further predict downstream task performance: tasks requiring similarity benefit from similarity-rich topics, whereas tasks requiring relatedness benefit from the converse, and excessive emphasis on either axis degrades performance on tasks aligned with the opposing semantic structure. Neither axis is uniformly beneficial. Measuring both therefore provides a practical, model-agnostic diagnostic for evaluating the semantic structure captured by topic models.
Figures & tables
| Model | Sim. ( ) | Rel. ( ) |
|---|---|---|
| Neural Scorer | 0.667 | 0.506 |
| BERT ( Devlin et al., 2019 ) | 0.229 | 0.239 |
| OpenAI ( OpenAI, 2024 ) | 0.347 | 0.182 |
| WordNet W–P ( Wu and Palmer, 1994 ) | 0.620 |
| Evaluation Axis | Kendall’s |
|---|---|
| Similarity | 0.768 |
| Relatedness | 0.743 |
| Shifted Normalized Gap | 0.732 |
| Relatedness | Similarity | ||
| Predictor | Event Mon. | Cat. Retr. | Syn-Aware Retr. |
| Single-factor partial | |||
| Similarity | .00 | .06 | .53 |
| Relatedness | .24 | .42 | .03 |
| Gap | .64 | .64 | .77 |
| Coherence | .49 | .41 | .08 |
Appendix figures & tables21 assets
Supplementary material from the paper’s appendix.
Appendix
| Word Pair | Score Type | Score 1 | Score 2 | Datasource 1 | Datasource 2 |
|---|---|---|---|---|---|
| breakfast / dinner | similarity | 0.941 | 0.149 | TxThmNorms | SimLex-999 |
| camel / desert | relatedness | 0.903 | 0.580 | TxThmNorms | MEN |
| LDA (Reuters) | BERT-KT (Reuters) | |
|---|---|---|
| Topic 1 | dollar, rate, exchange, currency, market | rise, increase, fall, decline, drop |
| Topic 2 | growth, government, economic, economy, inflation | president, chairman, minister, secretary, chief |
| Topic 3 | gold, reserves, production, ounces, mine | make, continue, reduce, support, remain |
| Task A | Task B | Metric Scores | |||||||
|---|---|---|---|---|---|---|---|---|---|
| Model | Nature | Hit@3 | Recall@3 | SCR | Sim. | Rel. | Gap | Coh. | Div. |
| lda | Non-PLM | 0.585 | 0.492 | 0.688 | 0.246 | 0.582 | 0.336 | 0.510 | 0.660 |
| bertopic | PLM-Aug. | 0.573 | 0.482 | 0.679 | 0.228 | 0.558 | 0.330 | 0.612 | 0.824 |
| nmf | Non-PLM | 0.526 | 0.456 | 0.642 | 0.251 | 0.572 | 0.321 | 0.547 | 0.618 |
| neurallda | Non-PLM | 0.523 | 0.441 | 0.624 | 0.224 | 0.537 | 0.313 | 0.417 | 0.998 |
| prodlda | Non-PLM | 0.520 | 0.445 | 0.662 | 0.233 | 0.532 | 0.299 | 0.629 | 0.868 |
| Model | Type | Reut. | 20NG | BBC | DBLP | ACL | M10 | NewsC | MN-DS | arXiv |
|---|---|---|---|---|---|---|---|---|---|---|
| PLM Augmented | ||||||||||
| TNTM | PLM Aug. | 0.94 | — | 0.96 | — | — | 0.70 | 0.94 | 0.98 | 0.97 |
| BERT-KT | PLM Aug. | 0.76 | 0.48 | 0.40 | 0.60 | 0.68 | 0.54 | 0.74 | 0.74 | 0.56 |
| T&M | PLM Aug. | 0.45 | 0.25 | 0.38 | 0.39 | — | 0.33 | — | — | — |
| Non-PLM Augmented | ||||||||||
| LDA | Non-PLM Aug. | 0.23 | 0.16 | 0.12 | 0.06 | 0.20 | 0.13 | 0.04 | 0.12 | 0.17 |
| approach/method | evaluate/assess | evaluation/assessment | identify/detect |
| identification/detection | discover/find | analysis/study | model/framework |
| method/technique | procedure/process | improve/enhance | develop/design |
| combine/integrate | predict/estimate | prediction/estimation | approximate/estimate |
| approximation/estimation | accuracy/precision | robust/stable | stability/consistency |
| verification/validation | goal/objective | goal/target | objective/target |
| task/work | handle/manage | assist/support | build/construct |
| increase/rise | decline/fall | reduce/lower | stop/halt |
| purchase/acquire | agreement/accord | company/firm | stock/share |
| lawsuit/suit | announce/declare | begin/start | benefit/advantage |
| change/alter | choose/select | close/shut | complete/finish |
| concern/worry | create/build | deal/agreement | delay/postpone |
| demand/request | deny/reject | dismiss/reject | estimate/forecast |
| expand/grow | fast/quick | final/ultimate | goal/objective |
| Task | ||||
|---|---|---|---|---|
| Event Monitoring | 46 | .58 | ||
| Category Retrieval | 46 | .68 | ||
| Synonym-Aware Retrieval | 38 | .79 |
| Model | Approach | Reference |
|---|---|---|
| LDA | Hierarchical Bayesian model; documents as mixtures of topics over vocabulary words | Blei et al. (2003) |
| LSI | Truncated SVD on the term–document matrix to extract latent semantic factors | Deerwester et al. (1990) |
| HDP | Nonparametric Bayesian extension of LDA; infers the number of topics automatically | Teh et al. (2006) |
| NMF | Factorises the term–document matrix under non-negativity constraints into additive topic components | Lee and Seung (1999) |
| NeuralLDA | VAE with logistic-normal prior for amortised LDA inference (AVITM, LDA variant) | Srivastava and Sutton (2017) |
| ProdLDA | Product-of-experts formulation in a VAE framework (AVITM, ProdLDA variant) | Srivastava and Sutton (2017) |
| Model | Corpus | Failure mode |
|---|---|---|
| CWTM | BBC News | Diversity collapse: topics near-identical across runs |
| ETM ( etm_fresh ) | DBLP | Diversity collapse: topic diversity on all runs |
| TNTM | DBLP | Word-count collapse: runs yield only 1–5 topics with 3 words |
| TNTM | ACL | Word-count collapse: runs yield only 1–4 topics with 3 words |
| Source Family | Sub-sources | Pairs |
|---|---|---|
| WordNet ( Miller, 1995 ) | synonyms (9.9K), cohyponyms (18.6K), hypernyms (4.6K), antonyms (1.3K) | 34,346 |
| ConceptNet 5.7 ( Speer et al., 2017 ) | RelatedTo (5.3K), AtLocation (2.1K), PartOf (1.7K), UsedFor (0.3K), Causes (0.2K) | 9,616 |
| Embedding neighbors | BERT top- cosine nearest neighbours (max 3/cluster, 20/word) | 14,961 |
| Controlled negatives | far-distance (5.0K), cross-domain (3.7K), cross-POS (3.7K) | 12,423 |
| Total | 71,346 |
| Configuration | mean | ||
|---|---|---|---|
| Prompt design (DeepSeek-V4) | |||
| Answer-first (scores, then reason) | 0.828 | 0.809 | 0.818 |
| Reason-first, 1 sentence | 0.873 | 0.835 | 0.854 |
| Reason-first, 2 sentences | 0.879 | 0.837 | 0.858 |
| Reason-first, 2 sent, simple prompt | 0.869 | 0.809 | 0.839 |
| Single annotator | |||
| Block | Setting | Value |
| Input | Word embeddings | OpenAI text-embedding-3-large , prompt-templated static, 3072D |
| Pair features (embedding) | ( D) | |
| Pair features (WordNet) | D: path sim, LCH, WuP, Lin, JCN, shortest path, same category, is-synonym (binary) | |
| Total input dimension | ||
| Architecture | Shared trunk | (Linear + GELU + Dropout) |
| Task heads | Similarity: (Sigmoid); Relatedness: (Sigmoid) |
| Subset | ||
|---|---|---|
| Noun–noun ( ) | 0.897 | 0.827 |
| Other ( ) | 0.925 | 0.836 |
| All ( ) | 0.916 | 0.836 |
| Embedding | Dim | MLP trunk | Internal (S/R) | Holdout (S/R) | External (S/R) | Mean |
| Count-based word embedding | ||||||
| GloVe 840B ( Pennington et al., 2014 ) | 300 | 256-64 | 0.885 / 0.811 | 0.841 / 0.811 | 0.520 / 0.437 | 0.718 |
| Large-model static embeddings | ||||||
| X2Static (RoBERTa-large) ( Gupta and Jaggi, 2021 ) | 1024 | 1024-512-128 | 0.877 / 0.808 | 0.841 / 0.781 | 0.532 / 0.421 | 0.710 |
| Model2Vec (potion-base) | 512 | 256-64 | 0.751 / 0.656 | 0.754 / 0.697 | 0.466 / 0.291 | 0.603 |
| WordLlama | 256 | 256-64 | 0.697 / 0.565 | 0.588 / 0.577 | 0.270 / 0.302 | 0.500 |
| Component | Setting | Pair-heldout (S,R) | Word-heldout (S,R) | External (Tx,Thm) |
|---|---|---|---|---|
| Input | (no interaction) | .736, .632 | .712, .684 | .606, -.006 |
| WordNet | OFF | .887, .814 | .843, .783 | .496, .420 |
| Architecture | Cross-attn hybrid | .648, .578 | .631, .627 | .202, -.010 |
| Corpus | Domain | Documents | Avg Tokens | Vocab Size |
| 20NewsGroups ( Lang, 1995 ) | Online forum posts (20 topics) | 18,248 | 69.4 | 24,381 |
| BBC News ( Greene and Cunningham, 2006 ) | News articles (5 categories) | 2,225 | 111.0 | 2,814 |
| Reuters-21578 ( Lewis, 1997 ) | Newswire (financial/economic) | 7,717 | 54.4 | 5,699 |
| DBLP ( Ley, 2002 ) | CS paper titles | 54,595 | 5.3 | 1,480 |
| M10 ( Lim and Buntine, 2015 ) | Scientific paper titles (10 areas) | 8,355 | 5.8 | 1,591 |
| ACL ( Bird et al., 2008 ) | NLP paper full texts | 10,551 | 1,661.0 | 8,868 |
| Component | Wall-clock | Hardware / notes |
|---|---|---|
| LLM annotation (71,346 pairs 3 providers) | h per provider, run in parallel | 1 CPU, 5 GB RAM; API-bound. Total API spend {\approx}\50$90$ . |
| Neural-scorer training | 17 min | 1 A100 80GB, 100 epochs, batch 32 |
| Scorer ablations + scaling sweep | GPU-hours total | Same hardware as above |
| Atlas topic-model training (1,567 runs across 17 topic models and a random baseline 9 corpora 10 seeds) | – GPU-hours total | GH200 single-GPU; CPU-only for co-occurrence models. Per-model wall time varies from min (Random, BERT-KT, LDA) to h (CombinedTM); Contextual Top2Vec additionally requires GB host RAM. |
| Atlas scoring (162 cells) | 48 min | 1 A100 80GB, 12 CPUs, 64 GB RAM |
| Downstream evaluation (regressions, retrieval tasks, traditional metrics) | GPU-hours | 4 CPUs, 32 GB RAM per task |
Topic Modeling
Topic modeling discovers recurring themes and document-topic structure in text, helping organize, compare, and track large collections.
See all 57 papers