ICMLJun, 2024

综合翻译聚焦特征解释:从表格数据中获取预测信号

TL;DR通过对注意力机制进行研究并将其建模为从 Gumbel-Softmax 分布中采样的潜在变量,我们提出了 InterpreTabNet,该模型通过 KL 散度正则化实现了注意力掩码中不同概念的学习,从而改善了预测结果的解释性和预测关键要素的确定性,并结合大型语言模型和提示工程方法来帮助解释我们模型中特征之间的相互依赖关系。在真实数据集上的综合实验表明,InterpreTabNet 在解释表格数据方面优于先前的方法,并同时保持了较高的准确性。