FuDoBa: Fusing Document and Knowledge Graph Based Representations with Bayesian Optimisation
Abstract
Abstract Building on the success of large language models (LLMs), LLM-based representations have dominated the document representation landscape, achieving strong performance on document embedding benchmarks. However, high-dimensional, computationally expensive LLM embeddings can be too generic or inefficient for domain-specific and resource-scarce applications. To address these limitations, we introduce FuDoBa—a Bayesian optimisation-based representation learning method that integrates LLM embeddings with domain-specific structured knowledge, sourced both locally and from external repositories such as WikiData. This fusion produces low-dimensional, task-relevant representations while reducing training complexity and yielding interpretable early-fusion weights for improved classification performance. We demonstrate the effectiveness of our approach on six datasets across two domains, showing that when paired with robust AutoML-based classifiers, our method performs on par with, or surpasses, proprietary LLM-only embedding baselines, while offering modality-wise interpretability and a smaller dimensional footprint.
Fields of Study & Subjects
Topic Modeling
Advanced Graph Neural Networks
Text and Document Classification Technologies
Publication details
- Journal / source
- Machine Learning
- Volume
- 115
- Issue
- 3
- Publication date
- 1 March 2026
- Language
- EN