← Search

Aria Masoomi

12 accepted papers

2025

DISCO: Disentangled Communication Steering for Large Language Models

NeurIPS 2025poster

A variety of recent methods guide large language model outputs via the inference-time addition of *steering vectors* to residual-stream or attention-head representations. In contrast, we propose to inject steering vectors directly into the query and value representation spaces within attention heads…

Cited by 0SourcecodeScholar
2025

OrdShap: Feature Position Importance for Sequential Black-Box Models

NeurIPS 2025poster

Sequential deep learning models excel in domains with temporal or sequential dependencies, but their complexity necessitates post-hoc feature attribution methods for understanding their predictions. While existing techniques quantify feature importance, they inherently assume fixed feature ordering…

Cited by 0SourceScholar
2024

Analyzing Explainer Robustness via Probabilistic Lipschitzness of Prediction Functions

AISTATS 2024poster

Machine learning methods have significantly improved in their predictive capabilities, but at the same time they are becoming more complex and less transparent. As a result, explainers are often relied on to provide interpretability to these black-box prediction models. As crucial diagnostics tools,…

2024

Boundary-Aware Uncertainty for Feature Attribution Explainers

AISTATS 2024poster

Post-hoc explanation methods have become a critical tool for understanding black-box classifiers in high-stakes applications. However, high-performing classifiers are often highly nonlinear and can exhibit complex behavior around the decision boundary, leading to brittle or misleading local explanat…

2023

Inv-Senet: Invariant Self Expression Network for Clustering Under Biased Data

ICASSP 2023accepted

Subspace clustering algorithms are used for understanding the cluster structure that explains the patterns prevalent in the dataset well. These methods are extensively used for data-exploration tasks in various areas of Natural Sciences. However, most of these methods fail to handle confounding attr…

Cited by 0SourceScholar
2023

SmoothHess: ReLU Network Feature Interactions via Stein's Lemma

NeurIPS 2023poster

Several recent methods for interpretability model feature interactions by looking at the Hessian of a neural network. This poses a challenge for ReLU networks, which are piecewise-linear and thus have a zero Hessian almost everywhere. We propose SmoothHess, a method of estimating second-order intera…

2022

Explanations of Black-Box Models based on Directional Feature Interactions

ICLR 2022spotlight

As machine learning algorithms are deployed ubiquitously to a variety of domains, it is imperative to make these often black-box models transparent. Several recent works explain black-box models by capturing the most influential features for prediction per instance; such explanation methods are uni…

2021

Reliable Estimation of KL Divergence using a Discriminator in Reproducing Kernel Hilbert Space

NeurIPS 2021spotlight

Estimating Kullback–Leibler (KL) divergence from samples of two distributions is essential in many machine learning problems. Variational methods using neural network discriminator have been proposed to achieve this task in a scalable manner. However, we noticed that most of these methods using neur…

Cited by 11SourcePDFScholar
2021

Revisiting Hilbert-Schmidt Information Bottleneck for Adversarial Robustness

NeurIPS 2021poster

We investigate the HSIC (Hilbert-Schmidt independence criterion) bottleneck as a regularizer for learning an adversarially robust deep neural network classifier. In addition to the usual cross-entropy loss, we add regularization terms for every intermediate layer to ensure that the latent representa…

2020

Instance-wise Feature Grouping

NeurIPS 2020poster

In many learning problems, the domain scientist is often interested in discovering the groups of features that are redundant and are important for classification. Moreover, the features that belong to each group, and the important feature groups may vary per sample. But what do we mean by feature…