← Search

Puneet K. Dokania

23 accepted papers

2025

Mixture of Experts Made Intrinsically Interpretable

ICML 2025poster

Neurons in large language models often exhibit \emph{polysemanticity}, simultaneously encoding multiple unrelated concepts and obscuring interpretability. Instead of relying on post-hoc methods, we present \textbf{MoE-X}, a mixture-of-experts (MoE) language model designed to be \emph{intrinsically}…

Cited by 0SourcePDFScholar
2024

RanDumb: Random Representations Outperform Online Continually Learned Representations

NeurIPS 2024poster

Continual learning has primarily focused on the issue of catastrophic forgetting and the associated stability-plasticity tradeoffs. However, little attention has been paid to the efficacy of continually learned representations, as representations are learned alongside classifiers throughout the lear…

Cited by 0SourcePDFScholar
2024

What Makes and Breaks Safety Fine-tuning? A Mechanistic Study

NeurIPS 2024poster

Safety fine-tuning helps align Large Language Models (LLMs) with human preferences for their safe deployment. To better understand the underlying factors that make models safe via safety fine-tuning, we design a synthetic data generation framework that captures salient aspects of an unsafe input by…

2023

Computationally Budgeted Continual Learning: What Does Matter?

CVPR 2023poster

Continual Learning (CL) aims to sequentially train models on streams of incoming data that vary in distribution by preserving previous knowledge while adapting to new data. Current CL literature focuses on restricted access to previously seen data, while imposing no constraints on the computational…

2023

Graph Inductive Biases in Transformers without Message Passing

ICML 2023poster

Transformers for graph data are increasingly widely studied and successful in numerous learning tasks. Graph inductive biases are crucial for Graph Transformers, and previous works incorporate them using message-passing modules and/or positional encodings. However, Graph Transformers that use messag…

2023

Query-Based Hard-Image Retrieval for Object Detection at Test Time

AAAI 2023technical

There is a longstanding interest in capturing the error behaviour of object detectors by finding images where their performance is likely to be unsatisfactory. In real-world applications such as autonomous driving, it is also crucial to characterise potential failures beyond simple requirements of d…

2023

Sample-Dependent Adaptive Temperature Scaling for Improved Calibration

AAAI 2023technical

It is now well known that neural networks can be wrong with high confidence in their predictions, leading to poor calibration. The most common post-hoc approach to compensate for this is to perform temperature scaling, which adjusts the confidences of the predictions on any input by scaling the logi…

2023

Towards Building Self-Aware Object Detectors via Reliable Uncertainty Quantification and Calibration

CVPR 2023poster

The current approach for testing the robustness of object detectors suffers from serious deficiencies such as improper methods of performing out-of-distribution detection and using calibration metrics which do not consider both localisation and classification quality. In this work, we address these…

2022

An Impartial Take to the CNN vs Transformer Robustness Contest

ECCV 2022poster

"Following the surge of popularity of Transformers in Computer Vision, several studies have attempted to determine whether they could be more robust to distribution shifts and provide better uncertainty estimates than Convolutional Neural Networks (CNNs). The almost unanimous conclusion is that they…

2022

Make Some Noise: Reliable and Efficient Single-Step Adversarial Training

NeurIPS 2022accept

Recently, Wong et al. (2020) showed that adversarial training with single-step FGSM leads to a characteristic failure mode named catastrophic overfitting (CO), in which a model becomes suddenly vulnerable to multi-step attacks. Experimentally they showed that simply adding a random perturbation prio…

2022

Using Mixup as a Regularizer Can Surprisingly Improve Accuracy & Out-of-Distribution Robustness

NeurIPS 2022accept

We show that the effectiveness of the well celebrated Mixup can be further improved if instead of using it as the sole learning objective, it is utilized as an additional regularizer to the standard cross-entropy loss. This simple change not only improves accuracy but also significantly improves the…

Cited by 103SourcePDFScholar
2021

