← Search

Abhineet Agarwal

7 accepted papers

2025

Proxy-SPEX: Sample-Efficient Interpretability via Sparse Feature Interactions in LLMs

NeurIPS 2025spotlight

Large Language Models (LLMs) have achieved remarkable performance by capturing complex interactions between input features. To identify these interactions, most existing approaches require enumerating all possible combinations of features up to a given order, causing them to scale poorly with the nu…

Cited by 0SourcecodeScholar
2025

SPEX: Scaling Feature Interaction Explanations for LLMs

ICML 2025poster

Large language models (LLMs) have revolutionized machine learning due to their ability to capture complex interactions between input features. Popular post-hoc explanation methods like SHAP provide *marginal* feature attributions, while their extensions to interaction importances only scale to small…

2024

ED-Copilot: Reduce Emergency Department Wait Time with Language Model Diagnostic Assistance

ICML 2024poster

In the emergency department (ED), patients undergo triage and multiple laboratory tests before diagnosis. This time-consuming process causes ED crowding which impacts patient mortality, medical errors, staff burnout, etc. This work proposes (time) *cost-effective diagnostic assistance* that leverage…

2024

Mutli-Armed Bandits with Network Interference

NeurIPS 2024poster

Online experimentation with interference is a common challenge in modern applications such as e-commerce and adaptive clinical trials in medicine. For example, in online marketplaces, the revenue of a good depends on discounts applied to competing goods. Statistical inference with interference is wi…

Cited by 5SourcePDFScholar
2023

Synthetic Combinations: A Causal Inference Framework for Combinatorial Interventions

NeurIPS 2023poster

We consider a setting where there are $N$ heterogeneous units and $p$ interventions. Our goal is to learn unit-specific potential outcomes for any combination of these $p$ interventions, i.e., $N \times 2^p$ causal parameters. Choosing a combination of interventions is a problem that naturally arise…

2022

A cautionary tale on fitting decision trees to data from additive models: generalization lower bounds

AISTATS 2022poster

Decision trees are important both as interpretable models amenable to high-stakes decision-making, and as building blocks of ensemble methods such as random forests and gradient boosting. Their statistical properties, however, are not well understood. The most cited prior works have focused on deriv…

2022

Hierarchical Shrinkage: Improving the accuracy and interpretability of tree-based models.

ICML 2022oral

Decision trees and random forests (RF) are a cornerstone of modern machine learning practice. Due to their tendency to overfit, trees are typically regularized by a variety of techniques that modify their structure (e.g. pruning). We introduce Hierarchical Shrinkage (HS), a post-hoc algorithm which…