arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2302.07253cs.LGcond-mat.dis-nncs.CVq-bio.NCstat.ML
收藏

能量Transformer

Energy Transformer

IBM Research(IBM研究院);Georgia Tech(佐治亚理工学院);RPI(伦斯勒理工学院);MIT-IBM Watson AI Lab(MIT-IBM沃森人工智能实验室)

Benjamin Hoover, Yuchen Liang, Bao Pham, Rameswar Panda, Hendrik Strobelt, Duen Horng Chau, Mohammed J. Zaki, Dmitry Krotov

AI总结:

本文融合注意力、能量模型与联想记忆,提出通过注意力层最小化 token 关系能量函数的 Energy Transformer(ET),并在图像补全、图异常检测与分类中验证其效果。

摘要:

我们的工作结合了机器学习中三种有前景范式的若干方面,即注意力机制、基于能量的模型和联想记忆。注意力是推动现代深度学习成功的核心动力,但其缺乏清晰的理论基础。基于能量的模型为判别式和生成式任务提供了一种有原则的方法,但能量泛函的设计并不简单。与此同时,Dense Associative Memory模型(稠密联想记忆模型)或Modern Hopfield Networks(现代Hopfield网络)具有成熟的理论基础,并允许对能量函数进行直观设计。我们提出了一种名为Energy Transformer(简称ET)的新颖架构,它使用一系列注意力层,这些层经过专门设计,用于最小化一个专门构造的能量函数,该函数负责表示token之间的关系。在本工作中,我们介绍了ET的理论基础,利用图像补全任务探索了其实证能力,并在图异常检测和图分类任务上取得了强劲的定量结果。

英文摘要:

Our work combines aspects of three promising paradigms in machine learning, namely, attention mechanism, energy-based models, and associative memory. Attention is the power-house driving modern deep learning successes, but it lacks clear theoretical foundations. Energy-based models allow a principled approach to discriminative and generative tasks, but the design of the energy functional is not straightforward. At the same time, Dense Associative Memory models or Modern Hopfield Networks have a well-established theoretical foundation, and allow an intuitive design of the energy function. We propose a novel architecture, called the Energy Transformer (or ET for short), that uses a sequence of attention layers that are purposely designed to minimize a specifically engineered energy function, which is responsible for representing the relationships between the tokens. In this work, we introduce the theoretical foundations of ET, explore its empirical capabilities using the image completion task, and obtain strong quantitative results on the graph anomaly detection and graph classification tasks.

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

↑