mlsanityData to 7 Oct 2026

NumCoKE: Ordinal-Aware Numerical Reasoning over Knowledge Graphs with Mixture-of-Experts and Contrastive Learning

arXiv 2411.12950 ↗First posted 20 Nov 2024cs.AIcited by 1

On the map: Mixture-of-experts → MoE large language models

Abstract (arXiv)

Knowledge graphs (KGs) serve as a vital backbone for a wide range of AI applications, including natural language understanding and recommendation. A promising yet underexplored direction is numerical reasoning over KGs, which involves inferring new facts by leveraging not only symbolic triples but also numerical attribute values (e.g., length, weight). However, existing methods fall short in two key aspects: (1) Incomplete semantic integration: Most models struggle to jointly encode entities, relations, and numerical attributes in a unified representation space, limiting their ability to extract relation-aware semantics from numeric information. (2) Ordinal indistinguishability: Due to subtle differences between close values and sampling imbalance, models often fail to capture fine-grained ordinal relationships (e.g., longer, heavier), especially in the presence of hard negatives. To address these challenges, we propose NumCoKE, a numerical reasoning framework for KGs based on Mixture-of-Experts and Ordinal Contrastive Embedding. To overcome (C1), we introduce a Mixture-of-Experts Knowledge-Aware (MoEKA) encoder that jointly aligns symbolic and numeric components into a shared semantic space, while dynamically routing attribute features to relation-specific experts. To handle (C2), we propose Ordinal Knowledge Contrastive Learning (OKCL), which constructs ordinal-aware positive and negative samples using prior knowledge, enabling the model to better discriminate subtle semantic shifts. Extensive experiments on three public KG benchmarks demonstrate that NumCoKE consistently outperforms competitive baselines across diverse attribute distributions, validating its superiority in both semantic integration and ordinal reasoning.

Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Placed in MoE large language models, one of 12 directions in Mixture-of-experts. Builds on 20 of its 43 references that are in the map. First posted on arXiv 20 Nov 2024.

Builds on

20 of 43 references.

References · 20

  1. 2017Modeling Relational Data with Graph Convolutional Networkscited by 1,684
  2. 2019RotatE: Knowledge Graph Embedding by Relational Rotation in Complex Spacecited by 893
  3. 2017Convolutional 2D Knowledge Graph Embeddingscited by 729
  4. 2019Composition-based Multi-Relational Graph Convolutional Networkscited by 395
  5. 2019TuckER: Tensor Factorization for Knowledge Graph Completioncited by 290
  6. 2018SimplE Embedding for Link Prediction in Knowledge Graphscited by 195
  7. 2020Low-Dimensional Hyperbolic Knowledge Graph Embeddingscited by 195
  8. 2018End-to-end Structure-Aware Convolutional Networks for Knowledge Base Completioncited by 136
  9. 2019Learning Hierarchy-Aware Knowledge Graph Embeddings for Link Predictioncited by 133
  10. 2021Self-supervised Heterogeneous Graph Neural Network with Co-contrastive Learningcited by 130
  11. 2015Order-Embeddings of Images and Languagecited by 112
  12. 2020A unifying mutual information view of metric learning: cross-entropy vs. pairwise lossescited by 100
8 more
  1. 2021Modeling Heterogeneous Hierarchies with Relation-specific Hyperbolic Conescited by 38
  2. 2018Incorporating Literals into Knowledge Graph Embeddingscited by 29
  3. 2020Bipartite Graph Embedding via Mutual Information Maximizationcited by 29
  4. 2017KBLRN : End-to-End Learning of Knowledge Base Representations with Latent, Relational, and Numerical Featurescited by 25
  5. 2020Self-Supervised Learning of Contextual Embeddings for Link Prediction in Heterogeneous Networkscited by 17
  6. 2017Multi-task Neural Network for Non-discrete Attribute Prediction in Knowledge Graphscited by 14
  7. 2023One-stage Low-resolution Text Recognition with High-resolution Knowledge Transfercited by 3
  8. 2023TransERR: Translation-based Knowledge Graph Embedding via Efficient Relation Rotationcited by 2

Built on by

Cited by year

Citing papers per year: 2025: 1.

1 paper.

Similar, not cited

Close in wording, no citation either way

  1. 2025CHAINSFORMER: Numerical Reasoning on Knowledge Graphs from a Chain Perspective
  2. 2022NumGLUE: A Suite of Fundamental yet Challenging Mathematical Reasoning Tasks
  3. 2025Exposing Numeracy Gaps: A Benchmark to Evaluate Fundamental Numerical Abilities in Large Language Models
  4. 2025Reliable Reasoning Path: Distilling Effective Guidance for LLM Reasoning with Knowledge Graphs
  5. 2023Exploring the Numerical Reasoning Capabilities of Language Models: A Comprehensive Analysis on Tabular Data
  6. 2025A Large Language Model Based Method for Complex Logical Reasoning over Knowledge Graphs
  7. 2025Mixture of Length and Pruning Experts for Knowledge Graphs Reasoning
  8. 2026ANI: Adaptive Numerical Injection for Unifying Semantic and Arithmetic Representations in Numerical Reasoning