Codebook-Guided Cross-Modal Knowledge Distillation for Structurally Heterogeneous Features
Organizations: Kyungpook National University · AX/PI Center, Samsung Electronics · Chung-Ang University, SNUAILAB · University of Seoul
Abstract
Cross-modal knowledge distillation transfers knowledge from a teacher modality to a student modality. Existing feature-level alignment methods typically assume that teacher and student features reside in structurally alignable representation spaces. However, this assumption does not hold when cross-modal features are structurally heterogeneous and lack clear unit-level correspondence, such as 2D spatial visual grids and 1D temporal audio sequences, thereby limiting the applicability of feature-level alignment. To address this challenge, we propose a cross-modal distillation framework that enables effective knowledge transfer across structurally heterogeneous feature spaces via a vector-quantized codebook. Specifically, teacher features are abstracted into a set of vector-form codes regardless of their original feature structure, and the selected codes serve as concept-level anchors for student learning. Code selection is guided by both task relevance and student compatibility, allowing the student to receive transferable teacher knowledge without requiring direct unit-level feature alignment. Experimental results across diverse cross-modal distillation scenarios demonstrate the effectiveness of the proposed framework on classification and semantic segmentation tasks.
Figures & tables
| Method | RAVDESS | VGG-Sound | CrisisMMD | MM-IMDB | ||||
|---|---|---|---|---|---|---|---|---|
| A V | V A | A V | V A | T I | I T | T I | I T | |
| w/o KD | 0.9208 | 0.6826 | 0.5031 | 0.6058 | 0.5403 | 0.5500 | 0.6645 | 0.7294 |
| General distillation baselines | ||||||||
| KLD | 0.9118 | 0.6972 | 0.5153 | 0.6176 | 0.5377 | 0.5578 | 0.6732 | 0.7310 |
| MLLD | 0.9292 | 0.6889 | 0.5135 | 0.6122 | 0.5357 | 0.5562 | 0.6699 | 0.7320 |
| FitNets | 0.9062 | 0.7146 | 0.4916 | 0.5962 | 0.5281 | 0.5507 | 0.6759 | 0.7296 |
| Method | Depth RGB | RGB Depth | ||||
|---|---|---|---|---|---|---|
| OA | CA | mIoU | OA | CA | mIoU | |
| w/o KD | 0.5582 | 0.2748 | 0.1850 | 0.5269 | 0.2100 | 0.1366 |
| MLLD | 0.5599 | 0.2792 | 0.1911 | 0.5250 | 0.2118 | 0.1399 |
| CRD | 0.5592 | 0.2677 | 0.1806 | 0.5261 | 0.2111 | 0.1385 |
| OFA | 0.5607 | 0.2881 | 0.1982 | 0.5242 | 0.2123 | 0.1409 |
| MGDFR | 0.5601 | 0.2737 | 0.1878 | 0.5213 | 0.2162 | 0.1442 |
| Code | Pred | RAVDESS | VGG-Sound | CrisisMMD | MM-IMDB | ||||
|---|---|---|---|---|---|---|---|---|---|
| Match | A V | V A | A V | V A | T I | I T | T I | I T | |
| - | - | 0.9208 | 0.6826 | 0.5031 | 0.6058 | 0.5403 | 0.5500 | 0.6645 | 0.7294 |
| ✓ | - | 0.9202 | 0.7333 | 0.5067 | 0.6098 | 0.5387 | 0.5534 | 0.6716 | 0.7363 |
| - | ✓ | 0.9285 | 0.7479 | 0.5294 | 0.6204 | 0.5367 | 0.5529 | 0.6770 | 0.7421 |
| ✓ | ✓ | 0.9431 | 0.7486 | 0.5299 | 0.6229 | 0.5433 | 0.5592 | 0.6768 | 0.7434 |
| Pred-Match Target | RAVDESS | VGG-Sound | CrisisMMD | MM-IMDB | ||||
|---|---|---|---|---|---|---|---|---|
| A V | V A | A V | V A | T I | I T | T I | I T | |
| Original branch | 0.8903 | 0.6201 | 0.5173 | 0.6171 | 0.5425 | 0.5539 | 0.6654 | 0.7351 |
| Codebook branch | 0.9431 | 0.7486 | 0.5299 | 0.6229 | 0.5433 | 0.5592 | 0.6768 | 0.7434 |
| Selection Criteria | RAVDESS | VGG-Sound | CrisisMMD | MM-IMDB | ||||
|---|---|---|---|---|---|---|---|---|
| A V | V A | A V | V A | T I | I T | T I | I T | |
| Random | 0.9292 | 0.7396 | 0.5252 | 0.6211 | 0.5353 | 0.5536 | 0.6718 | 0.7353 |
| Task relevance only | 0.9396 | 0.7361 | 0.5276 | 0.6233 | 0.5435 | 0.5558 | 0.6727 | 0.7351 |
| Student compatibility only | 0.9375 | 0.7417 | 0.5275 | 0.6222 | 0.5437 | 0.5565 | 0.6722 | 0.7346 |
| Combination (Ours) | 0.9431 | 0.7486 | 0.5299 | 0.6229 | 0.5433 | 0.5592 | 0.6768 | 0.7434 |
Appendix figures & tables16 assets
Supplementary material from the paper’s appendix.
Appendix
| Dataset | Epochs | Learning rate | Weight decay | Batch size | #Code | Code Dim. | ||
|---|---|---|---|---|---|---|---|---|
| RAVDESS | 100 | 64 | 512 | 128 | 0.25 | |||
| VGG-Sound | 100 | 128 | 512 | 128 | 0.25 | |||
| CrisisMMD | 100 | 64 | 256 | 128 | 0.25 | |||
| MM-IMDB | 100 | 64 | 512 | 64 | 0.25 | |||
| NYU-Depth V2 | 100 | 6 | 1024 | 256 | 0.25 |
| Class | # Samples |
|---|---|
| Drama | 2,681 |
| Comedy | 1,337 |
| Documentary | 918 |
| Horror | 417 |
| Western | 173 |
| Thriller | 143 |
| Method | RAVDESS | VGG-Sound | ||
|---|---|---|---|---|
| A V | V A | A V | V A | |
| w/o KD | 0.9208 0.0164 | 0.6826 0.0189 | 0.5031 0.0035 | 0.6058 0.0086 |
| KLD | 0.9118 0.0156 | 0.6972 0.0674 | 0.5153 0.0067 | 0.6176 0.0090 |
| MLLD | 0.9292 0.0140 | 0.6889 0.0461 | 0.5135 0.0056 | 0.6122 0.0047 |
| FitNets | 0.9062 0.0188 | 0.7146 0.0385 | 0.4916 0.0105 | 0.5962 0.0128 |
| RKD | 0.9236 0.0132 | 0.7014 0.0269 | 0.5066 0.0068 | 0.6085 0.0018 |
| Method | Depth RGB | RGB Depth | ||||
|---|---|---|---|---|---|---|
| OA | CA | mIoU | OA | CA | mIoU | |
| w/o KD | 0.5582 0.0112 | 0.2748 0.0218 | 0.1850 0.0154 | 0.5269 0.0070 | 0.2100 0.0155 | 0.1366 0.0110 |
| MLLD | 0.5599 0.0122 | 0.2792 0.0226 | 0.1911 0.0169 | 0.5250 0.0063 | 0.2118 0.0150 | 0.1399 0.0152 |
| CRD | 0.5592 0.0101 | 0.2677 0.0158 | 0.1806 0.0108 | 0.5261 0.0061 | 0.2111 0.0125 | 0.1385 0.0107 |
| OFA | 0.5607 0.0119 | 0.2881 0.0146 | 0.1982 0.0107 | 0.5242 0.0085 | 0.2123 0.0125 | 0.1409 0.0123 |
| MGDFR | 0.5601 0.0123 | 0.2737 0.0196 | 0.1878 0.0165 | 0.5213 0.0099 | 0.2162 0.0097 | 0.1442 0.0083 |
| Code | Pred-Match | RAVDESS | VGG-Sound | ||
|---|---|---|---|---|---|
| A V | V A | A V | V A | ||
| - | - | 0.9208 0.0164 | 0.6826 0.0189 | 0.5031 0.0035 | 0.6058 0.0086 |
| ✓ | - | 0.9202 0.0260 | 0.7333 0.0397 | 0.5067 0.0029 | 0.6098 0.0033 |
| - | ✓ | 0.9285 0.0234 | 0.7479 0.0175 | 0.5294 0.0013 | 0.6204 0.0056 |
| ✓ | ✓ | 0.9431 0.0097 | 0.7486 0.0231 | 0.5299 0.0017 | 0.6229 0.0059 |
| Pred-Match Target | RAVDESS | VGG-Sound | ||
|---|---|---|---|---|
| A V | V A | A V | V A | |
| Original branch | 0.8903 0.0140 | 0.6201 0.0788 | 0.5173 0.0024 | 0.6171 0.0079 |
| Codebook branch | 0.9431 0.0097 | 0.7486 0.0231 | 0.5299 0.0017 | 0.6229 0.0059 |
| Selection Criteria | RAVDESS | VGG-Sound | ||
|---|---|---|---|---|
| A V | V A | A V | V A | |
| Random | 0.9292 0.0134 | 0.7396 0.0214 | 0.5252 0.0049 | 0.6211 0.0051 |
| Task relevance only | 0.9396 0.0063 | 0.7361 0.0134 | 0.5276 0.0055 | 0.6233 0.0063 |
| Student compatibility only | 0.9375 0.0078 | 0.7417 0.0124 | 0.5275 0.0077 | 0.6222 0.0052 |
| Combination (Ours) | 0.9431 0.0097 | 0.7486 0.0231 | 0.5299 0.0017 | 0.6229 0.0059 |
| Method | RAVDESS | VGG-Sound | ||
|---|---|---|---|---|
| A V | V A | A V | V A | |
| Runner-up | 0.9271 0.0080 | 0.7351 0.0144 | 0.5219 0.0046 | 0.6126 0.0043 |
| Ours | 0.9420 0.0071 | 0.7556 0.0120 | 0.5273 0.0041 | 0.6158 0.0029 |
| p-value | 0.001 | 0.007 | 0.028 | 0.007 |
| Method | RAVDESS | VGG-Sound | CrisisMMD | MM-IMDB | ||||
|---|---|---|---|---|---|---|---|---|
| A V | V A | A V | V A | T I | I T | T I | I T | |
| KLD | 7.4 | 6.4 | 4.6 | 3.8 | 7.4 | 5.4 | 4.0 | 7.0 |
| MLLD | 4.8 | 7.2 | 4.4 | 5.8 | 8.6 | 5.0 | 5.8 | 6.4 |
| FitNets | 7.8 | 5.8 | 10.8 | 10.8 | 7.4 | 7.8 | 4.4 | 7.4 |
| RKD | 5.4 | 7.0 | 8.0 | 8.2 | 4.4 | 4.6 | 7.6 | 6.6 |
| CRD | 4.6 | 7.4 | 7.8 | 7.8 | 6.2 | 4.8 | 5.8 | 7.0 |
| Code | Pred-Match | RAVDESS | VGG-Sound | ||
|---|---|---|---|---|---|
| A V | V A | A V | V A | ||
| - | ✓ | 0.9274 0.0110 | 0.7469 0.0060 | 0.5246 0.0031 | 0.6129 0.0031 |
| ✓ | ✓ | 0.9420 0.0071 | 0.7556 0.0120 | 0.5273 0.0041 | 0.6158 0.0029 |
| p-value | 0.0031 | 0.0885 | 0.1215 | 0.0504 | |
| Selection Criteria | RAVDESS | VGG-Sound | ||
|---|---|---|---|---|
| A V | V A | A V | V A | |
| Random | 0.9316 0.0100 | 0.7423 0.0073 | 0.5242 0.0029 | 0.6125 0.0023 |
| Combination (Ours) | 0.9420 0.0071 | 0.7556 0.0120 | 0.5273 0.0041 | 0.6158 0.0029 |
| p-value | 0.0333 | 0.0205 | 0.0309 | 0.0287 |
| Method | V A | A V | ||||
|---|---|---|---|---|---|---|
| Time | Mem | TFLOPs | Time | Mem | TFLOPs | |
| w/o KD | 82.69 | 25.40 | – | 168.44 | 77.25 | – |
| KLD | 172.95 | 76.29 | 0.62 | 176.04 | 80.34 | 0.62 |
| MSE | 171.78 | 76.35 | 0.62 | 165.68 | 80.40 | 0.62 |
| CRD | 325.45 | 88.22 | 0.63 | 324.03 | 94.71 | 0.63 |
| FitNets | 338.17 | 78.26 | 1.24 | 339.19 | 83.05 | 1.24 |
| V A | A V | |||||||
|---|---|---|---|---|---|---|---|---|
| Stage-1 | Stage-2 | Stage-3 | Acc. | Stage-1 | Stage-2 | Stage-3 | Acc. | |
| 128 | 231.53 | 10.48 | 182.34 | 0.6202 | 224.55 | 10.56 | 179.26 | 0.5270 |
| 256 | 228.37 | 18.54 | 190.97 | 0.6213 | 225.65 | 18.65 | 193.02 | 0.5272 |
| 512 | 226.81 | 36.24 | 195.80 | 0.6229 | 225.70 | 34.87 | 191.72 | 0.5299 |
| 1024 | 226.90 | 79.93 | 191.29 | 0.6201 | 224.92 | 68.62 | 190.45 | 0.5283 |
| 2048 | 228.29 | 184.83 | 202.42 | 0.6205 | 221.61 | 131.78 | 199.46 | 0.5262 |
| Top-5 increased classes | Top-5 decreased classes | ||||
|---|---|---|---|---|---|
| Class | Freq. | Recall | Class | Freq. | Recall |
| other_furniture | 2.7% | +0.1807 | person | 0.3% | 0.1298 |
| mirror | 1.2% | +0.1519 | counter | 1.7% | 0.0478 |
| television | 0.7% | +0.1324 | bookshelf | 2.1% | 0.0398 |
| desk | 0.8% | +0.1315 | clothes | 0.8% | 0.0294 |
| curtain | 1.8% | +0.1311 | wall | 25.6% | 0.0274 |
| Assignment | A V | V A |
|---|---|---|
| Hard | 0.5299 0.0017 | 0.6229 0.0059 |
| Soft | 0.5254 0.0024 | 0.6208 0.0082 |