PPaperPicks

Mengdi Wang

Princeton University, Center for Statistics and Machine Learning, Department of Electrical and Computer Engineering, NJ, USA

44 papers at tracked venues · 38 at CORE A* · active 2024–2026

DBLP profile ↗ORCID 0000-0002-2101-9507 ↗Homepage ↗

Venues

NeurIPS ×16ICML ×9ICLR ×8EMNLP ×3AAAI ×2AISTATS ×1CVPR ×1EACL ×1ICCV ×1IJCAI ×1SIGKDD ×1

Frequent coauthors

Ling Yang
×9
Hui Yuan
×8
Jiahao Qiu
×8
Kaixuan Huang
×8
Minshuo Chen
×8
Souradip Chakraborty
×7
Furong Huang
×7
Amrit Singh Bedi
×6
Yue Wu
×6
Dinesh Manocha
×6
Huazheng Wang
×5
Jiacheng Guo
×5

Papers

  1. Jailbreaks as Inference-Time Alignment: A Framework for Understanding Safety Failures in LLMs
    EACL 2026 ·
    James Beetham
    ,
    Souradip Chakraborty
    , Mengdi Wang,
    Furong Huang
    ,
    Amrit Singh Bedi
    ,
    Mubarak Shah
  2. A Common Pitfall of Margin-based Language Model Alignment: Gradient Entanglement
    ICLR 2025 ·
    Hui Yuan
    ,
    Yifan Zeng
    ,
    Yue Wu
    ,
    Huazheng Wang
    , Mengdi Wang,
    Liu Leqi
  3. A First-order Generative Bilevel Optimization Framework for Diffusion Models
    ICML 2025 ·
    Quan Xiao
    ,
    Hui Yuan
    ,
    A F M Saif
    ,
    Gaowen Liu
    ,
    Ramana Rao Kompella
    , Mengdi Wang,
    Tianyi Chen
  4. Co-Evolving LLM Coder and Unit Tester via Reinforcement Learning
    NeurIPS 2025 ·
    Yinjie Wang
    ,
    Ling Yang
    ,
    Ye Tian
    ,
    Ke Shen
    , Mengdi Wang
  5. Collab: Controlled Decoding using Mixture of Agents for LLM Alignment
    ICLR 2025 ·
    Souradip Chakraborty
    ,
    Sujay Bhatt
    ,
    Udari Madhushani Sehwag
    ,
    Soumya Suvra Ghosal
    ,
    Jiahao Qiu
    , Mengdi Wang,
    Dinesh Manocha
    ,
    Furong Huang
    ,
    Alec Koppel
    ,
    Sumitra Ganesh
  6. DISC: Dynamic Decomposition Improves LLM Inference Scaling
    NeurIPS 2025 ·
    Jonathan Light
    ,
    Wei Cheng
    ,
    Benjamin Rivière
    ,
    Yue Wu
    ,
    Masafumi Oyamada
    , Mengdi Wang,
    Yisong Yue
    ,
    Santiago Paternain
    ,
    Haifeng Chen
  7. Deep Reinforcement Learning for Efficient and Fair Allocation of Healthcare Resources
    IJCAI 2025 ·
    Yikuan Li
    ,
    Chengsheng Mao
    ,
    Kaixuan Huang
    ,
    Hanyin Wang
    ,
    Zheng Yu
    , Mengdi Wang,
    Yuan Luo
  8. Diffusion Transformer Captures Spatial-Temporal Dependencies: A Theory for Gaussian Process Data
    ICLR 2025 ·
    Hengyu Fu
    ,
    Zehao Dou
    ,
    Jiawei Guo
    , Mengdi Wang,
    Minshuo Chen
  9. Does Thinking More Always Help? Mirage of Test-Time Scaling in Reasoning Models
    NeurIPS 2025 ·
    Soumya Suvra Ghosal
    ,
    Souradip Chakraborty
    ,
    Avinash Reddy
    ,
    Yifu Lu
    , Mengdi Wang,
    Dinesh Manocha
    ,
    Furong Huang
    ,
    Mohammad Ghavamzadeh
    ,
    Amrit Singh Bedi
  10. Emergent Symbolic Mechanisms Support Abstract Reasoning in Large Language Models
    ICML 2025 ·
    Yukang Yang
    ,
    Declan Campbell
    ,
    Kaixuan Huang
    , Mengdi Wang,
    Jonathan D. Cohen
    ,
    Taylor Whittington Webb
  11. EmoAgent: Assessing and Safeguarding Human-AI Interaction for Mental Health Safety
    EMNLP 2025 ·
    Jiahao Qiu
    ,
    Yinghui He
    ,
    Xinzhe Juan
    ,
    Yimin Wang
    ,
    Yuhan Liu
    ,
    Zixin Yao
    ,
    Yue Wu
    ,
    Xun Jiang
    ,
    Ling Yang
    , Mengdi Wang
  12. Immune: Improving Safety Against Jailbreaks in Multi-modal LLMs via Inference-Time Alignment
    CVPR 2025 ·
    Soumya Suvra Ghosal
    ,
    Souradip Chakraborty
    ,
    Vaibhav Singh
    ,
    Tianrui Guan
    , Mengdi Wang,
    Ahmad Beirami
    ,
    Furong Huang
    ,
    Alvaro Velasquez
    ,
    Dinesh Manocha
    ,
    Amrit Singh Bedi
  13. IterComp: Iterative Composition-Aware Feedback Learning from Model Gallery for Text-to-Image Generation
    ICLR 2025 ·
    Xinchen Zhang
    ,
    Ling Yang
    ,
    Guohao Li
    ,
    Yaqi Cai
    ,
    Jiake Xie
    ,
    Yong Tang
    ,
    Yujiu Yang
    , Mengdi Wang,
    Bin Cui
  14. MATH-Perturb: Benchmarking LLMs' Math Reasoning Abilities against Hard Perturbations
    ICML 2025 ·
    Kaixuan Huang
    ,
    Jiacheng Guo
    ,
    Zihao Li
    ,
    Xiang Ji
    ,
    Jiawei Ge
    ,
    Wenzhe Li
    ,
    Yingqing Guo
    ,
    Tianle Cai
    ,
    Hui Yuan
    ,
    Runzhe Wang
    ,
    Yue Wu
    ,
    Ming Yin
    ,
    Shange Tang
    ,
    Yangsibo Huang
    ,
    Chi Jin
    ,
    Xinyun Chen
    ,
    Chiyuan Zhang
    , Mengdi Wang
  15. MMaDA: Multimodal Large Diffusion Language Models
    NeurIPS 2025 ·
    Ling Yang
    ,
    Ye Tian
    ,
    Bowen Li
    ,
    Xinchen Zhang
    ,
    Ke Shen
    ,
    Yunhai Tong
    , Mengdi Wang
  16. Preacher: Paper-to-Video Agentic System
    ICCV 2025 ·
    Jingwei Liu
    ,
    Ling Yang
    ,
    Hao Luo
    ,
    Fan Wang
    ,
    Hongyan Li
    , Mengdi Wang
  17. ReasonFlux-PRM: Trajectory-Aware PRMs for Long Chain-of-Thought Reasoning in LLMs
    NeurIPS 2025 ·
    Jiaru Zou
    ,
    Ling Yang
    ,
    Jingwen Gu
    ,
    Jiahao Qiu
    ,
    Ke Shen
    ,
    Jingrui He
    , Mengdi Wang
  18. Rectified Diffusion: Straightness Is Not Your Need in Rectified Flow
    ICLR 2025 ·
    Fu-Yun Wang
    ,
    Ling Yang
    ,
    Zhaoyang Huang
    , Mengdi Wang,
    Hongsheng Li
  19. Securing the Language of Life: Inheritable Watermarks from DNA Language Models to Proteins
    NeurIPS 2025 ·
    Zaixi Zhang
    ,
    Ruofan Jin
    ,
    Le Cong
    , Mengdi Wang
  20. Temporal Consistency for LLM Reasoning Process Error Identification
    EMNLP 2025 ·
    Jiacheng Guo
    ,
    Yue Wu
    ,
    Jiahao Qiu
    ,
    Kaixuan Huang
    ,
    Xinzhe Juan
    ,
    Ling Yang
    , Mengdi Wang
  21. Towards Understanding Text Hallucination of Diffusion Models via Local Generation Bias
    ICLR 2025 ·
    Rui Lu
    ,
    Runzhe Wang
    ,
    Kaifeng Lyu
    ,
    Xitai Jiang
    ,
    Gao Huang
    , Mengdi Wang
  22. Training-Free Guidance Beyond Differentiability: Scalable Path Steering with Tree Search in Diffusion and Flow Models
    NeurIPS 2025 ·
    Yingqing Guo
    ,
    Yukang Yang
    ,
    Hui Yuan
    , Mengdi Wang
  23. TreeBoN: Enhancing Inference-Time Alignment with Speculative Tree-Search and Best-of-N Sampling
    EMNLP 2025 ·
    Jiahao Qiu
    ,
    Yifu Lu
    ,
    Yifan Zeng
    ,
    Jiacheng Guo
    ,
    Jiayi Geng
    ,
    Chenhao Zhu
    ,
    Xinzhe Juan
    ,
    Ling Yang
    ,
    Huazheng Wang
    ,
    Kaixuan Huang
    ,
    Yue Wu
    , Mengdi Wang
  24. A Theoretical Perspective for Speculative Decoding Algorithm
    NeurIPS 2024 ·
    Ming Yin
    ,
    Minshuo Chen
    ,
    Kaixuan Huang
    , Mengdi Wang
  25. Assessing the Brittleness of Safety Alignment via Pruning and Low-Rank Modifications
    ICML 2024 ·
    Boyi Wei
    ,
    Kaixuan Huang
    ,
    Yangsibo Huang
    ,
    Tinghao Xie
    ,
    Xiangyu Qi
    ,
    Mengzhou Xia
    ,
    Prateek Mittal
    , Mengdi Wang,
    Peter Henderson
  26. Conversational Dueling Bandits in Generalized Linear Models
    SIGKDD 2024 ·
    Shuhua Yang
    ,
    Hui Yuan
    ,
    Xiaoying Zhang
    , Mengdi Wang,
    Hong Zhang
    ,
    Huazheng Wang
  27. Fast Best-of-N Decoding via Speculative Rejection
    NeurIPS 2024 ·
    Hanshi Sun
    ,
    Momin Haider
    ,
    Ruiqi Zhang
    ,
    Huitao Yang
    ,
    Jiahao Qiu
    ,
    Ming Yin
    , Mengdi Wang,
    Peter L. Bartlett
    ,
    Andrea Zanette
  28. FlexSBDD: Structure-Based Drug Design with Flexible Protein Modeling
    NeurIPS 2024 ·
    Zaixi Zhang
    , Mengdi Wang,
    Qi Liu
  29. Global Convergence in Training Large-Scale Transformers
    NeurIPS 2024 ·
    Cheng Gao
    ,
    Yuan Cao
    ,
    Zihao Li
    ,
    Yihan He
    , Mengdi Wang,
    Han Liu
    ,
    Jason M. Klusowski
    ,
    Jianqing Fan
  30. Gradient Guidance for Diffusion Models: An Optimization Perspective
    NeurIPS 2024 ·
    Yingqing Guo
    ,
    Hui Yuan
    ,
    Yukang Yang
    ,
    Minshuo Chen
    , Mengdi Wang
  31. Information-Directed Pessimism for Offline Reinforcement Learning
    ICML 2024 ·
    Alec Koppel
    ,
    Sujay Bhatt
    ,
    Jiacheng Guo
    ,
    Joe Eappen
    , Mengdi Wang,
    Sumitra Ganesh
  32. Is Inverse Reinforcement Learning Harder than Standard Reinforcement Learning? A Theoretical Perspective
    ICML 2024 ·
    Lei Zhao
    , Mengdi Wang,
    Yu Bai
  33. MaxMin-RLHF: Alignment with Diverse Human Preferences
    ICML 2024 ·
    Souradip Chakraborty
    ,
    Jiahao Qiu
    ,
    Hui Yuan
    ,
    Alec Koppel
    ,
    Dinesh Manocha
    ,
    Furong Huang
    ,
    Amrit S. Bedi
    , Mengdi Wang
  34. Nonparametric Classification on Low Dimensional Manifolds using Overparameterized Convolutional Residual Networks
    NeurIPS 2024 ·
    Zixuan Zhang
    ,
    Kaiqi Zhang
    ,
    Minshuo Chen
    ,
    Yuma Takeda
    , Mengdi Wang,
    Tuo Zhao
    ,
    Yu-Xiang Wang
  35. Offline Multitask Representation Learning for Reinforcement Learning
    NeurIPS 2024 ·
    Haque Ishfaq
    ,
    Thanh Nguyen-Tang
    ,
    Songtao Feng
    ,
    Raman Arora
    , Mengdi Wang,
    Ming Yin
    ,
    Doina Precup
  36. One-Layer Transformer Provably Learns One-Nearest Neighbor In Context
    NeurIPS 2024 ·
    Zihao Li
    ,
    Yuan Cao
    ,
    Cheng Gao
    ,
    Yihan He
    ,
    Han Liu
    ,
    Jason M. Klusowski
    ,
    Jianqing Fan
    , Mengdi Wang
  37. PARL: A Unified Framework for Policy Alignment in Reinforcement Learning from Human Feedback
    ICLR 2024 ·
    Souradip Chakraborty
    ,
    Amrit Singh Bedi
    ,
    Alec Koppel
    ,
    Huazheng Wang
    ,
    Dinesh Manocha
    , Mengdi Wang,
    Furong Huang
  38. Policy Evaluation for Reinforcement Learning from Human Feedback: A Sample Complexity Analysis
    AISTATS 2024 ·
    Zihao Li
    ,
    Xiang Ji
    ,
    Minshuo Chen
    , Mengdi Wang
  39. Sample-Efficient Learning of POMDPs with Multiple Observations In Hindsight
    ICLR 2024 ·
    Jiacheng Guo
    ,
    Minshuo Chen
    ,
    Huan Wang
    ,
    Caiming Xiong
    , Mengdi Wang,
    Yu Bai
  40. Theoretical insights for diffusion guidance: A case study for Gaussian mixture models
    ICML 2024 ·
    Yuchen Wu
    ,
    Minshuo Chen
    ,
    Zihao Li
    , Mengdi Wang,
    Yuting Wei
  41. Theory of Consistency Diffusion Models: Distribution Estimation Meets Fast Sampling
    ICML 2024 ·
    Zehao Dou
    ,
    Minshuo Chen
    , Mengdi Wang,
    Zhuoran Yang
  42. Transfer Q-star : Principled Decoding for LLM Alignment
    NeurIPS 2024 ·
    Souradip Chakraborty
    ,
    Soumya Suvra Ghosal
    ,
    Ming Yin
    ,
    Dinesh Manocha
    , Mengdi Wang,
    Amrit Singh Bedi
    ,
    Furong Huang
  43. Tree Search-Based Evolutionary Bandits for Protein Sequence Optimization
    AAAI 2024 ·
    Jiahao Qiu
    ,
    Hui Yuan
    ,
    Jinghong Zhang
    ,
    Wentao Chen
    ,
    Huazheng Wang
    , Mengdi Wang
  44. Visual Adversarial Examples Jailbreak Aligned Large Language Models
    AAAI 2024 ·
    Xiangyu Qi
    ,
    Kaixuan Huang
    ,
    Ashwinee Panda
    ,
    Peter Henderson
    , Mengdi Wang,
    Prateek Mittal