Disentangling Latent Embeddings with Sparse Linear Concept Cones (SLiCC)
Organizations: Department of Electrical & Computer Engineering, University of Delaware, USA · Naval Surface Warfare Center, Panama City Division, Panama City, Florida, USA · Department of Computer & Information Sciences, University of Delaware, USA
Abstract
Pretrained vision models encode semantic information in latent embedding spaces that are useful for downstream tasks. We hypothesize that this information can be disentangled to separate the content of complex scenes by decomposing each embedding into multiple concept-specific components. We propose Sparse Linear Concept Cones (SLiCC), a supervised dictionary learning approach that estimates a linear synthesis model consisting of sparse, non-negative combinations of groups of dictionary atoms, whose group-wise activity is guided by multi-label information. Each concept-specific component is a non-negative combination of atoms associated with a label and lies in a corresponding concept cone. We introduce an alternating optimization algorithm and an autoencoder formulation of SLiCC. We apply SLiCC to three types of CLIP and DINOv2 embeddings across five datasets. For CLIP, we additionally exploit image-text alignment to enable a self-supervised SLiCC. We also introduce a fully unsupervised autoencoder that enforces group sparsity based on the norm of neuron activations. Quantitative results show that supervised SLiCC improves concept-filtered image retrieval over unfiltered embeddings, including retrieval of finer-grained information not provided during training. Beyond retrieval of individual concepts, the disentangled components enable concept deletion and recombination across images. We further explore conditional image generation using Image-to-Prompt, with qualitative results illustrating retrieval and generation from manipulated concept representations.
Figures & tables
| CLIP-RN50 | Double MNIST | MIRFlickr25K | MS COCO | Pascal VOC 2012 | CIFAR100 | ||||
|---|---|---|---|---|---|---|---|---|---|
| Gen. | Gen. | Sub. | Gen. | Sub. | Gen. | Sub. | Gen. | Sub. | |
| UF | 0.751 | 0.723 | 0.693 | 0.782 | 0.663 | 0.847 | 0.780 | 0.507 | 0.342 |
| SpLiCE | 0.391 | 0.646 | 0.561 | 0.707 | 0.584 | 0.792 | 0.758 | 0.389 | 0.222 |
| F-SpLiCE | 0.302 | 0.653 | 0.594 | 0.628 | 0.496 | 0.840 | 0.797 | 0.487 | 0.256 |
| S-SVD | 0.948 | 0.909 | 0.633 | 0.908 | 0.674 | 0.922 | 0.300 | 0.717 | 0.252 |
| S-SLiCC | 0.979 | 0.932 | 0.801 | 0.922 | 0.717 | 0.961 | 0.842 | 0.806 | 0.494 |
Appendix figures & tables4 assets
Supplementary material from the paper’s appendix.
Appendix
| Concept Cone | S-SLiCC-AE word caption | SS-SLiCC-AE word caption |
|---|---|---|
| person | granddaughter, girl, participants, woman, grandson | granddaughter, girl, grandma, woman, herself |
| vehicle | bike, bikes, boat, boats, motorcycle | truck, buses, automotive, vehicle, motorcycle |
| outdoor | benches, bench, nyc, brooklyn, lights | street, beach, lake, nature, nyc |
| animal | dogs, horses, horse, pups, cows | bear, dogs, pups, bears, horse |
| accessory | rainy, rain, rainfall, umbrellas, rains | luggage, umbrellas, suitcase, umbrella, rainy |
| sports | skate, baseball, tennis, skateboard, surf | baseball, tennis, surfers, skate, surf |
| Concept pair | RN50 | ViT-B/32 | ViT-L/14 | DINO | |||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Concept 1 | Concept 2 | Weighted | B1 | B2 | Weighted | B1 | B2 | Weighted | B1 | B2 | Weighted | B1 | B2 |
| 0 | 1 | 0.459 | 0.277 | 0.474 | 0.488 | 0.289 | 0.503 | 0.515 | 0.252 | 0.540 | 0.516 | 0.269 | 0.551 |
| 0 | 4 | 0.424 | 0.193 | 0.380 | 0.460 | 0.189 | 0.408 | 0.496 | 0.214 | 0.464 | 0.432 | 0.203 | 0.452 |
| 0 | 5 | 0.696 | 0.622 | 0.760 | 0.733 | 0.656 | 0.780 | 0.787 | 0.673 | 0.846 | 0.628 | 0.692 | 0.753 |
| 0 | 6 | 0.370 | 0.147 | 0.334 | 0.347 | 0.132 | 0.322 | 0.417 | 0.107 | 0.412 | 0.430 | 0.118 | 0.385 |
| 0 | 8 | 0.268 | 0.165 | 0.238 | 0.301 | 0.132 | 0.260 | 0.338 | 0.171 | 0.289 | 0.400 | 0.193 | 0.379 |
Disentangled Representation Learning
Disentangled representation learning separates distinct factors of variation so models can interpret, control, and generalize their predictions.
See all 234 papers