HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Label-Specific Document Representation for Multi-Label Text Classification

{Liping Jing Boli Chen Lin Xiao Xin Huang}

Label-Specific Document Representation for Multi-Label Text Classification

Abstract

Multi-label text classification (MLTC) aims to tag most relevant labels for the given document. In this paper, we propose a Label-Specific Attention Network (LSAN) to learn a label-specific document representation. LSAN takes advantage of label semantic information to determine the semantic connection between labels and document for constructing label-specific document representation. Meanwhile, the self-attention mechanism is adopted to identify the label-specific document representation from document content information. In order to seamlessly integrate the above two parts, an adaptive fusion strategy is proposed, which can effectively output the comprehensive label-specific document representation to build multi-label text classifier. Extensive experimental results demonstrate that LSAN consistently outperforms the state-of-the-art methods on four different datasets, especially on the prediction of low-frequency labels. The code and hyper-parameter settings are released to facilitate other researchers.

Benchmarks

BenchmarkMethodologyMetrics
multi-label-text-classification-on-aapdLSAN
P@1: 85.28
P@3: 61.12
P@5: 41.84
nDCG@3: 80.84
nDCG@5: 84.78

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Label-Specific Document Representation for Multi-Label Text Classification | Papers | HyperAI