← Search

Xinyang Yi

10 accepted papers

2024

Aligning Large Language Models with Recommendation Knowledge

NAACL 2024findings

Large language models (LLMs) have recently been used as backbones for recommender systems. However, their performance often lags behind conventional methods in standard tasks like retrieval. We attribute this to a mismatch between LLMs’ knowledge and the knowledge crucial for effective recommendatio…

2022

Improving Multi-Task Generalization via Regularizing Spurious Correlation

NeurIPS 2022accept

Multi-Task Learning (MTL) is a powerful learning paradigm to improve generalization performance via knowledge sharing. However, existing studies find that MTL could sometimes hurt generalization, especially when two tasks are less correlated. One possible reason that hurts generalization is spurious…

Cited by 33SourcePDFScholar
2021

Learning-to-Rank with Partitioned Preference: Fast Estimation for the Plackett-Luce Model

AISTATS 2021poster

We consider the problem of listwise learning-to-rank (LTR) on data with \textit{partitioned preference}, where a set of items are sliced into ordered and disjoint partitions, but the ranking of items within a partition is unknown. The Plackett-Luce (PL) model has been widely used in listwise LTR met…

Cited by 9SourcePDFScholar
2019

Efficient Training on Very Large Corpora via Gramian Estimation

ICLR 2019poster

We study the problem of learning similarity functions over very large corpora using neural network embedding models. These models are typically trained using SGD with random sampling of unobserved pairs, with a sample size that grows quadratically with the corpus size, making it expensive to scale.…

Cited by 51SourcePDFScholar
2016

Fast Algorithms for Robust PCA via Gradient Descent

NeurIPS 2016poster

We consider the problem of Robust PCA in the fully and partially observed settings. Without corruptions, this is the well-known matrix completion problem. From a statistical standpoint this problem has been recently well-studied, and conditions on when recovery is possible (how many observations do…

Cited by 329SourcePDFScholar
2016

More Supervision, Less Computation: Statistical-Computational Tradeoffs in Weakly Supervised Learning

NeurIPS 2016poster

We consider the weakly supervised binary classification problem where the labels are randomly flipped with probability $1-\alpha$. Although there exist numerous algorithms for this problem, it remains theoretically unexplored how the statistical accuracies and computational efficiency of these algor…

Cited by 6SourcePDFScholar
2015

Optimal Linear Estimation under Unknown Nonlinear Transform

NeurIPS 2015poster

Linear regression studies the problem of estimating a model parameter $\beta^* \in \R^p$, from $n$ observations $\{(y_i,x_i)\}_{i=1}^n$ from linear model $y_i = \langle \x_i,\beta^* \rangle + \epsilon_i$. We consider a significant generalization in which the relationship between $\langle x_i,\beta^*…

Cited by 37SourcePDFScholar