| 2026 | ACL | VOYAGER: A Training Free Approach for Generating Diverse Datasets using LLMs. | Avinash Amballa, Yashas Malur Saidutta, Chi-Heng Lin, Vivek Kulkarni, Srinivas Chappidi |
| 2025 | ICLR | MoDeGPT: Modular Decomposition for Large Language Model Compression. | Chi-Heng Lin, Shangqian Gao, James Seale Smith, Abhishek Patel, Shikhar Tuli, Yilin Shen, Hongxia Jin, Yen-Chang Hsu |
| 2025 | IJCNLP | MossNet: Mixture of State-Space Experts is a Multi-Head Attention. | Shikhar Tuli, James Seale Smith, Haris Jeelani, Chi-Heng Lin, Abhishek Patel, Vasili Ramanishka, Yen-Chang Hsu, Hongxia Jin |
| 2025 | NAACL | FlexiGPT: Pruning and Extending Large Language Models with Low-Rank Weight Sharing. | James Seale Smith, Chi-Heng Lin, Shikhar Tuli, Haris Jeelani, Shangqian Gao, Yilin Shen, Hongxia Jin, Yen-Chang Hsu |
| 2024 | ICML | Balanced Data, Imbalanced Spectra: Unveiling Class Disparities with Spectral Imbalance. | Chiraag Kaushik, Ran Liu, Chi-Heng Lin, Amrit Khera, Matthew Y. Jin, Wenrui Ma, Vidya Muthukumar, Eva L. Dyer |
| 2024 | NAACL | SLiM: Speculative Decoding with Hypothesis Reduction. | Chi-Heng Lin, Shikhar Tuli, James Seale Smith, Yen-Chang Hsu, Yilin Shen, Hongxia Jin |
| 2024 | NAACL | DynaMo: Accelerating Language Model Inference with Dynamic Multi-Token Sampling. | Shikhar Tuli, Chi-Heng Lin, Yen-Chang Hsu, Niraj K. Jha, Yilin Shen, Hongxia Jin |
| 2023 | ICML | Half-Hop: A graph upsampling approach for slowing down message passing. | Mehdi Azabou, Venkataramana Ganesh, Shantanu Thakoor, Chi-Heng Lin, Lakshmi Sathidevi, Ran Liu, Michal Valko, Petar Velickovic, Eva L. Dyer |
| 2022 | ICML | Provable Acceleration of Heavy Ball beyond Quadratics for a Class of Polyak-Lojasiewicz Functions when the Non-Convexity is Averaged-Out. | Jun-Kun Wang, Chi-Heng Lin, Andre Wibisono, Bin Hu |
| 2021 | ICML | Making transport more robust and interpretable by moving data through a small number of anchor points. | Chi-Heng Lin, Mehdi Azabou, Eva L. Dyer |
| 2021 | ICML | A Modular Analysis of Provable Acceleration via Polyak's Momentum: Training a Wide ReLU Network and a Deep Linear Network. | Jun-Kun Wang, Chi-Heng Lin, Jacob D. Abernethy |
| 2021 | UAI | Bayesian optimization for modular black-box systems with switching costs. | Chi-Heng Lin, Joseph D. Miano, Eva L. Dyer |
| 2020 | ICLR | Escaping Saddle Points Faster with Stochastic Momentum. | Jun-Kun Wang, Chi-Heng Lin, Jacob D. Abernethy |
| 2016 | GLOBECOM | Efficient Error-Resilient Multicasting for Multi-View 3D Videos in Wireless Network. | Chi-Heng Lin, De-Nian Yang, Ji-Tang Lee, Wanjiun Liao |