Mengdi Wang

331 papers A* 76A 12B 2C 2Misc 4Journal 219Unranked 15
YearRankTypeTitle / Venue / Authors
2026 J jnl
A Survey of Self-Evolving Agents: What, When, How, and Where to Evolve on the Path to Artificial Super Intelligence.
Trans. Mach. Learn. Res.
Huan-ang Gao, Jiayi Geng, Wenyue Hua, Mengkang Hu, Xinzhe Juan, Hongzhang Liu, Shilong Liu, Jiahao Qiu, Xuan Qi, Qihan Ren, Yiran Wu, Hongru Wang, Han Xiao, Yuhang Zhou, Shaokun Zhang, Jiayi Zhang, Jinyu Xiang, Yixiong Fang, Qiwen Zhao, Dongrui Liu, Cheng Qian, Zhenhailong Wang, Minda Hu, Huazheng Wang, Qingyun Wu, Heng Ji, Mengdi Wang
2026 J jnl
Diffusion Model for Manifold Data: Score Decomposition, Curvature, and Statistical Complexity.
CoRR
Zixuan Zhang, Kaixuan Huang, Tuo Zhao, Mengdi Wang, Minshuo Chen
2026 J jnl
Embodied LLM Agents Learn to Cooperate in Organized Teams.
IEEE Trans. Comput. Soc. Syst.
Xudong Guo, Kaixuan Huang, Jiale Liu, Wenhui Fan, Natalia Vélez, Qingyun Wu, Huazheng Wang, Thomas L. Griffiths, Mengdi Wang
2026 — conf
Jailbreaks as Inference-Time Alignment: A Framework for Understanding Safety Failures in LLMs.
EACL (Volume 1: Long Papers)
James Beetham, Souradip Chakraborty, Mengdi Wang, Furong Huang, Amrit Singh Bedi, Mubarak Shah
2026 J jnl
LatentChem: From Textual CoT to Latent Thinking in Chemical Reasoning.
CoRR
Xinwu Ye, Yicheng Mao, Jia Zhang, Yimeng Liu, Li Hao, Fang Wu, Zhiwei Li, Yuxuan Liao, Zehong Wang, Zhiyuan Liu, Zhenfei Yin, Li Yuan, Philip Torr, Huan Sun, Xiangxiang Zeng, Mengdi Wang, Le Cong, Shenghua Gao, Xiangru Tang
2026 J jnl
Structure-Based Drug Design with Geometric Deep Learning: A Comprehensive Survey.
ACM Comput. Surv.
Zaixi Zhang, Jiaxian Yan, Yining Huang, Qi Liu, Enhong Chen, Mengdi Wang, Marinka Zitnik
2025 A* conf
A Common Pitfall of Margin-based Language Model Alignment: Gradient Entanglement.
ICLR
Hui Yuan, Yifan Zeng, Yue Wu, Huazheng Wang, Mengdi Wang, Liu Leqi
2025 A* conf
A First-order Generative Bilevel Optimization Framework for Diffusion Models.
ICML
Quan Xiao, Hui Yuan, A. F. M. Saif, Gaowen Liu, Ramana Rao Kompella, Mengdi Wang, Tianyi Chen
2025 J jnl
A First-order Generative Bilevel Optimization Framework for Diffusion Models.
CoRR
Quan Xiao, Hui Yuan, A. F. M. Saif, Gaowen Liu, Ramana Kompella, Mengdi Wang, Tianyi Chen
2025 J jnl
A Survey of Self-Evolving Agents: On Path to Artificial Super Intelligence.
CoRR
Huan-ang Gao, Jiayi Geng, Wenyue Hua, Mengkang Hu, Xinzhe Juan, Hongzhang Liu, Shilong Liu, Jiahao Qiu, Xuan Qi, Yiran Wu, Hongru Wang, Han Xiao, Yuhang Zhou, Shaokun Zhang, Jiayi Zhang, Jinyu Xiang, Yixiong Fang, Qiwen Zhao, Dongrui Liu, Qihan Ren, Cheng Qian, Zhenhailong Wang, Minda Hu, Huazheng Wang, Qingyun Wu, Heng Ji, Mengdi Wang
2025 J jnl
ARMOR: High-Performance Semi-Structured Pruning via Adaptive Matrix Factorization.
CoRR
Lawrence Liu, Alexander Liu, Mengdi Wang, Tuo Zhao, Lin F. Yang
2025 J jnl
AgentDistill: Training-Free Agent Distillation with Generalizable MCP Boxes.
CoRR
Jiahao Qiu, Xinzhe Juan, Yimin Wang, Ling Yang, Xuan Qi, Tongcheng Zhang, Jiacheng Guo, Yifu Lu, Zixin Yao, Hongru Wang, Shilong Liu, Xun Jiang, Liu Leqi, Mengdi Wang
2025 J jnl
Alita-G: Self-Evolving Generative Agent for Agent Generation.
CoRR
Jiahao Qiu, Xuan Qi, Hongru Wang, Xinzhe Juan, Yimin Wang, Zelin Zhao, Jiayi Geng, Jiacheng Guo, Peihang Li, Jingzhe Shi, Shilong Liu, Mengdi Wang
2025 J jnl
Alita: Generalist Agent Enabling Scalable Agentic Reasoning with Minimal Predefinition and Maximal Self-Evolution.
CoRR
Jiahao Qiu, Xuan Qi, Tongcheng Zhang, Xinzhe Juan, Jiacheng Guo, Yifu Lu, Yimin Wang, Zixin Yao, Qihan Ren, Xun Jiang, Xing Zhou, Dongrui Liu, Ling Yang, Yue Wu, Kaixuan Huang, Shilong Liu, Hongru Wang, Mengdi Wang
2025 J jnl
AutoTool: Dynamic Tool Selection and Integration for Agentic Reasoning.
CoRR
Jiaru Zou, Ling Yang, Yunzhe Qi, Sirui Chen, Mengting Ai, Ke Shen, Jingrui He, Mengdi Wang
2025 J jnl
Co-Evolving LLM Coder and Unit Tester via Reinforcement Learning.
CoRR
Yinjie Wang, Ling Yang, Ye Tian, Ke Shen, Mengdi Wang
2025 A* conf
Collab: Controlled Decoding using Mixture of Agents for LLM Alignment.
ICLR
Souradip Chakraborty, Sujay Bhatt, Udari Madhushani Sehwag, Soumya Suvra Ghosal, Jiahao Qiu, Mengdi Wang, Dinesh Manocha, Furong Huang, Alec Koppel, Sumitra Ganesh
2025 J jnl
Collab: Controlled Decoding using Mixture of Agents for LLM Alignment.
CoRR
Souradip Chakraborty, Sujay Bhatt, Udari Madhushani Sehwag, Soumya Suvra Ghosal, Jiahao Qiu, Mengdi Wang, Dinesh Manocha, Furong Huang, Alec Koppel, Sumitra Ganesh
2025 J jnl
CryptoBench: A Dynamic Benchmark for Expert-Level Evaluation of LLM Agents in Cryptocurrency.
CoRR
Jiacheng Guo, Suozhi Huang, Zixin Yao, Yifan Zhang, Yifu Lu, Jiashuo Liu, Zihao Li, Nicholas Deng, Qixin Xiao, Jia Tian, Kanghong Zhan, Tianyi Li, Xiaochen Liu, Jason Ge, Chaoyang He, Kaixuan Huang, Lin Yang, Wenhao Huang, Mengdi Wang
2025 J jnl
CubeBench: Diagnosing Interactive, Long-Horizon Spatial Reasoning Under Partial Observations.
CoRR
Huan-ang Gao, Zikang Zhang, Tianwei Luo, Kaisen Yang, Xinzhe Juan, Jiahao Qiu, Tianxing Chen, Bingxiang He, Hao Zhao, Hao Zhou, Shilong Liu, Mengdi Wang
2025 J jnl
DISC: Dynamic Decomposition Improves LLM Inference Scaling.
CoRR
Jonathan Light, Wei Cheng, Yue Wu, Masafumi Oyamada, Mengdi Wang, Santiago Paternain, Haifeng Chen
2025 A* conf
Deep Reinforcement Learning for Efficient and Fair Allocation of Healthcare Resources.
IJCAI
Yikuan Li, Chengsheng Mao, Kaixuan Huang, Hanyin Wang, Zheng Yu, Mengdi Wang, Yuan Luo
2025 J jnl
Demystifying Reinforcement Learning in Agentic Reasoning.
CoRR
Zhaochen Yu, Ling Yang, Jiaru Zou, Shuicheng Yan, Mengdi Wang
2025 J jnl
Detecting Post-generation Edits to Watermarked LLM Outputs via Combinatorial Watermarking.
CoRR
Liyan Xie, Muhammad Siddeek, Mohamed Seif, Andrea J. Goldsmith, Mengdi Wang
2025 A* conf
Diffusion Transformer Captures Spatial-Temporal Dependencies: A Theory for Gaussian Process Data.
ICLR
Hengyu Fu, Zehao Dou, Jiawei Guo, Mengdi Wang, Minshuo Chen
2025 J jnl
Diffusion-Sharpening: Fine-tuning Diffusion Models with Denoising Trajectory Sharpening.
CoRR
Ye Tian, Ling Yang, Xinchen Zhang, Yunhai Tong, Mengdi Wang, Bin Cui
2025 J jnl
Does Thinking More always Help? Understanding Test-Time Scaling in Reasoning Models.
CoRR
Soumya Suvra Ghosal, Souradip Chakraborty, Avinash Reddy, Yifu Lu, Mengdi Wang, Dinesh Manocha, Furong Huang, Mohammad Ghavamzadeh, Amrit Singh Bedi
2025 A* conf
Emergent Symbolic Mechanisms Support Abstract Reasoning in Large Language Models.
ICML
Yukang Yang, Declan Campbell, Kaixuan Huang, Mengdi Wang, Jonathan D. Cohen, Taylor Whittington Webb
2025 J jnl
Emergent Symbolic Mechanisms Support Abstract Reasoning in Large Language Models.
CoRR
Yukang Yang, Declan Campbell, Kaixuan Huang, Mengdi Wang, Jonathan D. Cohen, Taylor W. Webb
2025 A* conf
EmoAgent: Assessing and Safeguarding Human-AI Interaction for Mental Health Safety.
EMNLP
Jiahao Qiu, Yinghui He, Xinzhe Juan, Yimin Wang, Yuhan Liu, Zixin Yao, Yue Wu, Xun Jiang, Ling Yang, Mengdi Wang
2025 J jnl
EmoAgent: Assessing and Safeguarding Human-AI Interaction for Mental Health Safety.
CoRR
Jiahao Qiu, Yinghui He, Xinzhe Juan, Yimin Wang, Yuhan Liu, Zixin Yao, Yue Wu, Xun Jiang, Ling Yang, Mengdi Wang
2025 J jnl
FragmentGPT: A Unified GPT Model for Fragment Growing, Linking, and Merging in Molecular Design.
CoRR
Xuefeng Liu, Songhao Jiang, Qinan Huang, Tinson Xu, Ian T. Foster, Mengdi Wang, Hening Lin, Rick Stevens
2025 J jnl
From Word to World: Can Large Language Models be Implicit Text-based World Models?
CoRR
Yixia Li, Hongru Wang, Jiahao Qiu, Zhenfei Yin, Dongdong Zhang, Cheng Qian, Zeping Li, Pony Ma, Guanhua Chen, Heng Ji, Mengdi Wang
2025 J jnl
FutureX: An Advanced Live Benchmark for LLM Agents in Future Prediction.
CoRR
Zhiyuan Zeng, Jiashuo Liu, Siyuan Chen, Tianci He, Yali Liao, Jinpeng Wang, Zaiyuan Wang, Yang Yang, Lingyue Yin, Mingren Yin, Zhenwei Zhu, Tianle Cai, Zehui Chen, Jiecao Chen, Yantao Du, Xiang Gao, Jiacheng Guo, Liang Hu, Jianpeng Jiao, Xiangsheng Li, Jingkai Liu, Shuang Ni, Zhoufutu Wen, Ge Zhang, Kaiyuan Zhang, Xin Zhou, Jose H. Blanchet, Xipeng Qiu, Mengdi Wang, Wenhao Huang
2025 J jnl
GenEnv: Difficulty-Aligned Co-Evolution Between LLM Agents and Environment Simulators.
CoRR
Jiacheng Guo, Ling Yang, Peter Chen, Qixin Xiao, Yinjie Wang, Xinzhe Juan, Jiahao Qiu, Ke Shen, Mengdi Wang
2025 J jnl
GeneBreaker: Jailbreak Attacks against DNA Language Models with Pathogenicity Guidance.
CoRR
Zaixi Zhang, Zhenghong Zhou, Ruofan Jin, Le Cong, Mengdi Wang
2025 J jnl
Generative AI for Biosciences: Emerging Threats and Roadmap to Biosecurity.
CoRR
Zaixi Zhang, Souradip Chakraborty, Amrit Singh Bedi, Emilin Mathew, Varsha Saravanan, Le Cong, Alvaro Velasquez, Sheng Lin-Gibson, Megan Blewett, Dan Hendrycs, Alex John London, Ellen Zhong, Ben Raphael, Jian Ma, Eric Xing, Russ Altman, George Church, Mengdi Wang
2025 J jnl
Genome-Bench: A Scientific Reasoning Benchmark from Real-World Expert Discussions.
CoRR
Ming Yin, Yuanhao Qu, Dyllan Liu, Ling Yang, Le Cong, Mengdi Wang
2025 A* conf
Immune: Improving Safety Against Jailbreaks in Multi-modal LLMs via Inference-Time Alignment.
CVPR
Soumya Suvra Ghosal, Souradip Chakraborty, Vaibhav Singh, Tianrui Guan, Mengdi Wang, Ahmad Beirami, Furong Huang, Alvaro Velasquez, Dinesh Manocha, Amrit Singh Bedi
2025 A* conf
IterComp: Iterative Composition-Aware Feedback Learning from Model Gallery for Text-to-Image Generation.
ICLR
Xinchen Zhang, Ling Yang, Guohao Li, Yaqi Cai, Jiake Xie, Yong Tang, Yujiu Yang, Mengdi Wang, Bin Cui
2025 J jnl
LabOS: The AI-XR Co-Scientist That Sees and Works With Humans.
CoRR
Le Cong, Zaixi Zhang, Xiaotong Wang, Yin Di, Ruofan Jin, Michal Gerasimiuk, Yinkai Wang, Ravi K. Dinesh, David Smerkous, Alex Smerkous, Xuekun Wu, Shilong Liu, Peishan Li, Yi Zhu, Simran Serrao, Ning Zhao, Imran A. Mohammad, John B. Sunwoo, Joseph C. Wu, Mengdi Wang
2025 J jnl
Latent Collaboration in Multi-Agent Systems.
CoRR
Jiaru Zou, Xiyuan Yang, Ruizhong Qiu, Gaotang Li, Katherine Tieu, Pan Lu, Ke Shen, Hanghang Tong, Yejin Choi, Jingrui He, James Zou, Mengdi Wang, Ling Yang
2025 A* conf
MATH-Perturb: Benchmarking LLMs' Math Reasoning Abilities against Hard Perturbations.
ICML
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
2025 J jnl
MATH-Perturb: Benchmarking LLMs' Math Reasoning Abilities against Hard Perturbations.
CoRR
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
2025 J jnl
MMaDA-Parallel: Multimodal Large Diffusion Language Models for Thinking-Aware Editing and Generation.
CoRR
Ye Tian, Ling Yang, Jiongfan Yang, Anran Wang, Yu Tian, Jiani Zheng, Haochen Wang, Zhiyang Teng, Zhuochen Wang, Yinjie Wang, Yunhai Tong, Mengdi Wang, Xiangtai Li
2025 J jnl
MMaDA: Multimodal Large Diffusion Language Models.
CoRR
Ling Yang, Ye Tian, Bowen Li, Xinchen Zhang, Ke Shen, Yunhai Tong, Mengdi Wang
2025 J jnl
Monte Carlo Tree Diffusion with Multiple Experts for Protein Design.
CoRR
Xuefeng Liu, Mingxuan Cao, Songhao Jiang, Xiao Luo, Xiaotian Duan, Mengdi Wang, Tobin R. Sosnick, Jinbo Xu, Rick Stevens
2025 J jnl
NoWag: A Unified Framework for Shape Preserving Compression of Large Language Models.
CoRR
Lawrence Liu, Inesh Chakrabarti, Yixiao Li, Mengdi Wang, Tuo Zhao, Lin F. Yang
2025 J jnl
OTC: Optimal Tool Calls via Reinforcement Learning.
CoRR
Hongru Wang, Cheng Qian, Wanjun Zhong, Xiusi Chen, Jiahao Qiu, Shijue Huang, Bowen Jin, Mengdi Wang, Kam-Fai Wong, Heng Ji
2025 J jnl
On Path to Multimodal Historical Reasoning: HistBench and HistAgent.
CoRR
Jiahao Qiu, Fulian Xiao, Yimin Wang, Yuchen Mao, Yijia Chen, Xinzhe Juan, Siran Wang, Xuan Qi, Tongcheng Zhang, Zixin Yao, Jiacheng Guo, Yifu Lu, Charles Argon, Jundi Cui, Daixin Chen, Junran Zhou, Shuyao Zhou, Zhanpeng Zhou, Ling Yang, Shilong Liu, Hongru Wang, Kaixuan Huang, Xun Jiang, Yuming Cao, Yue Chen, Yunfei Chen, Zhengyi Chen, Ruowei Dai, Mengqiu Deng, Jiye Fu, Yunting Gu, Zijie Guan, Zirui Huang, Xiaoyan Ji, Yumeng Jiang, Delong Kong, Haolong Li, Jiaqi Li, Ruipeng Li, Tianze Li, Zhuoran Li, Haixia Lian, Mengyue Lin, Xudong Liu, Jiayi Lu, Jinghan Lu, Wanyu Luo, Ziyue Luo, Zihao Pu, Zhi Qiao, Ruihuan Ren, Liang Wan, Ruixiang Wang, Tianhui Wang, Yang Wang, Zeyu Wang, Zihua Wang, Yujia Wu, Zhaoyi Wu, Hao Xin, Weiao Xing, Ruojun Xiong, Weijie Xu, Yao Shu, Xiao Yao, Xiaorui Yang, Yuchen Yang, Nan Yi, Jiadong Yu, Yangyuxuan Yu, Huiting Zeng, Danni Zhang, Yunjie Zhang, Zhaoyu Zhang, Zhiheng Zhang, Xiaofeng Zheng, Peirong Zhou, Linyan Zhong, Xiaoyin Zong, Ying Zhao, Zhenxin Chen, Lin Ding, Xiaoyu Gao, Bingbing Gong, Yichao Li, Yang Liao, Guang Ma, Tianyuan Ma, Xinrui Sun, Tianyi Wang, Han Xia, Ruobing Xian, Gen Ye, Tengfei Yu, Wentao Zhang, Yuxi Wang, Xi Gao, Mengdi Wang
2025 J jnl
On the Statistical Complexity for Offline and Low-Adaptive Reinforcement Learning with Structures.
CoRR
Ming Yin, Mengdi Wang, Yuxiang Wang
2025 J jnl
Physics Supernova: AI Agent Matches Elite Gold Medalists at IPhO 2025.
CoRR
Jiahao Qiu, Jingzhe Shi, Xinzhe Juan, Zelin Zhao, Jiayi Geng, Shilong Liu, Hongru Wang, Sanfeng Wu, Mengdi Wang
2025 J jnl
PoseX: AI Defeats Physics Approaches on Protein-Ligand Cross Docking.
CoRR
Yize Jiang, Xinze Li, Yuanyuan Zhang, Jin Han, Youjun Xu, Ayush Pandit, Zaixi Zhang, Mengdi Wang, Mengyang Wang, Chong Liu, Guang Yang, Yejin Choi, Wu-Jun Li, Tianfan Fu, Fang Wu, Junhong Liu
2025 J jnl
Preacher: Paper-to-Video Agentic System.
CoRR
Jingwei Liu, Ling Yang, Hao Luo, Fan Wang, Hongyan Li, Mengdi Wang
2025 J jnl
ReasonFlux-PRM: Trajectory-Aware PRMs for Long Chain-of-Thought Reasoning in LLMs.
CoRR
Jiaru Zou, Ling Yang, Jingwen Gu, Jiahao Qiu, Ke Shen, Jingrui He, Mengdi Wang
2025 J jnl
ReasonFlux: Hierarchical LLM Reasoning via Scaling Thought Templates.
CoRR
Ling Yang, Zhaochen Yu, Bin Cui, Mengdi Wang
2025 A* conf
Rectified Diffusion: Straightness Is Not Your Need in Rectified Flow.
ICLR
Fu-Yun Wang, Ling Yang, Zhaoyang Huang, Mengdi Wang, Hongsheng Li
2025 J jnl
Revolutionizing Reinforcement Learning Framework for Diffusion Large Language Models.
CoRR
Yinjie Wang, Ling Yang, Bowen Li, Ye Tian, Ke Shen, Mengdi Wang
2025 J jnl
STELLA: Self-Evolving LLM Agent for Biomedical Research.
CoRR
Ruofan Jin, Zaixi Zhang, Mengdi Wang, Le Cong
2025 J jnl
SafeProtein: Red-Teaming Framework and Benchmark for Protein Foundation Models.
CoRR
Jigang Fan, Zhenghong Zhou, Ruofan Jin, Le Cong, Mengdi Wang, Zaixi Zhang
2025 J jnl
ScoreFlow: Mastering LLM Agent Workflows via Score-based Preference Optimization.
CoRR
Yinjie Wang, Ling Yang, Guohao Li, Mengdi Wang, Bryon Aragam
2025 J jnl
Shallow Preference Signals: Large Language Model Aligns Even Better with Truncated Data?
CoRR
Xuan Qi, Jiahao Qiu, Xinzhe Juan, Yue Wu, Mengdi Wang
2025 — conf
Temporal Consistency for LLM Reasoning Process Error Identification.
EMNLP (Findings)
Jiacheng Guo, Yue Wu, Jiahao Qiu, Kaixuan Huang, Xinzhe Juan, Ling Yang, Mengdi Wang
2025 J jnl
Temporal Consistency for LLM Reasoning Process Error Identification.
CoRR
Jiacheng Guo, Yue Wu, Jiahao Qiu, Kaixuan Huang, Xinzhe Juan, Ling Yang, Mengdi Wang
2025 J jnl
Toward a Theory of Agents as Tool-Use Decision-Makers.
CoRR
Hongru Wang, Cheng Qian, Manling Li, Jiahao Qiu, Boyang Xue, Mengdi Wang, Heng Ji, Kam-Fai Wong
2025 A* conf
Towards Understanding Text Hallucination of Diffusion Models via Local Generation Bias.
ICLR
Rui Lu, Runzhe Wang, Kaifeng Lyu, Xitai Jiang, Gao Huang, Mengdi Wang
2025 J jnl
Towards Understanding Text Hallucination of Diffusion Models via Local Generation Bias.
CoRR
Rui Lu, Runzhe Wang, Kaifeng Lyu, Xitai Jiang, Gao Huang, Mengdi Wang
2025 J jnl
Training-Free Guidance Beyond Differentiability: Scalable Path Steering with Tree Search in Diffusion and Flow Models.
CoRR
Yingqing Guo, Yukang Yang, Hui Yuan, Mengdi Wang
2025 — conf
TreeBoN: Enhancing Inference-Time Alignment with Speculative Tree-Search and Best-of-N Sampling.
EMNLP (Findings)
Jiahao Qiu, Yifu Lu, Yifan Zeng, Jiacheng Guo, Jiayi Geng, Chenhao Zhu, Xinzhe Juan, Ling Yang, Huazheng Wang, Kaixuan Huang, Yue Wu, Mengdi Wang
2025 J jnl
Web World Models.
CoRR
Jichen Feng, Yifan Zhang, Chenggong Zhang, Yifu Lu, Shilong Liu, Mengdi Wang
2024 J jnl
A 5′ UTR language model for decoding untranslated regions of mRNA and function predictions.
Nat. Mac. Intell.
Yanyi Chu, Dan Yu, Yupeng Li, Kaixuan Huang, Yue Shen, Le Cong, Jason Zhang, Mengdi Wang
2024 J jnl
A Common Pitfall of Margin-based Language Model Alignment: Gradient Entanglement.
CoRR
Hui Yuan, Yifan Zeng, Yue Wu, Huazheng Wang, Mengdi Wang, Liu Leqi
2024 A* conf
A Theoretical Perspective for Speculative Decoding Algorithm.
NeurIPS
Ming Yin, Minshuo Chen, Kaixuan Huang, Mengdi Wang
2024 J jnl
A Theoretical Perspective for Speculative Decoding Algorithm.
CoRR
Ming Yin, Minshuo Chen, Kaixuan Huang, Mengdi Wang
2024 J jnl
AI Risk Management Should Incorporate Both Safety and Security.
CoRR
Xiangyu Qi, Yangsibo Huang, Yi Zeng, Edoardo Debenedetti, Jonas Geiping, Luxi He, Kaixuan Huang, Udari Madhushani, Vikash Sehwag, Weijia Shi, Boyi Wei, Tinghao Xie, Danqi Chen, Pin-Yu Chen, Jeffrey Ding, Ruoxi Jia, Jiaqi Ma, Arvind Narayanan, Weijie J. Su, Mengdi Wang, Chaowei Xiao, Bo Li, Dawn Song, Peter Henderson, Prateek Mittal
2024 J jnl
AIME: AI System Optimization via Multiple LLM Evaluators.
CoRR
Bhrij Patel, Souradip Chakraborty, Wesley A. Suttle, Mengdi Wang, Amrit Singh Bedi, Dinesh Manocha
2024 J jnl
Adversarial Attacks on Online Learning to Rank with Stochastic Click Models.
Trans. Mach. Learn. Res.
Zichen Wang, Rishab Balasubramanian, Hui Yuan, Chenyu Song, Mengdi Wang, Huazheng Wang
2024 J jnl
An Overview of Diffusion Models: Applications, Guided Generation, Statistical Rates and Optimization.
CoRR
Minshuo Chen, Song Mei, Jianqing Fan, Mengdi Wang
2024 A* conf
Assessing the Brittleness of Safety Alignment via Pruning and Low-Rank Modifications.
ICML
Boyi Wei, Kaixuan Huang, Yangsibo Huang, Tinghao Xie, Xiangyu Qi, Mengzhou Xia, Prateek Mittal, Mengdi Wang, Peter Henderson
2024 J jnl
Assessing the Brittleness of Safety Alignment via Pruning and Low-Rank Modifications.
CoRR
Boyi Wei, Kaixuan Huang, Yangsibo Huang, Tinghao Xie, Xiangyu Qi, Mengzhou Xia, Prateek Mittal, Mengdi Wang, Peter Henderson
2024 J jnl
Author Correction: A 5′ UTR language model for decoding untranslated regions of mRNA and function predictions.
Nat. Mac. Intell.
Yanyi Chu, Dan Yu, Yupeng Li, Kaixuan Huang, Yue Shen, Le Cong, Jason Zhang, Mengdi Wang
2024 J jnl
Boosting the Convergence of Reinforcement Learning-Based Auto-Pruning Using Historical Data.
IEEE Trans. Comput. Aided Des. Integr. Circuits Syst.
Jiandong Mu, Mengdi Wang, Feiwen Zhu, Jun Yang, Wei Lin, Wei Zhang
2024 J jnl
CRISPR-GPT: An LLM Agent for Automated Design of Gene-Editing Experiments.
CoRR
Kaixuan Huang, Yuanhao Qu, Henry Cousins, William A. Johnson, Di Yin, Mihir Shah, Denny Zhou, Russ B. Altman, Mengdi Wang, Le Cong
2024 J jnl
Contractual Reinforcement Learning: Pulling Arms with Invisible Hands.
CoRR
Jibang Wu, Siyu Chen, Mengdi Wang, Huazheng Wang, Haifeng Xu
2024 A* conf
Conversational Dueling Bandits in Generalized Linear Models.
KDD
Shuhua Yang, Hui Yuan, Xiaoying Zhang, Mengdi Wang, Hong Zhang, Huazheng Wang
2024 J jnl
Conversational Dueling Bandits in Generalized Linear Models.
CoRR
Shuhua Yang, Hui Yuan, Xiaoying Zhang, Mengdi Wang, Hong Zhang, Huazheng Wang
2024 J jnl
Deep reinforcement learning identifies personalized intermittent androgen deprivation therapy for prostate cancer.
Briefings Bioinform.
Yitao Lu, Qian Chu, Zhen Li, Mengdi Wang, Robert A. Gatenby, Qingpeng Zhang
2024 J jnl
Diffusion Model for Data-Driven Black-Box Optimization.
CoRR
Zihao Li, Hui Yuan, Kaixuan Huang, Chengzhuo Ni, Yinyu Ye, Minshuo Chen, Mengdi Wang
2024 J jnl
Diffusion Transformer Captures Spatial-Temporal Dependencies: A Theory for Gaussian Process Data.
CoRR
Hengyu Fu, Zehao Dou, Jiawei Guo, Mengdi Wang, Minshuo Chen
2024 J jnl
Double Duality: Variational Primal-Dual Policy Optimization for Constrained Reinforcement Learning.
CoRR
Zihao Li, Boyi Liu, Zhuoran Yang, Zhaoran Wang, Mengdi Wang
2024 J jnl
Efficient Reinforcement Learning With Impaired Observability: Learning to Act With Delayed and Missing State Observations.
IEEE Trans. Inf. Theory
Minshuo Chen, Jie Meng, Yu Bai, Yinyu Ye, H. Vincent Poor, Mengdi Wang
2024 J jnl
Embodied LLM Agents Learn to Cooperate in Organized Teams.
CoRR
Xudong Guo, Kaixuan Huang, Jiale Liu, Wenhui Fan, Natalia Vélez, Qingyun Wu, Huazheng Wang, Thomas L. Griffiths, Mengdi Wang
2024 A* conf
Fast Best-of-N Decoding via Speculative Rejection.
NeurIPS
Hanshi Sun, Momin Haider, Ruiqi Zhang, Huitao Yang, Jiahao Qiu, Ming Yin, Mengdi Wang, Peter L. Bartlett, Andrea Zanette
2024 J jnl
Fast Best-of-N Decoding via Speculative Rejection.
CoRR
Hanshi Sun, Momin Haider, Ruiqi Zhang, Huitao Yang, Jiahao Qiu, Ming Yin, Mengdi Wang, Peter L. Bartlett, Andrea Zanette
2024 A* conf
FlexSBDD: Structure-Based Drug Design with Flexible Protein Modeling.
NeurIPS
Zaixi Zhang, Mengdi Wang, Qi Liu
2024 J jnl
FoldMark: Protecting Protein Generative Models with Watermarking.
CoRR
Zaixi Zhang, Ruofan Jin, Kaidi Fu, Le Cong, Marinka Zitnik, Mengdi Wang
2024 A* conf
Global Convergence in Training Large-Scale Transformers.
NeurIPS
Cheng Gao, Yuan Cao, Zihao Li, Yihan He, Mengdi Wang, Han Liu, Jason M. Klusowski, Jianqing Fan
2024 J jnl
Global Convergence in Training Large-Scale Transformers.
CoRR
Cheng Gao, Yuan Cao, Zihao Li, Yihan He, Mengdi Wang, Han Liu, Jason M. Klusowski, Jianqing Fan
2024 A* conf
Gradient Guidance for Diffusion Models: An Optimization Perspective.
NeurIPS
Yingqing Guo, Hui Yuan, Yukang Yang, Minshuo Chen, Mengdi Wang
2024 J jnl
Gradient Guidance for Diffusion Models: An Optimization Perspective.
CoRR
Yingqing Guo, Hui Yuan, Yukang Yang, Minshuo Chen, Mengdi Wang
2024 J jnl
Immune: Improving Safety Against Jailbreaks in Multi-modal LLMs via Inference-Time Alignment.
CoRR
Soumya Suvra Ghosal, Souradip Chakraborty, Vaibhav Singh, Tianrui Guan, Mengdi Wang, Ahmad Beirami, Furong Huang, Alvaro Velasquez, Dinesh Manocha, Amrit Singh Bedi
2024 A* conf
Information-Directed Pessimism for Offline Reinforcement Learning.
ICML
Alec Koppel, Sujay Bhatt, Jiacheng Guo, Joe Eappen, Mengdi Wang, Sumitra Ganesh
2024 A* conf
Is Inverse Reinforcement Learning Harder than Standard Reinforcement Learning? A Theoretical Perspective.
ICML
Lei Zhao, Mengdi Wang, Yu Bai
2024 J jnl
IterComp: Iterative Composition-Aware Feedback Learning from Model Gallery for Text-to-Image Generation.
CoRR
Xinchen Zhang, Ling Yang, Guohao Li, Yaqi Cai, Jiake Xie, Yong Tang, Yujiu Yang, Mengdi Wang, Bin Cui
2024 J jnl
LIAR: Leveraging Alignment (Best-of-N) to Jailbreak LLMs in Seconds.
CoRR
James Beetham, Souradip Chakraborty, Mengdi Wang, Furong Huang, Amrit Singh Bedi, Mubarak Shah
2024 J jnl
Latent Diffusion Models for Controllable RNA Sequence Generation.
CoRR
Kaixuan Huang, Yukang Yang, Kaidi Fu, Yanyi Chu, Le Cong, Mengdi Wang
2024 J jnl
Long Term Memory: The Foundation of AI Self-Evolution.
CoRR
Xun Jiang, Feng Li, Han Zhao, Jiaying Wang, Jun Shao, Shihao Xu, Shu Zhang, Weiling Chen, Xavier Tang, Yize Chen, Mengyue Wu, Weizhi Ma, Mengdi Wang, Tianqiao Chen
2024 A* conf
MaxMin-RLHF: Alignment with Diverse Human Preferences.
ICML
Souradip Chakraborty, Jiahao Qiu, Hui Yuan, Alec Koppel, Dinesh Manocha, Furong Huang, Amrit S. Bedi, Mengdi Wang
2024 J jnl
MaxMin-RLHF: Towards Equitable Alignment of Large Language Models with Diverse Human Preferences.
CoRR
Souradip Chakraborty, Jiahao Qiu, Hui Yuan, Alec Koppel, Furong Huang, Dinesh Manocha, Amrit Singh Bedi, Mengdi Wang
2024 A* conf
Nonparametric Classification on Low Dimensional Manifolds using Overparameterized Convolutional Residual Networks.
NeurIPS
Zixuan Zhang, Kaiqi Zhang, Minshuo Chen, Yuma Takeda, Mengdi Wang, Tuo Zhao, Yu-Xiang Wang
2024 A* conf
Offline Multitask Representation Learning for Reinforcement Learning.
NeurIPS
Haque Ishfaq, Thanh Nguyen-Tang, Songtao Feng, Raman Arora, Mengdi Wang, Ming Yin, Doina Precup
2024 J jnl
Offline Multitask Representation Learning for Reinforcement Learning.
CoRR
Haque Ishfaq, Thanh Nguyen-Tang, Songtao Feng, Raman Arora, Mengdi Wang, Ming Yin, Doina Precup
2024 J jnl
On the Sample Complexity and Metastability of Heavy-tailed Policy Search in Continuous Control.
J. Mach. Learn. Res.
Amrit Singh Bedi, Anjaly Parayil, Junyu Zhang, Mengdi Wang, Alec Koppel
2024 A* conf
One-Layer Transformer Provably Learns One-Nearest Neighbor In Context.
NeurIPS
Zihao Li, Yuan Cao, Cheng Gao, Yihan He, Han Liu, Jason M. Klusowski, Jianqing Fan, Mengdi Wang
2024 J jnl
One-Layer Transformer Provably Learns One-Nearest Neighbor In Context.
CoRR
Zihao Li, Yuan Cao, Cheng Gao, Yihan He, Han Liu, Jason M. Klusowski, Jianqing Fan, Mengdi Wang
2024 A* conf
PARL: A Unified Framework for Policy Alignment in Reinforcement Learning from Human Feedback.
ICLR
Souradip Chakraborty, Amrit Singh Bedi, Alec Koppel, Huazheng Wang, Dinesh Manocha, Mengdi Wang, Furong Huang
2024 A conf
Policy Evaluation for Reinforcement Learning from Human Feedback: A Sample Complexity Analysis.
AISTATS
Zihao Li, Xiang Ji, Minshuo Chen, Mengdi Wang
2024 J jnl
Provable Statistical Rates for Consistency Diffusion Models.
CoRR
Zehao Dou, Minshuo Chen, Mengdi Wang, Zhuoran Yang
2024 J jnl
Rectified Diffusion: Straightness Is Not Your Need in Rectified Flow.
CoRR
Fu-Yun Wang, Ling Yang, Zhaoyang Huang, Mengdi Wang, Hongsheng Li
2024 J jnl
Redefining the Game: MVAE-DFDPnet's Low-Dimensional Embeddings for Superior Drug-Protein Interaction Predictions.
IEEE J. Biomed. Health Informatics
Liang-Yong Xia, Yu Wu, Longfei Zhao, Leying Chen, Shiyi Zhang, Mengdi Wang, Jie Luo
2024 J jnl
Regularized DeepIV with Model Selection.
CoRR
Zihao Li, Hui Lan, Vasilis Syrgkanis, Mengdi Wang, Masatoshi Uehara
2024 J jnl
Relative-Translation Invariant Wasserstein Distance.
CoRR
Binshuai Wang, Qiwei Di, Ming Yin, Mengdi Wang, Quanquan Gu, Peng Wei
2024 J jnl
SAIL: Self-Improving Efficient Online Alignment of Large Language Models.
CoRR
Mucong Ding, Souradip Chakraborty, Vibhu Agrawal, Zora Che, Alec Koppel, Mengdi Wang, Amrit S. Bedi, Furong Huang
2024 J jnl
Sample Complexity of Neural Policy Mirror Descent for Policy Optimization on Low-Dimensional Manifolds.
J. Mach. Learn. Res.
Zhenghao Xu, Xiang Ji, Minshuo Chen, Mengdi Wang, Tuo Zhao
2024 A* conf
Sample-Efficient Learning of POMDPs with Multiple Observations In Hindsight.
ICLR
Jiacheng Guo, Minshuo Chen, Huan Wang, Caiming Xiong, Mengdi Wang, Yu Bai
2024 J jnl
Scalable Normalizing Flows Enable Boltzmann Generators for Macromolecules.
CoRR
Joseph C. Kim, David Bloore, Karan Kapoor, Jun Feng, Ming-Hong Hao, Mengdi Wang
2024 J jnl
Self-Play with Adversarial Critic: Provable and Scalable Offline Alignment for Language Models.
CoRR
Xiang Ji, Sanjeev Kulkarni, Mengdi Wang, Tengyang Xie
2024 J jnl
SpecDec++: Boosting Speculative Decoding via Adaptive Candidate Lengths.
CoRR
Kaixuan Huang, Xudong Guo, Mengdi Wang
2024 J jnl
Teamwork Reinforcement Learning With Concave Utilities.
IEEE Trans. Mob. Comput.
Zheng Yu, Junyu Zhang, Zheng Wen, Andrea Tacchetti, Mengdi Wang, Ian Gemp
2024 J jnl
Theoretical Insights for Diffusion Guidance: A Case Study for Gaussian Mixture Models.
CoRR
Yuchen Wu, Minshuo Chen, Zihao Li, Mengdi Wang, Yuting Wei
2024 A* conf
Theoretical insights for diffusion guidance: A case study for Gaussian mixture models.
ICML
Yuchen Wu, Minshuo Chen, Zihao Li, Mengdi Wang, Yuting Wei
2024 A* conf
Theory of Consistency Diffusion Models: Distribution Estimation Meets Fast Sampling.
ICML
Zehao Dou, Minshuo Chen, Mengdi Wang, Zhuoran Yang
2024 J jnl
Transfer Q Star: Principled Decoding for LLM Alignment.
CoRR
Souradip Chakraborty, Soumya Suvra Ghosal, Ming Yin, Dinesh Manocha, Mengdi Wang, Amrit Singh Bedi, Furong Huang
2024 A* conf
Transfer Q-star : Principled Decoding for LLM Alignment.
NeurIPS
Souradip Chakraborty, Soumya Suvra Ghosal, Ming Yin, Dinesh Manocha, Mengdi Wang, Amrit Singh Bedi, Furong Huang
2024 A* conf
Tree Search-Based Evolutionary Bandits for Protein Sequence Optimization.
AAAI
Jiahao Qiu, Hui Yuan, Jinghong Zhang, Wentao Chen, Huazheng Wang, Mengdi Wang
2024 J jnl
Tree Search-Based Evolutionary Bandits for Protein Sequence Optimization.
CoRR
Jiahao Qiu, Hui Yuan, Jinghong Zhang, Wentao Chen, Huazheng Wang, Mengdi Wang
2024 J jnl
TreeBoN: Enhancing Inference-Time Alignment with Speculative Tree-Search and Best-of-N Sampling.
CoRR
Jiahao Qiu, Yifu Lu, Yifan Zeng, Jiacheng Guo, Jiayi Geng, Huazheng Wang, Kaixuan Huang, Yue Wu, Mengdi Wang
2024 J jnl
Unveil Conditional Diffusion Models with Classifier-free Guidance: A Sharp Statistical Theory.
CoRR
Hengyu Fu, Zhuoran Yang, Mengdi Wang, Minshuo Chen
2024 A* conf
Visual Adversarial Examples Jailbreak Aligned Large Language Models.
AAAI
Xiangyu Qi, Kaixuan Huang, Ashwinee Panda, Peter Henderson, Mengdi Wang, Prateek Mittal
2023 J jnl
1xN Pattern for Pruning Convolutional Neural Networks.
IEEE Trans. Pattern Anal. Mach. Intell.
Mingbao Lin, Yuxin Zhang, Yuchao Li, Bohong Chen, Fei Chao, Mengdi Wang, Shen Li, Yonghong Tian, Rongrong Ji
2023 J jnl
Actions Speak What You Want: Provably Sample-Efficient Reinforcement Learning of the Quantal Stackelberg Equilibrium from Strategic Feedbacks.
CoRR
Siyu Chen, Mengdi Wang, Zhuoran Yang
2023 J jnl
Adversarial Attacks on Online Learning to Rank with Stochastic Click Models.
CoRR
Zichen Wang, Rishab Balasubramanian, Hui Yuan, Chenyu Song, Mengdi Wang, Huazheng Wang
2023 J jnl
Aligning Agent Policy with Externalities: Reward Design via Bilevel RL.
CoRR
Souradip Chakraborty, Amrit Singh Bedi, Alec Koppel, Dinesh Manocha, Huazheng Wang, Furong Huang, Mengdi Wang
2023 A conf
Byzantine-Robust Online and Offline Distributed Reinforcement Learning.
AISTATS
Yiding Chen, Xuezhou Zhang, Kaiqing Zhang, Mengdi Wang, Xiaojin Zhu
2023 A* conf
Deep Reinforcement Learning for Cost-Effective Medical Diagnosis.
ICLR
Zheng Yu, Yikuan Li, Joseph C. Kim, Kaixuan Huang, Yuan Luo, Mengdi Wang
2023 J jnl
Deep Reinforcement Learning for Cost-Effective Medical Diagnosis.
CoRR
Zheng Yu, Yikuan Li, Joseph C. Kim, Kaixuan Huang, Yuan Luo, Mengdi Wang
2023 J jnl
Deep Reinforcement Learning for Efficient and Fair Allocation of Health Care Resources.
CoRR
Yikuan Li, Chengsheng Mao, Kaixuan Huang, Hanyin Wang, Zheng Yu, Mengdi Wang, Yuan Luo
2023 J jnl
Double Duality: Variational Primal-Dual Policy Optimization for Constrained Reinforcement Learning.
J. Mach. Learn. Res.
Zihao Li, Boyi Liu, Zhuoran Yang, Zhaoran Wang, Mengdi Wang
2023 A* conf
Effective Minkowski Dimension of Deep Nonparametric Regression: Function Approximation and Statistical Theories.
ICML
Zixuan Zhang, Minshuo Chen, Mengdi Wang, Wenjing Liao, Tuo Zhao
2023 J jnl
Effective Minkowski Dimension of Deep Nonparametric Regression: Function Approximation and Statistical Theories.
CoRR
Zixuan Zhang, Minshuo Chen, Mengdi Wang, Wenjing Liao, Tuo Zhao
2023 A* conf
Efficient RL with Impaired Observability: Learning to Act with Delayed and Missing State Observations.
NeurIPS
Minshuo Chen, Yu Bai, H. Vincent Poor, Mengdi Wang
2023 J jnl
Efficient RL with Impaired Observability: Learning to Act with Delayed and Missing State Observations.
CoRR
Minshuo Chen, Yu Bai, H. Vincent Poor, Mengdi Wang
2023 J jnl
Federated Multi-Level Optimization over Decentralized Networks.
CoRR
Shuoguang Yang, Xuezhou Zhang, Mengdi Wang
2023 J jnl
Is Inverse Reinforcement Learning Harder than Standard Reinforcement Learning?
CoRR
Lei Zhao, Mengdi Wang, Yu Bai
2023 J jnl
Knowledge Annotation for Intelligent Textbooks.
Technol. Knowl. Learn.
Mengdi Wang, Hung Chau, Khushboo Thaker, Peter Brusilovsky, Daqing He
2023 J jnl
Learning Good State and Action Representations for Markov Decision Process via Tensor Decomposition.
J. Mach. Learn. Res.
Chengzhuo Ni, Yaqi Duan, Munther A. Dahleh, Mengdi Wang, Anru R. Zhang
2023 A* conf
Learning Kernelized Contextual Bandits in a Distributed and Asynchronous Environment.
ICLR
Chuanhao Li, Huazheng Wang, Mengdi Wang, Hongning Wang
2023 J jnl
Nonparametric Classification on Low Dimensional Manifolds using Overparameterized Convolutional Residual Networks.
CoRR
Kaiqi Zhang, Zixuan Zhang, Minshuo Chen, Mengdi Wang, Tuo Zhao, Yu-Xiang Wang
2023 A* conf
Offline Reinforcement Learning with Differentiable Function Approximation is Provably Efficient.
ICLR
Ming Yin, Mengdi Wang, Yu-Xiang Wang
2023 A* conf
Posterior Sampling with Delayed Feedback for Reinforcement Learning with Linear Function Approximation.
NeurIPS
Nikki Lijing Kuang, Ming Yin, Mengdi Wang, Yu-Xiang Wang, Yian Ma
2023 J jnl
Posterior Sampling with Delayed Feedback for Reinforcement Learning with Linear Function Approximation.
CoRR
Nikki Lijing Kuang, Ming Yin, Mengdi Wang, Yu-Xiang Wang, Yi-An Ma
2023 J jnl
Primal-Dual First-Order Methods for Affinely Constrained Multi-block Saddle Point Problems.
SIAM J. Optim.
Junyu Zhang, Mengdi Wang, Mingyi Hong, Shuzhong Zhang
2023 J jnl
Provable Benefits of Policy Learning from Human Preferences in Contextual Bandit Problems.
CoRR
Xiang Ji, Huazheng Wang, Minshuo Chen, Tuo Zhao, Mengdi Wang
2023 A* conf
Provable Benefits of Representational Transfer in Reinforcement Learning.
COLT
Alekh Agarwal, Yuda Song, Wen Sun, Kaiwen Wang, Mengdi Wang, Xuezhou Zhang
2023 A* conf
Provably Efficient Representation Learning with Tractable Planning in Low-Rank POMDP.
ICML
Jiacheng Guo, Zihao Li, Huazheng Wang, Mengdi Wang, Zhuoran Yang, Xuezhou Zhang
2023 J jnl
Provably Efficient Representation Learning with Tractable Planning in Low-Rank POMDP.
CoRR
Jiacheng Guo, Zihao Li, Huazheng Wang, Mengdi Wang, Zhuoran Yang, Xuezhou Zhang
2023 J jnl
Reinforcement Learning with Human Feedback: Learning Dynamic Choices via Pessimism.
CoRR
Zihao Li, Zhuoran Yang, Mengdi Wang
2023 A* conf
Representation Learning for Low-rank General-sum Markov Games.
ICLR
Chengzhuo Ni, Yuda Song, Xuezhou Zhang, Zihan Ding, Chi Jin, Mengdi Wang
2023 A* conf
Reward-Directed Conditional Diffusion: Provable Distribution Estimation and Reward Improvement.
NeurIPS
Hui Yuan, Kaixuan Huang, Chengzhuo Ni, Minshuo Chen, Mengdi Wang
2023 J jnl
Reward-Directed Conditional Diffusion: Provable Distribution Estimation and Reward Improvement.
CoRR
Hui Yuan, Kaixuan Huang, Chengzhuo Ni, Minshuo Chen, Mengdi Wang
2023 A* conf
STEERING : Stein Information Directed Exploration for Model-Based Reinforcement Learning.
ICML
Souradip Chakraborty, Amrit S. Bedi, Alec Koppel, Mengdi Wang, Furong Huang, Dinesh Manocha
2023 J jnl
STEERING: Stein Information Directed Exploration for Model-Based Reinforcement Learning.
CoRR
Souradip Chakraborty, Amrit Singh Bedi, Alec Koppel, Mengdi Wang, Furong Huang, Dinesh Manocha
2023 J jnl
Sample Complexity of Neural Policy Mirror Descent for Policy Optimization on Low-Dimensional Manifolds.
CoRR
Zhenghao Xu, Xiang Ji, Minshuo Chen, Mengdi Wang, Tuo Zhao
2023 A* conf
Sample Complexity of Nonparametric Off-Policy Evaluation on Low-Dimensional Manifolds using Deep Networks.
ICLR
Xiang Ji, Minshuo Chen, Mengdi Wang, Tuo Zhao
2023 J jnl
Sample Complexity of Preference-Based Nonparametric Off-Policy Evaluation with Deep Networks.
CoRR
Zihao Li, Xiang Ji, Minshuo Chen, Mengdi Wang
2023 J jnl
Sample-Efficient Learning of POMDPs with Multiple Observations In Hindsight.
CoRR
Jiacheng Guo, Minshuo Chen, Huan Wang, Caiming Xiong, Mengdi Wang, Yu Bai
2023 J jnl
Scaling In-Context Demonstrations with Structured Attention.
CoRR
Tianle Cai, Kaixuan Huang, Jason D. Lee, Mengdi Wang
2023 A* conf
Score Approximation, Estimation and Distribution Recovery of Diffusion Models on Low-Dimensional Data.
ICML
Minshuo Chen, Kaixuan Huang, Tuo Zhao, Mengdi Wang
2023 J jnl
Score Approximation, Estimation and Distribution Recovery of Diffusion Models on Low-Dimensional Data.
CoRR
Minshuo Chen, Kaixuan Huang, Tuo Zhao, Mengdi Wang
2023 A* conf
Unified Off-Policy Learning to Rank: a Reinforcement Learning Perspective.
NeurIPS
Zeyu Zhang, Yi Su, Hui Yuan, Yiran Wu, Rishab Balasubramanian, Qingyun Wu, Huazheng Wang, Mengdi Wang
2023 J jnl
Unified Off-Policy Learning to Rank: a Reinforcement Learning Perspective.
CoRR
Zeyu Zhang, Yi Su, Hui Yuan, Yiran Wu, Rishab Balasubramanian, Qingyun Wu, Huazheng Wang, Mengdi Wang
2023 J jnl
Visual Adversarial Examples Jailbreak Large Language Models.
CoRR
Xiangyu Qi, Kaixuan Huang, Ashwinee Panda, Mengdi Wang, Prateek Mittal
2022 A* conf
Bandit Theory and Thompson Sampling-Guided Directed Evolution for Sequence Optimization.
NeurIPS
Hui Yuan, Chengzhuo Ni, Huazheng Wang, Xuezhou Zhang, Le Cong, Csaba Szepesvári, Mengdi Wang
2022 J jnl
Bandit Theory and Thompson Sampling-Guided Directed Evolution for Sequence Optimization.
CoRR
Hui Yuan, Chengzhuo Ni, Huazheng Wang, Xuezhou Zhang, Le Cong, Csaba Szepesvári, Mengdi Wang
2022 J jnl
Byzantine-Robust Online and Offline Distributed Reinforcement Learning.
CoRR
Yiding Chen, Xuezhou Zhang, Kaiqing Zhang, Mengdi Wang, Xiaojin Zhu
2022 A* conf
Communication Efficient Distributed Learning for Kernelized Contextual Bandits.
NeurIPS
Chuanhao Li, Huazheng Wang, Mengdi Wang, Hongning Wang
2022 J jnl
Communication Efficient Distributed Learning for Kernelized Contextual Bandits.
CoRR
Chuanhao Li, Huazheng Wang, Mengdi Wang, Hongning Wang
2022 A* conf
Decentralized Gossip-Based Stochastic Bilevel Optimization over Communication Networks.
NeurIPS
Shuoguang Yang, Xuezhou Zhang, Mengdi Wang
2022 J jnl
Decentralized Gossip-Based Stochastic Bilevel Optimization over Communication Networks.
CoRR
Shuoguang Yang, Xuezhou Zhang, Mengdi Wang
2022 J jnl
Efficient Reinforcement Learning in Block MDPs: A Model-free Representation Learning Approach.
CoRR
Xuezhou Zhang, Yuda Song, Masatoshi Uehara, Mengdi Wang, Alekh Agarwal, Wen Sun
2022 A* conf
Efficient Reinforcement Learning in Block MDPs: A Model-free Representation Learning approach.
ICML
Xuezhou Zhang, Yuda Song, Masatoshi Uehara, Mengdi Wang, Alekh Agarwal, Wen Sun
2022 J jnl
Energy System Digitization in the Era of AI: A Three-Layered Approach towards Carbon Neutrality.
CoRR
Le Xie, Tong Huang, Xiangtian Zheng, Yan Liu, Mengdi Wang, Vijay Vittal, P. R. Kumar, Srinivas Shakkottai, Yi Cui
2022 J jnl
Energy system digitization in the era of AI: A three-layered approach toward carbon neutrality.
Patterns
Le Xie, Tong Huang, Xiangtian Zheng, Yan Liu, Mengdi Wang, Vijay Vittal, P. R. Kumar, Srinivas Shakkottai, Yi Cui
2022 J jnl
Learning Markov Models Via Low-Rank Optimization.
Oper. Res.
Ziwei Zhu, Xudong Li, Mengdi Wang, Anru Zhang
2022 A* conf
Multi-Agent Reinforcement Learning with General Utilities via Decentralized Shadow Reward Actor-Critic.
AAAI
Junyu Zhang, Amrit Singh Bedi, Mengdi Wang, Alec Koppel
2022 J jnl
Near Sample-Optimal Reduction-based Policy Learning for Average Reward MDP.
CoRR
Jinghan Wang, Mengdi Wang, Lin F. Yang
2022 A* conf
Near-optimal Offline Reinforcement Learning with Linear Representation: Leveraging Variance Information with Pessimism.
ICLR
Ming Yin, Yaqi Duan, Mengdi Wang, Yu-Xiang Wang
2022 J jnl
Near-optimal Offline Reinforcement Learning with Linear Representation: Leveraging Variance Information with Pessimism.
CoRR
Ming Yin, Yaqi Duan, Mengdi Wang, Yu-Xiang Wang
2022 Misc conf
Neural Bandits for Protein Sequence Optimization.
CISS
Chenyu Wang, Joseph C. Kim, Le Cong, Mengdi Wang
2022 A* conf
Off-Policy Fitted Q-Evaluation with Differentiable Function Approximators: Z-Estimation and Inference Theory.
ICML
Ruiqi Zhang, Xuezhou Zhang, Chengzhuo Ni, Mengdi Wang
2022 J jnl
Off-Policy Fitted Q-Evaluation with Differentiable Function Approximators: Z-Estimation and Inference Theory.
CoRR
Ruiqi Zhang, Xuezhou Zhang, Chengzhuo Ni, Mengdi Wang
2022 J jnl
Offline Reinforcement Learning with Differentiable Function Approximation is Provably Efficient.
CoRR
Ming Yin, Mengdi Wang, Yu-Xiang Wang
2022 J jnl
Offline Stochastic Shortest Path: Learning, Evaluation and Towards Optimality.
CoRR
Ming Yin, Wenjing Chen, Mengdi Wang, Yu-Xiang Wang
2022 A conf
Offline stochastic shortest path: Learning, evaluation and towards optimality.
UAI
Ming Yin, Wenjing Chen, Mengdi Wang, Yu-Xiang Wang
2022 J jnl
Optimal Estimation of Off-Policy Policy Gradient via Double Fitted Iteration.
CoRR
Chengzhuo Ni, Ruiqi Zhang, Xiang Ji, Xuezhou Zhang, Mengdi Wang
2022 A* conf
Optimal Estimation of Policy Gradient via Double Fitted Iteration.
ICML
Chengzhuo Ni, Ruiqi Zhang, Xiang Ji, Xuezhou Zhang, Mengdi Wang
2022 J jnl
Parameter-Efficient Sparsity for Large Language Models Fine-Tuning.
CoRR
Yuchao Li, Fuli Luo, Chuanqi Tan, Mengdi Wang, Songfang Huang, Shen Li, Junjie Bai
2022 J jnl
Provable Benefits of Representational Transfer in Reinforcement Learning.
CoRR
Alekh Agarwal, Yuda Song, Wen Sun, Kaiwen Wang, Mengdi Wang, Xuezhou Zhang
2022 J jnl
Provably Efficient Reinforcement Learning for Online Adaptive Influence Maximization.
CoRR
Kaixuan Huang, Yu Wu, Xuezhou Zhang, Shenyinying Tu, Qingyun Wu, Mengdi Wang, Huazheng Wang
2022 J jnl
Representation Learning for General-sum Low-rank Markov Games.
CoRR
Chengzhuo Ni, Yuda Song, Xuezhou Zhang, Chi Jin, Mengdi Wang
2022 J jnl
Sample Complexity of Nonparametric Off-Policy Evaluation on Low-Dimensional Manifolds using Deep Networks.
CoRR
Xiang Ji, Minshuo Chen, Mengdi Wang, Tuo Zhao
2021 J jnl
1×N Block Pattern for Network Sparsity.
CoRR
Mingbao Lin, Yuchao Li, Yuxin Zhang, Bohong Chen, Fei Chao, Mengdi Wang, Shen Li, Jun Yang, Rongrong Ji
2021 C conf
Beyond Cumulative Returns via Reinforcement Learning over State-Action Occupancy Measures.
ACC
Junyu Zhang, Amrit Singh Bedi, Mengdi Wang, Alec Koppel
2021 J jnl
Boosting the Convergence of Reinforcement Learning-based Auto-pruning Using Historical Data.
CoRR
Jiandong Mu, Mengdi Wang, Feiwen Zhu, Jun Yang, Wei Lin, Wei Zhang
2021 A* conf
Bootstrapping Fitted Q-Evaluation for Off-Policy Inference.
ICML
Botao Hao, Xiang Ji, Yaqi Duan, Hao Lu, Csaba Szepesvári, Mengdi Wang
2021 J jnl
Bootstrapping Statistical Inference for Off-Policy Evaluation.
CoRR
Botao Hao, Xiang Ji, Yaqi Duan, Hao Lu, Csaba Szepesvári, Mengdi Wang
2021 J jnl
Cautious Reinforcement Learning via Distributional Risk in the Dual Domain.
IEEE J. Sel. Areas Inf. Theory
Junyu Zhang, Amrit Singh Bedi, Mengdi Wang, Alec Koppel
2021 A conf
Contrastive Multi-document Question Generation.
EACL
Woon Sang Cho, Yizhe Zhang, Sudha Rao, Asli Celikyilmaz, Chenyan Xiong, Jianfeng Gao, Mengdi Wang, Bill Dolan
2021 A conf
Generalization Bounds for Stochastic Saddle Point Problems.
AISTATS
Junyu Zhang, Mingyi Hong, Mengdi Wang, Shuzhong Zhang
2021 — conf
Intermittent Communications in Decentralized Shadow Reward Actor-Critic.
CDC
Amrit Singh Bedi, Alec Koppel, Mengdi Wang, Junyu Zhang
2021 B conf
Learning Good State and Action Representations via Tensor Decomposition.
ISIT
Chengzhuo Ni, Anru R. Zhang, Yaqi Duan, Mengdi Wang
2021 J jnl
Learning Good State and Action Representations via Tensor Decomposition.
CoRR
Chengzhuo Ni, Anru Zhang, Yaqi Duan, Mengdi Wang
2021 J jnl
MARL with General Utilities via Decentralized Shadow Reward Actor-Critic.
CoRR
Junyu Zhang, Amrit Singh Bedi, Mengdi Wang, Alec Koppel
2021 A* conf
On the Convergence and Sample Efficiency of Variance-Reduced Policy Gradient Method.
NeurIPS
Junyu Zhang, Chengzhuo Ni, Zheng Yu, Csaba Szepesvári, Mengdi Wang
2021 J jnl
On the Convergence and Sample Efficiency of Variance-Reduced Policy Gradient Method.
CoRR
Junyu Zhang, Chengzhuo Ni, Zheng Yu, Csaba Szepesvári, Mengdi Wang
2021 J jnl
On the Sample Complexity and Metastability of Heavy-tailed Policy Search in Continuous Control.
CoRR
Amrit Singh Bedi, Anjaly Parayil, Junyu Zhang, Mengdi Wang, Alec Koppel
2021 A conf
Online Sparse Reinforcement Learning.
AISTATS
Botao Hao, Tor Lattimore, Csaba Szepesvári, Mengdi Wang
2021 J jnl
Optimal policy evaluation using kernel-based temporal difference methods.
CoRR
Yaqi Duan, Mengdi Wang, Martin J. Wainwright
2021 A* conf
Sparse Feature Selection Makes Batch Reinforcement Learning More Sample Efficient.
ICML
Botao Hao, Yaqi Duan, Tor Lattimore, Csaba Szepesvári, Mengdi Wang
2021 A* conf
Towards Compact CNNs via Collaborative Compression.
CVPR
Yuchao Li, Shaohui Lin, Jianzhuang Liu, Qixiang Ye, Mengdi Wang, Fei Chao, Fan Yang, Jincheng Ma, Qi Tian, Rongrong Ji
2021 J jnl
Towards Compact CNNs via Collaborative Compression.
CoRR
Yuchao Li, Shaohui Lin, Jianzhuang Liu, Qixiang Ye, Mengdi Wang, Fei Chao, Fan Yang, Jincheng Ma, Qi Tian, Rongrong Ji
2021 J jnl
Voting-Based Multiagent Reinforcement Learning for Intelligent IoT.
IEEE Internet Things J.
Yue Xu, Zengde Deng, Mengdi Wang, Wenjun Xu, Anthony Man-Cho So, Shuguang Cui
2021 J jnl
You Only Compress Once: Towards Effective and Elastic BERT Compression via Exploit-Explore Stochastic Nature Gradient.
CoRR
Shaokun Zhang, Xiawu Zheng, Chenyi Yang, Yuchao Li, Yan Wang, Fei Chao, Mengdi Wang, Shen Li, Jun Yang, Rongrong Ji
2020 A* conf
A History-Based Auto-Tuning Framework for Fast and High-Performance DNN Design on GPU.
DAC
Jiandong Mu, Mengdi Wang, Lanbo Li, Jun Yang, Wei Lin, Wei Zhang
2020 J jnl
A Single Timescale Stochastic Approximation Method for Nested Stochastic Optimization.
SIAM J. Optim.
Saeed Ghadimi, Andrzej Ruszczynski, Mengdi Wang
2020 J jnl
Adaptive Low-Nonnegative-Rank Approximation for State Aggregation of Markov Chains.
SIAM J. Matrix Anal. Appl.
Yaqi Duan, Mengdi Wang, Zaiwen Wen, Yaxiang Yuan
2020 J jnl
Bridging Exploration and General Function Approximation in Reinforcement Learning: Provably Efficient Kernel and Neural Value Iterations.
CoRR
Zhuoran Yang, Chi Jin, Zhaoran Wang, Mengdi Wang, Michael I. Jordan
2020 J jnl
Cautious Reinforcement Learning via Distributional Risk in the Dual Domain.
CoRR
Junyu Zhang, Amrit Singh Bedi, Mengdi Wang, Alec Koppel
2020 J jnl
Concept Annotation for Intelligent Textbooks.
CoRR
Mengdi Wang, Hung Chau, Khushboo Thaker, Peter Brusilovsky, Daqing He
2020 A conf
Fast Training of Deep Learning Models over Multiple GPUs.
Middleware
Xiaodong Yi, Ziyue Luo, Chen Meng, Mengdi Wang, Guoping Long, Chuan Wu, Jun Yang, Wei Lin
2020 A* conf
Generalized Leverage Score Sampling for Neural Networks.
NeurIPS
Jason D. Lee, Ruoqi Shen, Zhao Song, Mengdi Wang, Zheng Yu
2020 J jnl
Generalized Leverage Score Sampling for Neural Networks.
CoRR
Jason D. Lee, Ruoqi Shen, Zhao Song, Mengdi Wang, Zheng Yu
2020 A* conf
High-Dimensional Sparse Linear Bandits.
NeurIPS
Botao Hao, Tor Lattimore, Mengdi Wang
2020 J jnl
High-Dimensional Sparse Linear Bandits.
CoRR
Botao Hao, Tor Lattimore, Mengdi Wang
2020 C conf
Improved Sample Complexity for Stochastic Compositional Variance Reduced Gradient.
ACC
Tianyi Lin, Chengyou Fan, Mengdi Wang, Michael I. Jordan
2020 A* conf
Minimax-Optimal Off-Policy Evaluation with Linear Function Approximation.
ICML
Yaqi Duan, Zeyu Jia, Mengdi Wang
2020 J jnl
Minimax-Optimal Off-Policy Evaluation with Linear Function Approximation.
CoRR
Yaqi Duan, Mengdi Wang
2020 A* conf
Model-Based Reinforcement Learning with Value-Targeted Regression.
ICML
Alex Ayoub, Zeyu Jia, Csaba Szepesvári, Mengdi Wang, Lin Yang
2020 — conf
Model-Based Reinforcement Learning with Value-Targeted Regression.
L4DC
Zeyu Jia, Lin Yang, Csaba Szepesvári, Mengdi Wang
2020 J jnl
Model-Based Reinforcement Learning with Value-Targeted Regression.
CoRR
Alex Ayoub, Zeyu Jia, Csaba Szepesvári, Mengdi Wang, Lin F. Yang
2020 J jnl
Online Sparse Reinforcement Learning.
CoRR
Botao Hao, Tor Lattimore, Csaba Szepesvári, Mengdi Wang
2020 J jnl
Picasso: A Sparse Learning Library for High Dimensional Data Analysis in R and Python.
CoRR
Jason Ge, Xingguo Li, Haoming Jiang, Han Liu, Tong Zhang, Mengdi Wang, Tuo Zhao
2020 A* conf
Provably Efficient Reinforcement Learning with Kernel and Neural Function Approximations.
NeurIPS
Zhuoran Yang, Chi Jin, Zhaoran Wang, Mengdi Wang, Michael I. Jordan
2020 J jnl
Randomized Linear Programming Solves the Markov Decision Problem in Nearly Linear (Sometimes Sublinear) Time.
Math. Oper. Res.
Mengdi Wang
2020 A* conf
Reinforcement Learning in Feature Space: Matrix Bandit, Kernels, and Regret Bound.
ICML
Lin Yang, Mengdi Wang
2020 A conf
Sketching Transformed Matrices with Applications to Natural Language Processing.
AISTATS
Yingyu Liang, Zhao Song, Mengdi Wang, Lin Yang, Xin Yang
2020 J jnl
Sketching Transformed Matrices with Applications to Natural Language Processing.
CoRR
Yingyu Liang, Zhao Song, Mengdi Wang, Lin F. Yang, Xin Yang
2020 A conf
Solving Discounted Stochastic Two-Player Games with Near-Optimal Time and Sample Complexity.
AISTATS
Aaron Sidford, Mengdi Wang, Lin Yang, Yinyu Ye
2020 J jnl
Sparse Feature Selection Makes Batch Reinforcement Learning More Sample Efficient.
CoRR
Botao Hao, Yaqi Duan, Tor Lattimore, Csaba Szepesvári, Mengdi Wang
2020 J jnl
Spectral State Compression of Markov Processes.
IEEE Trans. Inf. Theory
Anru Zhang, Mengdi Wang
2020 A* conf
Variational Policy Gradient Method for Reinforcement Learning with General Utilities.
NeurIPS
Junyu Zhang, Alec Koppel, Amrit Singh Bedi, Csaba Szepesvári, Mengdi Wang
2020 J jnl
Variational Policy Gradient Method for Reinforcement Learning with General Utilities.
CoRR
Junyu Zhang, Alec Koppel, Amrit Singh Bedi, Csaba Szepesvári, Mengdi Wang
2019 J jnl
Approximation Hardness for A Class of Sparse Optimization Problems.
J. Mach. Learn. Res.
Yichen Chen, Yinyu Ye, Mengdi Wang
2019 J jnl
Blessing of massive scale: spatial graphical model estimation with a total cardinality constraint approach.
Math. Program.
Ethan X. Fang, Han Liu, Mengdi Wang
2019 — conf
Characterizing Deep Learning Training Workloads on Alibaba-PAI.
IISWC
Mengdi Wang, Chen Meng, Guoping Long, Chuan Wu, Jun Yang, Wei Lin, Yangqing Jia
2019 J jnl
Characterizing Deep Learning Training Workloads on Alibaba-PAI.
CoRR
Mengdi Wang, Chen Meng, Guoping Long, Chuan Wu, Jun Yang, Wei Lin, Yangqing Jia
2019 J jnl
Continuous Control with Contexts, Provably.
CoRR
Simon S. Du, Ruosong Wang, Mengdi Wang, Lin F. Yang
2019 J jnl
Feature-Based Q-Learning for Two-Player Stochastic Games.
CoRR
Zeyu Jia, Lin F. Yang, Mengdi Wang
2019 A* conf
Learning low-dimensional state embeddings and metastable clusters from time series data.
NeurIPS
Yifan Sun, Yaqi Duan, Hao Gong, Mengdi Wang
2019 J jnl
Learning low-dimensional state embeddings and metastable clusters from time series data.
CoRR
Yifan Sun, Yaqi Duan, Hao Gong, Mengdi Wang
2019 — conf
Learning to Control in Metric Space with Optimal Regret.
Allerton
Chengzhuo Ni, Lin F. Yang, Mengdi Wang
2019 J jnl
Learning to Control in Metric Space with Optimal Regret.
CoRR
Lin F. Yang, Chengzhuo Ni, Mengdi Wang
2019 B conf
Maximum Likelihood Tensor Decomposition of Markov Decision Process.
ISIT
Chengzhuo Ni, Mengdi Wang
2019 J jnl
Multilevel Stochastic Gradient Methods for Nested Composition Optimization.
SIAM J. Optim.
Shuoguang Yang, Mengdi Wang, Ethan X. Fang
2019 A conf
Online Factorization and Partition of Complex Networks by Random Walk.
UAI
Lin F. Yang, Zheng Yu, Vladimir Braverman, Tuo Zhao, Mengdi Wang
2019 J jnl
Picasso: A Sparse Learning Library for High Dimensional Data Analysis in R and Python.
J. Mach. Learn. Res.
Jason Ge, Xingguo Li, Haoming Jiang, Han Liu, Tong Zhang, Mengdi Wang, Tuo Zhao
2019 J jnl
RL4health: Crowdsourcing Reinforcement Learning for Knee Replacement Pathway Optimization.
CoRR
Hao Lu, Mengdi Wang
2019 J jnl
Reinforcement Leaning in Feature Space: Matrix Bandit, Kernels, and Regret Bound.
CoRR
Lin F. Yang, Mengdi Wang
2019 A* conf
Sample-Optimal Parametric Q-Learning Using Linearly Additive Features.
ICML
Lin Yang, Mengdi Wang
2019 J jnl
Sample-Optimal Parametric Q-Learning with Linear Transition Models.
CoRR
Lin F. Yang, Mengdi Wang
2019 A* conf
Smart Roles: Inferring Professional Roles in Email Networks.
KDD
Di Jin, Mark Heimann, Tara Safavi, Mengdi Wang, Wei Lee, Lindsay Snider, Danai Koutra
2019 J jnl
Solving Discounted Stochastic Two-Player Games with Near-Optimal Time and Sample Complexity.
CoRR
Aaron Sidford, Mengdi Wang, Lin F. Yang, Yinyu Ye
2019 A* conf
State Aggregation Learning from Markov Transition Data.
NeurIPS
Yaqi Duan, Zheng Tracy Ke, Mengdi Wang
2019 J jnl
Unsupervised Common Question Generation from Multiple Documents using Reinforced Contrastive Coordinator.
CoRR
Woon Sang Cho, Yizhe Zhang, Sudha Rao, Asli Celikyilmaz, Chenyan Xiong, Jianfeng Gao, Mengdi Wang, Bill Dolan
2019 J jnl
Voting-Based Multi-Agent Reinforcement Learning.
CoRR
Yue Xu, Zengde Deng, Mengdi Wang, Wenjun Xu, Anthony Man-Cho So, Shuguang Cui
2018 J jnl
A bird's-eye view on coherence, and a worm's-eye view on cohesion.
CoRR
Woon Sang Cho, Pengchuan Zhang, Yizhe Zhang, Xiujun Li, Michel Galley, Mengdi Wang, Jianfeng Gao
2018 J jnl
Diffusion Approximations for Online Principal Component Estimation and Global Convergence.
CoRR
Chris Junchi Li, Mengdi Wang, Han Liu, Tong Zhang
2018 A* conf
Dimensionality Reduction for Stationary Time Series via Stochastic Nonconvex Optimization.
NeurIPS
Minshuo Chen, Lin Yang, Mengdi Wang, Tuo Zhao
2018 J jnl
Dimensionality Reduction for Stationary Time Series via Stochastic Nonconvex Optimization.
CoRR
Minshuo Chen, Lin Yang, Mengdi Wang, Tuo Zhao
2018 — conf
Efficient Deep Learning Inference Based on Model Compression.
CVPR Workshops
Qing Zhang, Mengru Zhang, Mengdi Wang, Wanchen Sui, Chen Meng, Jun Yang, Weidan Kong, Xiaoyuan Cui, Wei Lin
2018 A* conf
Estimation of Markov Chain via Rank-constrained Likelihood.
ICML
Xudong Li, Mengdi Wang, Anru Zhang
2018 J jnl
Estimation of Markov Chain via Rank-constrained Likelihood.
CoRR
Xudong Li, Mengdi Wang, Anru Zhang
2018 J jnl
Graph-Adaptive Pruning for Efficient Inference of Convolutional Neural Networks.
CoRR
Mengdi Wang, Qing Zhang, Jun Yang, Xiaoyuan Cui, Wei Lin
2018 J jnl
Improved Oracle Complexity for Stochastic Compositional Variance Reduced Gradient.
CoRR
Tianyi Lin, Chenyou Fan, Mengdi Wang, Michael I. Jordan
2018 A conf
Minimax-Optimal Privacy-Preserving Sparse PCA in Distributed Systems.
AISTATS
Jason Ge, Zhaoran Wang, Mengdi Wang, Han Liu
2018 A* conf
Near-Optimal Time and Sample Complexities for Solving Markov Decision Processes with a Generative Model.
NeurIPS
Aaron Sidford, Mengdi Wang, Xian Wu, Lin Yang, Yinyu Ye
2018 J jnl
Near-optimal stochastic approximation for online principal component estimation.
Math. Program.
Chris Junchi Li, Mengdi Wang, Han Liu, Tong Zhang
2018 A* conf
Scalable Bilinear Learning Using State and Action Features.
ICML
Yichen Chen, Lihong Li, Mengdi Wang
2018 J jnl
Scalable Bilinear π Learning Using State and Action Features.
CoRR
Yichen Chen, Lihong Li, Mengdi Wang
2018 J jnl
State Aggregation Learning from Markov Transition Data.
CoRR
Yaqi Duan, Zheng Tracy Ke, Mengdi Wang
2018 J jnl
State Compression of Markov Processes via Empirical Low-Rank Estimation.
CoRR
Anru Zhang, Mengdi Wang
2018 A* conf
Variance Reduced Value Iteration and Faster Algorithms for Solving Markov Decision Processes.
SODA
Aaron Sidford, Mengdi Wang, Xian Wu, Yinyu Ye
2017 J jnl
Accelerating Stochastic Composition Optimization.
J. Mach. Learn. Res.
Mengdi Wang, Ji Liu, Ethan X. Fang
2017 J jnl
Deep Primal-Dual Reinforcement Learning: Accelerating Actor-Critic using Bellman Duality.
CoRR
Woon Sang Cho, Mengdi Wang
2017 — conf
Diffusion Approximations for Online Principal Component Estimation and Global Convergence.
NIPS
Chris Junchi Li, Mengdi Wang, Tong Zhang
2017 J jnl
Dynamic Factorization and Partition of Complex Networks.
CoRR
Lin F. Yang, Vladimir Braverman, Tuo Zhao, Mengdi Wang
2017 A conf
Finite-sum Composition Optimization via Variance Reduced Gradient Descent.
AISTATS
Xiangru Lian, Mengdi Wang, Ji Liu
2017 J jnl
Lower Bound On the Computational Complexity of Discounted Markov Decision Problems.
CoRR
Yichen Chen, Mengdi Wang
2017 J jnl
Primal-Dual π Learning: Sample Complexity and Sublinear Run Time for Ergodic Markov Decision Problems.
CoRR
Mengdi Wang
2017 J jnl
Randomized Linear Programming Solves the Discounted Markov Decision Problem In Nearly-Linear Running Time.
CoRR
Mengdi Wang
2017 J jnl
Stochastic compositional gradient descent: algorithms for minimizing compositions of expected-value functions.
Math. Program.
Mengdi Wang, Ethan X. Fang, Han Liu
2017 A* conf
Strong NP-Hardness for Sparse Optimization with Concave Penalty Functions.
ICML
Yichen Chen, Dongdong Ge, Mengdi Wang, Zizhuo Wang, Yinyu Ye, Hao Yin
2017 — conf
The Signals and Noise: Actionable Information in Improvised Social Media Channels During a Disaster.
WebSci
Xingsheng He, Di Lu, Drew Margolin, Mengdi Wang, Salma El Idrissi, Yu-Ru Lin
2017 J jnl
Vanishing Price of Decentralization in Large Coordinative Nonconvex Optimization.
SIAM J. Optim.
Mengdi Wang
2017 J jnl
Variance Reduced Value Iteration and Faster Algorithms for Solving Markov Decision Processes.
CoRR
Aaron Sidford, Mengdi Wang, Xian Wu, Yinyu Ye
2016 Misc conf
A stochastic compositional gradient method using Markov samples.
WSC
Mengdi Wang, Ji Liu
2016 — conf
Accelerating Stochastic Composition Optimization.
NIPS
Mengdi Wang, Ji Liu, Ethan X. Fang
2016 — conf
An online primal-dual method for discounted Markov decision processes.
CDC
Mengdi Wang, Yichen Chen
2016 — conf
Link Prediction via Multi-hashing Framework.
SBP-BRiMS
Mengdi Wang, Yu-Ru Lin
2016 J jnl
Stochastic First-Order Methods with Random Constraint Projection.
SIAM J. Optim.
Mengdi Wang, Dimitri P. Bertsekas
2016 J jnl
Stochastic Primal-Dual Methods and Sample Complexity of Reinforcement Learning.
CoRR
Yichen Chen, Mengdi Wang
2016 — conf
TeleLink: Link Prediction in Social Network Based on Multiplex Cohesive Structures.
SBP-BRiMS
Di Jin, Mengdi Wang, Yu-Ru Lin
2015 J jnl
A Distributed Tracking Algorithm for Reconstruction of Graph Signals.
CoRR
Xiaohan Wang, Mengdi Wang, Yuantao Gu
2015 J jnl
A Distributed Tracking Algorithm for Reconstruction of Graph Signals.
IEEE J. Sel. Top. Signal Process.
Xiaohan Wang, Mengdi Wang, Yuantao Gu
2015 Misc conf
Averaging random projection: A fast online solution for large-scale constrained stochastic optimization.
ICASSP
Jialin Liu, Yuantao Gu, Mengdi Wang
2015 J jnl
Incremental constraint projection methods for variational inequalities.
Math. Program.
Mengdi Wang, Dimitri P. Bertsekas
2015 J jnl
Random Multi-Constraint Projection: Stochastic Gradient Methods for Convex Optimization with Many Constraints.
CoRR
Mengdi Wang, Yichen Chen, Jialin Liu, Yuantao Gu
2014 Misc conf
Learning distributed jointly sparse systems by collaborative LMS.
ICASSP
Yuantao Gu, Mengdi Wang
2014 — conf
Multi-task nonconvex optimization with total budget constraint: A distributed algorithm using Monte Carlo estimates.
DSP
Mengdi Wang, Yunjian Xu, Yuntao Gu
2014 J jnl
Stabilization of Stochastic Iterative Methods for Singular and Nearly Singular Linear Systems.
Math. Oper. Res.
Mengdi Wang, Dimitri P. Bertsekas
2013 — —
Stochastic methods for large-scale linear problems, variational inequalities, and convex optimization.
Mengdi Wang