← Search

Sriram Sankararaman

5 accepted papers

2025

CACTI: Leveraging Copy Masking and Contextual Information to Improve Tabular Data Imputation

ICML 2025spotlight

We present CACTI, a masked autoencoding approach for imputing tabular data that leverages the structure in missingness patterns and contextual information. Our approach employs a novel median truncated copy masking training strategy that encourages the model to learn from empirical patterns of missi…

Cited by 0SourcePDFScholar
2025

Raptor: Scalable Train-Free Embeddings for 3D Medical Volumes Leveraging Pretrained 2D Foundation Models

ICML 2025spotlight

Current challenges in developing foundational models for volumetric imaging data, such as magnetic resonance imaging (MRI), stem from the computational complexity of state-of-the-art architectures in high dimensions and curating sufficiently large datasets of volumes. To address these challenges, we…

2021

CONTRA: Contrarian statistics for controlled variable selection

AISTATS 2021poster

The holdout randomization test (HRT) discovers a set of covariates most predictive of a response. Given the covariate distribution, HRTs can explicitly control the false discovery rate (FDR). However, if this distribution is unknown and must be estimated from data, HRTs can inflate the FDR. To allev…

Cited by 4SourcePDFScholar
2021

Marginal Contribution Feature Importance - an Axiomatic Approach for Explaining Data

ICML 2021spotlight

In recent years, methods were proposed for assigning feature importance scores to measure the contribution of individual features. While in some cases the goal is to understand a specific model, in many cases the goal is to understand the contribution of certain properties (features) to a real-world…

2020

Explaining Groups of Points in Low-Dimensional Representations

ICML 2020poster

A common workflow in data exploration is to learn a low-dimensional representation of the data, identify groups of points in that representation, and examine the differences between the groups to determine what they represent. We treat this workflow as an interpretable machine learning problem by le…