Search Results for author: Yihe Dong

Found 14 papers, 10 papers with code

Koopman Neural Forecaster for Time Series with Temporal Distribution Shifts

1 code implementation7 Oct 2022 Rui Wang, Yihe Dong, Sercan Ö. Arik, Rose Yu

Temporal distributional shifts, with underlying dynamics changing over time, frequently occur in real-world time series and pose a fundamental challenge for deep neural networks (DNNs).

Time Series Time Series Forecasting

COSTAR: Improved Temporal Counterfactual Estimation with Self-Supervised Learning

1 code implementation1 Nov 2023 Chuizheng Meng, Yihe Dong, Sercan Ö. Arik, Yan Liu, Tomas Pfister

Estimation of temporal counterfactual outcomes from observed history is crucial for decision-making in many domains such as healthcare and e-commerce, particularly when randomized controlled trials (RCTs) suffer from high cost or impracticality.

counterfactual Decision Making +2

Attention is Not All You Need: Pure Attention Loses Rank Doubly Exponentially with Depth

1 code implementation5 Mar 2021 Yihe Dong, Jean-Baptiste Cordonnier, Andreas Loukas

Attention-based architectures have become ubiquitous in machine learning, yet our understanding of the reasons for their effectiveness remains limited.

Inductive Bias

Scalable Nearest Neighbor Search for Optimal Transport

1 code implementation ICML 2020 Arturs Backurs, Yihe Dong, Piotr Indyk, Ilya Razenshteyn, Tal Wagner

Our extensive experiments, on real-world text and image datasets, show that Flowtree improves over various baselines and existing methods in either running time or accuracy.

Data Structures and Algorithms

Learning Space Partitions for Nearest Neighbor Search

1 code implementation ICLR 2020 Yihe Dong, Piotr Indyk, Ilya Razenshteyn, Tal Wagner

Space partitions of $\mathbb{R}^d$ underlie a vast and important class of fast nearest neighbor search (NNS) algorithms.

General Classification graph partitioning +1

CoinPress: Practical Private Mean and Covariance Estimation

3 code implementations NeurIPS 2020 Sourav Biswas, Yihe Dong, Gautam Kamath, Jonathan Ullman

We present simple differentially private estimators for the mean and covariance of multivariate sub-Gaussian data that are accurate at small sample sizes.

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

1 code implementation NeurIPS 2019 Yihe Dong, Samuel B. Hopkins, Jerry Li

In robust mean estimation the goal is to estimate the mean $\mu$ of a distribution on $\mathbb{R}^d$ given $n$ independent samples, an $\varepsilon$-fraction of which have been corrupted by a malicious adversary.

Outlier Detection

COPT: Coordinated Optimal Transport for Graph Sketching

1 code implementation9 Mar 2020 Yihe Dong, Will Sawin

We introduce COPT, a novel distance metric between graphs defined via an optimization routine, computing a coordinated pair of optimal transport maps simultaneously.

Graph Classification

A Study of Performance of Optimal Transport

1 code implementation3 May 2020 Yihe Dong, Yu Gao, Richard Peng, Ilya Razenshteyn, Saurabh Sawlani

We investigate the problem of efficiently computing optimal transport (OT) distances, which is equivalent to the node-capacitated minimum cost maximum flow problem in a bipartite graph.

SANNS: Scaling Up Secure Approximate k-Nearest Neighbors Search

no code implementations3 Apr 2019 Hao Chen, Ilaria Chillotti, Yihe Dong, Oxana Poburinnaya, Ilya Razenshteyn, M. Sadegh Riazi

In this paper, we introduce SANNS, a system for secure $k$-NNS that keeps client's query and the search result confidential.

Clustering Face Recognition +1

COPT: Coordinated Optimal Transport on Graphs

no code implementations NeurIPS 2020 Yihe Dong, Will Sawin

We introduce COPT, a novel distance metric between graphs defined via an optimization routine, computing a coordinated pair of optimal transport maps simultaneously.

Graph Classification

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

no code implementations6 Apr 2023 Yihe Dong, Sercan O. Arik

Feature selection has been widely used to alleviate compute requirements during training, elucidate model interpretability, and improve model generalizability.

feature selection

LANISTR: Multimodal Learning from Structured and Unstructured Data

no code implementations26 May 2023 Sayna Ebrahimi, Sercan O. Arik, Yihe Dong, Tomas Pfister

Multimodal large-scale pretraining has shown impressive performance for unstructured data including language, image, audio, and video.

Time Series

Cannot find the paper you are looking for? You can Submit a new open access paper.