Yihe Dong

According to our database, Yihe Dong authored at least 21 papers between 2018 and 2024.

Collaborative distances:

• Dijkstra number of four.
• Erdős number of three.

2024

AdaRank: Disagreement Based Module Rank Prediction for Low-rank Adaptation.

Yihe Dong

CoRR, 2024

Learned Feature Importance Scores for Automated Feature Engineering.

Yihe Dong, Sercan Ö. Arik, Nathanael C. Yoder, Tomas Pfister

CoRR, 2024

2023

Learning Space Partitions for Nearest Neighbor Search.

Yihe Dong, Piotr Indyk, Ilya P. Razenshteyn, Tal Wagner

IEEE Data Eng. Bull., 2023

COSTAR: Improved Temporal Counterfactual Estimation with Self-Supervised Learning.

Chuizheng Meng, Yihe Dong, Sercan Ö. Arik, Yan Liu, Tomas Pfister

CoRR, 2023

LANISTR: Multimodal Learning from Structured and Unstructured Data.

Sayna Ebrahimi, Sercan Ö. Arik, Yihe Dong, Tomas Pfister

CoRR, 2023

SLM: End-to-end Feature Selection via Sparse Learnable Masks.

Yihe Dong, Sercan Ö. Arik

CoRR, 2023

Koopman Neural Operator Forecaster for Time-series with Temporal Distributional Shifts.

Rui Wang, Yihe Dong, Sercan Ö. Arik, Rose Yu

Proceedings of the Eleventh International Conference on Learning Representations, 2023

2022

Koopman Neural Forecaster for Time Series with Temporal Distribution Shifts.

Rui Wang, Yihe Dong, Sercan Ö. Arik, Rose Yu

CoRR, 2022

2021

Attention is not all you need: pure attention loses rank doubly exponentially with depth.

Yihe Dong, Jean-Baptiste Cordonnier, Andreas Loukas

Proceedings of the 38th International Conference on Machine Learning, 2021

2020

HNHN: Hypergraph Networks with Hyperedge Neurons.

Yihe Dong, Will Sawin, Yoshua Bengio

CoRR, 2020

A Study of Performance of Optimal Transport.

Yihe Dong, Yu Gao, Richard Peng, Ilya P. Razenshteyn, Saurabh Sawlani

CoRR, 2020

Social Media Big Data-Based Research on the Influencing Factors of Insomnia and Spatiotemporal Evolution.

Yu Liu, Qinyao Luo, Hang Shen, Sida Zhuang, Chen Xu, Yihe Dong, Yukai Sun, Shaochen Wang, Hao Deng

IEEE Access, 2020

COPT: Coordinated Optimal Transport on Graphs.

Yihe Dong, Will Sawin

Proceedings of the Advances in Neural Information Processing Systems 33: Annual Conference on Neural Information Processing Systems 2020, 2020

CoinPress: Practical Private Mean and Covariance Estimation.

Sourav Biswas, Yihe Dong, Gautam Kamath, Jonathan R. Ullman

Proceedings of the Advances in Neural Information Processing Systems 33: Annual Conference on Neural Information Processing Systems 2020, 2020

Scalable Nearest Neighbor Search for Optimal Transport.

Arturs Backurs, Yihe Dong, Piotr Indyk, Ilya P. Razenshteyn, Tal Wagner

Proceedings of the 37th International Conference on Machine Learning, 2020

2019

SANNS: Scaling Up Secure Approximate k-Nearest Neighbors Search.

Hao Chen, Ilaria Chillotti, Yihe Dong, Oxana Poburinnaya, Ilya P. Razenshteyn, M. Sadegh Riazi

IACR Cryptol. ePrint Arch., 2019

Scalable Nearest Neighbor Search for Optimal Transport.

Yihe Dong, Piotr Indyk, Ilya P. Razenshteyn, Tal Wagner

CoRR, 2019

Learning Sublinear-Time Indexing for Nearest Neighbor Search.

Yihe Dong, Piotr Indyk, Ilya P. Razenshteyn, Tal Wagner

CoRR, 2019

Quantum Entropy Scoring for Fast Robust Mean Estimation and Improved Outlier Detection.

Yihe Dong, Samuel B. Hopkins, Jerry Li

Proceedings of the Advances in Neural Information Processing Systems 32: Annual Conference on Neural Information Processing Systems 2019, 2019

2018

NLP and Large-Scale Information Retrieval on Mathematical Texts.

Yihe Dong

Proceedings of the Mathematical Software - ICMS 2018, 2018

NLP-Based Detection of Mathematics Subject Classification.

Yihe Dong

Proceedings of the Mathematical Software - ICMS 2018, 2018