← Search

Vidit Jain

3 accepted papers

2024

Enhancing Tail Performance in Extreme Classifiers by Label Variance Reduction

ICLR 2024poster

Extreme Classification (XC) architectures, which utilize a massive One-vs-All (OvA) classifier layer at the output, have demonstrated remarkable performance on problems with large label sets. Nonetheless, these architectures falter on tail labels with few representative samples. This phenomenon has…

Cited by 7SourcePDFScholar
2022

Neural Models for Output-Space Invariance in Combinatorial Problems

ICLR 2022poster

Recently many neural models have been proposed to solve combinatorial puzzles by implicitly learning underlying constraints using their solved instances, such as sudoku or graph coloring (GCP). One drawback of the proposed architectures, which are often based on Graph Neural Networks (GNN) (Zhou et…

Cited by 4SourcePDFScholar
2019

Deep Latent Factor Model for Predicting Drug Target Interactions

ICASSP 2019accepted

In drug target interaction (DTI) the interactions of some (a subset) drugs on some (a subset) targets are known. The goal is to predict the interactions of all drugs on all targets. One approach is to formulate this as a matrix completion problem, where the matrix of interactions having drugs along…

Cited by 0SourceScholar