357 citations · 681 across the 28 of their papers we have counts for
19 papers · 1 filter
Representer Point Selection for Explaining Regularized High-dimensional Models
Che-Ping Tsai, Jiong Zhang, Eli Chien +3
We introduce a novel class of sample-based explanations we term high-dimensional representers, that can be used to explain the predictions of a regularized high-dimensional model i…
PINA: Leveraging Side Information in eXtreme Multi-label Classification via Predicted Instance Neighborhood Aggregation
Eli Chien, Jiong Zhang, Cho-Jui Hsieh +4
The eXtreme Multi-label Classification~(XMC) problem seeks to find relevant labels from an exceptionally large label space. Most of the existing XMC learners focus on the extractio…
Uncertainty in Extreme Multi-label Classification
Jyun-Yu Jiang, Wei-Cheng Chang, Jiong Zhong +2
Uncertainty quantification is one of the most crucial tasks to obtain trustworthy and reliable machine learning models for decision making. However, most research in this domain ha…
ELIAS: End-to-End Learning to Index and Search in Large Output Spaces
Nilesh Gupta, Patrick H. Chen, Hsiang-Fu Yu +2
Extreme multi-label classification (XMC) is a popular framework for solving many real-world problems that require accurate prediction from a very large number of potential output c…
FINGER: Fast Inference for Graph-based Approximate Nearest Neighbor Search
Patrick H. Chen, Chang Wei-cheng, Yu Hsiang-fu +2
Approximate K-Nearest Neighbor Search (AKNNS) has now become ubiquitous in modern applications, for example, as a fast search procedure with two tower deep learning models. Graph-b…
Extreme Zero-Shot Learning for Extreme Text Classification
Yuanhao Xiong, Wei-Cheng Chang, Cho-Jui Hsieh +2
The eXtreme Multi-label text Classification (XMC) problem concerns finding most relevant labels for an input text instance from a large label set. However, the XMC setup faces two…
Node Feature Extraction by Self-Supervised Multi-scale Neighborhood Prediction
Eli Chien, Wei-Cheng Chang, Cho-Jui Hsieh +4
Learning on graphs has attracted significant attention in the learning community due to numerous real-world applications. In particular, graph neural networks (GNNs), which take nu…
Fast Multi-Resolution Transformer Fine-tuning for Extreme Multi-label Text Classification
Jiong Zhang, Wei-cheng Chang, Hsiang-fu Yu +1
Extreme multi-label text classification (XMC) seeks to find relevant labels from an extreme large label collection for a given text input. Many real-world applications can be formu…
Enterprise-Scale Search: Accelerating Inference for Sparse Extreme Multi-Label Ranking Trees
Philip A. Etter, Kai Zhong, Hsiang-Fu Yu +2
Tree-based models underpin many modern semantic search engines and recommender systems due to their sub-linear inference times. In industrial applications, these models operate at…
PECOS: Prediction for Enormous and Correlated Output Spaces
Hsiang-Fu Yu, Kai Zhong, Jiong Zhang +2
Many large-scale applications amount to finding relevant results from an enormous output space of potential candidates. For example, finding the best matching product from a large…
Extreme Multi-label Classification from Aggregated Labels
Yanyao Shen, Hsiang-fu Yu, Sujay Sanghavi +1
Extreme multi-label classification (XMC) is the problem of finding the relevant labels for an input, from a very large universe of possible labels. We consider XMC in the setting w…
Learning to Encode Position for Transformer with Continuous Dynamical Model
Xuanqing Liu, Hsiang-Fu Yu, Inderjit Dhillon +1
We introduce a new way of learning to encode position information for non-recurrent models, such as Transformer models. Unlike RNN and LSTM, which contain inductive bias by loading…
Multiresolution Transformer Networks: Recurrence is Not Essential for Modeling Hierarchical Structure
Vikas K. Garg, Inderjit S. Dhillon, Hsiang-Fu Yu
The architecture of Transformer is based entirely on self-attention, and has been shown to outperform models that employ recurrence on sequence transduction tasks such as machine t…
Graph DNA: Deep Neighborhood Aware Graph Encoding for Collaborative Filtering
Liwei Wu, Hsiang-Fu Yu, Nikhil Rao +2
In this paper, we consider recommender systems with side information in the form of graphs. Existing collaborative filtering algorithms mainly utilize only immediate neighborhood i…
AutoAssist: A Framework to Accelerate Training of Deep Neural Networks
Jiong Zhang, Hsiang-fu Yu, Inderjit S. Dhillon
Deep neural networks have yielded superior performance in many applications; however, the gradient computation in a deep model with millions of instances lead to a lengthy training…
Taming Pretrained Transformers for Extreme Multi-label Text Classification
Wei-Cheng Chang, Hsiang-Fu Yu, Kai Zhong +2
We consider the extreme multi-label text classification (XMC) problem: given an input text, return the most relevant labels from a large label collection. For example, the input te…
High-dimensional Time Series Prediction with Missing Values
Hsiang-Fu Yu, Nikhil Rao, Inderjit S. Dhillon
High-dimensional time series prediction is needed in applications as diverse as demand forecasting and climatology. Often, such applications require methods that are both highly sc…
PASSCoDe: Parallel ASynchronous Stochastic dual Co-ordinate Descent
Cho-Jui Hsieh, Hsiang-Fu Yu, Inderjit S. Dhillon
Stochastic Dual Coordinate Descent (SDCD) has become one of the most efficient ways to solve the family of -regularized empirical risk minimization problems, including line…
Large-scale Multi-label Learning with Missing Labels
Hsiang-Fu Yu, Prateek Jain, Purushottam Kar +1
The multi-label classification problem has generated significant interest in recent years. However, existing approaches do not adequately address two key challenges: (a) the abilit…