← Search

Himanshu Jain

8 accepted papers

2025

Efficient and Asymptotically Unbiased Constrained Decoding for Large Language Models

AISTATS 2025poster

In real-world applications of large language models, outputs are often required to be confined: selecting items from predefined product or document sets, generating phrases that comply with safety standards, or conforming to specialized formatting styles. To control the generation, constrained decod…

Cited by 0SourceScholar
2023

SpecTr: Fast Speculative Decoding via Optimal Transport

NeurIPS 2023poster

Autoregressive sampling from large language models has led to state-of-the-art results in several natural language tasks. However, autoregressive sampling generates tokens one at a time making it slow, and even prohibitive in certain tasks. One way to speed up sampling is *speculative decoding*: use…

Cited by 86SourcePDFScholar
2023

Teacher Guided Training: An Efficient Framework for Knowledge Transfer

ICLR 2023poster

The remarkable performance gains realized by large pretrained models, e.g., GPT-3, hinge on the massive amounts of data they are exposed to during training. Analogously, distilling such large models to compact models for efficient deployment also necessitates a large amount of (labeled or unlabeled)…

Cited by 2SourcePDFScholar
2023

Treeformer: Dense Gradient Trees for Efficient Attention Computation

ICLR 2023poster

Standard inference and training with transformer based architectures scale quadratically with input sequence length. This is prohibitively large for a variety of applications especially in web-page translation, query-answering etc. Consequently, several approaches have been developed recently to spe…

Cited by 6SourcePDFScholar
2021

Long-tail learning via logit adjustment

ICLR 2021spotlight

Real-world classification problems typically exhibit an imbalanced or long-tailed label distribution, wherein many labels have only a few associated samples. This poses a challenge for generalisation on such labels, and also makes naive learning biased towards dominant labels. In this paper, we pr…

2020

Adversarial robustness via robust low rank representations

NeurIPS 2020poster

Adversarial robustness measures the susceptibility of a classifier to imperceptible perturbations made to the inputs at test time. In this work we highlight the benefits of natural low rank representations that often exist for real data such as images, for training neural networks with certified rob…

Cited by 27SourcePDFScholar
2017

Human Shape From Silhouettes Using Generative HKS Descriptors and Cross-Modal Neural Networks

CVPR 2017spotlight

In this work, we present a novel method for capturing human body shape from a single scaled silhouette. We combine deep correlated features capturing different 2D views, and embedding spaces based on 3D cues in a novel convolutional neural network (CNN) based architecture. We first train a CNN to fi…

Cited by 127PDFScholar
2015

Sparse Local Embeddings for Extreme Multi-label Classification

NeurIPS 2015poster

The objective in extreme multi-label learning is to train a classifier that can automatically tag a novel data point with the most relevant subset of labels from an extremely large label set. Embedding based approaches make training and prediction tractable by assuming that the training label matrix…

Cited by 562SourcePDFScholar