learned-clustering

EMAGN 是一种用于交通预测的高效多注意力图网络，通过学习聚类实现自注意力的线性化，将复杂度从二次降至线性，同时保持接近全注意力模型的准确性，并在训练时间、推理时间和 GPU 内存方面显著降低。