HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Task-oriented Word Embedding for Text Classification

{He-Yan Huang Yang Gao Qian Liu Yuxin Tian Xiaochi Wei Luyang Liu}

Task-oriented Word Embedding for Text Classification

Abstract

Distributed word representation plays a pivotal role in various natural language processing tasks. In spite of its success, most existing methods only consider contextual information, which is suboptimal when used in various tasks due to a lack of task-specific features. The rational word embeddings should have the ability to capture both the semantic features and task-specific features of words. In this paper, we propose a task-oriented word embedding method and apply it to the text classification task. With the function-aware component, our method regularizes the distribution of words to enable the embedding space to have a clear classification boundary. We evaluate our method using five text classification datasets. The experiment results show that our method significantly outperforms the state-of-the-art methods.

Benchmarks

BenchmarkMethodologyMetrics
sentiment-analysis-on-imdbToWE-SG
Accuracy: 90.8
sentiment-analysis-on-sst-2-binaryToWE-CBOW
Accuracy: 78.8
text-classification-on-ag-newsToWE-SG
Error: 14.0

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
Task-oriented Word Embedding for Text Classification | Papers | HyperAI