Justin Singh Kang

dblp:271/4534 · DBLP profile ↗
← Back
7ranked-venue papers
3as first author
6since 2021 · last 2025
—ORCID · none

Domains — the database's venue-derived domains; a paper can count in several

Artificial intelligence and machine learning · 5 · 2 first-author · 5 since 2021Applied, interdisciplinary, general and emerging computing · 2 · 1 first-author · 1 since 2021

Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.

Artificial intelligence
5 papers
Trustworthy machine learning · 85% Generative modeling · 6% Learning theory · 6%
Interdisciplinary, comprehensive, and emerging computing
1 paper
Bioinformatics and computational biology · 100%
Theoretical computer science
1 paper
Algorithms and data structures · 100%

Topics — the 12 heaviest of 13, each with the papers that count most for it

TopicWeightPapersLastEvidence papers
Machine learning › Trustworthy machine learning
interpretability
3.442025
SHAP zero Explains Biological Sequence Models with Near-zero Marginal Cost for Future Queries · NeurIPS 2025
ProxySPEX: Inference-Efficient Interpretability via Sparse Feature Interactions in LLMs · NeurIPS 2025
SPEX: Scaling Feature Interaction Explanations for LLMs · ICML 2025
Machine learning › Trustworthy machine learning › interpretability › attribution methods
feature interaction attribution
1.722025
ProxySPEX: Inference-Efficient Interpretability via Sparse Feature Interactions in LLMs · NeurIPS 2025
SPEX: Scaling Feature Interaction Explanations for LLMs · ICML 2025
Machine learning › Trustworthy machine learning › language model interpretability
large language model explanation
0.912025
SPEX: Scaling Feature Interaction Explanations for LLMs · ICML 2025
Machine learning › Trustworthy machine learning › interpretability › shapley value
shapley value estimation
0.912025
ProxySPEX: Inference-Efficient Interpretability via Sparse Feature Interactions in LLMs · NeurIPS 2025
Machine learning › Trustworthy machine learning › interpretability › shapley value
shapley value explanation
0.912025
SHAP zero Explains Biological Sequence Models with Near-zero Marginal Cost for Future Queries · NeurIPS 2025
Bioinformatics and computational biology
sequence analysis
0.912025
SHAP zero Explains Biological Sequence Models with Near-zero Marginal Cost for Future Queries · NeurIPS 2025
Bioinformatics and computational biology › sequence analysis › sequence modeling
sequence model interpretation
0.912025
SHAP zero Explains Biological Sequence Models with Near-zero Marginal Cost for Future Queries · NeurIPS 2025
Machine learning › Trustworthy machine learning › interpretability › feature interpretation
feature interaction detection
0.812024
Learning to Understand: Identifying Interactions via the Möbius Transform · NeurIPS 2024
Machine learning › Generative modeling
conditional generative model
0.712023
Learning a 1-layer conditional generative model in total variation · NeurIPS 2023
Machine learning › Learning theory
generalization bounds
0.712023
Learning a 1-layer conditional generative model in total variation · NeurIPS 2023
Machine learning › Trustworthy machine learning
language model interpretability
0.312025
ProxySPEX: Inference-Efficient Interpretability via Sparse Feature Interactions in LLMs · NeurIPS 2025
Machine learning › Deep learning architectures and training
ReLU networks
0.212023
Learning a 1-layer conditional generative model in total variation · NeurIPS 2023

Methods — techniques the papers use, named apart from their topics

