arrow
返回

BoostXML: Gradient Boosting for Extreme Multilabel Text Classification With Tail Labels

delete2024-11-01
delete0
PRE
AI
李
李凤志 (Fengzhi Li)
Y
Yuan Zuo
H
Hao Lin
J
Junjie Wu *
DOI:10.1109/TNNLS.2023.3285294delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Multilabel learning involving hundreds of thousands or even millions of labels is referred to as extreme multilabel learning (XML), in which the labels often follow a powerlaw distribution with the majority occurring in very few data points as tail labels. Recent years have witnessed the intensive use of deep-learning methods for high-performance XML, but they are typically optimized for the head labels with abundant training instances and less consider the performance on tail labels, which, however, like the needles in haystacks, are often the focus of attention in real-life applications. In light of this, we present BoostXML, a deep learning-based XML method for extreme multilabel text classification, enhanced greatly by gradient boosting. In BoostXML, we pay more attention to tail labels in each Boosting Step by optimizing the residual mostly from unfitted training instances with tail labels. A Corrective Step is further proposed to avoid the mismatching between the text encoder and weak learners during optimization, which reduces the risk of falling into local optima and improves model performance. A Pretraining Step is also introduced in the initial stage of BoostXML to avoid exorbitant bias to tail labels. Extensive experiments on five benchmark datasets with stateof-the-art baselines demonstrate the advantage of BoostXML in tail-label prediction.
Keyword:
Corrective step
deep learning
extreme multilabel learning (XML)
gradient boosting
tail labels

期刊

IEEE Transactions on Neural Networks and Learning Systems 封面图
IEEE Transactions on Neural Networks and Learning Systems
IF:
8.9
论文数:
7.6K
被引数:
7.2W

机构

B
Beihang University
学者数:
5.2W
论文数: 4.1W
被引数: 37
引用论文

引用论文

EFFICACY OF TWO CANINE DISTEMPER VACCINES IN WILD NEARCTIC RIVER OTTERS (LONTRA CANADENSIS)
err2014-09-01
err0
PREAI
errSteven T. Peper; Randall L. Peper; George V. Kollias; Robert P. Brooks; Sadie S. Stevens; Thomas L. Serfass
err分享
err收藏
Strengthening of RC dapped-end beams under torsional moment
err2014-10-01
err0
PREAI
errAhmed Mohamed Atta; Tarek Fawzy El-Shafiey
err分享
err收藏
Introduction
err2005-08-25
err0
PREAI
errMary Coleman; Catalina Betancur
err分享
err收藏
Cracks at mortar-stone interfaces
err1995-10-01
err0
PREAI
errD.M. Tan; E.K. Tschegg; H. Rotter; H.O.K. Kirchner
err分享
err收藏
err分享
err收藏
Bonsai: diverse and shallow trees for extreme multi-label classification
err2020-08-23
err89
errOAAI
errKhandagale, Sujay; Xiao, Han; Babbar, Rohit
err分享
err收藏
学者 查看更多内容