CLJul 8, 2024

Open-world Multi-label Text Classification with Extremely Weak Supervision

arXiv:2407.05609v125 citationsh-index: 13
Originality Highly original
AI Analysis

It addresses the problem of multi-label classification without labels or known label spaces for users in text analysis, representing a novel method for a known bottleneck.

The paper tackles open-world multi-label text classification with extremely weak supervision, where only brief descriptions are provided, and introduces X-MLClass, which improves ground-truth label space coverage by 40% on the AAPD dataset and achieves the best end-to-end classification accuracy.

We study open-world multi-label text classification under extremely weak supervision (XWS), where the user only provides a brief description for classification objectives without any labels or ground-truth label space. Similar single-label XWS settings have been explored recently, however, these methods cannot be easily adapted for multi-label. We observe that (1) most documents have a dominant class covering the majority of content and (2) long-tail labels would appear in some documents as a dominant class. Therefore, we first utilize the user description to prompt a large language model (LLM) for dominant keyphrases of a subset of raw documents, and then construct a (initial) label space via clustering. We further apply a zero-shot multi-label classifier to locate the documents with small top predicted scores, so we can revisit their dominant keyphrases for more long-tail labels. We iterate this process to discover a comprehensive label space and construct a multi-label classifier as a novel method, X-MLClass. X-MLClass exhibits a remarkable increase in ground-truth label space coverage on various datasets, for example, a 40% improvement on the AAPD dataset over topic modeling and keyword extraction methods. Moreover, X-MLClass achieves the best end-to-end multi-label classification accuracy.

Code Implementations1 repo
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes