TopicTag: Automatic Annotation of NMF Topic Models Using Chain of Thought and Prompt Tuning with LLMs

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Wanna, Selma, Barron, Ryan, Solovyev, Nick, Eren, Maksim E., Bhattarai, Manish, Rasmussen, Kim, Alexandrov, Boian S.
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866909272305041408
author Wanna, Selma
Barron, Ryan
Solovyev, Nick
Eren, Maksim E.
Bhattarai, Manish
Rasmussen, Kim
Alexandrov, Boian S.
author_facet Wanna, Selma
Barron, Ryan
Solovyev, Nick
Eren, Maksim E.
Bhattarai, Manish
Rasmussen, Kim
Alexandrov, Boian S.
contents Topic modeling is a technique for organizing and extracting themes from large collections of unstructured text. Non-negative matrix factorization (NMF) is a common unsupervised approach that decomposes a term frequency-inverse document frequency (TF-IDF) matrix to uncover latent topics and segment the dataset accordingly. While useful for highlighting patterns and clustering documents, NMF does not provide explicit topic labels, necessitating subject matter experts (SMEs) to assign labels manually. We present a methodology for automating topic labeling in documents clustered via NMF with automatic model determination (NMFk). By leveraging the output of NMFk and employing prompt engineering, we utilize large language models (LLMs) to generate accurate topic labels. Our case study on over 34,000 scientific abstracts on Knowledge Graphs demonstrates the effectiveness of our method in enhancing knowledge management and document organization.
format Preprint
id arxiv_https___arxiv_org_abs_2407_19616
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle TopicTag: Automatic Annotation of NMF Topic Models Using Chain of Thought and Prompt Tuning with LLMs
Wanna, Selma
Barron, Ryan
Solovyev, Nick
Eren, Maksim E.
Bhattarai, Manish
Rasmussen, Kim
Alexandrov, Boian S.
Machine Learning
Artificial Intelligence
Computation and Language
Topic modeling is a technique for organizing and extracting themes from large collections of unstructured text. Non-negative matrix factorization (NMF) is a common unsupervised approach that decomposes a term frequency-inverse document frequency (TF-IDF) matrix to uncover latent topics and segment the dataset accordingly. While useful for highlighting patterns and clustering documents, NMF does not provide explicit topic labels, necessitating subject matter experts (SMEs) to assign labels manually. We present a methodology for automating topic labeling in documents clustered via NMF with automatic model determination (NMFk). By leveraging the output of NMFk and employing prompt engineering, we utilize large language models (LLMs) to generate accurate topic labels. Our case study on over 34,000 scientific abstracts on Knowledge Graphs demonstrates the effectiveness of our method in enhancing knowledge management and document organization.
title TopicTag: Automatic Annotation of NMF Topic Models Using Chain of Thought and Prompt Tuning with LLMs
topic Machine Learning
Artificial Intelligence
Computation and Language
url https://arxiv.org/abs/2407.19616