Interpretable Deep Clustering for Tabular Data
Not on the map. Why?
Abstract (arXiv)
Clustering is a fundamental learning task widely used as a first step in data analysis. For example, biologists use cluster assignments to analyze genome sequences, medical records, or images. Since downstream analysis is typically performed at the cluster level, practitioners seek reliable and interpretable clustering models. We propose a new deep-learning framework for general domain tabular data that predicts interpretable cluster assignments at the instance and cluster levels. First, we present a self-supervised procedure to identify the subset of the most informative features from each data point. Then, we design a model that predicts cluster assignments and a gate matrix that provides cluster-level feature selection. Overall, our model provides cluster assignments with an indication of the driving feature for each sample and each cluster. We show that the proposed method can reliably predict cluster assignments in biological, text, image, and physics tabular datasets. Furthermore, using previously proposed metrics, we verify that our model leads to interpretable results at a sample and cluster level. Our code is available at https://github.com/jsvir/idc.
Key facts. Cited by 19 papers in the map (data through 7 Oct 2026). Builds on 25 of its 72 references that are in the map. First posted on arXiv 7 Jun 2023.
Builds on
25 of 72 references.
References · 25
- 2017A Unified Approach to Interpreting Model Predictionscited by 6,488
- 2017Axiomatic Attribution for Deep Networkscited by 3,350
- 2015Unsupervised Deep Embedding for Clustering Analysiscited by 763
- 2018Towards Robust Interpretability with Self-Explaining Neural Networkscited by 505
- 2016Variational Deep Embedding: An Unsupervised and Generative Approach to Clusteringcited by 219
- 2017OpenML Benchmarking Suitescited by 169
- 2020Learning Diverse and Discriminative Representations via the Principle of Maximal Coding Rate Reductioncited by 141
- 2020Frequency Bias in Neural Networks for Input of Non-Uniform Densitycited by 140
- 2021SPICE: Semantic Pseudo-labeling for Image Clusteringcited by 104
- 2018SpectralNet: Spectral Clustering using Deep Neural Networkscited by 94
- 2019Concrete Autoencoders for Differentiable Feature Selection and Reconstructioncited by 90
- 2023Synthcity: facilitating innovative use cases of synthetic data in different data modalitiescited by 76
13 more
- 2022Neural Manifold Clustering and Embeddingcited by 44
- 2021You Never Cluster Alonecited by 38
- 2019From Clustering to Cluster Explanations via Neural Networkscited by 34
- 2021Locally Sparse Neural Networks for Tabular Biomedical Datacited by 34
- 2021Pseudo-supervised Deep Subspace Clusteringcited by 29
- 2018XAI Beyond Classification: Interpretable Neural Clusteringcited by 24
- 2022Where to Pay Attention in Sparse Training for Feature Selection?cited by 15
- 2019Deep clustering with concrete k-meanscited by 11
- 2021Deep Unsupervised Feature Selection by Discarding Nuisance and Correlated Featurescited by 11
- 2021Interpretable Clustering via Multi-Polytope Machinescited by 11
- 2022Composite Feature Selection using Deep Ensemblescited by 11
- 2021Support Recovery with Stochastic Gates: Theory and Application for Linear Modelscited by 8
- 2021Feature selection revisited in the single-cell eracited by 7
Built on by
Citing papers per year: 2024: 6; 2025: 6; 2026: 7.
19 papers.
Sort:
- 2025Representation Learning for Tabular Data: A Comprehensive Surveycited by 36
- 2024Interpretable Clustering: A Surveycited by 8
- 2025ZEUS: Zero-shot Embeddings for Unsupervised Separation of Tabular Datacited by 6
- 2025ICR-RL: Deep Reinforcement Learning via In-Context Regressioncited by 5
- 2025Hybrid Autoencoders for Tabular Data: Leveraging Model-Based Augmentation in Low-Label Settingscited by 5
- 2024Unveiling Multiple Descents in Unsupervised Autoencoderscited by 3
- 2024NeurCAM: Interpretable Neural Clustering via Additive Modelscited by 3
- 2025NetBurst: Event-Centric Forecasting of Bursty, Intermittent Time Seriescited by 2
- 2024Self Supervised Correlation-based Permutations for Multi-View Clusteringcited by 1
- 2024Spectral Self-supervised Feature Selectioncited by 1
- 2025Distributionally Robust Feature Selectioncited by 1
- 2024Conditional Deep Canonical Time Warpingcited by 0
7 more
- 2026i-IF-Learn: Iterative Feature Selection and Unsupervised Learning for High-Dimensional Complex Datacited by 0
- 2026Weight-Informed Self-Explaining Clustering for Mixed-Type Tabular Datacited by 0
- 2026Beyond Statistical Co-occurrence: Unlocking Intrinsic Semantics for Tabular Data Clusteringcited by 0
- 2026Cohort Organized Learning: Clustering Through Agreementcited by 0
- 2026Ensemble of Unsupervised Deep Learning for Clustering Imbalanced Tabular Datacited by 0
- 2026Population-Robust Feature Selection via Generalized Welfare Optimizationcited by 0
- 2026GRACE: LLM-Grounded Semantic Metric Spaces for Scalable Mixed-Data Clusteringcited by 0
- 2026GRACE: LLM-Grounded Semantic Metric Spaces for Scalable Mixed-Data Clustering8 Aug 2026
- 2026Population-Robust Feature Selection via Generalized Welfare Optimization3 Aug 2026
- 2026Ensemble of Unsupervised Deep Learning for Clustering Imbalanced Tabular Data31 Jul 2026
- 2026Cohort Organized Learning: Clustering Through Agreement19 Jun 2026
- 2026Beyond Statistical Co-occurrence: Unlocking Intrinsic Semantics for Tabular Data Clustering13 Apr 2026
- 2026Weight-Informed Self-Explaining Clustering for Mixed-Type Tabular Data7 Apr 2026
- 2026i-IF-Learn: Iterative Feature Selection and Unsupervised Learning for High-Dimensional Complex Data25 Mar 2026
- 2025Hybrid Autoencoders for Tabular Data: Leveraging Model-Based Augmentation in Low-Label Settings10 Nov 2025
- 2025NetBurst: Event-Centric Forecasting of Bursty, Intermittent Time Series25 Oct 2025
- 2025Distributionally Robust Feature Selection24 Oct 2025
- 2025ICR-RL: Deep Reinforcement Learning via In-Context Regression14 Sep 2025
- 2025ZEUS: Zero-shot Embeddings for Unsupervised Separation of Tabular Data15 May 2025
7 more
- 2025Representation Learning for Tabular Data: A Comprehensive Survey17 Apr 2025
- 2024Conditional Deep Canonical Time Warping24 Dec 2024
- 2024Interpretable Clustering: A Survey1 Sep 2024
- 2024NeurCAM: Interpretable Neural Clustering via Additive Models23 Aug 2024
- 2024Spectral Self-supervised Feature Selection12 Jul 2024
- 2024Unveiling Multiple Descents in Unsupervised Autoencoders17 Jun 2024
- 2024Self Supervised Correlation-based Permutations for Multi-View Clustering26 Feb 2024
- 2025Representation Learning for Tabular Data: A Comprehensive Surveycited by 36
- 2024Interpretable Clustering: A Surveycited by 8
- 2025ZEUS: Zero-shot Embeddings for Unsupervised Separation of Tabular Datacited by 6
- 2025ICR-RL: Deep Reinforcement Learning via In-Context Regressioncited by 5
- 2025Hybrid Autoencoders for Tabular Data: Leveraging Model-Based Augmentation in Low-Label Settingscited by 5
- 2024Unveiling Multiple Descents in Unsupervised Autoencoderscited by 3
- 2024NeurCAM: Interpretable Neural Clustering via Additive Modelscited by 3
- 2025NetBurst: Event-Centric Forecasting of Bursty, Intermittent Time Seriescited by 2
- 2024Self Supervised Correlation-based Permutations for Multi-View Clusteringcited by 1
- 2024Spectral Self-supervised Feature Selectioncited by 1
- 2025Distributionally Robust Feature Selectioncited by 1
- 2024Conditional Deep Canonical Time Warpingcited by 0
12 most cited of 19 All 19 (JSON)
- 2026GRACE: LLM-Grounded Semantic Metric Spaces for Scalable Mixed-Data Clustering8 Aug 2026
- 2026Population-Robust Feature Selection via Generalized Welfare Optimization3 Aug 2026
- 2026Ensemble of Unsupervised Deep Learning for Clustering Imbalanced Tabular Data31 Jul 2026
- 2026Cohort Organized Learning: Clustering Through Agreement19 Jun 2026
- 2026Beyond Statistical Co-occurrence: Unlocking Intrinsic Semantics for Tabular Data Clustering13 Apr 2026
- 2026Weight-Informed Self-Explaining Clustering for Mixed-Type Tabular Data7 Apr 2026
- 2026i-IF-Learn: Iterative Feature Selection and Unsupervised Learning for High-Dimensional Complex Data25 Mar 2026
- 2025Hybrid Autoencoders for Tabular Data: Leveraging Model-Based Augmentation in Low-Label Settings10 Nov 2025
- 2025NetBurst: Event-Centric Forecasting of Bursty, Intermittent Time Series25 Oct 2025
- 2025Distributionally Robust Feature Selection24 Oct 2025
- 2025ICR-RL: Deep Reinforcement Learning via In-Context Regression14 Sep 2025
- 2025ZEUS: Zero-shot Embeddings for Unsupervised Separation of Tabular Data15 May 2025
12 newest of 19 All 19 (JSON)