← Search

Su-In Lee

18 accepted papers

2026

SurrogateSHAP: Training-Free Contributor Attribution for Text-to-Image (T2I) Models

ICML 2026poster

As Text-to-Image (T2I) diffusion models are increasingly used in real-world creative workflows, a principled framework for valuing contributors who provide a collection of data is essential for fair compensation and sustainable data marketplaces. While the Shapley value offers a theoretically ground…

Cited by 0SourceScholar
2025

An Efficient Framework for Crediting Data Contributors of Diffusion Models

ICLR 2025poster

As diffusion models are deployed in real-world settings and their performance driven by training data, appraising the contribution of data contributors is crucial to creating incentives for sharing quality data and to implementing policies for data compensation. Depending on the use case, model perf…

Cited by 0SourcePDFScholar
2025

CellCLIP - Learning Perturbation Effects in Cell Painting via Text-Guided Contrastive Learning

NeurIPS 2025poster

High-content screening (HCS) assays based on high-throughput microscopy techniques such as Cell Painting have enabled the interrogation of cells' morphological responses to perturbations at an unprecedented scale. The collection of such data promises to facilitate a better understanding of the relat…

Cited by 0SourcecodeScholar
2024

Estimating Conditional Mutual Information for Dynamic Feature Selection

ICLR 2024poster

Dynamic feature selection, where we sequentially query features to make accurate predictions with a minimal budget, is a promising paradigm to reduce feature acquisition costs and provide transparency into the prediction process. The problem is challenging, however, as it requires both making predic…

2024

Stochastic Amortization: A Unified Approach to Accelerate Feature and Data Attribution

NeurIPS 2024poster

Many tasks in explainable machine learning, such as data valuation and feature attribution, perform expensive computation for each data point and are intractable for large datasets. These methods require efficient approximations, and although amortizing the process by learning a network to directly…

Cited by 6SourcePDFScholar
2023

Contrastive Corpus Attribution for Explaining Representations

ICLR 2023poster

Despite the widespread use of unsupervised models, very few methods are designed to explain them. Most explanation methods explain a scalar model output. However, unsupervised models output representation vectors, the elements of which are not good candidates to explain because they lack semantic me…

2023

Learning to Estimate Shapley Values with Vision Transformers

ICLR 2023top-25%

Transformers have become a default architecture in computer vision, but understanding what drives their predictions remains a challenging problem. Current explanation approaches rely on attention values or input gradients, but these provide a limited view of a model’s dependencies. Shapley values of…

2023

Learning to Maximize Mutual Information for Dynamic Feature Selection

ICML 2023poster

Feature selection helps reduce data acquisition costs in ML, but the standard approach is to train models with static feature subsets. Here, we consider the dynamic feature selection (DFS) problem where a model sequentially queries features based on the presently available information. DFS is often…

2022

FastSHAP: Real-Time Shapley Value Estimation

ICLR 2022poster

Although Shapley values are theoretically appealing for explaining black-box models, they are costly to calculate and thus impractical in settings that involve large, high-dimensional models. To remedy this issue, we introduce FastSHAP, a new method for estimating Shapley values in a single forward…

Cited by 172SourcePDFScholar
2022

Moment Matching Deep Contrastive Latent Variable Models

AISTATS 2022poster

In the contrastive analysis (CA) setting, machine learning practitioners are specifically interested in discovering patterns that are enriched in a target dataset as compared to a background dataset generated from sources of variation irrelevant to the task at hand. For example, a biomedical data an…

2021

Improving KernelSHAP: Practical Shapley Value Estimation Using Linear Regression

AISTATS 2021poster

The Shapley value concept from cooperative game theory has become a popular technique for interpreting ML models, but efficiently estimating these values remains challenging, particularly in the model-agnostic setting. Here, we revisit the idea of estimating Shapley values via linear regression to u…

2020

Understanding Global Feature Contributions With Additive Importance Measures

NeurIPS 2020poster

Understanding the inner workings of complex machine learning models is a long-standing problem and most recent research has focused on local interpretability. To assess the role of individual input features in a global sense, we explore the perspective of defining feature importance through the pred…