Symmetry and the Entropy of Small-World Structures and Graphs. - Ioannis Kontoyiannis; Yi Heng Lim; Katia Papakonstantinopoulou; Wojciech Szpankowski (ISIT, 2021)

Reducing the Long Tail Losses in Scientific Emulations with Active Learning. - Yi Heng Lim; Muhammad Firmansyah Kasim (CoRR, 2021)

Constants of motion network. - Muhammad Firmansyah Kasim; Yi Heng Lim (NeurIPS, 2022)

Compression and symmetry of small-world graphs and structures. - Ioannis Kontoyiannis; Yi Heng Lim; Katia Papakonstantinopoulou; Wojciech Szpankowski (Commun. Inf. Syst., 2022)

Unifying physical systems' inductive biases in neural ODE using dynamics constraints. - Yi Heng Lim; Muhammad Firmansyah Kasim (Trans. Mach. Learn. Res., 2023)

Parallelizing non-linear sequential models over the sequence length. - Yi Heng Lim; Qi Zhu; Joshua Selfridge; Muhammad Firmansyah Kasim (ICLR, 2024)