Liheng Ma

Liheng Ma

PhD Candidate · McGill University & Mila

I am a PhD candidate in Electrical & Computer Engineering at McGill University and Mila, advised by Prof. Mark Coates. I received my M.Sc. in Computer Science from the same institutions, supervised by Prof. Reihaneh Rabbany and Dr. Adriana Romero‑Soriano, and was a visiting student with the Torr Vision Group at the University of Oxford, working with Prof. Philip Torr.

My doctoral research focuses on expressive neural network architectures for learning real-world signals with geometric symmetries and temporal dynamics—mainly graphs (permutation symmetry group) and time-series (temporal dynamics). I develop neural networks with strong perception capacity, expressivity, and efficiency.

More recently, I have explored RL post-training for LLM reasoning and efficient world-action models. My current research focuses on developing low-latency world action models. My work has been published at ICML, AISTATS, AAAI, AAMAS, KDD, etc., and has accumulated over 1,000 citations.

liheng.ma at mail.mcgill.ca

Selected Publications

* indicates equal contribution.

arXiv 2026

Faster-WAM: Do World Action Models Need Deep Action Modules?

L. Ma*, R. H. Yang*, Z. Zhang*, M. Clemente, Z. Hu, T. Cao, Y. Zhang

arXiv 2026

Rethinking Groups in Critic-Free RLVR

Y. Wu*, L. Ma*, L. Xiao, M. Li, X. Wang, Y. Zhang, J.-Y. Nie

AAMAS 2026

Advancing Multi-Agent RAG system with Minimalist Reinforcement Learning

Y. Wu*, L. Ma*, M. Li*, J. Zhou, J. Hao, H. Leung, I. King, Y. Zhang, J.-Y. Nie

TMLR 2026

Plain Transformers Can Be Powerful Graph Learners

L. Ma, S. Pal, Y. Zhang, P. H. S. Torr, M. Coates

arXiv 2025

It Takes Two: Your GRPO Is Secretly DPO

Y. Wu*, L. Ma*, L. Ding, M. Li, X. Wang, K. Chen, Z. Su, Z. Zhang, C. Huang, Y. Zhang, M. Coates, J.-Y. Nie

ICML 2025

SKOLR: Structured Koopman Operator Linear RNN for Time-Series Forecasting

Y. Zhang, L. Ma, A. Valkanas, B. N. Oreshkin, M. Coates

ICML 2024

CKGConv: General Graph Convolution with Continuous Kernels

L. Ma, S. Pal, Y. Zhang, J. Zhou, Y. Zhang, M. Coates

AISTATS 2024

Multi-Resolution Time-Series Transformer for Long-term Forecasting

Y. Zhang*, L. Ma*, S. Pal, Y. Zhang, M. Coates

ICML 2023

Graph Inductive Biases in Transformers without Message Passing

L. Ma*, C. Lin*, D. Lim, A. Romero-Soriano, P. K. Dokania, M. Coates, P. H. S. Torr, S.-N. Lim

PAKDD 2021

Graph Attention Networks with Positional Embeddings

L. Ma, R. Rabbany, A. Romero-Soriano

AAAI 2021

Knowledge-Enhanced Top-K Recommendation in Poincaré Ball

C. Ma, L. Ma, Y. Zhang, H. Wu, X. Liu, M. Coates

KDD 2020

Probabilistic Metric Learning with Adaptive Margin for Top-K Recommendation

C. Ma, L. Ma, Y. Zhang, R. Tang, X. Liu, M. Coates

AAAI 2020

Memory Augmented Graph Neural Networks for Sequential Recommendation

C. Ma, L. Ma, Y. Zhang, J. Sun, X. Liu, M. Coates

Academic Service

Reviewer

ICML 2023–2026

Reviewer

NeurIPS 2023–2026

Reviewer

ICLR 2024–2026

Reviewer

LoG 2023–2025

Program Committee

AAAI 2025

Co-organizer

Learning on Graphs Conference Montreal Local Meetup (LOG-MTL) 2024

Reviewer

Neural Networks since 2025

Reviewer

IEEE TPAMI since 2025

Reviewer

ACL ARR since 2026