← Search

Rishabh Tiwari

13 accepted papers

2026

Let's (not) just put things in Context: Test-time Training for Long-context LLMs

ICLR 2026poster

Advances in training and architectural design have enabled LLMs with million-token context windows, yet in practice these models often read far more than they can reliably use. While inference-time compute scaling—typically via “thinking tokens”—can help on short multi-step reasoning tasks, our cont…

Cited by 0SourcecodeScholar
2026

LoSA: Locality Aware Sparse Attention in Diffusion Language Models

ICML 2026poster

Block-wise diffusion language models (DLMs) generate multiple tokens in parallel, offering a promising alternative to autoregressive decoding. However, their inference efficiency remains bottlenecked by memory-bound attention in long-context scenarios. Naïve sparse attention is ineffective for DLMs …

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

The Art of Scaling Reinforcement Learning Compute for LLMs

ICLR 2026oral

Reinforcement learning (RL) has become central to training large language models (LLMs), yet the field lacks predictive scaling methodologies comparable to those established for pre-training. Despite rapidly rising compute budgets, there is no principled understanding of how to evaluate algo…

Cited by 0SourceScholar
2026

V1: Unifying Generation and Self-Verification for Parallel Reasoners

ICML 2026poster

Test-time scaling for complex reasoning tasks shows that leveraging inference-time compute, for example by independently sampling and aggregating multiple solutions, results in significantly better task outcomes. However, a critical bottleneck is _verification_: sampling is only effective if correct…

Cited by 0SourceScholar
2025

QuantSpec: Self-Speculative Decoding with Hierarchical Quantized KV Cache

ICML 2025poster

Large Language Models (LLMs) are increasingly being deployed on edge devices for long-context settings, creating a growing need for fast and efficient long-context inference. In these scenarios, the Key-Value (KV) cache is the primary bottleneck in terms of both GPU memory and latency, as the full K…

Cited by 0SourcePDFScholar
2025

Why Do Multi-Agent LLM Systems Fail?

NeurIPS 2025spotlight

Despite enthusiasm for Multi-Agent LLM Systems (MAS), their performance gains on popular benchmarks are often minimal. This gap highlights a critical need for a principled understanding of why MAS fail. Addressing this question requires systematic identification and analysis of failure patterns. We…

Cited by 0SourcecodeScholar
2023

Interactive Concept Bottleneck Models

AAAI 2023technical

Concept bottleneck models (CBMs) are interpretable neural networks that first predict labels for human-interpretable concepts relevant to the prediction task, and then predict the final label based on the concept label predictions. We extend CBMs to interactive prediction settings where the model ca…

2023

On Designing Light-Weight Object Trackers Through Network Pruning: Use CNNS or Transformers?

ICASSP 2023accepted

Object trackers deployed on low-power devices need to be light-weight, however, most of the current state-of-the-art (SOTA) methods rely on using compute-heavy backbones built using CNNs or Transformers. Large sizes of such models do not allow their deployment in low-power conditions and designing c…

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
2022

GCR: Gradient Coreset Based Replay Buffer Selection for Continual Learning

CVPR 2022poster

Continual learning (CL) aims to develop techniques by which a single model adapts to an increasing number of tasks encountered sequentially, thereby potentially leveraging learnings across tasks in a resource-efficient manner. A major challenge for CL systems is catastrophic forgetting, where earlie…

Cited by 157PDFScholar
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…