Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Neural Topic Modeling with Large Language Models in the Loop

About

Topic modeling is a fundamental task in natural language processing, allowing the discovery of latent thematic structures in text corpora. While Large Language Models (LLMs) have demonstrated promising capabilities in topic discovery, their direct application to topic modeling suffers from issues such as incomplete topic coverage, misalignment of topics, and inefficiency. To address these limitations, we propose LLM-ITL, a novel LLM-in-the-loop framework that integrates LLMs with Neural Topic Models (NTMs). In LLM-ITL, global topics and document representations are learned through the NTM. Meanwhile, an LLM refines these topics using an Optimal Transport (OT)-based alignment objective, where the refinement is dynamically adjusted based on the LLM's confidence in suggesting topical words for each set of input words. With the flexibility of being integrated into many existing NTMs, the proposed approach enhances the interpretability of topics while preserving the efficiency of NTMs in learning topics and document representations. Extensive experiments demonstrate that LLM-ITL helps NTMs significantly improve their topic interpretability while maintaining the quality of document representation. Our code and datasets are available at https://github.com/Xiaohao-Yang/LLM-ITL

Xiaohao Yang, He Zhao, Weijie Xu, Yuanyuan Qi, Jueqing Lu, Dinh Phung, Lan Du• 2024

Related benchmarks

TaskDatasetResultRank
Topic ModelingBothering
UT Score70
44
Topic ModelingAskAcademia
UT0.715
44
Topic ModelingTeslaModel3
UT Score65
44
Topic ModelingYelp 5Cate
Topic Coherence (TC)45.4
14
Topic ModelingAmazon-10Cate
TC Score0.434
14
Goal-relevance EvaluationTeslaModel3 (test)
GS47.36
11
Goal-relevance EvaluationAskAcademia (test)
GS41.04
11
Goal-relevance EvaluationBothering (test)
Goal Score36.53
11
Topic ModelingTeslaModel3 (test)
Cp-0.0419
11
Topic ModelingAskAcademia (test)
Cp0.0561
11
Showing 10 of 11 rows

Other info

Follow for update