HGAP: Boosting Permutation Invariant and Permutation Equivariant in Multi-Agent Reinforcement Learning via Graph Attention Network. - Bor-Jiun Lin; Chun-Yi Lee (ICML, 2024)

Enhancing Memory and Imagination Consistency in Diffusion-based World Models via Linear-Time Sequence Modeling. - Jia-Hua Lee; Bor-Jiun Lin; Wei-Fang Sun; Chun-Yi Lee (CoRR, 2025)