LLM-TAKE: Theme Aware Keyword Extraction Using Large Language Models

TL;DR

This paper introduces LLM-TAKE, a framework using large language models for theme-aware keyword extraction that improves accuracy and diversity in product metadata analysis.

Contribution

It presents a novel LLM-based framework with two variations for extractive and abstractive theme generation in e-commerce, addressing limitations of traditional models.

Findings

Enhanced accuracy over benchmark models

Improved diversity in keyword extraction

Effective in e-commerce product metadata

Abstract

Keyword extraction is one of the core tasks in natural language processing. Classic extraction models are notorious for having a short attention span which make it hard for them to conclude relational connections among the words and sentences that are far from each other. This, in turn, makes their usage prohibitive for generating keywords that are inferred from the context of the whole text. In this paper, we explore using Large Language Models (LLMs) in generating keywords for items that are inferred from the items textual metadata. Our modeling framework includes several stages to fine grain the results by avoiding outputting keywords that are non informative or sensitive and reduce hallucinations common in LLM. We call our LLM-based framework Theme-Aware Keyword Extraction (LLM TAKE). We propose two variations of framework for generating extractive and abstractive themes for products in an E commerce setting. We perform an extensive set of experiments on three real data sets and show that our modeling framework can enhance accuracy based and diversity based metrics when compared with benchmark models.