本文提出了 STD-PLM,一個基於預訓練語言模型(PLM)的統一框架,用於空間時間預測和插值任務。
首先,作者設計了空間時間嵌入模塊,包括基於拓撲結構的節點嵌入和基於週期性的時間嵌入,以充分捕捉空間時間數據的異質性。
其次,作者設計了空間標記器和時間標記器,將空間時間數據轉換為序列標記,使 PLM 能夠理解數據中的空間、時間和空間時間相關性。
最後,作者提出了砂漏注意力模塊,不僅可以顯著提高模型的效率,還可以進一步捕捉非成對和高階的空間時間相關性。
實驗結果表明,STD-PLM 在空間時間預測和插值任務上都表現出了出色的性能,同時在少樣本和零樣本學習任務上也展現了強大的泛化能力。
In un'altra lingua
dal contenuto originale
arxiv.org
Approfondimenti chiave tratti da
by YiHeng Huang... alle arxiv.org 09-11-2024
https://arxiv.org/pdf/2407.09096.pdfDomande più approfondite