SSQA: Sibling-Selective Quadtree Attention for Hierarchical Modeling in Perception Tasks
Authors: Yufan Chen, Arnav Bali, Angela Liu, Laura Zheng, Ming C. Lin
Organizations: Thomas Lord Department of Computer Science, University of Southern California, Los Angeles, CA, USA. · South River High School, Edgewater, MD, USA. · Department of Computer Science, University of Maryland, College Park, MD, USA.
Perception tasks for navigation in robotics, including aerial platforms such as drones and autonomous driving systems, are inherently structured. Drone-mounted cameras typically capture sky above, terrain below, and obstacles or man-made structures in between, while driving data often contains organized road layouts, lane markings, and surrounding agents. Motivated by these axis-aligned structural priors, we note that such information is typically more structured than in generic image tasks. We hypothesize that processing information in a quadtree-esque manner can not only model features effectively in a hierarchical manner, but also offers an efficient linear-time alternative to vanilla attention mechanisms, which run in quadratic time. In this paper, we propose Sibling-Selective Quadtree Attention (SSQA), which models image tokens hierarchically as a structured, full quadtree. We show analytical complexity analysis that guarantees linear-time feature modeling, in addition to empirical experiments comparing inference speeds with other popular modeling approaches, such as Mamba 2 and Quadtree Attention. Our results, benchmarked across several tasks, show that we achieve results at least as good, if not notably better, as others at a fraction of the computational costs.
Figures & tables
Fig. 1 : Mean Epoch Wall Times (s) across 50 epochs on IllusionAnimals classification (224 2 inputs) on A100 GPU. Despite other methods also modeling attention in linear time, a practical caveat is the lack of compute optimization in practice. As a result, some methods are even slower than vanilla ViT [ 6 ] out-of-the-box from official implementations. SSQA is both easy to integrate and requires no additional practical optimizations, our results show that SSQA maintains or even improves performance over other hierarchical modeling baselines.
Fig. 2 : Top-Down vs. Bottom-Up Gating; Squares indicate leaf nodes. Our work has a key difference from previous work in hierarchical quadtree-based attention approaches: bottom-up gating. Previous works (a) avoid subdivisions into tokens with smaller granularity if possible by selecting top-k tokens with highest saliency during training. As a result, tokens on the same level outside of top-k saliency neglect hierarchical reasoning, even if there agents present in those patches . For example, in a driving scene, one patch may have a higher density of pedestrians, but it shouldn’t imply less attention on the single agent falling outside of the same patch. In our approach, we model features from bottom-up, ensuring that features are modeled hierarchically across the entire image . Doing so not only improves performance on various benchmarked tasks, but also maintains linear complexity with respect to number of tokens (shown in § III-G ).
Method
Params (M) ↓
FLOPs (G) ↓
Acc ↑
Avg Time (s) ↓
Baseline-ViT [ 28 ]
90.52
33.70
98.25%
53.17
ViT-LinAtt [ 29 ]
90.49
33.74
98.25%
54.65
Mamba 2 [ 30 ]
107.43
28.77
98.41%
67.70
SSQA (ours)
36.18
4.29
98.65%
26.19
TABLE I : Compute vs. Validation performance for image classification: upto 3x fewer parameters, 8x fewer FLOPs, 2.08x mean epoch time.
Fig. 3 : Training Dynamics . SSQA (a) converges faster and (b) reaches target accuracy earlier than the ViT baseline.
Method
Params (M)
FLOPs (G) ↓
Acc
Baseline-ViT [ 28 ]
90.52
33.70
98.25%
ViT-LinAtt [ 29 ]
90.51
33.74
98.25%
Mamba 2 [ 31 ]
107.43
28.77
98.41%
Swin TRXFM [ 1 ]
32.24
8.75
99.05%
MLP-Mixer [ 32 ]
63.84
25.24
98.02%
ConvNeXT V2 [ 33 ]
32.54
8.92
98.33%
TABLE II : Thorough Compute vs. Accuracy for Classification on IllusionAnimals (50 epochs, 2242 ).
Fig. 4 : Model Efficiency . SSQA is substantially lighter by reducing the number of parameters by nearly 3x and more compute-efficient by up to a factor of 8x for classification test.
Variant
Params (M)
FLOPs (G)
Acc
Δ Acc
SSQA (full; k=3 )
36.18
4.29
98.65%
+0.00
w/o CIF (OFF)
36.18
4.29
98.22%
−0.43
NO-GATE (no CIF gate)
36.18
4.29
98.53%
−0.12
w/o local FFN
36.18
1.52
98.29%
−0.36
k=2
36.18
4.29
98.40%
−0.25
k=4 (no top- k )
36.18
4.29
98.37%
−0.28
TABLE III : Ablation on SSQA components (IllusionAnimals, 50 epochs, 2242 ). We report best validation accuracy for one run (seed=42) and include compute for reference; key ablations additionally report mean ± std over 3 seeds (42/43/44) when available.
Method
Params (M)
FLOPs (G) ↓
mIoU
Baseline-ViT [ 28 ]
24.94
49.63
0.3349
ViT-LinAtt [ 29 ]
23.98
43.22
0.2932
Mamba 2 [ 31 ]
14.11
48.22
0.3518
Swin TRXFM [ 1 ]
35.49
53.60
0.3307
MLP-Mixer [ 32 ]
74.72
148.15
0.3432
ConvNeXT V2 [ 33 ]
33.13
24.64
0.2803
TABLE IV : Compute vs. mIoU for Segmentation on LoveDA ( 5122 )
Fig. 5 : Segmentation Model Performance Comparison . SSQA has the lowest FLOPs 1.54x lower than the second lowest flops model and still be able to achieve the second best accuracy with only 0.16 percentage apart.
Fig. 6 : CARLA Evaluation Score . Benchmark evaluation score across 14 scenarios (0–13) for Baseline [ 27 ] vs. Quadtreen ATTN [ 24 ] vs. SSQA (higher is better).