← Search

Justin Singh Kang

4 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

SHAP zero Explains Biological Sequence Models with Near-zero Marginal Cost for Future Queries

NeurIPS 2025poster

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 Shap…

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

Learning to Understand: Identifying Interactions via the Möbius Transform

NeurIPS 2024poster

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 use…

Cited by 3SourcePDFScholar