Dong, Yihe

9 publications

ICML 2025 Metadata Conditioning Accelerates Language Model Pre-Training Tianyu Gao, Alexander Wettig, Luxi He, Yihe Dong, Sadhika Malladi, Danqi Chen

NeurIPSW 2023 IngesTables: Scalable and Efficient Training of LLM-Enabled Tabular Foundation Models Scott Yak, Yihe Dong, Javier Gonzalvo, Sercan Arik

ICLR 2023 Koopman Neural Operator Forecaster for Time-Series with Temporal Distributional Shifts Rui Wang, Yihe Dong, Sercan O Arik, Rose Yu

ICML 2021 Attention Is Not All You Need: Pure Attention Loses Rank Doubly Exponentially with Depth Yihe Dong, Jean-Baptiste Cordonnier, Andreas Loukas

NeurIPS 2020 COPT: Coordinated Optimal Transport on Graphs Yihe Dong, Will Sawin

NeurIPS 2020 CoinPress: Practical Private Mean and Covariance Estimation Sourav Biswas, Yihe Dong, Gautam Kamath, Jonathan Ullman

ICLR 2020 Learning Space Partitions for Nearest Neighbor Search Yihe Dong, Piotr Indyk, Ilya Razenshteyn, Tal Wagner

ICML 2020 Scalable Nearest Neighbor Search for Optimal Transport Arturs Backurs, Yihe Dong, Piotr Indyk, Ilya Razenshteyn, Tal Wagner

NeurIPS 2019 Quantum Entropy Scoring for Fast Robust Mean Estimation and Improved Outlier Detection Yihe Dong, Samuel Hopkins, Jerry Li