Graph-Convolutional Networks: Named Entity Recognition and Large Language Model Embedding in Document Clustering - Ecole Normale Supérieure paris-Saclay
Pré-Publication, Document De Travail Année : 2024

Graph-Convolutional Networks: Named Entity Recognition and Large Language Model Embedding in Document Clustering

Imed Keraghel
Mohamed Nadif

Résumé

Recent advances in machine learning, particularly Large Language Models (LLMs) such as BERT and GPT, provide rich contextual embeddings that improve text representation. However, current document clustering approaches often ignore the deeper relationships between named entities (NEs) and the potential of LLM embeddings. This paper proposes a novel approach that integrates Named Entity Recognition (NER) and LLM embeddings within a graph-based framework for document clustering. The method builds a graph with nodes representing documents and edges weighted by named entity similarity, optimized using a graph-convolutional network (GCN). This ensures a more effective grouping of semantically related documents. Experimental results indicate that our approach outperforms conventional co-occurrence-based methods in clustering, notably for documents rich in named entities.
Fichier principal
Vignette du fichier
ECIR_OCT_2024_VF.pdf (1.85 Mo) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)
licence
Domaine public

Dates et versions

hal-04848839 , version 1 (19-12-2024)

Licence

Domaine public

Identifiants

  • HAL Id : hal-04848839 , version 1

Citer

Imed Keraghel, Mohamed Nadif. Graph-Convolutional Networks: Named Entity Recognition and Large Language Model Embedding in Document Clustering. 2024. ⟨hal-04848839⟩
0 Consultations
0 Téléchargements

Partager

More