Dec, 2023

基于 Transformer 嵌入和层次感知 注意力结构的金融交易分层分类

TL;DR本研究提出了一种基于 Transformer 的 Two-headed DragoNet 模型,用于金融交易的分层多标签分类。我们的模型基于一系列 Transformer 编码层,从两个简短的文本描述(商家名称和业务活动)生成上下文嵌入,然后经过一个上下文融合层和两个输出头根据两级分类法(宏观和微观类别)对交易进行分类。最后,我们提出的基于分类法的注意力层修正了违反给定分类法定义的层次规则的预测。在基于卡片的数据集上,我们的提出方法在宏观类别分类的实验中表现优于传统的机器学习方法,达到了 93% 的 F1 分数;在活期账户数据集上达到了 95% 的 F1 分数。