Interaction Modeling with Multiplex Attention - Sun, Fan-Yun; Kauvar, Isaac; Zhang, Ruohan; Li, Jiachen; Kochenderfer, Mykel J; Wu, Jiajun; Haber, Nick (Advances in Neural Information Processing Systems, 2022)