166 citations · 514 across the 44 of their papers we have counts for
49 papers
Read What Matters: Query-Adaptive Quantization for KV Caches
Siddharth Bhandari, Lucas Gretta, Krishna Balasubramanian +1
KV-cache entries are stored before their future queries are known, but each decoding query needs precision in different places. We study this mismatch using separate budgets for re…
A Dimension-Free Bound on the Poincaré Constant of Isotropic Log-Concave Measures
Krishnakumar Balasubramanian, Shiva Kasiviswanathan
The Kannan--Lovász--Simonovits (KLS) conjecture asserts that isotropic log-concave probability measures have Poincaré constants bounded by a universal constant, independently of di…
Counterfactual Generation via Flow Matching: Coupling-Sensitive End-to-End Rates
Yunrui Guan, Krishnakumar Balasubramanian, Shiva Prasad Kasiviswanathan
Counterfactual generation seeks to sample outcomes under a hypothetical intervention or decision using observational data collected under the factual assignment mechanism. We devel…
Group Adaptive Clipping Policy Optimization
Sheng Jia, Xiao Wang, Shiva Prasad Kasiviswanathan +1
Group relative policy optimization for reinforcement learning with verifiable rewards (RLVR) typically uses a fixed importance-sampling (IS) ratio clipping boundary across all roll…
Sample Complexity of Multicalibration for Multilevel Properties
Jiuyao Lu, Krishnakumar Balasubramanian, Aleksandr Podkopaev +1
Calibration requires a predictor to be unbiased after conditioning on its own predictions. Multicalibration asks for this guarantee simultaneously across a collection of groups. Ma…
FoundCause: Causal Discovery with Latent Confounders from Observational Data
Patrick Blöbaum, Krishnakumar Balasubramanian, Shiva Prasad Kasiviswanathan
Causal discovery from observational data remains challenging due to the need to recover directed structure and latent confounding without interventions. We propose FoundCause, an a…