HyperAIHyperAI

Command Palette

Search for a command to run...

层次文本分类与强化标签分配

Yuning Mao Jingjing Tian Jiawei Han Xiang Ren

摘要

现有的层次文本分类(HTC)方法试图在模型训练过程中捕捉标签层次结构,但在推理时要么对每个标签做出局部决策,要么完全忽略层次信息。为了解决训练与推理之间的不匹配问题,并以更加系统的方式建模标签依赖关系,我们将HTC表述为一个马尔可夫决策过程,并提出通过深度强化学习来学习一种标签分配策略(Label Assignment Policy),以确定对象的放置位置以及何时停止分配过程。所提出的HiLAP方法在训练和推理阶段均以一致的方式探索层次结构,并作出相互依赖的决策。作为通用框架,HiLAP可以结合不同的神经编码器作为基础模型进行端到端的训练。在五个公开数据集和四种基础模型上的实验表明,HiLAP在宏F1指标上比扁平分类器平均提高了33.4%,并且大幅优于当前最先进的HTC方法。数据和代码可在https://github.com/morningmoni/HiLAP 获取。


用 AI 构建 AI

从创意到上线——通过免费 AI 协同编码、开箱即用的环境和最优惠的 GPU 价格,加速您的 AI 开发。

AI 协同编码
开箱即用的 GPU
最优定价

HyperAI Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供