Search Results for author: Suyang Dai

Found 2 papers, 1 papers with code

HAXMLNet: Hierarchical Attention Network for Extreme Multi-Label Text Classification

no code implementations24 Mar 2019 Ronghui You, Zihan Zhang, Suyang Dai, Shanfeng Zhu

Extreme multi-label text classification (XMTC) addresses the problem of tagging each text with the most relevant labels from an extreme-scale label set.

General Classification Multi Label Text Classification +2

AttentionXML: Label Tree-based Attention-Aware Deep Model for High-Performance Extreme Multi-Label Text Classification

3 code implementations NeurIPS 2019 Ronghui You, Zihan Zhang, Ziye Wang, Suyang Dai, Hiroshi Mamitsuka, Shanfeng Zhu

We propose a new label tree-based deep learning model for XMTC, called AttentionXML, with two unique features: 1) a multi-label attention mechanism with raw text as input, which allows to capture the most relevant part of text to each label; and 2) a shallow and wide probabilistic label tree (PLT), which allows to handle millions of labels, especially for "tail labels".

General Classification Multi-Label Text Classification +3

Cannot find the paper you are looking for? You can Submit a new open access paper.