Sep, 2023

树形交叉注意力

TL;DR通过树形交叉关注,Tree Cross Attention (TCA) 模块以对数复杂度 O(log(N)) 用于推理时从一组上下文标记中检索信息,与 Cross Attention 相比,TCA 在各种分类和不确定性回归任务上表现相当并且更加高效。