Title: Escaping Low-Rank Traps: Interpretable Visual Concept Learning via Implicit Vector Quantization
Primary area: interpretability and explainable AI
Keywords: Concept Bottleneck Models; Visual Concept Learning; Vision-Language Models; Representational Collapse; Interpretability

Abstract: Concept Bottleneck Models (CBMs) achieve interpretability by interposing a human-understandable concept layer between perception and label prediction. 
The foundation of CBMs lies in the many-to-many mapping that translates high-dimensional visual features to a set of discrete concepts. 
However, we identify a critical and pervasive challenge that undermines this process: \emph{representational collapse}, where visual patch features degenerate into a low-rank subspace during training, severely degrading the quality of learned concept activation vectors, thus hindering both model interpretability and downstream performance.
To address these issues, we propose Implicit Vector Quantization (IVQ), a lightweight regularizer that maintains high-rank, diverse representations throughout training. 
Rather than imposing a hard bottleneck via direct quantization, IVQ learns a codebook prior that anchors semantic information in visual features, allowing it to act as a proxy objective.
To further exploit these high-rank concept-aware features, we propose Magnet Attention, which dynamically aggregates patch-level features into visual concept prototypes, explicitly modeling the many-to-many vision–concept correspondence.
Extensive experimental results show that our approach effectively prevents representational collapse and achieves state-of-the-art performance on eight diverse benchmarks.
Our experiments further probe the low-rank phenomenon in representational collapse, finding that IVQ mitigates the information bottleneck and yields cross-modal representations with clearer, more interpretable consistency.

TL;DR: We utilize implicit vector quantisation to address the Representational Collapse issue, while delivering better visual concept learning.