← Search

Udbhav Bamba

6 accepted papers

2026

DOT-MoE: Differentiable Optimal Transport for MoEfication

ICML 2026poster

The scaling of Large Language Models (LLMs) has driven significant performance gains but created substantial challenges in inference efficiency. While Mixture of Experts (MoEs) architectures address this by decoupling model size from inference cost, training MoEs from scratch is often unstable and c…

Cited by 0SourceScholar
2026

Reward Under Attack: Analyzing the Robustness and Hackability of Process Reward Models

ICML 2026poster

Process Reward Models (PRMs) are rapidly becoming the backbone of LLM reasoning pipelines, yet we demonstrate that state-of-the-art PRMs are systematically exploitable under optimization pressure. We introduce a three-tiered diagnostic framework that applies increasing adversarial pressure to quanti…

Cited by 0SourceScholar
2026

S2D: Selective Spectral Decay for Quantization-Friendly Conditioning of Neural Activations

CVPR 2026

Activation outliers in large-scale transformer models pose a fundamental challenge to model quantization, creating excessively large ranges that cause severe accuracy drops during quantization. We empirically observe that outlier severity intensifies with pre-training scale (e.g., progressing from C

Cited by 0SourceScholar
2026

XRPO: Pushing the Limits of GRPO with Targeted Exploration and Exploitation

ICML 2026poster

Reinforcement learning algorithms such as GRPO have driven recent advances in large language model (LLM) reasoning. While scaling the number of rollouts stabilizes training, existing approaches suffer from limited exploration on challenging prompts and leave informative feedback signals underexploit…

Cited by 0SourceScholar
2022

Dynamic Kernel Selection for Improved Generalization and Memory Efficiency in Meta-Learning

CVPR 2022poster

Gradient based meta-learning methods are prone to overfit on the meta-training set, and this behaviour is more prominent with large and complex networks. Moreover, large networks restrict the application of meta-learning models on low-power edge devices. While choosing smaller networks avoid these i…

Cited by 7PDFcodeScholar
2021

ChipNet: Budget-Aware Pruning with Heaviside Continuous Approximations

ICLR 2021poster

Structured pruning methods are among the effective strategies for extracting small resource-efficient convolutional neural networks from their dense counterparts with minimal loss in accuracy. However, most existing methods still suffer from one or more limitations, that include 1) the need for trai…