Weakly supervised concept Bottleneck Learning for Robust Two stage Object centric visual reasoning
Organizations: University of Lübeck, Germany · Ulm University, Germany · University of Bamberg, Germany
Abstract
Two-stage neuro-symbolic architectures provide an elegant paradigm for visual problem solving by cleanly separating connectionist perception of predefined symbols from possibly later defined relational reasoning thereon. However, anchoring high-level predicates into visual frames typically necessitates annotations that are expensive to acquire. In this work, we introduce the Dynamic Orthogonal Concept Bottleneck (D-OCB), an object-centric slot- VAE framework designed to extract human-aligned symbolic predicates under extremely weak supervision. D-OCB eliminates the arduous manual tuning of loss-balancing coef- ficients by dynamically learning optimal hyperparameter allocations during training. To infuse prior knowledge on independence of concept categories, in addition to standard re- construction self-supervision we penalize correlation across concept subspaces. Crucially, to combat the instability of very low supervision regimes, D-OCB incorporates a dynamic di- mensionality allocation mechanism; this adaptive formulation allows well-represented con- cepts to yield latent dimensions to underperforming concepts that are lagging behind, effectively preventing representation collapse and significantly improving overall concept accuracy. Through an extensive empirical evaluation, we demonstrate that our framework achieves high concept alignment and downstream visual reasoning accuracy using minimal label budgets, matching or outperforming end-to-end paradigms.
Figures & tables
| CNN Backbone | DINO Backbone | |||||||||||
| Sup. | Method | Size | Shape | Color | Material | Mean | Size | Shape | Color | Material | Mean | |
| 1% | Vanilla+SA | |||||||||||
| CEM | ||||||||||||
| ProbCBM | ||||||||||||
| SlotFormer | ||||||||||||
| SlotVAE-NoVAE | ||||||||||||
| Backbone | Sup. (%) | Dx | Localization | Sex | Age | Mean |
|---|---|---|---|---|---|---|
| CNN | 1 | |||||
| 5 | ||||||
| 15 | ||||||
| DINO | 1 | |||||
| 5 | ||||||
| 15 |
| ID | Rule Type | Description | |
| CLEVR | R1 | 4-way Conjunction | The scene contains an object that is simultaneously large, red, metal, and a cube. |
| R2 | 3-way Conjunction | The scene contains a small rubber sphere. | |
| R3 | Spatial Relation | A blue cylinder is spatially to the left of a red object (evaluated via 3D x-coordinates). | |
| R4 | Cardinality | The scene contains metal objects; requires the model to count distinct slots satisfying this constraint. | |
| R5 | Proximity Relation | A large green object is within a continuous Euclidean distance threshold ( ) of a yellow object. | |
| HAM | R1 | Simple | The lesion is malignant. |
| Backb. | Percept. Model | Decision Tree (DT) | Bayes Net (BN) | NS-CL | DILP | ||||||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| 1% | 15% | 25% | 50% | 75% | 1% | 15% | 25% | 50% | 75% | 1% | 15% | 25% | 50% | 75% | 1% | 15% | 25% | 50% | 75% | ||
| CNN | D-OCB (Ours) | 63.1 | 70.8 | 72.0 | 71.9 | 69.8 | 85.0 | 71.6 | 72.1 | 71.8 | 71.8 | 62.2 | 71.0 | 71.5 | 71.3 | 71.2 | 67.1 | 97.0 | 97.1 | 97.4 | 97.1 |
| SlotVAE-KL | 56.3 | 33.3 | 66.6 | 67.7 | 71.6 | 52.9 | 32.7 | 83.3 | 84.5 | 85.5 | 45.9 | 31.4 | 50.1 | 74.0 | 74.2 | 45.0 | 47.1 | 69.5 | 80.8 | 82.1 | |
| SlotVAE-NoVAE | 55.6 | 48.3 | 59.8 | 68.1 | 69.4 | 69.3 | 53.6 | 70.0 | 60.2 | 61.5 | 61.4 | 56.7 | 66.6 | 72.2 | 72.9 | 70.9 | 74.6 | 89.0 | 95.6 | 96.0 | |
| Vanilla+SA | 51.3 | 51.8 | 54.1 | 55.5 | 58.9 | 51.4 | 53.6 | 53.9 | 53.8 | 52.4 | 62.2 | 60.3 | 63.8 | 62.7 | 66.8 | 64.2 | 76.1 | 85.2 | 85.2 | 92.1 | |
| DINO | D-OCB (Ours) | 67.9 | 68.3 | 70.3 | 74.0 | 69.9 | 55.6 | 71.1 | 71.0 | 71.0 | 71.0 | 70.0 | 70.8 | 70.6 | 70.6 | 70.8 | 93.9 | 96.9 | 97.3 | 97.4 | 97.6 |
| BB | Sup% | Model | DILP (%) | Decision Tree (%) | Bayes Net (%) | NS-CL (%) | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| R1 | R2 | R3 | R1 | R2 | R3 | R1 | R2 | R3 | R1 | R2 | R3 | |||
| CNN | 1 | D-OCB (Ours) | ||||||||||||
| Slot-VAE | - | - | - | |||||||||||
| 15 | D-OCB (Ours) | |||||||||||||
| Slot-VAE | - | - | - | |||||||||||
| 75 | D-OCB (Ours) | |||||||||||||
| CLEVR-Hans3 Benchmark | ||||
|---|---|---|---|---|
| Method | Sup. | Train Acc | Test Acc | OOD Gap |
| E2E CNN Basel. | 0% | |||
| NeSy | 100% | |||
| NeSy-XIL | 100% | |||
| ILP | 100% | |||
| NEUMANN | 100% | |||
Concept Bottleneck Models
Concept Bottleneck Models route predictions through human-understandable concepts that people can inspect, correct, or intervene on.
See all 88 papers