← Search

Harit Vishwakarma

12 accepted papers

2025

Prune 'n Predict: Optimizing LLM Decision-making with Conformal Prediction

ICML 2025poster

Large language models (LLMs) are empowering decision-making in several applications, including tool or API usage and answering multiple-choice questions (MCQs). However, incorrect outputs pose significant risks in high-stakes domains like healthcare and finance. To quantify LLM uncertainty and there…

Cited by 0SourcePDFScholar
2025

Rethinking Confidence Scores and Thresholds in Pseudolabeling-based SSL

ICML 2025poster

Modern semi-supervised learning (SSL) methods rely on pseudolabeling and consistency regularization. Pseudolabeling is typically performed by comparing the model's confidence scores and a predefined threshold. While several heuristics have been proposed to improve threshold selection, the underlyin…

Cited by 0SourcePDFScholar
2024

OTTER: Effortless Label Distribution Adaptation of Zero-shot Models

NeurIPS 2024poster

Popular zero-shot models suffer due to artifacts inherited from pretraining. One particularly detrimental issue, caused by unbalanced web-scale pretraining data, is mismatched label distribution. Existing approaches that seek to repair the label distribution are not suitable in zero-shot settings, a…

2024

Pearls from Pebbles: Improved Confidence Functions for Auto-labeling

NeurIPS 2024poster

Auto-labeling is an important family of techniques that produce labeled training sets with minimum manual annotation. A prominent variant, threshold-based auto-labeling (TBAL), works by finding thresholds on a model's confidence scores above which it can accurately automatically label unlabeled data…

Cited by 2SourcePDFScholar
2024

Taming False Positives in Out-of-Distribution Detection with Human Feedback

AISTATS 2024poster

Robustness to out-of-distribution (OOD) samples is crucial for the safe deployment of machine learning models in the open world. Recent works have focused on designing scoring functions to quantify OOD uncertainty. Setting appropriate thresholds for these scoring functions for OOD detection is chall…

2023

Promises and Pitfalls of Threshold-based Auto-labeling

NeurIPS 2023spotlight

Creating large-scale high-quality labeled datasets is a major bottleneck in supervised machine learning workflows. Threshold-based auto-labeling (TBAL), where validation data obtained from humans is used to find a confidence threshold above which the data is machine-labeled, reduces reliance on manu…

2022

Universalizing Weak Supervision

ICLR 2022poster

Weak supervision (WS) frameworks are a popular way to bypass hand-labeling large datasets for training data-hungry models. These approaches synthesize multiple noisy but cheaply-acquired estimates of labels into a set of high-quality pseudo-labels for downstream training. However, the synthesis tech…

Cited by 43SourcePDFScholar
2020

Attack of the Tails: Yes, You Really Can Backdoor Federated Learning

NeurIPS 2020poster

Due to its decentralized nature, Federated Learning (FL) lends itself to adversarial attacks in the form of backdoors during training. The goal of a backdoor is to corrupt the performance of the trained model on specific sub-tasks (e.g., by classifying green cars as frogs). A range of FL backdoor at…

2020

Optimal Lottery Tickets via Subset Sum: Logarithmic Over-Parameterization is Sufficient

NeurIPS 2020spotlight

The strong lottery ticket hypothesis (LTH) postulates that one can approximate any target neural network by only pruning the weights of a sufficiently over-parameterized random network. A recent work by Malach et al. [MYSS20] establishes the first theoretical analysis for the strong LTH: one can pr…

2019

Quantum Embedding of Knowledge for Reasoning

NeurIPS 2019poster

Statistical Relational Learning (SRL) methods are the most widely used techniques to generate distributional representations of the symbolic Knowledge Bases (KBs). These methods embed any given KB into a vector space by exploiting statistical similarities among its entities and predicates but withou…