You can find my publications on Google Scholar.
Research Interests
Selected Works
Deep Delta Learning
arXiv:2601.00417
[RPG] On the Design of KL-Regularized Policy Gradient Algorithms for LLM Reasoning
Conference on Neural Information Processing Systems (NeurIPS 2025 MATH-AI Workshop); See also Thinking Machines Tinker and DeepSeek V3.2
[TPA] Tensor Product Attention Is All You Need
Conference on Neural Information Processing Systems (NeurIPS 2025 Spotlight)
[GPM] Beyond Bradley-Terry Models: A General Preference Model for Language Model Alignment
International Conference on Machine Learning (ICML 2025)
[AutoMathText] Autonomous Data Selection with Zero-shot Generative Classifiers for Mathematical Texts
Findings of the Association for Computational Linguistics (ACL 2025 Findings)
[MMIQC] Augmenting Math Word Problems via Iterative Question Composing
AAAI Conference on Artificial Intelligence (AAAI 2025)
[Proposer-Verifier] Cumulative Reasoning with Large Language Models
Transactions on Machine Learning Research (TMLR)
(* denotes equal contribution)
Recent Publications
Tensor Product Attention Is All You Need
Conference on Neural Information Processing Systems (NeurIPS 2025 Spotlight)
Autonomous Data Selection with Zero-shot Generative Classifiers for Mathematical Texts
Findings of the Association for Computational Linguistics (ACL 2025 Findings)
Beyond Bradley-Terry Models: A General Preference Model for Language Model Alignment
International Conference on Machine Learning (ICML 2025)
Augmenting Math Word Problems via Iterative Question Composing
AAAI Conference on Artificial Intelligence (AAAI 2025)
Beyond Squared Error: Exploring Loss Design for Enhanced Training of Generative Flow Networks
International Conference on Learning Representations (ICLR 2025 Spotlight)
(* denotes equal contribution)
Professional Activities
Teaching
- Teaching Assistant, Machine Learning for Yao class, IIIS, Tsinghua University
Academic Services
- Conference Reviewer: NeurIPS, ICLR, ICML, COLM, AAAI, AISTATS
- Journal Reviewer: ACM TKDD, Neural Computing, Neural Networks