A Continuous Mapping For Augmentation Design

NeurIPS 2021poster

Automated data augmentation (ADA) techniques have played an important role in boosting the performance of deep models. Such techniques mostly aim to optimize a parameterized distribution over a discrete augmentation space. Thus, are restricted by the discretization of the search space which normally…

Cited by 5SourcePDFScholar
2021

No Cost Likelihood Manipulation at Test Time for Making Better Mistakes in Deep Networks

ICLR 2021poster

There has been increasing interest in building deep hierarchy-aware classifiers that aim to quantify and reduce the severity of mistakes, and not just reduce the number of errors. The idea is to exploit the label hierarchy (e.g., the WordNet ontology) and consider graph distances as a proxy for mist…

2021

Progressive Skeletonization: Trimming more fat from a network at initialization

ICLR 2021poster

Recent studies have shown that skeletonization (pruning parameters) of networks at initialization provides all the practical benefits of sparsity both at inference and training time, while only marginally degrading their performance. However, we observe that beyond a certain level of sparsity (appro…

2020

GDumb: A Simple Approach that Questions Our Progress in Continual Learning

ECCV 2020poster

We discuss a general formulation for the Continual Learning (CL) problem for classification---a learning task where a stream provides samples to a learner and the goal of the learner, depending on the samples it receives, is to continually upgrade its knowledge about the old classes and learn new on…

2020

Stable Rank Normalization for Improved Generalization in Neural Networks and GANs

ICLR 2020spotlight

Exciting new work on generalization bounds for neural networks (NN) given by Bartlett et al. (2017); Neyshabur et al. (2018) closely depend on two parameter- dependant quantities: the Lipschitz constant upper bound and the stable rank (a softer version of rank). Even though these bounds typically ha…

Cited by 55SourceScholar
2019

Interactive Sketch & Fill: Multiclass Sketch-to-Image Translation

ICCV 2019poster

We propose an interactive GAN-based sketch-to-image translation method that helps novice users easily create images of simple objects. The user starts with a sparse sketch and a desired object category, and the network then recommends its plausible completion(s) and shows a corresponding synthesized…

Cited by 166PDFcodeScholar
2019

Proximal Mean-Field for Neural Network Quantization

ICCV 2019poster

Compressing large Neural Networks (NN) by quantizing the parameters, while maintaining the performance is highly desirable due to reduced memory and time complexity. In this work, we cast NN quantization as a discrete labelling problem, and by examining relaxations, we design an efficient iterative…

Cited by 24PDFcodeScholar
2018

FlipDial: A Generative Model for Two-Way Visual Dialogue

CVPR 2018poster

We present FlipDial, a generative model for Visual Dialogue that simultaneously plays the role of both participants in a visually-grounded dialogue. Given context in the form of an image and an associated caption summarising the contents of the image, FlipDial learns both to answer questions and put…

Cited by 48SourcePDFScholar
2018

Multi-Agent Diverse Generative Adversarial Networks

CVPR 2018poster

We propose MAD-GAN, an intuitive generalization to the Generative Adversarial Networks (GANs) and its conditional variants to address the well known problem of mode collapse. First, MAD-GAN is a multi-agent GAN architecture incorporating multiple generators and one discriminator. Second, to enforce…

Cited by 429SourcePDFScholar
2018

Riemannian Walk for Incremental Learning: Understanding Forgetting and Intransigence

ECCV 2018poster

Incremental learning (IL) has received a lot of attention recently, however, the literature lacks a precise problem definition, proper evaluation settings, and metrics tailored specifically for the IL problem. One of the main objectives of this work is to fill these gaps so as to provide a common gr…

2015

Parsimonious Labeling

ICCV 2015poster

We propose a new family of discrete energy minimization problems, which we call parsimonious labeling. Our energy function consists of unary potentials and high-order clique potentials. While the unary potentials are arbitrary, the clique potentials are proportional to the diversity of the set of un…

Cited by 13PDFScholar