arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2305.11684cs.LG

用于表格学习的自强化注意力机制

Self-Reinforcement Attention Mechanism For Tabular Learning

Sorbonne Paris Nord University(巴黎北索邦大学);Paris-Saclay University(巴黎萨克雷大学);Groupe BPCE(法国BPCE集团)

Kodjo Mawuena Amekoe, Mohamed Djallel Dilmi, Hanene Azzag, Mustapha Lebbah, Zaineb Chelly Dagdia, Gregoire Jaffre

AI总结:

本文提出自强化注意力机制(SRA),通过权重向量学习可解释表示并调整原始输入,在合成与真实不平衡数据上与基线模型端到端结合,有效提升表格学习性能。

摘要:

除了机器学习模型的高准确率外,在现实问题(如欺诈检测、信用评分)中吸引许多研究人员的,是寻找数据中的隐藏模式;特别是在处理其具有挑战性的不平衡特征时。可解释性也是伴随所使用的机器学习模型的一项关键要求。在这方面,通常更倾向于使用内在可解释的模型,而不是在大多数情况下是黑盒模型的复杂模型。此外,在一些高风险领域,使用线性模型来处理表格数据,即使必须牺牲性能。在本文中,我们引入了自强化注意力(SRA),这是一种新颖的注意力机制,它提供特征的相关性作为权重向量,用于学习可理解的表示。然后,该权重通过逐元素向量乘法来强化或减少原始输入的某些分量。我们在合成和真实世界的不平衡数据上的结果表明,我们提出的 SRA 块在与基线模型的端到端组合中是有效的。

英文摘要:

Apart from the high accuracy of machine learning models, what interests many researchers in real-life problems (e.g., fraud detection, credit scoring) is to find hidden patterns in data; particularly when dealing with their challenging imbalanced characteristics. Interpretability is also a key requirement that needs to accompany the used machine learning model. In this concern, often, intrinsically interpretable models are preferred to complex ones, which are in most cases black-box models. Also, linear models are used in some high-risk fields to handle tabular data, even if performance must be sacrificed. In this paper, we introduce Self-Reinforcement Attention (SRA), a novel attention mechanism that provides a relevance of features as a weight vector which is used to learn an intelligible representation. This weight is then used to reinforce or reduce some components of the raw input through element-wise vector multiplication. Our results on synthetic and real-world imbalanced data show that our proposed SRA block is effective in end-to-end combination with baseline models.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