Contrastive Learning Of Structured World Models

Paper : Contrastive Learning Of Structured World Models Code : official 摘要 作者在強化學習任務中,提出了使用GNN來建模多物體之間的關係的方法,使用transE的思路來進行狀態的預測,並通過Contrastive Loss 來進行訓練,提出 C-SWM 模型。作者在雅達利遊戲,多物體物理模擬等任務上達到了SOTA。 方法 C
相關文章
相關標籤/搜索