DC-ViT: Modulating Spatial and Channel Interactions for Multi-Channel Images
Organizations: University of Surrey
Abstract
Training and evaluation in multi-channel imaging (MCI) remains challenging due to heterogeneous channel configurations arising from varying staining protocols, sensor types, and acquisition settings. This heterogeneity limits the applicability of fixed-channel encoders commonly used in general computer vision. Recent Multi-Channel Vision Transformers (MC-ViTs) address this by enabling flexible channel inputs, typically by jointly encoding patch tokens from all channels within a unified attention space. However, unrestricted token interactions across channels can lead to feature dilution, reducing the ability to preserve channel-specific semantics that are critical in MCI data. To address this, we propose Decoupled Vision Transformer (DC-ViT), which explicitly regulates information sharing using Decoupled Self-Attention (DSA), which decomposes token updates into two complementary pathways: spatial updates that model intra-channel structure, and channel-wise updates that adaptively integrate cross-channel information. This decoupling mitigates informational collapse while allowing selective inter-channel interaction. To further exploit these enhanced channel-specific representations, we introduce Decoupled Aggregation (DAG), which allows the model to learn task-specific channel importances. Extensive experiments across three MCI benchmarks demonstrate consistent improvements over existing MC-ViT approaches.
Figures & tables
| Method DC-ViT | CHAMMI | JUMP-CP | So2Sat | |||
|---|---|---|---|---|---|---|
| Avg. | Full | Partial | Full | Partial | ||
| ChAdaViT | ✗ | 63.88 | 65.03 | 42.15 | 56.98 | 12.38 |
| ✓ | 68.91 | 69.53 | 25.09 | 62.56 | 10.63 | |
| ChannelViT | ✗ | 64.90 | 67.51 | 56.49 | 61.03 | 46.16 |
| ✓ | 71.72 | 73.26 | 61.90 | 64.97 | 48.61 | |
| DiChaViT | ✗ | 69.68 | 69.19 | 57.98 | 63.36 | 47.76 |
Appendix figures & tables2 assets
Supplementary material from the paper’s appendix.
Appendix
| DC-ViT | Sampling | CHAMMI | JUMP-CP | So2Sat | ||
|---|---|---|---|---|---|---|
| Avg. | Full | Partial | Full | Partial | ||
| ✗ | Shared | 73.13 | 85.23 | 66.85 | 65.14 | 50.01 |
| Random | 74.63 | 90.73 | 68.05 | 67.44 | 52.11 | |
| ✓ | Shared | 76.33 | 86.11 | 70.02 | 67.57 | 53.11 |
| Method | DC-ViT | Average OOD | WTC | HPA | CP | |||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Mean | WTC | HPA | CP | Task1 | Task2 | Task1 | Task2 | Task3 | Task1 | Task2 | Task3 | Task4 | ||
| ChAda- | ✗ | 50.82 | 67.18 | 60.67 | 24.60 | 77.58 | 67.18 | 87.49 | 75.94 | 45.41 | 83.92 | 45.58 | 21.94 | 6.28 |
| -ViT | ✓ | 56.05 | 72.06 | 67.76 | 28.32 | 78.01 | 72.06 | 93.54 | 88.42 | 47.09 | 86.09 | 56.22 | 22.61 | 6.14 |
| Channel- | ✗ | 52.54 | 67.58 | 62.35 | 27.81 | 78.36 | 67.58 | 83.93 | 76.73 | 47.97 | 77.70 | 55.16 | 21.89 | 6.38 |
| -ViT | ✓ | 56.93 | 73.04 | 70.40 | 27.35 | 82.00 | 73.04 | 92.01 | 87.92 | 52.87 | 85.95 | 54.31 | 22.28 | 5.45 |
| DiCha- | ✗ | 55.36 | 75.18 | 64.36 | 26.53 | 80.87 | 75.18 | 88.08 | 79.26 | 49.45 | 84.08 | 53.03 | 20.95 | 5.60 |