Yiming Yang
Professor
Studies machine learning for text, graphs and time series, including LLM reasoning and neural architecture search.
Yiming Yang's lab at CMU works on accelerators, chips & architecture, time series, anomaly & spatio-temporal data and reasoning & agents. 52 papers since 2023, 32 at top venues, mostly ICLR and ICML. Applications close .
At a glance
- papers, 2023–2026
- 14/ year
- at T1 venues
- 62%
- PhD students
- —
- MS students
- —
- alumni listed
- —
Papers by yearfrom the lab homepage, 2023–2026
- 10 papers2026 · T1 6 · collecting
- 17 papers2025 · T1 15
- 12 papers2024 · T1 5
- 13 papers2023 · T1 6
T1 venuesOtherVenues: ICLR, ICML, NeurIPS, EMNLP, AAAI
Research focuspaper titles that use each topic's words
- Reasoning & agents6 papers
- Time series, anomaly & spatio-temporal data4 papers
- Accelerators, chips & architecture0 papers
- llm reasoning0 papers
- neural architecture0 papers
Paper list
From the lab homepage.
202610 papers
- AAAI (T1)CO-Bench: Benchmarking Language Model Agents in Algorithm Search for Combinatorial Optimization (opens in a new tab)
- Transactions on Machine Learning Research (TMLR), .CodePDE: An Inference Framework for LLM-driven PDE Solver Generation (opens in a new tab)
- ICLR (T1)CoMind: Toward Community-Driven Agents for Machine Learning Engineering (opens in a new tab)
- ICLR (T1)FrontierCO: Real-World and Large-Scale Evaluation of Machine Learning Solvers for Combinatorial Optimization (opens in a new tab)
- Weihua Du, Hailei Gong, Zhan Ling, Kang Liu, Lingfeng Shen, Xuesong Yao, Yufei Xu, Dingyuan Shi, Yiming Yang, Jiecao Chen, ICLR .Generalizable End-to-End Tool-Use RL with Synthetic CodeGym (opens in a new tab)
- ICLR (T1)Sample Complexity and Representation Ability of Test-time Scaling Paradigms (opens in a new tab)
- Weiwei Sun, Miao Lu, Zhan Ling, Kang Liu, Xuesong Yao, Yiming Yang, Jiecao Chen, ICMLScaling Long-Horizon Agent via Context Folding (opens in a new tab)
- ICML (T1)Unsupervised Diffusion for Combinatorial Optimization via Adjoint Matching
- Yixin Huang, Shengyu Feng, Yiming Yang, ICMLUnsupervised Neural Leagevin Sampler for Mixed Integer Linear Programming
- ICLR (T1)ZeroGR: A Generalizable and Scalable Framework for Zero-shot Generative Retrieval (opens in a new tab)
202517 papers
- Future Generation Computer Systems (Volume 175, article number 108075Alternative Mixed Integer Linear Programming Optimization for Joint Job Scheduling and Data Allocation in Grid Computing (opens in a new tab)
- NAACL (T1)Direct Preference Optimization of Video Large Multimodal Models from Language Model Reward (opens in a new tab)
- Weiwei Sun, Haokun Liu, Nikhil Kandpal, Colin Raffel, Yiming Yang, NeurIPS .Enhancing Training Data Attribution with Representational Optimization (opens in a new tab)
- NAACL (T1)Few-shot Personalization of LLMs with Mis-aligned Responses (opens in a new tab)
- ACL (T1)Improve Vision Language Model Chain-of-thought Reasoning (opens in a new tab)
- NeurIPS (T1)Improving Retrieval-Augmented Generation through Multi-Agent Reinforcement Learning (opens in a new tab)
- ICLR (T1)Inference Scaling Laws: An Empirical Analysis of Compute-Optimal Inference for LLM Problem-Solving (opens in a new tab)
- ICML (T1)Large Language Models are Demonstration Pre-Selectors for Themselves (opens in a new tab)
- ICLR (T1)Lean-STaR: Learning to Interleave Thinking and Proving (opens in a new tab)
- ICML (T1)Maximum Update Parametrization and Zero-Shot Hyperparameter Transfer for Fourier Neural Operators (opens in a new tab)
- ICML (T1)Optimizing Temperature for Language Models with Multi-Sample Inference (opens in a new tab)
- ICML (T1)Regularized Langevin Dynamics for Combinatorial Optimization (opens in a new tab)
202412 papers
- SC24 AI4S Workshop.AI Surrogate Model for Distributed Computing Workloads (opens in a new tab)
- 13088-13110Aligning Large Multimodal Models with Factually Augmented RLHF (opens in a new tab)
- —CoRR abs/2403.04307 (opens in a new tab)
- NearIPSEasy-to-Hard Generalization: Scalable Alignment Beyond Human Supervision (opens in a new tab)
- ICLR (T1)Functional Interpolation for Relative Positions improves Long Context Transformers (opens in a new tab)
- 659-673Generation-driven Contrastive Self-training for Zero-shot Text Classification with Instruction-following LLM (opens in a new tab)
- ICML (T1)In-Context Principle Learning from Mistakes (opens in a new tab)
- 2278-2286Learning a Fourier Transform for Linear Relative Positional Encodings in Transformers (opens in a new tab)
- ICLR (T1)Learning Performance-Improving Code Edits (opens in a new tab)
- EMNLP (T1)Learning to Correct for QA Reasoning with Black-box LLMs (opens in a new tab)
- SIGIR KEYNOTE.Representation Learning and Information Retrieval (opens in a new tab)
- ICLR (T1)SALMON: Self-Alignment with Instructable Reward Models (opens in a new tab)
202313 papers
- 33135-33155A Neural PDE Solver with Temporal Stencil Modeling (opens in a new tab)
- ICML’ workshop.Accelerating Diffusion-based Combinatorial Optimization Solvers by Progressive Distillation (opens in a new tab)
- EMNLP (T1)Active Retrieval Augmented Generation (opens in a new tab)
- Findings of EMNLPCompleQA: Benchmarking the Impacts of Knowledge Graph Completion Methods on Question Answering
- NeurIPS (T1)DIFUSCO: Graph-based Diffusion Solvers for Combinatorial Optimization (opens in a new tab)
- EMNLP (T1)Let's Sample Step by Step: Adaptive-Consistency for Efficient Reasoning with LLMs (opens in a new tab)
- EACLLong-tailed Extreme Multi-label Text Classification with Generated Pseudo Label Descriptions (opens in a new tab)
- 10764-10799PAL: Program-aided Language Models (opens in a new tab)
- 14897-14911PESCO: Prompt-enhanced Self Contrastive Learning for Zero-shot Text Classification (opens in a new tab)
- NeurIPS (T1)Principle-Driven Self-Alignment of Language Models from Scratch with Minimal Human Supervision (opens in a new tab)
- ICLR (T1)Recitation-Augmented Language Models (opens in a new tab)
- 2334-2338Retrieval-Enhanced Generative Model for Large-Scale Knowledge Graph Completion (opens in a new tab)
Where alumni went
Alumni are not listed on the lab homepage.
Apply to CMUFall 2027
CMU SCS Computer Science PhD
- Round
- Single round
- Opens
- GRE
- See program page
- English
- TOEFL / IELTS / Duolingo (F-1/J-1 non-native speakers)
- Fee
- USD 100 (USD 80 by early deadline)
Confirmed on the program page.
Similar labsnearest by research summary
- USStanford
Emma Brunskill
- Reinforcement learning
- GBImperial
Francesca Toni
- Trustworthy & safe ML
- CNTsinghua
Jie Tang
- Large language models
- USGeorgia Tech
Ling Liu
- Trustworthy & safe ML
- USCMU
Louis-Philippe Morency
- Signal processing & information theory
- USMichigan
Elizabeth Bondi-Kelly
- Reasoning & neuro-symbolic AI
Your notes
Save this lab to keep notes.
Data
Some numbers could not be verified:
- The member list could not be found on the lab site.
- The alumni list could not be found on the lab site.
- Collected
- Verified
- Papers
- The lab homepage
- Members
- Not collected
- Alumni
- Not collected
Tiers are venue classifications, not lab rankings.
CMUcloses