← Search

Jiong Zhang

10 accepted papers

2023

PINA: Leveraging Side Information in eXtreme Multi-label Classification via Predicted Instance Neighborhood Aggregation

ICML 2023poster

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 extraction of semantic features from input query text. However, conventional XMC studies usually neglect the side information of…

2023

Representer Point Selection for Explaining Regularized High-dimensional Models

ICML 2023poster

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 in terms of importance weights for each of the training samples. Our workhorse is a novel representer theorem for gener…

Cited by 4SourcePDFScholar
2022

Node Feature Extraction by Self-Supervised Multi-scale Neighborhood Prediction

ICLR 2022poster

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 \emph{numerical} node features and graph structure as inputs, have been shown to achieve state-of-the-art performance on v…

2021

Fast Multi-Resolution Transformer Fine-tuning for Extreme Multi-label Text Classification

NeurIPS 2021poster

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 formulated as XMC problems, such as recommendation systems, document tagging and semantic search. Recently, transformer based…

2019

AutoAssist: A Framework to Accelerate Training of Deep Neural Networks

NeurIPS 2019poster

Deep neural networks have yielded superior performance in many contemporary applications. However, the gradient computation in a deep model with millions of instances leads to a lengthy training process even with modern GPU/TPU hardware acceleration. In this paper, we propose AutoAssist, a simple f…

2019

Extreme Stochastic Variational Inference: Distributed Inference for Large Scale Mixture Models

AISTATS 2019poster

Mixture of exponential family models are among the most fundamental and widely used statistical models. Stochastic variational inference (SVI), the state-of-the-art algorithm for parameter estimation in such models is inherently serial. Moreover, it requires the parameters to fit in the memory of a…

Cited by 5SourcePDFScholar
2018

Learning Long Term Dependencies via Fourier Recurrent Units

ICML 2018oral

It is a known fact that training recurrent neural networks for tasks that have long term dependencies is challenging. One of the main reasons is the vanishing or exploding gradient problem, which prevents gradient information from propagating to early layers. In this paper we propose a simple recurr…

2018

Stabilizing Gradients for Deep Neural Networks via Efficient SVD Parameterization

ICML 2018oral

Vanishing and exploding gradients are two of the main obstacles in training deep neural networks, especially in capturing long range dependencies in recurrent neural networks (RNNs). In this paper, we present an efficient parametrization of the transition matrix of an RNN that allows us to stabilize…

2017

Scalable Convex Multiple Sequence Alignment via Entropy-Regularized Dual Decomposition

AISTATS 2017poster

Multiple Sequence Alignment (MSA) is one of the fundamental tasks in biological sequence analysis that underlies applications such as phylogenetic trees, profiles, and structure prediction. The task, however, is NP-hard, and the current practice resorts to heuristic and local-search methods. Recentl…

Cited by 1SourcePDFScholar
2016

A Convex Atomic-Norm Approach to Multiple Sequence Alignment and Motif Discovery

ICML 2016poster

Multiple Sequence Alignment and Motif Discovery, known as NP-hard problems, are two fundamental tasks in Bioinformatics. Existing approaches to these two problems are based on either local search methods such as Expectation Maximization (EM), Gibbs Sampling or greedy heuristic methods. In this work,…

Cited by 14SourcePDFScholar