arXiv · Machine Learning· Yuen Chen, Yulun Wu, Samuel Sharpe, Igor Melnyk, Nam H. Nguyen, Furong Huang, C. Bayan Bruss, Rizal Fathony·· 9 小时前AI 评分33
BRIDGE:弥合序列与图建模鸿沟的端到端学习架构
Bridging the Divide: End-to-End Sequence-Graph Learning
AI 导读
BRIDGE提出端到端架构,在单一目标下联合训练序列模型与图模块,让梯度跨组件传递,以学习适配任务的表示。其TOKENXATTN层通过token级交叉注意力,在相邻序列的特定事件间传递信息。在关系预测和欺诈检测任务中,BRIDGE在排序与分类指标上均优于静态图模型、时序图方法和仅使用序列的基线。
来源:arXiv · Machine Learning · arxiv.org