shapley value · 3.3sparse fourier transform · 2.6model sketching · 1.7group testing · 1.5masked inference · 0.9interaction sparsity · 0.9gradient boosted trees · 0.9channel decoding · 0.9SHAP · 0.9sample complexity · 0.7
YearPublicationVenuePosition
2025 SPEX: Scaling Feature Interaction Explanations for LLMs
abstract
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 input lengths ($\approx 20$). We propose Spectral Explainer (SPEX), a model-agnostic interaction attribution algorithm that efficiently scales to large input lengths ($\approx 1000)$. SPEX exploits underlying natural sparsity among interactions—common in real-world data—and applies a sparse Fourier transform using a channel decoding algorithm to efficiently identify important interactions. We perform experiments across three difficult long-context datasets that require LLMs to utilize interactions between inputs to complete the task. For large inputs, SPEX outperforms marginal attribution methods by up to 20% in terms of faithfully reconstructing LLM outputs. Further, SPEX successfully identifies key features and interactions that strongly influence model output. For one of our datasets, HotpotQA, SPEX provides interactions that align with human annotations. Finally, we use our model-agnostic approach to generate explanations to demonstrate abstract reasoning in closed-source LLMs (GPT-4o mini) and compositional reasoning in vision-language models.
Justin Singh Kang, Landon Butler, Abhineet Agarwal, Yigit Efe Erginbas, Ramtin Pedarsani, Bin Yu 0001, Kannan Ramchandran
ICML1
2025 ProxySPEX: Inference-Efficient Interpretability via Sparse Feature Interactions in LLMs
abstract
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 number of inputs $n$. Recently, Kang et al. (2025) proposed SPEX, an information-theoretic approach that uses interaction sparsity to scale to $n \approx 10^3$ features. SPEX greatly improves upon prior methods but requires tens of thousands of model inferences, which can be prohibitive for large models. In this paper, we observe that LLM feature interactions are often *hierarchical*—higher-order interactions are accompanied by their lower-order subsets—which enables more efficient discovery. To exploit this hierarchy, we propose ProxySPEX, an interaction attribution algorithm that first fits gradient boosted trees to masked LLM outputs and then extracts the important interactions. Experiments across four challenging high-dimensional datasets show that ProxySPEX more faithfully reconstructs LLM outputs by 20\% over marginal attribution approaches while using *$10\times$ fewer inferences* than SPEX. By accounting for interactions, ProxySPEX efficiently identifies the most influential features, providing a scalable approximation of their Shapley values. Further, we apply ProxySPEX to two interpretability tasks. *Data attribution*, where we identify interactions among CIFAR-10 training samples that influence test predictions, and *mechanistic interpretability*, where we uncover interactions between attention heads, both within and across layers, on a question-answering task. The ProxySPEX algorithm is available at <https://github.com/mmschlk/shapiq>.
Landon Butler, Abhineet Agarwal, Justin Singh Kang, Yigit Efe Erginbas, Bin Yu 0001, Kannan Ramchandran
NeurIPS3
2025 SHAP zero Explains Biological Sequence Models with Near-zero Marginal Cost for Future Queries
abstract
The growing adoption of machine learning models for biological sequences has intensified the need for interpretable predictions, with Shapley values emerging as a theoretically grounded standard for model explanation. While effective for local explanations of individual input sequences, scaling Shapley-based interpretability to extract global biological insights requires evaluating thousands of sequences—incurring exponential computational cost per query. We introduce SHAP zero, a novel algorithm that amortizes the cost of Shapley value computation across large-scale biological datasets. After a one-time model sketching step, SHAP zero enables near-zero marginal cost for future queries by uncovering an underexplored connection between Shapley values, high-order feature interactions, and the sparse Fourier transform of the model. Applied to models of guide RNA efficacy, DNA repair outcomes, and protein fitness, SHAP zero explains predictions orders of magnitude faster than existing methods, recovering rich combinatorial interactions previously inaccessible at scale. This work opens the door to principled, efficient, and scalable interpretability for black-box sequence models in biology.
Darin Tsui, Aryan Musharaf, Yigit Efe Erginbas, Justin Singh Kang, Amirali Aghazadeh
NeurIPS4
2024 Learning to Understand: Identifying Interactions via the Möbius Transform
abstract
One of the key challenges in machine learning is to find interpretable representations of learned functions. The Möbius transform is essential for this purpose, as its coefficients correspond to unique *importance scores* for *sets of input variables*. This transform is closely related to widely used game-theoretic notions of importance like the *Shapley* and *Bhanzaf value*, but it also captures crucial higher-order interactions. Although computing the Möbius Transform of a function with $n$ inputs involves $2^n$ coefficients, it becomes tractable when the function is *sparse* and of *low-degree* as we show is the case for many real-world functions. Under these conditions, the complexity of the transform computation is significantly reduced. When there are $K$ non-zero coefficients, our algorithm recovers the Möbius transform in $O(Kn)$ samples and $O(Kn^2)$ time asymptotically under certain assumptions, the first non-adaptive algorithm to do so. We also uncover a surprising connection between group testing and the Möbius transform. For functions where all interactions involve at most $t$ inputs, we use group testing results to compute the Möbius transform with $O(Kt\log n)$ sample complexity and $O(K\mathrm{poly}(n))$ time. A robust version of this algorithm withstands noise and maintains this complexity. This marks the first $n$ sub-linear query complexity, noise-tolerant algorithm for the Möbius transform. While our algorithms are conceptualized in an idealized setting, they indicate that the Möbius transform is a potent tool for interpreting deep learning models.
Justin Singh Kang, Yigit Efe Erginbas, Landon Butler, Ramtin Pedarsani, Kannan Ramchandran
NeurIPS1
2023 Efficiently Computing Sparse Fourier Transforms of q-ary Functions
abstract
Fourier transformations of pseudo-Boolean functions are popular tools for analyzing functions of binary sequences. Real-world functions often have structures that manifest in a sparse Fourier transform, and previous works have shown that under the assumption of sparsity the transform can be computed efficiently. But what if we want to compute the Fourier transform of functions defined over a q-ary alphabet? These types of functions arise naturally in many areas including biology. A typical workaround is to encode the q-ary sequence in binary however, this approach is computationally inefficient and fundamentally incompatible with the existing sparse Fourier transform techniques. Herein, we develop a sparse Fourier transform algorithm specifically for q-ary functions of length n sequences, dubbed q-SFT, which provably computes an S-sparse transform with vanishing error as qn→ ∞ in O(Sn) function evaluations and O(Sn2log q) computations, where S = qnδfor some δ2) and a computational complexity of O(Sn3) with the same asymptotic guarantees. We present numerical simulations on synthetic and real-world RNA data, demonstrating the scalability of q-SFT to massively high dimensional q-ary functions.
Yigit Efe Erginbas, Justin Singh Kang, Amirali Aghazadeh, Kannan Ramchandran
ISIT2
2023 Learning a 1-layer conditional generative model in total variation
abstract
A conditional generative model is a method for sampling from a conditional distribution $p(y \mid x)$. For example, one may want to sample an image of a cat given the label ``cat''. A feed-forward conditional generative model is a function $g(x, z)$ that takes the input $x$ and a random seed $z$, and outputs a sample $y$ from $p(y \mid x)$. Ideally the distribution of outputs $(x, g(x, z))$ would be close in total variation to the ideal distribution $(x, y)$. Generalization bounds for other learning models require assumptions on the distribution of $x$, even in simple settings like linear regression with Gaussian noise. We show these assumptions are unnecessary in our model, for both linear regression and single-layer ReLU networks. Given samples $(x, y)$, we show how to learn a 1-layer ReLU conditional generative model in total variation. As our result has no assumption on the distribution of inputs $x$, if we are given access to the internal activations of a deep generative model, we can compose our 1-layer guarantee to progressively learn the deep model using a near-linear number of samples.
Ajil Jalal, Justin Singh Kang, Ananya Uppal, Kannan Ramchandran, Eric Price 0001
NeurIPS2
2020 Minimum Feedback for Collision-Free Scheduling in Massive Random Access
abstract
This paper considers a massive random access scenario where a small random set of k active users out of a larger number of n total potential users seek to transmit data to a base station. Specifically, we examine an approach in which the base station first determines the set of active users based on an uplink pilot phase, then broadcasts a common feedback message to all the active users for the scheduling of their subsequent data transmissions. Our main question is: What is the minimum amount of common feedback needed to schedule k users in k slots while completely avoiding collisions? Instead of a naive scheme of using k log(n) feedback bits, this paper presents upper and lower bounds to show that the minimum number of required common feedback bits scales linearly in k, plus an additive term that scales only as Θ(log log(n)). The achievability proof is based on a random coding argument. We further connect the problem of constructing a minimal length feedback code to that of finding a minimal set of complete k-partite subgraphs that form an edge covering of a k-uniform complete hypergraph with n vertices. Moreover, the problem is also equivalent to that of finding a minimal perfect hashing family, thus allowing leveraging the explicit perfect hashing code constructions for achieving collision-free massive random access.
Justin Singh Kang, Wei Yu 0001
ISIT1