Structured, flexible, and robust: benchmarking and improving large language models towards more human-like behavior in out-of-distribution reasoning tasks. - Katherine M. Collins; Catherine Wong; Jiahai Feng; Megan Wei; Josh Tenenbaum 001 (CogSci, 2022)

The Neuro-Symbolic Inverse Planning Engine (NIPE): Modeling Probabilistic Social Inferences from Linguistic Inputs. - Lance Ying; Katherine M. Collins; Megan Wei; Cedegao E. Zhang; Tan Zhi-Xuan; Adrian Weller; Joshua B. Tenenbaum; Lionel Wong (CoRR, 2023)

Do Music Generation Models Encode Music Theory? - Megan Wei; Michael Freeman; Chris Donahue; Chen Sun 002 (ISMIR, 2024)

Prevailing Research Areas for Music AI in the Era of Foundation Models. - Megan Wei; Mateusz Modrzejewski; Aswin Sivaraman; Dorien Herremans (CoRR, 2024)

Language-Informed Synthesis of Rational Agent Models for Grounded Theory-of-Mind Reasoning On-The-Fly. - Lance Ying; Ryan Truong; Katherine M. Collins; Cedegao E. Zhang; Megan Wei; Tyler Brooke-Wilson; Tan Zhi-Xuan; Lionel Wong; Joshua B. Tenenbaum (CoRR, 2025)