Publications

Leveraging Online Olympiad-Level Math Problems for LLMs Training and Contamination-Resistant Evaluation

Sadegh Mahdavi, Muchen Li, Kaiwen Liu, Christos Thrampoulidis, Leonid Sigal, Renjie Liao

PDF Code

Leveraging Environment Interaction for Automated PDDL Translation and Planning with Large Language Models

Sadegh Mahdavi, Raquel Aoki, Keyi Tang, Yanshuai Cao

PDF Code DOI

From Graph Diffusion to Graph Classification

Jia Jun Cheng Xian, Sadegh Mahdavi, Renjie Liao, Oliver Schulte

PDF

Memorization Capacity of Multi-Head Attention in Transformers

Sadegh Mahdavi, Renjie Liao, Christos Thrampoulidis

PDF Code

Revisiting the Equivalence of In-Context Learning and Gradient Descent: The Impact of Data Distribution

Sadegh Mahdavi, Renjie Liao, Christos Thrampoulidis

DOI

Towards Better Out-of-Distribution Generalization of Neural Algorithmic Reasoning Tasks

Sadegh Mahdavi, Kevin Swersky, Thomas Kipf, Milad Hashemi, Christos Thrampoulidis, Renjie Liao

PDF Code